import json
from pathlib import Path
from locale import getdefaultlocale


class I18nAuto:
    LANGUAGE_PATH = "./assets/i18n/languages/"

    def __init__(self, language=None):
        language = language or getdefaultlocale()[0]

        if self._language_exists(language):
            self.language = language
        else:
            lang_prefix = language[:2]
            available_languages = self._get_available_languages()
            matching_languages = [
                lang for lang in available_languages if lang.startswith(lang_prefix)
            ]

            self.language = matching_languages[0] if matching_languages else "en_US"

        self.language_map = self._load_language_list()

    def _load_language_list(self):
        try:
            file_path = Path(self.LANGUAGE_PATH) / f"{self.language}.json"
            with open(file_path, "r", encoding="utf-8") as f:
                return json.load(f)
        except FileNotFoundError:
            raise FileNotFoundError(
                f"Failed to load language file for {self.language}. Check if the correct .json file exists."
            )

    def _get_available_languages(self):
        language_files = [path.stem for path in Path(self.LANGUAGE_PATH).glob("*.json")]
        return language_files

    def _language_exists(self, language):
        return (Path(self.LANGUAGE_PATH) / f"{language}.json").exists()

    def __call__(self, key):
        return self.language_map.get(key, key)

    def print(self):
        print(f"Using Language: {self.language}")