Исправил много чего в фронте

handlers/start.py

@@ -1,62 +1,52 @@
 from aiogram import Router, types
 from aiogram.filters import Command
-from aiogram.types import Message, CallbackQuery
-import logging
-from instences.config import BASE_URL_FASTAPI
-import aiohttp
-from keyboard.keyboards import main_keyboard
 from aiogram.enums.parse_mode import ParseMode
+import logging
+import aiohttp
+
+from aiogram.types import Message, CallbackQuery
+
+from instences.config import BASE_URL_FASTAPI
+from keyboard.keyboards import main_keyboard
 from .referrals import _build_referral_text
 
 router = Router()
 logger = logging.getLogger(__name__)
 
 
-async def call_api(method, endpoint, data=None):
-    """
-    Выполняет HTTP-запрос к FastAPI.
-    :param method: HTTP метод (GET, POST, и т.д.)
-    :param endpoint: конечная точка API
-    :param data: тело запроса (если необходимо)
-    :return: JSON-ответ или "ERROR" при неуспехе
-    """
+async def call_api(method: str, endpoint: str, data: dict | None = None):
+    """Мини-обёртка для запросов к backend API."""
     url = f"{BASE_URL_FASTAPI}{endpoint}"
-    logger.info(f"Инициализация запроса: {method} {url} с данными {data}")
+    logger.info(f"API {method} {url} data={data}")
 
     try:
         async with aiohttp.ClientSession() as session:
-            async with session.request(method, url, json=data) as response:
-                logger.info(
-                    f"Получен ответ от {url}: статус {response.status}"
-                )
-
-                if response.status in {200, 201}:
-                    result = await response.json()
-                    logger.debug(f"Ответ JSON: {result}")
-                    return result
-                if response.status == 404:
-                    logger.debug(f"Код {response.status}, возвращаю ничего")
-                    return None
+            async with session.request(method, url, json=data) as resp:
+                logger.info(f"API response {resp.status} {resp.reason}")
+                if resp.status in {200, 201}:
+                    return await resp.json()
+                if resp.status in {400, 404}:
+                    try:
+                        return await resp.json()
+                    except Exception:
+                        return None
                 logger.error(
-                    f"Ошибка в запросе: статус {response.status}, причина {response.reason}"
+                    f"Unexpected status {resp.status}: {await resp.text()}"
                 )
                 return "ERROR"
     except Exception as e:
-        logger.exception(f"Исключение при выполнении запроса к {url}: {e}")
+        logger.exception(f"API exception {url}: {e}")
         return "ERROR"
 
 
 def _welcome_text(username: str | None) -> str:
-    """
-    Текст приветствия в /start и в главном меню.
-    Имя пока не используем — оставляем сигнатуру на будущее.
-    """
+    """Текст приветствия в /start и в главном меню."""
     return "🥚 Lark Security\n\nВыберите действие из меню ниже."
 
 
 def _parse_referrer_id(message: Message) -> int | None:
-    """
-    Достаём ref_<telegram_id> из /start.
+    """Достаём ref_<telegram_id> из /start.
+
     Примеры:
       /start
       /start ref_123456789
@@ -82,86 +72,117 @@ def _parse_referrer_id(message: Message) -> int | None:
 
 @router.message(Command("start"))
 async def start_command(message: Message):
-    """
-    Обработчик команды /start.
-    Добавлена обработка реферального payload'а ref_<telegram_id>.
-    """
-    logger.info(
-        f"Получена команда /start от пользователя: "
-        f"{message.from_user.id} ({message.from_user.username}) | text='{message.text}'"
-    )
+    """/start с обработкой реферального параметра.
 
+    Логика:
+    - проверяем, есть ли пользователь в БД по telegram_id;
+    - если нет — создаём через /user/create;
+    - если есть корректный ref_ и пользователь новый — вызываем
+      /user/{referrer_id}/add_referral с invited_id = telegram_id.
+    """
     user_id = message.from_user.id
+    username = message.from_user.username
     referrer_id = _parse_referrer_id(message)
 
+    logger.info(
+        f"[start] Команда /start от {user_id} (@{username}), "
+        f"text={message.text!r}, referrer_id={referrer_id}"
+    )
+
     try:
