from aiogram import Bot, Dispatcher
from aiogram.client.default import DefaultBotProperties
from aiogram.utils.callback_answer import CallbackAnswerMiddleware
from aiogram.utils.i18n import I18n
from fastapi import FastAPI
from fastapi.applications import Lifespan, AppType
from logging import getLogger
from secrets import token_hex

from .config import Config
from .fsm.db_storage import DbStorage
from .middleware.telegram import AuthMiddleware, I18nMiddleware, ResetStateMiddleware
from .model.user import UserBase
from .model.entity_metadata import EntityMetadata


class QBotApp(FastAPI):

    bot: Bot
    dp: Dispatcher
    config: Config

    logger = getLogger(__name__)

    def __init__[UserType: UserBase](self,
                user_class: type[UserType] | None = None,
                config: Config | None = None,
                lifespan: Lifespan[AppType] | None = None,
                *args, 
                **kwargs):

        if config is None:
            config = Config()

        if user_class is None:
            from .model.default_user import DefaultUser
            user_class = DefaultUser

        self.user_class = user_class
        self.entity_metadata: EntityMetadata = user_class.entity_metadata
        self.config = config
        self.lifespan = lifespan
        self.bot = Bot(token = self.config.TELEGRAM_BOT_TOKEN, default = DefaultBotProperties(parse_mode = "HTML"))
        
        dp = Dispatcher(storage = DbStorage())

        i18n = I18n(path = "locales", default_locale = "en", domain = "messages")
        i18n_middleware = I18nMiddleware(user_class = user_class, i18n = i18n)
        i18n_middleware.setup(dp)
        # dp.callback_query.middleware(ResetStateMiddleware())
        dp.callback_query.middleware(CallbackAnswerMiddleware())

        from .bot.handlers.start import router as start_router
        dp.include_router(start_router)

        from .bot.handlers.menu.main import router as main_menu_router
        auth = AuthMiddleware(user_class = user_class)
        main_menu_router.message.middleware.register(auth)
        main_menu_router.callback_query.middleware.register(auth)
        dp.include_router(main_menu_router)

        self.dp = dp

        self.bot_auth_token = token_hex(128)

        from .lifespan import default_lifespan

        super().__init__(lifespan = default_lifespan, *args, **kwargs)

        from .api_route.telegram import router as telegram_router
        self.include_router(telegram_router, prefix = "/api/telegram", tags = ["telegram"])