import os from typing import Any import asyncpg from aiogram import F, Router from aiogram.enums import ParseMode from aiogram.fsm.context import FSMContext from aiogram.types import ( CallbackQuery, InlineKeyboardButton, InlineQuery, InlineQueryResultArticle, InputTextMessageContent, Message, ) from aiogram.utils.keyboard import InlineKeyboardBuilder from config import ( DATABASE_URL, INLINE_MODE, INSTRUCTIONS_BUTTON, NEWS_MESSAGE, REFERRAL_OFFERS, RENEWAL_PLANS, TRIAL_TIME, USERNAME_BOT, ) from database import get_balance, get_key_count, get_last_payments, get_referral_stats, get_trial from handlers.buttons.profile import ( ADD_SUB, BALANCE, BALANCE_HISTORY, GIFTS, INSTRUCTIONS, INVITE, MAIN_MENU, MY_SUBS, PAYMENT, ) from handlers.texts import get_referral_link, invite_message_send, profile_message_send from keyboards.admin.panel_kb import AdminPanelCallback from logger import logger import html from .utils import edit_or_send_message router = Router() @router.callback_query(F.data == "profile") @router.message(F.text == "/profile") async def process_callback_view_profile( callback_query_or_message: Message | CallbackQuery, state: FSMContext, admin: bool, ): if isinstance(callback_query_or_message, CallbackQuery): chat_id = callback_query_or_message.message.chat.id username = html.escape(callback_query_or_message.from_user.full_name) target_message = callback_query_or_message.message else: chat_id = callback_query_or_message.chat.id username = html.escape(callback_query_or_message.from_user.full_name) target_message = callback_query_or_message image_path = os.path.join("img", "profile.jpg") logger.info(f"Переход в профиль. Используется изображение: {image_path}") key_count = await get_key_count(chat_id) balance = await get_balance(chat_id) or 0 conn = await asyncpg.connect(DATABASE_URL) try: trial_status = await get_trial(chat_id, conn) profile_message = profile_message_send(username, chat_id, int(balance), key_count) if key_count == 0: profile_message += ( "\n
🔧 Нажмите кнопку ➕ Подписка, чтобы настроить VPN-подключение
" ) else: profile_message += f"\n
{NEWS_MESSAGE}
" builder = InlineKeyboardBuilder() if trial_status == 0 or key_count == 0: builder.row(InlineKeyboardButton(text=ADD_SUB, callback_data="create_key")) else: builder.row(InlineKeyboardButton(text=MY_SUBS, callback_data="view_keys")) builder.row(InlineKeyboardButton(text=BALANCE, callback_data="balance")) builder.row( InlineKeyboardButton(text=INVITE, callback_data="invite"), InlineKeyboardButton(text=GIFTS, callback_data="gifts"), ) if INSTRUCTIONS_BUTTON: builder.row(InlineKeyboardButton(text=INSTRUCTIONS, callback_data="instructions")) if admin: builder.row( InlineKeyboardButton(text="🔧 Администратор", callback_data=AdminPanelCallback(action="admin").pack()) ) builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="start")) await edit_or_send_message( target_message=target_message, text=profile_message, reply_markup=builder.as_markup(), media_path=image_path, disable_web_page_preview=False, force_text=True, ) finally: await conn.close() @router.callback_query(F.data == "balance") async def balance_handler(callback_query: CallbackQuery, session: Any): result = await session.fetchrow( "SELECT balance FROM connections WHERE tg_id = $1", callback_query.from_user.id, ) balance = result["balance"] if result else 0.0 balance = int(balance) builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=PAYMENT, callback_data="pay")) builder.row(InlineKeyboardButton(text=BALANCE_HISTORY, callback_data="balance_history")) builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile")) text = f"Управление вашим балансом 💰\n\nВаш баланс: {balance}" await edit_or_send_message( target_message=callback_query.message, text=text, reply_markup=builder.as_markup(), media_path=None, disable_web_page_preview=False, ) @router.callback_query(F.data == "balance_history") async def balance_history_handler(callback_query: CallbackQuery, session: Any): builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=PAYMENT, callback_data="pay")) builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile")) records = await get_last_payments(callback_query.from_user.id, session) if records: history_text = "📊 Последние 3 операции с балансом:\n\n" for record in records: amount = record["amount"] payment_system = record["payment_system"] status = record["status"] date = record["created_at"].strftime("%Y-%m-%d %H:%M:%S") history_text += ( f"Сумма: {amount}₽\n" f"Способ оплаты: {payment_system}\n" f"Статус: {status}\n" f"Дата: {date}\n\n" ) else: history_text = "❌ У вас пока нет операций с балансом." await edit_or_send_message( target_message=callback_query.message, text=history_text, reply_markup=builder.as_markup(), media_path=None, disable_web_page_preview=False, ) @router.message(F.text == "/tariffs") @router.callback_query(F.data == "view_tariffs") async def view_tariffs_handler(callback_query: CallbackQuery): builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text="👤 Личный кабинет", callback_data="profile")) image_path = os.path.join("img", "tariffs.jpg") tariffs_message = "🚀 Доступные тарифы VPN:\n\n" + "\n".join( [ f"{months} {'месяц' if months == '1' else 'месяца' if int(months) in [2, 3, 4] else 'месяцев'}: " f"{RENEWAL_PLANS[months]['price']} " f"{'💳' if months == '1' else '🌟' if months == '3' else '🔥' if months == '6' else '🚀'} рублей" for months in sorted(RENEWAL_PLANS.keys(), key=int) ] ) await edit_or_send_message( target_message=callback_query.message, text=tariffs_message, reply_markup=builder.as_markup(), media_path=image_path, disable_web_page_preview=False, ) @router.callback_query(F.data == "invite") @router.message(F.text == "/invite") async def invite_handler(callback_query_or_message: Message | CallbackQuery): chat_id = None if isinstance(callback_query_or_message, CallbackQuery): chat_id = callback_query_or_message.message.chat.id target_message = callback_query_or_message.message else: chat_id = callback_query_or_message.chat.id target_message = callback_query_or_message referral_link = get_referral_link(chat_id) referral_stats = await get_referral_stats(chat_id) invite_message = invite_message_send(referral_link, referral_stats) image_path = os.path.join("img", "pic_invite.jpg") builder = InlineKeyboardBuilder() if INLINE_MODE: builder.button(text="👥 Пригласить друга", switch_inline_query="invite") else: invite_text = f"\nПриглашаю тебя пользоваться действительно быстрым VPN вместе:\n\n{referral_link}" builder.button(text="👥 Пригласить друга", switch_inline_query=invite_text) builder.button(text="👤 Личный кабинет", callback_data="profile") builder.adjust(1) await edit_or_send_message( target_message=target_message, text=invite_message, reply_markup=builder.as_markup(), media_path=image_path, disable_web_page_preview=False, ) @router.inline_query(F.query.in_(["referral", "ref", "invite"])) async def inline_referral_handler(inline_query: InlineQuery): referral_link = f"https://t.me/{USERNAME_BOT}?start=referral_{inline_query.from_user.id}" results: list[InlineQueryResultArticle] = [] for index, offer in enumerate(REFERRAL_OFFERS): description = offer["description"][:64] message_text = offer["message"].format(trial_time=TRIAL_TIME)[:4096] builder = InlineKeyboardBuilder() builder.row(InlineKeyboardButton(text=offer["title"], url=referral_link)) results.append( InlineQueryResultArticle( id=str(index), title=offer["title"], description=description, input_message_content=InputTextMessageContent(message_text=message_text, parse_mode=ParseMode.HTML), reply_markup=builder.as_markup(), ) ) await inline_query.answer(results=results, cache_time=86400, is_personal=True)