import html from aiogram import Router from aiogram.types import Message from bot.api.asf import get_all_bots, send_command from bot.filters import AdminFilter from bot.logging_utils import describe_user, get_logger from bot.services.redeem import ( extract_keys, redeem_all_bots_keys, redeem_single_bot_keys, ) from bot.state import ( bot_redeem_messages, bot_redeem_users, console_users, redeem_messages, redeem_users, ) from bot.ui.formatters import format_bot_ui, get_asf_status_text from bot.ui.keyboards import ( bot_details_keyboard, delete_message_keyboard, main_keyboard, ) router = Router() logger = get_logger(__name__) async def handle_console_message(message: Message) -> bool: if message.from_user.id not in console_users: # type: ignore[union-attr] return False command = message.text.strip() # type: ignore[union-attr] logger.info( "Пользователь %s отправил консольную команду: %s", describe_user(message.from_user), command, ) try: await message.delete() except Exception: logger.debug( "Не удалось удалить сообщение с консольной командой от %s", describe_user(message.from_user), exc_info=True, ) try: response = send_command(f"!{command}") if response.status_code != 200: logger.warning( "ASF вернул HTTP %s на консольную команду '%s' от %s", response.status_code, command, describe_user(message.from_user), ) text = f"Ошибка HTTP {response.status_code}" else: data = response.json() if not data.get("Success"): logger.warning( "ASF вернул неуспешный ответ на консольную команду '%s' от %s", command, describe_user(message.from_user), ) text = "Ошибка ASF" else: logger.info( "Консольная команда '%s' от %s выполнена успешно", command, describe_user(message.from_user), ) result = html.escape(str(data.get("Result", "Нет ответа"))) text = f"Команда: {command}\n\nОтвет:\n{result}" await message.answer(text, reply_markup=delete_message_keyboard()) except Exception as error: logger.exception( "Ошибка при выполнении консольной команды '%s' от %s", command, describe_user(message.from_user), ) await message.answer(f"Ошибка: {error}") return True async def handle_single_bot_redeem(message: Message) -> bool: if message.from_user.id not in bot_redeem_users: # type: ignore[union-attr] return False bot_name = bot_redeem_users.pop(message.from_user.id) # type: ignore[union-attr] menu_message = bot_redeem_messages.pop(message.from_user.id, None) # type: ignore[union-attr] logger.info( "Пользователь %s отправил ключи для бота %s", describe_user(message.from_user), bot_name, ) if menu_message: try: response = get_all_bots() bot_data = response.json().get("Result", {}).get(bot_name) if bot_data: await menu_message.edit_text( format_bot_ui(bot_data), disable_web_page_preview=True, reply_markup=bot_details_keyboard( bot_name, bool(bot_data.get("HasMobileAuthenticator")) ), ) except Exception: logger.debug( "Не удалось восстановить меню бота %s после ввода ключей от %s", bot_name, describe_user(message.from_user), exc_info=True, ) try: await message.bot.delete_message( chat_id=message.chat.id, message_id=message.message_id ) except Exception: logger.debug( "Не удалось удалить сообщение с ключами для бота %s от %s", bot_name, describe_user(message.from_user), exc_info=True, ) keys = extract_keys(message.text or "") if not keys: logger.info( "В сообщении от %s для бота %s не найдено ключей", describe_user(message.from_user), bot_name, ) await message.answer("Ключи не найдены") return True logger.info( "Для бота %s получено %s ключ(ей) от %s", bot_name, len(keys), describe_user(message.from_user), ) checking = await message.answer( f"Аккаунт: {bot_name}\nНайдено ключей: {len(keys)}\nНачинаю активацию..." ) await checking.edit_text(await redeem_single_bot_keys(bot_name, keys)) return True async def handle_all_bots_redeem(message: Message) -> bool: if message.from_user.id not in redeem_users: # type: ignore[union-attr] return False redeem_users.discard(message.from_user.id) # type: ignore[union-attr] menu_message = redeem_messages.pop(message.from_user.id, None) # type: ignore[union-attr] logger.info( "Пользователь %s отправил ключи для активации на всех ботах", describe_user(message.from_user), ) if menu_message: try: await menu_message.edit_text( get_asf_status_text(), reply_markup=main_keyboard() ) except Exception: logger.debug( "Не удалось восстановить главное меню после массовой активации для %s", describe_user(message.from_user), exc_info=True, ) try: await message.bot.delete_message( chat_id=message.chat.id, message_id=message.message_id ) except Exception: logger.debug( "Не удалось удалить сообщение с массовой активацией от %s", describe_user(message.from_user), exc_info=True, ) keys = extract_keys(message.text or "") if not keys: logger.info( "В сообщении от %s для массовой активации не найдено ключей", describe_user(message.from_user), ) await message.answer("Ключи не найдены") return True logger.info( "Для массовой активации получено %s ключ(ей) от %s", len(keys), describe_user(message.from_user), ) checking = await message.answer( f"Найдено ключей: {len(keys)}\nНачинаю активацию..." ) try: await checking.edit_text(await redeem_all_bots_keys(keys)) except Exception as error: logger.exception( "Ошибка при массовой активации ключей от %s", describe_user(message.from_user), ) await checking.edit_text(f"Ошибка: {error}") return True @router.message(AdminFilter()) async def text_router(message: Message) -> None: logger.debug( "Получено текстовое сообщение от %s для маршрутизации", describe_user(message.from_user), ) if await handle_console_message(message): return if await handle_single_bot_redeem(message): return await handle_all_bots_redeem(message)