Initial commit

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

bot/middleware.py

@@ -0,0 +1,102 @@
+"""Мидлварь для проверки доступа пользователей к боту."""
+
+import logging
+from aiogram import BaseMiddleware
+from aiogram.types import Message, CallbackQuery
+from aiogram.filters import Command
+from database.database import db
+
+logger = logging.getLogger(__name__)
+
+
+class AccessCheckMiddleware(BaseMiddleware):
+    """Мидлварь проверяет, имеет ли пользователь доступ к боту."""
+
+    # Команды, которые всегда разрешены (без проверки доступа)
+    ALLOWED_COMMANDS = {"start", "menu"}
+
+    async def __call__(self, handler, event, data):
+        # Определяем тип события и извлекаем user_id
+        user_id = None
+        username = None
+        display_name = None
+
+        if isinstance(event, Message):
+            user_id = event.from_user.id
+            username = event.from_user.username
+            display_name = event.from_user.full_name
+            # Проверяем, является ли это разрешённой командой
+            if isinstance(event, Message) and event.text:
+                text = event.text.strip().lower()
+                for cmd in self.ALLOWED_COMMANDS:
+                    # Проверяем /cmd и /cmd@username
+                    if text == f"/{cmd}" or text.startswith(f"/{cmd}@"):
+                        return await handler(event, data)
+
+        elif isinstance(event, CallbackQuery):
+            user_id = event.from_user.id
+            username = event.from_user.username
+            display_name = event.from_user.full_name
+
+        if user_id is None:
+            return await handler(event, data)
+
+        # Проверяем доступ
+        access_info = await db.check_user_access(user_id)
+
+        if access_info["has_access"]:
+            # Сохраняем информацию о пользователе в data для обработчиков
+            data["user_access"] = access_info
+            return await handler(event, data)
+
+        # Доступ запрещён — отправляем сообщение
+        reason = access_info["reason"]
+        user_info = access_info["user_info"]
+
+        # Формируем сообщение в зависимости от причины
+        if reason == "not_registered":
+            text = (
+                "🔒 <b>Доступ запрещён</b>\n\n"
+                "У вас нет доступа к этому боту. "
+                "Обратитесь к администратору для получения доступа."
+            )
+        elif reason == "deactivated":
+            text = (
+                "🔒 <b>Доступ заблокирован</b>\n\n"
+                "Ваш доступ к боту был отозван администратором."
+            )
+        elif reason == "expired":
+            expires_str = "не указан"
+            if user_info and user_info.get("access_expires_at"):
+                from datetime import datetime
+                try:
+                    exp_date = datetime.fromisoformat(user_info["access_expires_at"])
+                    expires_str = exp_date.strftime("%d.%m.%Y %H:%M")
+                except Exception:
+                    pass
+            text = (
+                "⏰ <b>Срок доступа истёк</b>\n\n"
+                f"Ваш доступ к боту истёк ({expires_str}).\n"
+                "Обратитесь к администратору для продления."
+            )
+        elif reason == "generations_limit":
+            used = user_info.get("used_generations", "?") if user_info else "?"
+            max_gens = user_info.get("max_generations", "?") if user_info else "?"
+            text = (
+                "🎨 <b>Лимит генераций исчерпан</b>\n\n"
+                f"Вы использовали все {max_gens} генераций.\n"
+                "Обратитесь к администратору для увеличения лимита."
+            )
+        else:
+            text = "🔒 <b>Доступ запрещён</b>"
+
+        if isinstance(event, Message):
+            await event.answer(text, parse_mode="HTML")
+        elif isinstance(event, CallbackQuery):
+            await event.answer("Доступ запрещён", show_alert=True)
+            try:
+                await event.message.answer(text, parse_mode="HTML")
+            except Exception:
+                pass
+
+        return None