import json
import logging
import os
import typing as t

from app.classes.shared.console import console
from app.classes.shared.helpers import helper

logger = logging.getLogger(__name__)

class Translation:
    def __init__(self):
        self.translations_path = os.path.join(helper.root_dir, 'app', 'translations')
        self.cached_translation = None
        self.cached_translation_lang = None

    def get_language_file(self, language: str):
        return os.path.join(self.translations_path, str(language) + '.json')

    def translate(self, page, word, language):
        fallback_language = 'en_EN'

        translated_word = self.translate_inner(page, word, language)
        if translated_word is None:
            translated_word = self.translate_inner(page, word, fallback_language)

        if translated_word:
            if isinstance(translated_word, dict):
                # JSON objects
                return json.dumps(translated_word)
            elif isinstance(translated_word, str):
                # Basic strings
                return translated_word
            elif hasattr(translated_word, '__iter__'):
                # Multiline strings
                return '\n'.join(translated_word)
        return 'Error while getting translation'

    def translate_inner(self, page, word, language) -> t.Union[t.Any, None]:
        language_file = self.get_language_file(language)
        try:
            if not self.cached_translation:
                with open(language_file, 'r', encoding='utf-8') as f:
                    data = json.load(f)
                    self.cached_translation = data
            elif self.cached_translation_lang != language:
                with open(language_file, 'r', encoding='utf-8') as f:
                    data = json.load(f)
                    self.cached_translation = data
                    self.cached_translation_lang = language
            else:
                data = self.cached_translation

            try:
                translated_page = data[page]
            except KeyError:
                logger.error(f'Translation File Error: page {page} does not exist for lang {language}')
                console.error(f'Translation File Error: page {page} does not exist for lang {language}')
                return None

            try:
                translated_word = translated_page[word]
                return translated_word
            except KeyError:
                logger.error(f'Translation File Error: word {word} does not exist on page {page} for lang {language}')
                console.error(f'Translation File Error: word {word} does not exist on page {page} for lang {language}')
                return None

        except Exception as e:
            logger.critical(f'Translation File Error: Unable to read {language_file} due to {e}')
            console.critical(f'Translation File Error: Unable to read {language_file} due to {e}')
            return None


translation = Translation()