i18n.py

# i18n.py
import os
import json
from typing import Dict
from PyQt5 import QtCore

from constants import BASE_DIR, KEY_UI_LANGUAGE
from utils import get_settings


LANG_DIR = os.path.join(BASE_DIR, "lang")


class TranslationManager:
    """
    Простой менеджер переводов:
    - грузит lang/ru.json и lang/en.json (если есть);
    - хранит текущий язык в QSettings (ключ ui/language);
    - даёт метод translate(key) и глобальную функцию _().
    """
    def __init__(self) -> None:
        # Базовые встроенные fallback-переводы для runtime/UI-критичных строк.
        # JSON-файлы по-прежнему имеют приоритет и мёрджатся поверх этих значений.
        self._translations: Dict[str, Dict[str, str]] = {
            "ru": {
                "app_title": "XTimer",
                "settings_title": "Настройки XTimer",
                "dlg_sound_read_error": "Не удалось прочитать выбранный файл.",
                "dlg_sound_too_large": "Файл слишком большой ({size_mb:.1f} MB). Максимум: {max_mb} MB.",
                "dlg_sound_too_long": "Файл слишком длинный ({duration_s:.1f} s). Максимум: {max_s} s.",
                "update_status_available": "Найдена версия {tag}",
                "update_status_uptodate": "У вас уже последняя версия.",
                "update_status_failed": "Ошибка проверки обновления: {stage}",
                "tray_update_available_title": "XTimer",
                "tray_update_available_message": "Новая версия {tag} доступна. Нажмите уведомление, чтобы открыть релиз.",
                "tray_update_uptodate_title": "XTimer",
                "tray_update_uptodate_message": "Вы используете последнюю версию.",
                "app_already_running": "XTimer уже запущен.",
                "startup_failed": "XTimer не удалось запустить. Подробности смотрите в логе.",
            },
            "en": {
                "app_title": "XTimer",
                "settings_title": "XTimer Settings",
                "dlg_sound_read_error": "Cannot read selected file.",
                "dlg_sound_too_large": "File is too large ({size_mb:.1f} MB). Max allowed: {max_mb} MB.",
                "dlg_sound_too_long": "File is too long ({duration_s:.1f} s). Max allowed: {max_s} s.",
                "update_status_available": "Version {tag} is available",
                "update_status_uptodate": "You already have the latest version.",
                "update_status_failed": "Update check failed: {stage}",
                "tray_update_available_title": "XTimer",
                "tray_update_available_message": "A new version {tag} is available. Click the notification to open the release page.",
                "tray_update_uptodate_title": "XTimer",
                "tray_update_uptodate_message": "You are using the latest version.",
                "app_already_running": "XTimer is already running.",
                "startup_failed": "XTimer failed to start. See the log for details.",
            },
        }
        self._default_lang = "en"   # первый запуск
        self._fallback_lang = "en"  # страховка переводов

        # Подгружаем JSON-файлы, если существуют
        self._load_json("ru")
        self._load_json("en")

        # Определяем текущий язык из настроек
        settings = get_settings()
        lang_code = settings.value(KEY_UI_LANGUAGE, self._default_lang, type=str)
        if lang_code not in self._translations:
            lang_code = self._default_lang
        self._current_lang = lang_code

    def _load_json(self, code: str) -> None:
        """
        Загружает lang/<code>.json и мёрджит в словарь переводов.
        Если файла нет или JSON битый — тихо игнорируем.
        """
        path = os.path.join(LANG_DIR, f"{code}.json")
        if not os.path.isfile(path):
            return

        try:
            with open(path, "r", encoding="utf-8") as f:
                data = json.load(f)
            if isinstance(data, dict):
                # все ключи/значения приводим к строкам
                self._translations[code].update(
                    {str(k): str(v) for k, v in data.items()}
                )
        except Exception:
            # никаких падений из-за переводов
            pass

    def set_language(self, code: str) -> None:
        """
        Смена языка: 'ru' или 'en'. Сохраняем в QSettings.
        """
        if code not in self._translations:
            code = self._default_lang

        self._current_lang = code
        settings = get_settings()
        settings.setValue(KEY_UI_LANGUAGE, code)

    def get_language(self) -> str:
        return self._current_lang
    
    def format(self, key: str, **kwargs) -> str:
        """
        Возвращает перевод и безопасно форматирует его через str.format(**kwargs).
        Если форматирование не удалось, возвращает сырой template без падения.
        """
        template = self.translate(key)
        try:
            return template.format(**kwargs)
        except Exception:
            return template
    
    def translate(self, key: str) -> str:
        """
        Вернуть перевод по ключу:
        1) текущий язык
        2) fallback (en)
        3) сам key, если ничего не найдено
        """
        lang = self._current_lang

        if key in self._translations.get(lang, {}):
            return self._translations[lang][key]

        if key in self._translations.get(self._fallback_lang, {}):
            return self._translations[self._fallback_lang][key]

        return key


# Глобальный экземпляр менеджера
_manager = TranslationManager()


def set_language(code: str) -> None:
    _manager.set_language(code)


def get_language() -> str:
    return _manager.get_language()

def tr(key: str, **kwargs) -> str:
    return _manager.format(key, **kwargs)

def _(key: str) -> str:
    """
    Глобальная функция-переводчик:
        _("settings_title") -> "Настройки XTimer" / "XTimer Settings"
    """
    return _manager.translate(key)