ASF-Control-Bot/bot/handlers/twofa.py

import asyncio

from aiogram import Router
from aiogram.types import CallbackQuery

from bot.api.asf import accept_confirmations, get_confirmations
from bot.filters import AdminCallbackFilter
from bot.logging_utils import describe_user, get_logger
from bot.services.twofa import auto_update_2fa, stop_twofa_task
from bot.state import twofa_tasks
from bot.ui.keyboards import back_keyboard, confirmations_keyboard

router = Router()
logger = get_logger(__name__)


@router.callback_query(
    lambda c: c.data
    and c.data.startswith("2fa_")
    and not c.data.startswith("2fa_refresh_"),
    AdminCallbackFilter(),
)
async def twofa_handler(callback: CallbackQuery) -> None:
    await callback.answer()
    if not callback.data:
        logger.debug(
            "Колбэк открытия 2FA пришел без data от %s",
            describe_user(callback.from_user),
        )
        return
    bot_name = callback.data.replace("2fa_", "")
    message_id = callback.message.message_id  # type: ignore[union-attr]
    logger.info(
        "Пользователь %s открыл 2FA для бота %s",
        describe_user(callback.from_user),
        bot_name,
    )
    await stop_twofa_task(message_id)
    await callback.message.edit_text(  # type: ignore[union-attr]
        f"🔐 {bot_name}\n\nЗагрузка 2FA...",
        reply_markup=back_keyboard(f"bot_{bot_name}"),
    )
    task = asyncio.create_task(auto_update_2fa(callback.message, bot_name))
    twofa_tasks[message_id] = task
    logger.info(
        "Запущено фоновое обновление 2FA для бота %s в сообщении %s",
        bot_name,
        message_id,
    )


@router.callback_query(
    lambda c: c.data
    and c.data.startswith("confirm_")
    and not c.data.startswith("confirm_list_")
    and not c.data.startswith("confirm_all_"),
    AdminCallbackFilter(),
)
async def confirm_trades(callback: CallbackQuery) -> None:
    await callback.answer()
    bot_name = callback.data.replace("confirm_", "")  # type: ignore[union-attr]
    logger.info(
        "Пользователь %s подтвердил сделки для бота %s",
        describe_user(callback.from_user),
        bot_name,
    )
    try:
        status = await accept_confirmations(bot_name)
        if status != 200:
            logger.warning(
                "Не удалось подтвердить сделки для бота %s: HTTP %s",
                bot_name,
                status,
            )
            await callback.answer("Ошибка HTTP", show_alert=True)
            return
        logger.info("Сделки для бота %s подтверждены", bot_name)
        await callback.answer("Подтверждено", show_alert=True)
        await twofa_handler(callback)
    except Exception as error:
        logger.exception(
            "Ошибка при подтверждении сделок для бота %s",
            bot_name,
        )
        await callback.message.edit_text(f"Ошибка: {error}")  # type: ignore[union-attr]


@router.callback_query(
    lambda c: c.data and c.data.startswith("confirm_list_"), AdminCallbackFilter()
)
async def confirm_list(callback: CallbackQuery) -> None:
    await stop_twofa_task(callback.message.message_id)  # type: ignore[union-attr]
    await callback.answer()
    if not callback.data:
        logger.debug(
            "Колбэк списка подтверждений пришел без data от %s",
            describe_user(callback.from_user),
        )
        return
    bot_name = callback.data.replace("confirm_list_", "")
    logger.info(
        "Пользователь %s открыл список подтверждений для бота %s",
        describe_user(callback.from_user),
        bot_name,
    )
    try:
        confirmations = await get_confirmations(bot_name)
        if not confirmations:
            logger.info("Для бота %s подтверждений не найдено", bot_name)
            text = "Подтверждений нет"
        else:
            logger.info(
                "Для бота %s найдено %s подтверждени(й)",
                bot_name,
                len(confirmations),
            )
            text = f"🔐 {bot_name}\n\nПодтверждения:\n\n"
            for conf in confirmations:
                text += f"- {conf['type_name']}\nID: <code>{conf['id']}</code>\n\n"
        await callback.message.edit_text(text, reply_markup=confirmations_keyboard(bot_name))  # type: ignore[union-attr]
    except Exception as error:
        logger.exception(
            "Ошибка при загрузке списка подтверждений для бота %s",
            bot_name,
        )
        await callback.message.edit_text(f"Ошибка: {error}")  # type: ignore[union-attr]


@router.callback_query(
    lambda c: c.data and c.data.startswith("confirm_all_"), AdminCallbackFilter()
)
async def confirm_all(callback: CallbackQuery) -> None:
    await stop_twofa_task(callback.message.message_id)  # type: ignore[union-attr]
    await callback.answer()
    if not callback.data:
        logger.debug(
            "Колбэк подтверждения всех сделок пришел без data от %s",
            describe_user(callback.from_user),
        )
        return
    bot_name = callback.data.replace("confirm_all_", "")
    logger.info(
        "Пользователь %s запустил подтверждение всех сделок для бота %s",
        describe_user(callback.from_user),
        bot_name,
    )
    try:
        status = await accept_confirmations(bot_name)
        if status != 200:
            logger.warning(
                "Не удалось подтвердить все сделки для бота %s: HTTP %s",
                bot_name,
                status,
            )
            await callback.answer("Ошибка HTTP", show_alert=True)
            return
        confirmations = await get_confirmations(bot_name)
        if not confirmations:
            logger.info("После подтверждения у бота %s не осталось сделок", bot_name)
            text = f"🔐 {bot_name}\n\nПодтверждений больше нет"
        else:
            logger.info(
                "После подтверждения у бота %s осталось %s подтверждени(й)",
                bot_name,
                len(confirmations),
            )
            text = f"🔐 {bot_name}\n\nПодтверждения:\n\n"
            for conf in confirmations:
                text += f"- {conf['type_name']}\nID: <code>{conf['id']}</code>\n\n"
        await callback.message.edit_text(text, reply_markup=confirmations_keyboard(bot_name))  # type: ignore[union-attr]
        await callback.answer("Все подтверждено", show_alert=True)
    except Exception as error:
        logger.exception(
            "Ошибка при подтверждении всех сделок для бота %s",
            bot_name,
        )
        await callback.message.edit_text(f"Ошибка: {error}")  # type: ignore[union-attr]