from typing import Any, Awaitable, Callable, Dict

from aiogram.dispatcher.middlewares.base import BaseMiddleware
from aiogram.types import TelegramObject

from bot.modules.settings import UserSettings


class SettingsInjectorMiddleware(BaseMiddleware):
    async def __call__(
        self,
        handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]],
        event: TelegramObject,
        data: Dict[str, Any],
    ):
        if not hasattr(event, "from_user") and (
            not hasattr(event, "inline_query") or event.inline_query is None
        ):
            return await handler(event, data)
        elif hasattr(event, "inline_query") and event.inline_query is not None:
            settings = UserSettings(event.inline_query.from_user.id)
            data["settings"] = settings
        else:
            settings = UserSettings(event.from_user.id)
            data["settings"] = settings

        return await handler(event, data)