# - *- coding: utf- 8 - *-
import time
from typing import Any, Awaitable, Callable, Dict, Union

from aiogram import BaseMiddleware
from aiogram.dispatcher.flags import get_flag
from aiogram.types import Message, User
from cachetools import TTLCache


# Антиспам
class ThrottlingMiddleware(BaseMiddleware):
    def __init__(self, default_rate: Union[int, float] = 1) -> None:
        self.default_rate = default_rate

        self.users = TTLCache(maxsize=10_000, ttl=600)

    async def __call__(self, handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]], event: Message, data):
        this_user: User = data.get("event_from_user")

        if get_flag(data, "rate") is not None:
            self.default_rate = get_flag(data, "rate")

        if self.default_rate == 0:
            return await handler(event, data)

        if this_user.id not in self.users:
            self.users[this_user.id] = {
                'last_throttled': int(time.time()),
                'count_throttled': 0,
                'now_rate': self.default_rate,
            }

            return await handler(event, data)
        else:
            if int(time.time()) - self.users[this_user.id]['last_throttled'] >= self.users[this_user.id]['now_rate']:
                self.users.pop(this_user.id)

                return await handler(event, data)
            else:
                self.users[this_user.id]['last_throttled'] = int(time.time())

                if self.users[this_user.id]['count_throttled'] == 0:
                    self.users[this_user.id]['count_throttled'] = 1
                    self.users[this_user.id]['now_rate'] = self.default_rate + 2

                    return await handler(event, data)
                elif self.users[this_user.id]['count_throttled'] == 1:
                    self.users[this_user.id]['count_throttled'] = 2
                    self.users[this_user.id]['now_rate'] = self.default_rate + 3

                    await event.reply(
                        "<b>❗ Пожалуйста, не спамьте.\n"
                        "❗ Please, do not spam.</b>",
                    )
                elif self.users[this_user.id]['count_throttled'] == 2:
                    self.users[this_user.id]['count_throttled'] = 3
                    self.users[this_user.id]['now_rate'] = self.default_rate + 5

                    await event.reply(
                        "<b>❗ Бот не будет отвечать до прекращения спама.\n"
                        "❗ The bot will not respond until the spam stops.</b>",
                    )