Solo_bot/handlers/admin/users/keyboard.py

from datetime import datetime, timezone

import asyncpg

from aiogram.filters.callback_data import CallbackData
from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
from aiogram.utils.keyboard import InlineKeyboardBuilder

from config import DATABASE_URL, HWID_RESET_BUTTON
from database import get_clusters, get_tariffs, get_tariffs_for_cluster
from handlers.buttons import BACK

from ..panel.keyboard import build_admin_back_btn


class AdminUserEditorCallback(CallbackData, prefix="admin_users"):
    action: str
    tg_id: int
    data: str | int | None = None
    edit: bool = False


class AdminUserKeyEditorCallback(CallbackData, prefix="admin_users_key"):
    action: str
    tg_id: int
    data: str
    month: int | None = None
    edit: bool = False


def build_user_edit_kb(tg_id: int, key_records: list, is_banned: bool = False) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()
    current_time = datetime.now(tz=timezone.utc)

    builder.button(
        text="➕ Создать ключ",
        callback_data=AdminUserEditorCallback(action="users_create_key", tg_id=tg_id).pack(),
    )

    for record in key_records:
        email = record["email"]
        expiry = datetime.fromtimestamp(record["expiry_time"] / 1000, tz=timezone.utc)
        days = (expiry - current_time).days
        builder.button(
            text=f"🔑 {email} ({'<1' if days < 1 else days} дн.)",
            callback_data=AdminUserEditorCallback(action="users_key_edit", tg_id=tg_id, data=str(email)).pack(),
        )

    builder.button(
        text="✉️ Сообщение", callback_data=AdminUserEditorCallback(action="users_send_message", tg_id=tg_id).pack()
    )
    builder.button(
        text="💸 Изменить баланс",
        callback_data=AdminUserEditorCallback(action="users_balance_edit", tg_id=tg_id).pack(),
    )
    builder.button(
        text="🤝 Выгрузить рефералов",
        callback_data=AdminUserEditorCallback(action="users_export_referrals", tg_id=tg_id).pack(),
    )
    builder.button(
        text="♻️ Восстановить триал",
        callback_data=AdminUserEditorCallback(action="users_trial_restore", tg_id=tg_id).pack(),
    )
    builder.button(
        text="❌ Удалить клиента", callback_data=AdminUserEditorCallback(action="users_delete_user", tg_id=tg_id).pack()
    )
    builder.button(
        text="✅ Разблокировать" if is_banned else "🚫 Заблокировать",
        callback_data=AdminUserEditorCallback(action="users_unban" if is_banned else "users_ban", tg_id=tg_id).pack(),
    )
    builder.row(build_editor_btn("🔄 Обновить данные", tg_id, edit=True))
    builder.row(build_admin_back_btn())
    builder.adjust(1)
    return builder.as_markup()


def build_users_balance_change_kb(tg_id: int) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()
    builder.button(
        text=BACK,  # todo: fix magic text was set
        callback_data=AdminUserEditorCallback(action="users_balance_edit", tg_id=tg_id).pack(),
    )
    return builder.as_markup()


async def build_users_balance_kb(tg_id: int) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()

    conn = await asyncpg.connect(DATABASE_URL)
    try:
        tariffs = await get_tariffs(conn)
        for tariff in tariffs:
            months = tariff["duration_days"] // 30
            if months < 1:
                continue
            price = tariff["price_rub"]
            builder.row(
                InlineKeyboardButton(
                    text=f"+ {price}₽ ({months} мес.)",
                    callback_data=AdminUserEditorCallback(action="users_balance_add", tg_id=tg_id, data=price).pack(),
                ),
                InlineKeyboardButton(
                    text=f"- {price}₽ ({months} мес.)",
                    callback_data=AdminUserEditorCallback(action="users_balance_add", tg_id=tg_id, data=-price).pack(),
                ),
            )
    finally:
        await conn.close()

    builder.row(
        InlineKeyboardButton(
            text="💵 Добавить",
            callback_data=AdminUserEditorCallback(action="users_balance_add", tg_id=tg_id).pack(),
        ),
        InlineKeyboardButton(
            text="💵 Вычесть",
            callback_data=AdminUserEditorCallback(action="users_balance_take", tg_id=tg_id).pack(),
        ),
    )

    builder.row(
        InlineKeyboardButton(
            text="💵 Установить баланс",
            callback_data=AdminUserEditorCallback(action="users_balance_set", tg_id=tg_id).pack(),
        )
    )

    builder.row(build_editor_back_btn(tg_id, True))

    return builder.as_markup()


def build_users_key_show_kb(tg_id: int, email: str) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()
    builder.button(
        text=BACK,  # todo: fix magic text was set
        callback_data=AdminUserEditorCallback(action="users_key_edit", tg_id=tg_id, data=email, edit=True).pack(),
    )
    return builder.as_markup()


