12 files changed, 248 insertions, 15 deletions

diff --git a/alembic/versions/1627487324fd_init_database.py b/alembic/versions/c0c4d0fbcee2_init_database.py
index f25277f..8f61846 100644
--- a/alembic/versions/1627487324fd_init_database.py
+++ b/alembic/versions/c0c4d0fbcee2_init_database.py
@@ -1,8 +1,8 @@
 """init database
 
-Revision ID: 1627487324fd
+Revision ID: c0c4d0fbcee2
 Revises:
-Create Date: 2026-03-23 18:33:08.493629
+Create Date: 2026-03-23 21:19:28.195907
 
 """
 
@@ -13,7 +13,7 @@ import sqlalchemy as sa
 from alembic import op
 
 # revision identifiers, used by Alembic.
-revision: str = "1627487324fd"
+revision: str = "c0c4d0fbcee2"
 down_revision: Union[str, Sequence[str], None] = None
 branch_labels: Union[str, Sequence[str], None] = None
 depends_on: Union[str, Sequence[str], None] = None
@@ -49,7 +49,12 @@ def upgrade() -> None:
         sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
         sa.Column("user_id", sa.Integer(), nullable=False),
         sa.Column("invoice_id", sa.Integer(), nullable=False),
-        sa.Column("receipt_file_id", sa.String(), nullable=False),
+        sa.Column("receipt_file", sa.JSON(), nullable=False),
+        sa.Column(
+            "status",
+            sa.Enum("PENDING", "ACCEPTED", "REJECTED", name="paymentstatus"),
+            nullable=False,
+        ),
         sa.Column("datetime", sa.DateTime(), nullable=False),
         sa.ForeignKeyConstraint(
             ["invoice_id"], ["invoice.id"], name=op.f("fk_payment_invoice_id_invoice")
diff --git a/codebook.toml b/codebook.toml
index a599a2c..e1cb340 100644
--- a/codebook.toml
+++ b/codebook.toml
@@ -3,10 +3,12 @@ words = [
     "aiohttp",
     "aiosqlite",
     "asyncio",
+    "clb",
     "isort",
     "pycache",
     "pydantic",
     "pyright",
     "tablename",
     "venv",
+    "скриншот",
 ]
diff --git a/handlers/admin/add_user.py b/handlers/admin/add_user.py
index c58af2b..1d19834 100644
--- a/handlers/admin/add_user.py
+++ b/handlers/admin/add_user.py
@@ -95,5 +95,5 @@ async def set_vpn_link(
     session.add(User(id=data.user_id, vpn_link=msg.text, datetime=datetime.now(UTC)))
     await session.flush()
 
-    await msg.answer("Пользователь добавлен.")
+    await msg.answer("Пользователь добавлен.", reply_markup=ReplyKeyboardRemove())
     await state.clear()
diff --git a/handlers/user/__init__.py b/handlers/user/__init__.py
index ac4c0a3..4b43427 100644
--- a/handlers/user/__init__.py
+++ b/handlers/user/__init__.py
@@ -3,10 +3,12 @@ from aiogram import Router
 # isort: off
 from . import info
 from . import vpn_link
+from . import pay_invoice
 # isort: on
 
 router = Router(name="user")
 router.include_routers(
     info.router,
     vpn_link.router,
+    pay_invoice.router,
 )
diff --git a/handlers/user/pay_invoice.py b/handlers/user/pay_invoice.py
new file mode 100644
index 0000000..98f80a6
--- /dev/null
+++ b/handlers/user/pay_invoice.py
@@ -0,0 +1,160 @@
+from datetime import UTC, datetime
+
+from aiogram import Bot, F, Router
+from aiogram.enums import ButtonStyle
+from aiogram.exceptions import TelegramAPIError
+from aiogram.fsm.context import FSMContext
+from aiogram.fsm.state import State, StatesGroup
+from aiogram.types import (
+    CallbackQuery,
+    InlineKeyboardButton,
+    InlineKeyboardMarkup,
+    KeyboardButton,
+    Message,
+    ReplyKeyboardMarkup,
+    ReplyKeyboardRemove,
+)
+from pydantic import BaseModel
+from sqlalchemy import and_, select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from libs.fsm import get_data, set_data
+from libs.user import mention
+from models import Payment, PaymentStatus, ReceiptFile, ReceiptFileType, User, UserRole
+from models.callback_data import PayInvoiceClb, PaymentStatusClb
+
+router = Router(name="pay_invoice")
+
+
+class PayInvoiceStates(StatesGroup):
+    receipt = State()
+
+
+class PayInvoiceData(BaseModel):
+    invoice_id: int
+
+
+CANCEL_BUTTON = "Отмена оплаты"
+
+
+@router.callback_query(PayInvoiceClb.filter())
+async def button(
+    clb: CallbackQuery,
+    bot: Bot,
+    state: FSMContext,
+    callback_data: PayInvoiceClb,
+    session: AsyncSession,
+) -> None:
+    payment = await session.scalar(
+        select(Payment).where(
+            and_(
+                Payment.user_id == clb.from_user.id,
+                Payment.invoice_id == callback_data.invoice_id,
+                Payment.status != PaymentStatus.REJECTED,
+            )
+        )
+    )
+
+    if payment is not None:
+        await clb.answer(
+            "Вы уже оплатили данный счёт.",
+            show_alert=True,
+        )
+        return
+
+    await bot.send_message(
+        clb.from_user.id,
+        "Укажите подтверждение оплаты (скриншот, pdf чека и т.п.)",
+        reply_markup=ReplyKeyboardMarkup(
+            keyboard=[[KeyboardButton(text=CANCEL_BUTTON, style=ButtonStyle.DANGER)]],
+            resize_keyboard=True,
+        ),
+    )
+
+    await state.set_state(PayInvoiceStates.receipt)
+    await set_data(state, PayInvoiceData(invoice_id=callback_data.invoice_id))
+
+    await clb.answer()
+
+
+@router.message(PayInvoiceStates.receipt, F.text == CANCEL_BUTTON)
+async def cancel(msg: Message, state: FSMContext) -> None:
+    await msg.answer(
+        "Отправка подтверждений оплаты отменена.",
+        reply_markup=ReplyKeyboardRemove(),
+    )
+    await state.clear()
+
+
+@router.message(PayInvoiceStates.receipt)
+async def receipt(
+    msg: Message,
+    bot: Bot,
+    state: FSMContext,
+    session: AsyncSession,
+) -> None:
+    if msg.document is not None:
+        receipt_file = ReceiptFile(
+            type=ReceiptFileType.DOCUMENT,
+            file_id=msg.document.file_id,
+        )
+    elif msg.photo is not None:
+        receipt_file = ReceiptFile(
+            type=ReceiptFileType.PHOTO,
+            file_id=max(msg.photo, key=lambda p: (p.width, p.height)).file_id,
+        )
+    else:
+        await msg.answer("Вы должны прислать файл или фото.")
+        return
+
+    data = await get_data(state, PayInvoiceData)
+    payment = Payment(
+        user_id=msg.chat.id,
+        invoice_id=data.invoice_id,
+        receipt_file=receipt_file,
+        datetime=datetime.now(UTC),
+    )
+    session.add(payment)
+    await session.flush()
+
+    await msg.answer(
+        "Файл подтверждения оплаты прикреплен.",
+        reply_markup=ReplyKeyboardRemove(),
+    )
+    await state.clear()
+
+    admin_ids = await session.scalars(
+        select(User.id).where(User.role == UserRole.ADMIN)
+    )
+    reply_markup = InlineKeyboardMarkup(
+        inline_keyboard=[
+            [
+                InlineKeyboardButton(
+                    text="Подтвердить",
+                    callback_data=PaymentStatusClb(
+                        payment_id=payment.id,
+                        payment_status=PaymentStatus.ACCEPTED,
+                    ).pack(),
+                    style=ButtonStyle.SUCCESS,
+                ),
+                InlineKeyboardButton(
+                    text="Отклонить",
+                    callback_data=PaymentStatusClb(
+                        payment_id=payment.id,
+                        payment_status=PaymentStatus.REJECTED,
+                    ).pack(),
+                    style=ButtonStyle.DANGER,
+                ),
+            ]
+        ]
+    )
+
+    for admin_id in admin_ids:
+        try:
+            await bot.send_message(
+                admin_id,
+                f"Новое подтверждение оплаты:\nПользователь: {mention(msg.chat)}",
+            )
+            await receipt_file.send(bot, admin_id, reply_markup=reply_markup)
+        except TelegramAPIError as e:
+            print(e)
diff --git a/libs/__init__.py b/libs/__init__.py
index d8ed122..65f8ab7 100644
--- a/libs/__init__.py
+++ b/libs/__init__.py
@@ -1,7 +1,8 @@
-from . import fsm, msg, storage
+from . import fsm, msg, storage, user
 
 __all__ = [
     "storage",
     "fsm",
     "msg",
+    "user",
 ]
diff --git a/libs/msg.py b/libs/msg.py
index 2e9e16b..a00dd0c 100644
--- a/libs/msg.py
+++ b/libs/msg.py
@@ -7,7 +7,7 @@ from aiogram.exceptions import TelegramAPIError, TelegramRetryAfter
 from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
 
 from models import RichText
-from models.callback_data import PayInvoiceData
+from models.callback_data import PayInvoiceClb
 
 
 async def publish_announcement(
@@ -34,7 +34,7 @@ async def send_invoice(
     rich_text: RichText,
     invoice_id: int,
 ) -> AsyncGenerator[int]:
-    callback_data = PayInvoiceData(invoice_id=invoice_id).pack()
+    callback_data = PayInvoiceClb(invoice_id=invoice_id).pack()
     reply_markup = InlineKeyboardMarkup(
         inline_keyboard=[
             [
diff --git a/libs/user.py b/libs/user.py
new file mode 100644
index 0000000..b201ce9
--- /dev/null
+++ b/libs/user.py
@@ -0,0 +1,5 @@
+from aiogram.types import Chat, User
+
+
+def mention(user: User | Chat) -> str:
+    return f'<a href="tg://user?id={user.id}">{user.full_name}</a>'
diff --git a/models/__init__.py b/models/__init__.py
index 0547429..f26ee74 100644
--- a/models/__init__.py
+++ b/models/__init__.py
@@ -1,9 +1,9 @@
 # isort: off
 from .base import BaseTable
 from .rich_text import RichText
-from .user import User
+from .user import User, UserRole
 from .invoce import Invoice
-from .payment import Payment
+from .payment import Payment, PaymentStatus, ReceiptFile, ReceiptFileType
 from .announcement import Announcement
 from . import callback_data
 # isort: on
@@ -11,8 +11,12 @@ from . import callback_data
 __all__ = [
     "BaseTable",
     "User",
+    "UserRole",
     "Invoice",
     "Payment",
+    "PaymentStatus",
+    "ReceiptFile",
+    "ReceiptFileType",
     "RichText",
     "Announcement",
     "callback_data",
diff --git a/models/callback_data.py b/models/callback_data.py
index d3e6d61..137c4fa 100644
--- a/models/callback_data.py
+++ b/models/callback_data.py
@@ -1,5 +1,12 @@
 from aiogram.filters.callback_data import CallbackData
 
+from models import PaymentStatus
 
-class PayInvoiceData(CallbackData, prefix="pay_invoice"):
+
+class PayInvoiceClb(CallbackData, prefix="pay_invoice"):
     invoice_id: int
+
+
+class PaymentStatusClb(CallbackData, prefix="payment_status"):
+    payment_id: int
+    payment_status: PaymentStatus
diff --git a/models/payment.py b/models/payment.py
index 2b1cb90..afae642 100644
--- a/models/payment.py
+++ b/models/payment.py
@@ -1,16 +1,64 @@
 from datetime import datetime
+from enum import StrEnum, auto
 
-from sqlalchemy import ForeignKey
+from aiogram import Bot
+from aiogram.types import Message, ReplyMarkupUnion
+from pydantic import BaseModel
+from sqlalchemy import JSON, ForeignKey
 from sqlalchemy.orm import Mapped, mapped_column
 
 from models import BaseTable, Invoice, User
 
 
+class ReceiptFileType(StrEnum):
+    PHOTO = auto()
+    DOCUMENT = auto()
+
+
+class ReceiptFile(BaseModel):
+    type: ReceiptFileType
+    file_id: str
+
+    async def send(
+        self,
+        bot: Bot,
+        user_id: int,
+        reply_markup: ReplyMarkupUnion | None = None,
+    ) -> Message:
+        if self.type == ReceiptFileType.DOCUMENT:
+            return await bot.send_document(
+                user_id,
+                document=self.file_id,
+                reply_markup=reply_markup,
+            )
+        else:
+            return await bot.send_photo(
+                user_id,
+                photo=self.file_id,
+                reply_markup=reply_markup,
+            )
+
+
+class PaymentStatus(StrEnum):
+    PENDING = auto()
+    ACCEPTED = auto()
+    REJECTED = auto()
+
+
 class Payment(BaseTable):
     __tablename__ = "payment"
 
     id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
     user_id: Mapped[int] = mapped_column(ForeignKey(User.id))
     invoice_id: Mapped[int] = mapped_column(ForeignKey(Invoice.id))
-    receipt_file_id: Mapped[str]
+    __receipt_file: Mapped[str] = mapped_column("receipt_file", JSON())
+    status: Mapped[PaymentStatus] = mapped_column(default=PaymentStatus.PENDING)
     datetime: Mapped[datetime]
+
+    @property
+    def receipt_file(self) -> ReceiptFile:
+        return ReceiptFile.model_validate_json(self.__receipt_file)
+
+    @receipt_file.setter
+    def receipt_file(self, value: ReceiptFile) -> None:
+        self.__receipt_file = value.model_dump_json()
diff --git a/models/suggest.py b/models/suggest.py
index a76a004..bd628bb 100644
--- a/models/suggest.py
+++ b/models/suggest.py
@@ -4,8 +4,7 @@ from sqlalchemy import JSON
 from sqlalchemy.orm import Mapped, mapped_column
 from sqlalchemy.sql.schema import ForeignKey
 
-from models import RichText, User
-from models.base import BaseTable
+from models import BaseTable, RichText, User
 
 
 class Suggest(BaseTable):