# - *- coding: utf- 8 - *-
from aiogram import BaseMiddleware
from cachetools import TTLCache

from tgbot.data.config import BOT_USER_CACHE_TTL
from tgbot.database import Referralx, Settingsx, Userx
from tgbot.utils.const_functions import clear_html
from tgbot.utils.misc.bot_logging import bot_logger


class ExistsUserMiddleware(BaseMiddleware):
    # Создание кеша пользователей
    def __init__(self, cache_ttl: int = BOT_USER_CACHE_TTL) -> None:
        self.users = Userx()
        self.referrals = Referralx()
        self.cache = TTLCache(maxsize=10_000, ttl=cache_ttl)

    @staticmethod
    def _get_referrer_id(event) -> int | None:
        event_text = getattr(event, "text", None)

        if not isinstance(event_text, str):
            return None

        start_data = event_text.split(maxsplit=1)

        if len(start_data) != 2 or start_data[0] != "/start":
            return None

        referral_data = start_data[1]

        if not referral_data.startswith("r_") or not referral_data[2:].isdigit():
            return None

        return int(referral_data[2:])

    # Добавление или обновление пользователя перед handler-ом
    async def __call__(self, handler, event, data):
        this_user = data.get("event_from_user")

        if this_user is not None and not this_user.is_bot:
            user_id = this_user.id
            user_login = (this_user.username or "").lower()
            user_name = clear_html(this_user.first_name) or ""
            user_surname = clear_html(this_user.last_name) or ""
            user_fullname = clear_html(this_user.full_name) or " ".join(filter(None, [user_name, user_surname]))

            user_data = (
                user_login,
                user_name,
                user_surname,
                user_fullname,
            )

            cached_user = self.cache.get(user_id)
            existing_user = None

            if cached_user is None:
                existing_user = await self.users.get(user_id=user_id)

            is_new_user = cached_user is None and existing_user is None

            if cached_user is None or cached_user["data"] != user_data:
                user = await self.users.upsert(
                    user_id=user_id,
                    user_login=user_data[0],
                    user_name=user_data[1],
                    user_surname=user_data[2],
                    user_fullname=user_data[3],
                )
            else:
                user = cached_user["user"]

            referrer_id = self._get_referrer_id(event)

            if is_new_user and referrer_id is not None:
                registration_status = await self.referrals.register_referral(
                    referral_id=user_id,
                    referrer_id=referrer_id,
                )

                if registration_status == "ok":
                    user = await self.users.get_required(user_id=user_id)
                    get_settings = await Settingsx().get()
                    bot = data.get("bot") or getattr(event, "bot", None)

                    if bot is not None:
                        notifications = (
                            (
                                referrer_id,
                                "<b>🤝 Новый реферал</b>\n\n"
                                f"▪️ Пользователь: <a href='tg://user?id={user_id}'>{user.user_name}</a>\n"
                                f"▪️ Начислено: <code>{get_settings.referral_bonus_rub}₽</code>",
                            ),
                            (
                                user_id,
                                "<b>🤝 Реферальный бонус</b>\n\n"
                                f"▪️ Вам начислено: <code>{get_settings.referral_bonus_rub}₽</code>",
                            ),
                        )

                        for recipient_id, notification_text in notifications:
                            try:
                                await bot.send_message(recipient_id, notification_text)
                            except Exception:
                                bot_logger.warning(
                                    "Не удалось отправить уведомление о реферальной регистрации",
                                    exc_info=True,
                                )

            self.cache[user_id] = {"data": user_data, "user": user}

            data["User"] = user

        return await handler(event, data)