import os import urllib.parse from typing import Any from aiogram import F, Router from aiogram.types import CallbackQuery, InlineKeyboardButton, Message from aiogram.utils.keyboard import InlineKeyboardBuilder from config import ( CONNECT_MACOS, CONNECT_WINDOWS, DOWNLOAD_MACOS, DOWNLOAD_PC, HAPP_CRYPTOLINK, REMNAWAVE_WEBAPP, SUPPORT_CHAT_URL, WEBHOOK_HOST, ) from core.bootstrap import MODES_CONFIG from database import get_key_details, get_subscription_link from handlers.buttons import ( BACK, CONNECT_MACOS_BUTTON, CONNECT_WINDOWS_BUTTON, DOWNLOAD_MACOS_BUTTON, DOWNLOAD_PC_BUTTON, MAIN_MENU, PC_MACOS, PC_PC, SUPPORT, TV_CONTINUE, ) from handlers.texts import ( CHOOSE_DEVICE_TEXT, CONNECT_TV_TEXT, INSTRUCTIONS, INSTRUCTION_MACOS, INSTRUCTION_PC, KEY_MESSAGE, ROUTER_MESSAGE, SUBSCRIPTION_DETAILS_TEXT, ) from handlers.utils import edit_or_send_message, is_full_remnawave_cluster from hooks.processors import process_remnawave_webapp_override router = Router() @router.callback_query(F.data == "instructions") @router.message(F.text == "/instructions") async def send_instructions(callback_query_or_message: CallbackQuery | Message): instructions_message = INSTRUCTIONS image_path = os.path.join("img", "instructions.jpg") builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=SUPPORT, url=SUPPORT_CHAT_URL)) builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile")) if isinstance(callback_query_or_message, CallbackQuery): target_message = callback_query_or_message.message else: target_message = callback_query_or_message await edit_or_send_message( target_message=target_message, text=instructions_message, reply_markup=builder.as_markup(), media_path=image_path, ) @router.callback_query(F.data.startswith("connect_pc|")) async def process_connect_pc(callback_query: CallbackQuery, session: Any): key_name = callback_query.data.split("|")[1] key_link = await get_subscription_link(session, key_name) if not key_link: builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile")) await edit_or_send_message( target_message=callback_query.message, text="❌ Ключ не найден. Проверьте имя ключа. 🔍", reply_markup=builder.as_markup(), media_path=None, ) return builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=PC_PC, callback_data=f"windows_menu|{key_name}")) builder.row(InlineKeyboardButton(text=PC_MACOS, callback_data=f"macos_menu|{key_name}")) builder.row(InlineKeyboardButton(text=BACK, callback_data=f"connect_device|{key_name}")) await edit_or_send_message( target_message=callback_query.message, text=CHOOSE_DEVICE_TEXT, reply_markup=builder.as_markup(), media_path=None, ) @router.callback_query(F.data.startswith("windows_menu|")) async def process_windows_menu(callback_query: CallbackQuery, session: Any): key_name = callback_query.data.split("|")[1] key_link = await get_subscription_link(session, key_name) if not key_link: await callback_query.message.answer("❌ Ошибка: ключ не найден.") return key_message_text = KEY_MESSAGE.format(key_link) instruction_message = f"{key_message_text}{INSTRUCTION_PC}" builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=DOWNLOAD_PC_BUTTON, url=DOWNLOAD_PC)) if "happ://crypt" in key_link: processed_link = urllib.parse.quote(key_link, safe="") windows_url = f"{WEBHOOK_HOST}/?url={processed_link}" else: processed_link = key_link windows_url = f"{CONNECT_WINDOWS}{processed_link}" builder.row(InlineKeyboardButton(text=CONNECT_WINDOWS_BUTTON, url=windows_url)) builder.row(InlineKeyboardButton(text=SUPPORT, url=SUPPORT_CHAT_URL)) builder.row(InlineKeyboardButton(text=BACK, callback_data=f"connect_pc|{key_name}")) await edit_or_send_message( target_message=callback_query.message, text=instruction_message, reply_markup=builder.as_markup(), media_path=None, ) @router.callback_query(F.data.startswith("macos_menu|")) async def process_macos_menu(callback_query: CallbackQuery, session: Any): key_name = callback_query.data.split("|")[1] key_link = await get_subscription_link(session, key_name) if not key_link: await callback_query.message.answer("❌ Ошибка: ключ не найден.") return key_message_text = KEY_MESSAGE.format(key_link) instruction_message = f"{key_message_text}{INSTRUCTION_MACOS}" builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=DOWNLOAD_MACOS_BUTTON, url=DOWNLOAD_MACOS)) if "happ://crypt" in key_link: processed_link = urllib.parse.quote(key_link, safe="") macos_url = f"{WEBHOOK_HOST}/?url={processed_link}" else: processed_link = key_link macos_url = f"{CONNECT_MACOS}{processed_link}" builder.row(InlineKeyboardButton(text=CONNECT_MACOS_BUTTON, url=macos_url)) builder.row(InlineKeyboardButton(text=SUPPORT, url=SUPPORT_CHAT_URL)) builder.row(InlineKeyboardButton(text=BACK, callback_data=f"connect_pc|{key_name}")) await edit_or_send_message( target_message=callback_query.message, text=instruction_message, reply_markup=builder.as_markup(), media_path=None, ) @router.callback_query(F.data.startswith("connect_tv|")) async def process_connect_tv(callback_query: CallbackQuery, session: Any): key_name = callback_query.data.split("|", 1)[1] record = await get_key_details(session, key_name) final_link = None is_full_remnawave = False use_webapp = False is_remnawave_webapp = False if record: server_name = record.get("server_id") final_link = record.get("key") or record.get("remnawave_link") if server_name: is_full_remnawave = await is_full_remnawave_cluster(server_name, session) remnawave_webapp_enabled = bool(MODES_CONFIG.get("REMNAWAVE_WEBAPP_ENABLED", REMNAWAVE_WEBAPP)) happ_cryptolink_enabled = bool(MODES_CONFIG.get("HAPP_CRYPTOLINK_ENABLED", HAPP_CRYPTOLINK)) use_webapp = remnawave_webapp_enabled if is_full_remnawave and final_link and remnawave_webapp_enabled and not happ_cryptolink_enabled: use_webapp = await process_remnawave_webapp_override( remnawave_webapp=remnawave_webapp_enabled, final_link=final_link, session=session, ) is_remnawave_webapp = bool(is_full_remnawave and final_link and use_webapp and not happ_cryptolink_enabled) if not final_link: await callback_query.answer("❌ Ссылка подписки не найдена", show_alert=True) return back_callback = f"view_key|{key_name}" if is_remnawave_webapp else f"connect_device|{key_name}" builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=TV_CONTINUE, callback_data=f"continue_tv|{key_name}")) builder.row(InlineKeyboardButton(text=BACK, callback_data=back_callback)) builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile")) text = CONNECT_TV_TEXT.format(subscription_link=final_link) await edit_or_send_message( target_message=callback_query.message, text=text, reply_markup=builder.as_markup(), media_path=None, disable_web_page_preview=True, ) @router.callback_query(F.data.startswith("continue_tv|")) async def process_continue_tv(callback_query: CallbackQuery, session: Any): key_name = callback_query.data.split("|")[1] key_link = await get_subscription_link(session, key_name) message_text = SUBSCRIPTION_DETAILS_TEXT.format(subscription_link=key_link) builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=BACK, callback_data=f"connect_tv|{key_name}")) builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile")) await edit_or_send_message( target_message=callback_query.message, text=message_text, reply_markup=builder.as_markup(), media_path=None, ) @router.callback_query(F.data.startswith("connect_router|")) async def process_connect_router(callback_query: CallbackQuery, session: Any): key_name = callback_query.data.split("|")[1] key_link = await get_subscription_link(session, key_name) if not key_link: builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile")) await edit_or_send_message( target_message=callback_query.message, text="❌ Ключ не найден.", reply_markup=builder.as_markup(), media_path=None, ) return message_text = ROUTER_MESSAGE.format(subscription_link=key_link) builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=BACK, callback_data=f"view_key|{key_name}")) await edit_or_send_message( target_message=callback_query.message, text=message_text, reply_markup=builder.as_markup(), media_path=None, )