from fastapi import Request, Depends
from babel.support import Translations
import logging

logger = logging.getLogger(__name__)


def get_translations(domain: str = "api_v1") -> callable:
    """
    Создаёт зависимость для загрузки переводов на основе заголовка Accept-Language
    или параметра запроса 'locale'. По умолчанию используется русский язык.

    Args:
        domain (str): Домен переводов (например, 'api_v1', 'services'). По умолчанию 'api_v1'.

    Returns:
        callable: Функция, возвращающая объект Translations для указанной локали и домена.
    """

    def translations_dependency(request: Request) -> Translations:
        # Проверяем параметр запроса 'locale', если он есть
        locale = request.query_params.get("locale")
        if not locale:
            # Если параметр отсутствует, используем заголовок Accept-Language
            locale = request.headers.get("Accept-Language", "ru").split(",")[0].split("-")[0]

        try:
            translations = Translations.load("locales", [locale], domain)
            if not translations._catalog:
                logger.warning(f"Translations for locale {locale} and domain {domain} not found, falling back to ru")
                translations = Translations.load("locales", ["ru"], domain)
            logger.debug(f"Loaded translations for locale {locale} and domain {domain}")
            return translations
        except Exception as e:
            logger.error(f"Failed to load translations for locale {locale} and domain {domain}: {str(e)}")
            # Fallback на русский
            return Translations.load("locales", ["ru"], domain)

    return translations_dependency


def _(msgid: str, domain: str = "api_v1", request: Request = None) -> str:
    """
    Переводит строку msgid для указанного домена и текущей локали.

    Args:
        msgid (str): Строка для перевода.
        domain (str): Домен переводов (например, 'api_v1', 'services'). По умолчанию 'api_v1'.
        request (Request, optional): Объект запроса FastAPI для получения локали.

    Returns:
        str: Переведенная строка.
    """
    if request is None:
        # Язык, который будет отображаться по умолчанию
        translations = Translations.load("locales", ["ru"], domain)
    else:
        translations = get_translations(domain)(request)
    return translations.gettext(msgid)