First commit

bot/middlewares/__init__.py

@@ -0,0 +1,55 @@
+from aiogram import Dispatcher, Bot
+
+from configs import ImportantID
+from .error_mdw import ErrorHandlingMiddleware
+from .logging_mdw import LoggingMiddleware
+from .msg_mdw import MessageCounterMiddleware
+from .referal_mdw import ReferralMiddleware
+from .spam_mdw import RateLimitMiddleware
+from .subscription_mdw import SubscriptionMiddleware
+from .time_mdw import TimingMiddleware
+from .ban_user_mdw import BanCheckMiddleware
+
+# Настройки экспорта
+__all__ = (
+    "LoggingMiddleware",
+    "SubscriptionMiddleware",
+    "RateLimitMiddleware",
+    "ErrorHandlingMiddleware",
+    "TimingMiddleware",
+    "MessageCounterMiddleware",
+    "setup_middlewares",
+    "ReferralMiddleware",
+    "BanCheckMiddleware",
+)
+
+
+def setup_middlewares(dp: Dispatcher, bot: Bot, channel_ids: list[int | str] = None) -> None:
+    """
+    Регистрирует все middleware в диспетчере.
+    """
+    channel_ids: list = channel_ids or []
+
+    # Middleware для ВСЕХ событий (update level)
+    middlewares_updates: list = [
+        TimingMiddleware(),  # Замер времени
+        LoggingMiddleware(),  # Логирование
+        ErrorHandlingMiddleware(admin_ids=ImportantID.ADMIN_ID),  # Обработка ошибок
+    ]
+
+    # Middleware только для СООБЩЕНИЙ (message level)
+    middlewares_msg: list = [
+        BanCheckMiddleware(),
+        # RateLimitMiddleware(rate_limit=3, time_period=5.0),  # Антифлуд
+        # SubscriptionMiddleware(bot=bot, channel_ids=channel_ids),  # Проверка подписки
+        MessageCounterMiddleware(),  # Подсчет сообщений
+        ReferralMiddleware(),  # Проверка реф-ссылок
+    ]
+
+    # Регистрируем middleware для всех событий
+    for middleware in middlewares_updates:
+        dp.update.middleware(middleware)
+
+    # Регистрируем middleware только для сообщений
+    for middleware in middlewares_msg:
+        dp.message.middleware(middleware)