feat(referral): add referral system with withdrawals

tgbot/middlewares/middleware_user.py

@@ -3,16 +3,37 @@ from aiogram import BaseMiddleware
 from cachetools import TTLCache
 
 from tgbot.data.config import BOT_USER_CACHE_TTL
-from tgbot.database import Userx
+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")
@@ -32,6 +53,12 @@ class ExistsUserMiddleware(BaseMiddleware):
             )
 
             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(
@@ -41,10 +68,48 @@ class ExistsUserMiddleware(BaseMiddleware):
                     user_surname=user_data[2],
                     user_fullname=user_data[3],
                 )
-                self.cache[user_id] = {"data": user_data, "user": user}
             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)