Add referral system and user balance tracking

Introduce balance field to User model with Alembic migration. Add
referral menu, admin /refpay command, and deep link sharing. Update
Dockerfile to multi-stage build and add Postgres service to compose.
Fix deep link parsing logic and conditional proxy initialization.

handlers/__init__.py

@@ -1,5 +1,6 @@
+from .referals import router as referal_router
 from .menus import router as menus_router
 from .support import router as support_router
 from .buy import router as buy_router
 
-routers = [menus_router, support_router, buy_router]
+routers = [referal_router, menus_router, support_router, buy_router]

handlers/menus.py

@@ -12,7 +12,7 @@ from texts import GENERAL_PROCESSING, build_main_menu_text, FAQ_TEXT
 router = Router()
 
 
-@router.message(CommandStart(deep_link=True, deep_link_encoded=False))
+@router.message(CommandStart(deep_link=True, deep_link_encoded=True))
 async def fetch_referal(
     msg: Message,
     command: CommandObject,
@@ -22,12 +22,12 @@ async def fetch_referal(
 ):
     await state.clear()
 
-    data = command.args.split()
+    data = command.args.split("_")
     if not data:
         await standard_start(msg, command, state)
         return
 
-    referal = data[0]
+    referal = int(data[1])
     user = await deps.user_service.add_user(
         session, user_id=msg.from_user.id, referal=referal, rw_sdk=deps.rw_sdk
     )
@@ -41,7 +41,6 @@ async def fetch_referal(
 @router.message(Command(commands=["start", "cancel"]))
 async def standard_start(
     msg: Message,
-    command: CommandObject,
     state: FSMContext,
     session: AsyncSession,
     deps: DependenciesDTO,

handlers/referals.py

@@ -0,0 +1,63 @@
+from aiogram import Router, F
+from aiogram.filters import Command, CommandObject
+from aiogram.utils.deep_linking import create_start_link
+from aiogram.types import CallbackQuery, Message
+from aiogram.fsm.context import FSMContext
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from keyboards.client import referals_kb
+from misc.utils import format_share_deep_link
+from schemas.di import DependenciesDTO
+from texts import GENERAL_PROCESSING, REFERALS_MENU
+from config import settings
+
+router = Router()
+
+
+@router.callback_query(F.data == "referal")
+async def referal_menu(
+    cb: CallbackQuery, state: FSMContext, session: AsyncSession, deps: DependenciesDTO
+):
+    await state.clear()
+
+    user = await deps.user_repository.get_user_by_id(session, cb.from_user.id)
+    referal_link = await create_start_link(cb.bot, f"ref_{user.id}", encode=True)
+    share_link = format_share_deep_link(referal_link)
+
+    await cb.message.edit_text(
+        REFERALS_MENU.format(balance=user.balance, link=referal_link),
+        reply_markup=referals_kb(share_link),
+    )
+
+
+@router.message(F.chat.id == settings.admin_group_id, Command("refpay"))
+async def refpay(
+    msg: Message,
+    command: CommandObject,
+    state: FSMContext,
+    session: AsyncSession,
+    deps: DependenciesDTO,
+):
+    await state.clear()
+
+    await msg.delete()
+    status_msg = await msg.answer(GENERAL_PROCESSING)
+    user = await deps.user_service.fetch_user_by_thread(session, msg.message_thread_id)
+    referal = await deps.user_repository.get_user_by_id(session, user.referal_id)
+    if not referal:
+        await status_msg.edit_text("❌ У пользователя нет реферала.")
+        return
+
+    amount = command.args
+
+    if not amount.isdigit():
+        await status_msg.edit_text(
+            "❌ Укажите в аргументах количество средств, которые необходимо перевести рефералу пользователя."
+        )
+        return
+
+    amount = int(amount)
+    await deps.user_repository.increase_user_balance(session, user.id, amount)
+    await status_msg.edit_text(
+        f"<b>💸 Баланс реферала ({referal.id}) пополнен на {amount}₽</b>"
+    )

handlers/support.py

@@ -51,6 +51,7 @@ async def received_message(
     F.message_thread_id.is_not(None),
     F.from_user.is_bot == False,  # noqa: E712
     ~Command("close"),
+    ~Command("refpay"),
     F.text,
 )
 async def handle_admin_message(