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

from typing import Any, Awaitable, Callable, Dict

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)