from typing import List, Tuple
from src.annotations import (
    Annotation,
    EmailAnnotation,
)
from src.string_replacements import replace_and_update
from src.replacers.interface import ReplacerInterface
import random
import string

def random_char(char_num):
       return ''.join(random.choice(string.ascii_letters) for _ in range(char_num))

def random_email():
    return random_char(7)+"@gmail.com"

class EmailReplacer(ReplacerInterface):
    def __init__(self):
        pass
    
    def replace(self, text: str, detections: List[Tuple[int, int, Annotation]]) -> Tuple[str, List[Tuple[int, int, Annotation]]]:
        replacements = []
        not_processed = []
        
        already_replaced = dict()
        
        for item in detections:
            start, end, detection = item
            
            if isinstance(detection, EmailAnnotation):
                if text[start:end] not in already_replaced:
                    already_replaced[text[start:end]] = random_email()
                    
                replacements.append((start, end, already_replaced[text[start:end]]))
            else:
                not_processed.append(item)
            
        return replace_and_update(text, replacements, not_processed)