-        # 1. Проверяем/создаём пользователя в базе
-        user_data = await call_api("GET", f"/user/{user_id}")
-        if not user_data:
-            logger.debug(
-                "Пользователь не найден в базе, создаем новую запись."
-            )
-            await call_api(
+        # 1. Проверяем, есть ли пользователь в БД
+        existing = await call_api("GET", f"/user/{user_id}")
+        user_exists = existing not in (None, "ERROR")
+
+        # 2. Если пользователя нет — создаём
+        if not user_exists:
+            logger.debug(f"[start] Пользователь {user_id} не найден, создаю.")
+            create_result = await call_api(
                 "POST",
                 "/user/create",
                 {"telegram_id": user_id},
             )
-
-        # 2. Если есть реферер и он не сам пользователь — отправляем инфу в бекенд
-        if referrer_id and referrer_id != user_id:
-            payload = {
-                "referrer_id": referrer_id,
-                "telegram_id": user_id,
-            }
-            logger.info(
-                f"Отправка данных о реферале в бекенд: {payload}"
-            )
-            result = await call_api(
-                "POST",
-                "/user/referrals/track",
-                payload,
-            )
-            if result == "ERROR":
+            if create_result == "ERROR":
                 logger.error(
-                    f"Не удалось зафиксировать реферала в бекенде: {payload}"
+                    f"[start] Не удалось создать пользователя {user_id} в БД"
                 )
-        elif referrer_id == user_id:
-            logger.info(
-                "Обнаружена попытка самореферала, запрос в бекенд не отправляем."
-            )
 
-        # 3. Приветственное сообщение и главное меню
-        logger.debug("Отправка приветственного сообщения пользователю.")
+        # 3. Обработка рефералки, если параметр есть
+        if referrer_id is not None:
+            # 3.1. Самореферал
+            if referrer_id == user_id:
+                logger.info(
+                    f"[start] Пользователь {user_id} попытался зайти "
+                    f"по своей реферальной ссылке."
+                )
+                await message.answer(
+                    "Нельзя переходить по своей же реферальной ссылке."
+                )
+
+            # 3.2. Пользователь уже зарегистрирован в боте
+            elif user_exists:
+                logger.info(
+                    f"[start] Пользователь {user_id} уже есть в БД, "
+                    f"реферальная ссылка {referrer_id} не сработает."
+                )
+                await message.answer(
+                    "Вы уже зарегистрированы в боте, "
+                    "реферальная ссылка не сработает."
+                )
+
+            # 3.3. Новый пользователь + чужая рефералка → регистрируем реферал
+            else:
+                payload = {
+                    "invited_id": user_id,
+                }
+                logger.info(
+                    f"[start] Фиксирую реферала в бекенде: "
+                    f"referrer_id={referrer_id}, payload={payload}"
+                )
+                result = await call_api(
+                    "POST",
+                    f"/user/{referrer_id}/add_referral",
+                    payload,
+                )
+                if result == "ERROR":
+                    logger.error(
+                        f"[start] Ошибка при фиксации реферала через "
+                        f"/user/{referrer_id}/add_referral: referrer={referrer_id}, "
+                        f"invited={user_id}"
+                    )
+                await message.answer("Вы вошли по реферальной ссылке.")
+
+        # 4. В любом случае показываем главное меню
         await message.answer(
-            _welcome_text(message.from_user.username),
+            _welcome_text(username),
             reply_markup=main_keyboard(),
         )
-        logger.info("Приветственное сообщение отправлено.")
+        logger.info(f"[start] Главное меню отправлено пользователю {user_id}.")
+
     except Exception as e:
         logger.exception(
-            f"Ошибка при обработке команды /start для пользователя "
-            f"{user_id}: {e}"
+            f"[start] Ошибка при обработке /start для пользователя {user_id}: {e}"
         )
         await message.answer("Произошла ошибка. Попробуйте позже.")
 
 
 @router.message(Command("referrals"))
 async def referrals_menu_command(message: Message):
-    """
-    Команда /referrals из бокового меню Telegram.
+    """Команда /referrals из бокового меню Telegram.
+
     Показывает текст реферальной программы.
     """
     logger.info(
-        f"Получена команда /referrals от пользователя: "
-        f"{message.from_user.id} ({message.from_user.username})"
+        f"[start] Команда /referrals от {message.from_user.id} "
+        f"(@{message.from_user.username})"
     )
     try:
         text = await _build_referral_text(message.bot, message.from_user.id)
         await message.answer(text, parse_mode=ParseMode.HTML)
-        logger.info("Реферальная программа отправлена пользователю.")
+        logger.info("[start] Реферальная программа отправлена пользователю.")
     except Exception as e:
         logger.exception(
-            f"Ошибка при обработке команды /referrals для пользователя "
+            f"[start] Ошибка при обработке /referrals для пользователя "
             f"{message.from_user.id}: {e}"
         )
         await message.answer("Произошла ошибка. Попробуйте позже.")
@@ -169,25 +190,26 @@ async def referrals_menu_command(message: Message):
 
 @router.callback_query(lambda callback: callback.data == "base")
 async def start_callback_handler(callback: CallbackQuery):
-    """
-    Обработчик callback_query с data="base".
-    Возвращает пользователя в главное меню.
-    """
+    """Callback с data="base" — возврат в главное меню."""
     try:
-        user_data = await call_api("GET", f"/user/{callback.from_user.id}")
-        if not user_data:
+        user_id = callback.from_user.id
+        username = callback.from_user.username
+        logger.info(f"[start] callback 'base' от {user_id} (@{username})")
+
+        user_data = await call_api("GET", f"/user/{user_id}")
+        if user_data in (None, "ERROR"):
             await call_api(
                 "POST",
                 "/user/create",
-                {"telegram_id": callback.from_user.id},
+                {"telegram_id": user_id},
             )
 
         await callback.message.edit_text(
-            _welcome_text(callback.from_user.username),
+            _welcome_text(username),
             reply_markup=main_keyboard(),
         )
     except Exception as e:
-        logger.exception(f"Ошибка при обработке callback с data='base': {e}")
+        logger.exception(f"[start] Ошибка при обработке callback 'base': {e}")
         await callback.message.answer("Произошла ошибка. Попробуйте позже.")
     finally:
         await callback.answer()