from collections.abc import Awaitable, Callable
from typing import Any, Dict, Optional, TypedDict, Union

from aiogram import BaseMiddleware
from aiogram.types import CallbackQuery, InlineQuery, Message, TelegramObject, User

from logger import logger


class UserInfo(TypedDict):
    user_id: int | None
    username: str | None
    action: str | None


class LoggingMiddleware(BaseMiddleware):
    """Middleware для логирования действий пользователя."""

    async def __call__(
        self,
        handler: Callable[[TelegramObject, dict[str, Any]], Awaitable[Any]],
        event: TelegramObject,
        data: dict[str, Any],
    ) -> Any:
        user_info = self._extract_user_info(event)

        if user_info["user_id"]:
            logger.info(
                f"Активность пользователя - "
                f"ID пользователя: {user_info['user_id']}, "
                f"Имя пользователя: {user_info['username'] or 'Не указано'}, "
                f"Действие: {user_info['action'] or 'Неизвестно'}"
            )

        return await handler(event, data)

    def _extract_user_info(self, event: TelegramObject) -> UserInfo:
        """Извлекает информацию о пользователе из различных типов событий.

        Args:
            event: Событие Telegram

        Returns:
            Словарь с информацией о пользователе
        """
        result: UserInfo = {"user_id": None, "username": None, "action": None}

        # Обработка общих атрибутов для разных типов событий
        if hasattr(event, "from_user") and isinstance(event.from_user, User):
            result["user_id"] = event.from_user.id
            result["username"] = event.from_user.username

            # Определение типа действия в зависимости от типа события
            if isinstance(event, Message):
                result["action"] = f"Сообщение: {event.text}"
            elif isinstance(event, CallbackQuery):
                result["action"] = f"Обратный вызов: {event.data}"
            elif isinstance(event, InlineQuery):
                result["action"] = f"Inline запрос: {event.query}"

        return result