import json
import os
from typing import TypedDict, Any

from ..constants import PROFILE_NATIVE_SKILLS_PATH, SKILL_PATH

SKILL_NAME_SUFFIX = "_skill"


def normalize_skill_name(skill_name: str) -> str:
    return (
        skill_name
        if skill_name.endswith(SKILL_NAME_SUFFIX)
        else f"{skill_name}{SKILL_NAME_SUFFIX}"
    )


class MemoryOptions(TypedDict, total=False):
    name: str
    default_memory: Any


class Memory:
    def __init__(self, options: MemoryOptions):
        self.name = options['name']
        self.default_memory = options['default_memory'] if 'default_memory' in options else None
        self.memory_path = os.path.join(
            PROFILE_NATIVE_SKILLS_PATH,
            os.path.basename(SKILL_PATH),
            'memory',
            f'{self.name}.json'
        )
        self.__is_from_another_skill = False

        if ':' in self.name and self.name.count(':') == 2:
            self.__is_from_another_skill = True
            _, skill_name, memory_name = self.name.split(':')
            self.memory_path = os.path.join(
                PROFILE_NATIVE_SKILLS_PATH,
                normalize_skill_name(skill_name),
                'memory',
                memory_name + '.json'
            )

    def clear(self) -> None:
        """
        Clear the memory and set it to the default memory value
        """
        if not self.__is_from_another_skill:
            self.write(self.default_memory)
        else:
            raise ValueError(f'You cannot clear the memory "{self.name}" as it belongs to another skill')

    def read(self):
        """
        Read the memory
        """
        if self.__is_from_another_skill and not os.path.exists(self.memory_path):
            raise ValueError(f'You cannot read the memory "{self.name}" as it belongs to another skill which hasn\'t written to this memory yet')

        try:
            if not os.path.exists(self.memory_path):
                self.clear()

            with open(self.memory_path, 'r') as f:
                return json.load(f)
        except Exception as e:
            print(f'Error while reading memory for "{self.name}": {e}')
            raise e

    def write(self, memory):
        """
        Write the memory
        :param memory: The memory to write
        """
        if not self.__is_from_another_skill:
            try:
                os.makedirs(os.path.dirname(self.memory_path), exist_ok=True)
                with open(self.memory_path, 'w') as f:
                    json.dump(memory, f, indent=2)

                return memory
            except Exception as e:
                print(f'Error while writing memory for "{self.name}": {e}')
                raise e
        else:
            raise ValueError(f'You cannot write into the memory "{self.name}" as it belongs to another skill')