import datetime
import json
import os
import re
from .chat_processor import ChatProcessor

PATTERN  = re.compile(r"(.*)\(([0-9]+)\)$")

class JsonfileArchiver(ChatProcessor):
    """
    JsonfileArchiver saves chat data as text of JSON lines.

    Parameter:
    ----------
    save_path : str :
        save path of file.If a file with the same name exists, 
        it is automatically saved under a different name
        with suffix '(number)'
    """
    def __init__(self,save_path):
        super().__init__()
        self.save_path = self._checkpath(save_path)
        self.line_counter = 0
                
    def process(self,chat_components: list):
        """
        Returns
        ----------
        dict : 
            save_path : str :
                Actual save path of file.
            total_lines : int :
                count of total lines written to the file.
        """
        if chat_components is None: return
        with open(self.save_path, mode='a', encoding = 'utf-8') as f:
            for component in chat_components:
                if component is None: continue
                chatdata = component.get('chatdata')
                if chatdata is None: continue
                for action in chatdata:
                    if action is None: continue
                    json_line = json.dumps(action, ensure_ascii = False)
                    f.writelines(json_line+'\n')
                    self.line_counter+=1
            return { "save_path"  : self.save_path,
                     "total_lines": self.line_counter }
    
    def _checkpath(self, filepath):
        splitter = os.path.splitext(os.path.basename(filepath))
        body = splitter[0]
        extention = splitter[1]
        newpath = filepath
        counter = 0
        while os.path.exists(newpath):
            match = re.search(PATTERN,body)
            if match:
                counter=int(match[2])+1
                num_with_bracket = f'({str(counter)})'
                body = f'{match[1]}{num_with_bracket}'
            else:
                body = f'{body}({str(counter)})'
            newpath = os.path.join(os.path.dirname(filepath),body+extention)
        return newpath