async def build_users_key_expiry_kb(tg_id: int, email: str) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()

    conn = await asyncpg.connect(DATABASE_URL)
    try:
        record = await conn.fetchrow("SELECT server_id FROM keys WHERE email = $1", email)
        if not record:
            builder.row(
                InlineKeyboardButton(
                    text="⚠️ Сервер не найден",
                    callback_data=AdminUserEditorCallback(action="users_key_edit", tg_id=tg_id, data=email).pack(),
                )
            )
            return builder.as_markup()

        server_id = record["server_id"]

        tariffs = await get_tariffs_for_cluster(conn, server_id)
        for tariff in tariffs:
            months = tariff["duration_days"] // 30
            if months < 1:
                continue
            builder.row(
                InlineKeyboardButton(
                    text=f"+ {months} мес.",
                    callback_data=AdminUserKeyEditorCallback(
                        action="add", tg_id=tg_id, data=email, month=months
                    ).pack(),
                ),
                InlineKeyboardButton(
                    text=f"- {months} мес.",
                    callback_data=AdminUserKeyEditorCallback(
                        action="add", tg_id=tg_id, data=email, month=-months
                    ).pack(),
                ),
            )
    finally:
        await conn.close()

    builder.row(
        InlineKeyboardButton(
            text="⏳ Добавить дни",
            callback_data=AdminUserKeyEditorCallback(action="add", tg_id=tg_id, data=email).pack(),
        ),
        InlineKeyboardButton(
            text="⏳ Вычесть дни",
            callback_data=AdminUserKeyEditorCallback(action="take", tg_id=tg_id, data=email).pack(),
        ),
    )

    builder.row(
        InlineKeyboardButton(
            text="⏳ Установить дату истечения",
            callback_data=AdminUserKeyEditorCallback(action="set", tg_id=tg_id, data=email).pack(),
        )
    )
    builder.row(
        InlineKeyboardButton(
            text=BACK,
            callback_data=AdminUserEditorCallback(action="users_key_edit", tg_id=tg_id, data=email).pack(),
        )
    )

    return builder.as_markup()


def build_user_delete_kb(tg_id: int):
    builder = InlineKeyboardBuilder()
    builder.button(
        text="❌ Да, удалить!",
        callback_data=AdminUserEditorCallback(action="users_delete_user_confirm", tg_id=tg_id).pack(),
    )
    builder.row(build_editor_back_btn(tg_id, True))
    builder.adjust(1)
    return builder.as_markup()


def build_user_key_kb(tg_id: int, email: str) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()
    builder.button(
        text=BACK, callback_data=AdminUserEditorCallback(action="users_key_edit", tg_id=tg_id, data=email).pack()
    )
    builder.adjust(1)
    return builder.as_markup()


def build_key_edit_kb(key_details: dict, email: str) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()
    builder.button(
        text="⏳ Время истечения",
        callback_data=AdminUserEditorCallback(
            action="users_expiry_edit", data=email, tg_id=key_details["tg_id"]
        ).pack(),
    )
    builder.button(
        text="🔄 Перевыпустить",
        callback_data=AdminUserEditorCallback(action="users_update_key", data=email, tg_id=key_details["tg_id"]).pack(),
    )
    builder.button(
        text="❌ Удалить",
        callback_data=AdminUserEditorCallback(action="users_delete_key", data=email, tg_id=key_details["tg_id"]).pack(),
    )
    builder.button(
        text="📊 Трафик",
        callback_data=AdminUserEditorCallback(action="users_traffic", data=email, tg_id=key_details["tg_id"]).pack(),
    )
    builder.button(
        text="♻️ Сбросить трафик",
        callback_data=AdminUserEditorCallback(
            action="users_reset_traffic", data=email, tg_id=key_details["tg_id"]
        ).pack(),
    )
    if HWID_RESET_BUTTON:
        builder.button(
            text="💻 HWID",
            callback_data=AdminUserEditorCallback(
                action="users_hwid_menu", data=email, tg_id=key_details["tg_id"]
            ).pack(),
        )
    builder.row(build_editor_back_btn(key_details["tg_id"], True))
    builder.adjust(1)
    return builder.as_markup()


def build_hwid_menu_kb(email: str, tg_id: int) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()
    builder.button(
        text="♻️ Сбросить HWID",
        callback_data=AdminUserEditorCallback(action="users_hwid_reset", data=email, tg_id=tg_id).pack(),
    )
    builder.button(
        text="🔙 Назад",
        callback_data=AdminUserEditorCallback(action="users_key_edit", data=email, tg_id=tg_id).pack(),
    )
    builder.adjust(1)
    return builder.as_markup()


def build_key_delete_kb(tg_id: int, email: str) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()
    builder.row(
        InlineKeyboardButton(
            text="✅ Да, удалить",
            callback_data=AdminUserEditorCallback(action="users_delete_key_confirm", data=email, tg_id=tg_id).pack(),
        )
    )
    builder.row(build_editor_back_btn(tg_id))
    builder.adjust(1)
    return builder.as_markup()


def build_editor_kb(tg_id: int, edit: bool = False) -> InlineKeyboardMarkup:
    return build_editor_singleton_kb(BACK, tg_id, edit)


def build_editor_singleton_kb(text: str, tg_id: int, edit: bool = False) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()
    builder.row(build_editor_btn(text, tg_id, edit))
    return builder.as_markup()


def build_editor_back_btn(tg_id: int, edit: bool = False) -> InlineKeyboardButton:
    return build_editor_btn(BACK, tg_id, edit)


def build_editor_btn(text: str, tg_id: int, edit: bool = False) -> InlineKeyboardButton:
    return InlineKeyboardButton(
        text=text, callback_data=AdminUserEditorCallback(action="users_editor", tg_id=tg_id, edit=edit).pack()
    )


async def build_cluster_selection_kb(session, tg_id: int, email: str, action: str) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()
    clusters = await get_clusters(session)

    for cluster_id in clusters:
        builder.button(text=cluster_id, callback_data=f"{action}|{tg_id}|{email}|{cluster_id}")

    builder.button(text=BACK, callback_data=f"edit_user_key|{tg_id}|{email}")
    builder.adjust(1)
    return builder.as_markup()