src.handlers.info

  1"""Информационные сообщения"""
  2
  3import logging
  4from contextlib import suppress
  5from datetime import timedelta
  6from typing import Union
  7
  8from aiogram import Bot, F, Router
  9from aiogram.exceptions import TelegramBadRequest
 10from aiogram.filters.command import Command
 11from aiogram.types import CallbackQuery, Message
 12from aiogram.utils.formatting import Bold, as_list, as_marked_section
 13
 14import kb
 15import text
 16from core.config import settings
 17from core.err import bot_except
 18from core.exceptions import BaseBotError, DatabaseError, WireguardError
 19from db.models import UserData
 20from db.utils import get_user, test_server_speed
 21from handlers.utils import find_user
 22from wg.utils import WgServerTools
 23
 24logger = logging.getLogger()
 25router = Router()
 26router.message.filter(F.chat.type == "private")
 27
 28
 29async def more_help_info(callback: CallbackQuery):
 30    """Отправляет дополнительную информацию о помощи.
 31
 32    Args:
 33        callback (CallbackQuery): Событие обратного вызова, содержащее информацию о запросе.
 34
 35    Отправляет сообщение с предложением задать дополнительные вопросы и кнопкой для доступа к меню помощи.
 36    """
 37    await callback.message.answer(
 38        "Если у вас есть дополнительные вопросы, не стесняйтесь спрашивать! Мы здесь, чтобы помочь вам. 🚀",
 39        reply_markup=kb.get_help_menu(
 40            callback.from_user.full_name, callback.from_user.id
 41        ),
 42    )
 43
 44
 45async def change_help_page(message: Message, pages: list, page: int, prefix: str):
 46    """Изменяет страницу помощи.
 47
 48    Args:
 49        message (Message): Сообщение, которое будет изменено.
 50        pages (list): Список страниц помощи.
 51        page (int): Индекс текущей страницы.
 52        prefix (str): Префикс для кнопок навигации.
 53
 54    Обновляет текст сообщения с новой страницей помощи и кнопками навигации.
 55    """
 56    with suppress(TelegramBadRequest):
 57        await message.edit_text(
 58            pages[page], reply_markup=kb.get_help_book_keyboard(pages, page, prefix)
 59        )
 60
 61
 62async def post_help_book(
 63    callback: CallbackQuery, book: list, step: str, start_message: str, prefix: str
 64):
 65    """Отправляет книгу помощи пользователю.
 66
 67    Args:
 68        callback (CallbackQuery): Событие обратного вызова, содержащее информацию о запросе.
 69        book (list): Список шагов помощи.
 70        step (str): Текущий шаг.
 71        start_message (str): Сообщение, отправляемое в начале.
 72        prefix (str): Префикс для кнопок навигации.
 73
 74    Отправляет сообщения с шагами помощи в зависимости от текущего шага.
 75    """
 76    if step == "start":
 77        await callback.message.answer(start_message)
 78
 79        await callback.message.answer(
 80            book[0],
 81            reply_markup=kb.get_help_book_keyboard(pages=book, page=0, prefix=prefix),
 82        )
 83    elif step.isdigit():
 84        await change_help_page(
 85            callback.message, pages=book, page=int(step), prefix=prefix
 86        )
 87    else:
 88        for step in book:
 89            await callback.message.answer(step)
 90        await more_help_info(callback)
 91
 92
 93@router.message(Command("help"))
 94@router.message(F.text == "Помощь")
 95@router.callback_query(F.data == "main_help")
 96@bot_except
 97async def help_me(trigger: Union[Message, CallbackQuery]):
 98    """Обработчик команды помощи.
 99
100    Args:
101        trigger (Union[Message, CallbackQuery]): Сообщение или событие обратного вызова, инициировавшее команду.
102
103    Отправляет сообщение с предложением помощи и кнопками для доступа к меню помощи.
104    """
105    await getattr(trigger, "message", trigger).answer(
106        "Чем вам помочь?",
107        reply_markup=kb.get_help_menu(
108            trigger.from_user.full_name, trigger.from_user.id
109        ),
110    )
111
112
113@router.callback_query(F.data == "bot_info")
114@bot_except
115async def bot_info(callback: CallbackQuery):
116    """Отправляет информацию о боте.
117
118    Args:
119        callback (CallbackQuery): Событие обратного вызова, содержащее информацию о запросе.
120
121    Отправляет сообщение с информацией о боте и кнопкой для доступа к дополнительной помощи.
122    """
123    await callback.message.answer(text.BOT_INFO, reply_markup=kb.static_join_button)
124    await more_help_info(callback)
125
126
127@router.callback_query(F.data.startswith("first_help_info_"))
128@bot_except
129async def next_help(callback: CallbackQuery):
130    """Обработчик перехода к следующему шагу помощи.
131
132    Args:
133        callback (CallbackQuery): Событие обратного вызова, содержащее информацию о запросе.
134
135    Отправляет сообщения с шагами помощи в зависимости от текущего шага.
136    """
137    current_step: str = callback.data.split("_")[-1]
138
139    await post_help_book(
140        callback,
141        book=text.BOT_STEPS,
142        step=current_step,
143        start_message="🤖 <b>Что делать дальше? Вот краткий алгоритм работы с нашим ботом и WireGuard:</b>",
144        prefix="first_help_info",
145    )
146
147
148@router.callback_query(F.data == "wg_help_info")
149@bot_except
150async def wg_help(callback: CallbackQuery):
151    """Отправляет информацию о платформе для установки WireGuard.
152
153    Args:
154        callback (CallbackQuery): Событие обратного вызова, содержащее информацию о запросе.
155
156    Отправляет сообщение с вопросом о платформе для установки WireGuard и кнопками для выбора.
157    """
158    await callback.message.answer(
159        "На какую платформу вы хотите установить WireGuard?",
160        reply_markup=kb.static_wg_platform_keyboard,
161    )
162
163
164@router.callback_query(F.data.startswith("wg_help_info_"))
165@bot_except
166async def wg_help_platform(callback: CallbackQuery):
167    """Обработчик помощи по установке WireGuard на выбранной платформе.
168
169    Args:
170        callback (CallbackQuery): Событие обратного вызова, содержащее информацию о запросе.
171
172    Отправляет сообщения с шагами помощи для установки WireGuard на выбранной платформе.
173    """
174    *_, current_platform, current_step = callback.data.split("_")
175
176    await post_help_book(
177        callback,
178        book=text.WG_STEPS[current_platform],
179        step=current_step,
180        start_message=f"🛠️ <b>Настройка WireGuard на {current_platform}:</b>",
181        prefix=f"wg_help_info_{current_platform}",
182    )
183
184
185@router.callback_query(F.data.startswith("error_help_info"))
186@bot_except
187async def error_help(callback: CallbackQuery):
188    """Обработчик помощи по устранению ошибок.
189
190    Args:
191        callback (CallbackQuery): Событие обратного вызова, содержащее информацию о запросе.
192
193    Отправляет сообщения с шагами помощи для устранения ошибок.
194    """
195    current_step: str = callback.data.split("_")[-1]
196
197    await post_help_book(
198        callback,
199        book=text.BOT_ERROR_STEP,
200        step=current_step,
201        start_message="📋 <b>Не волнуйтесь, вот инструкции по устранению проблем с DanVPN</b>",
202        prefix="error_help_info",
203    )
204
205
206@router.message(Command("time"))
207@bot_except
208async def started(message: Message, started_at):
209    """Отправляет время начала работы бота.
210
211    Args:
212        message (Message): Сообщение, инициировавшее команду.
213        started_at: Время начала работы бота.
214
215    Отправляет сообщение с временем начала работы бота.
216    """
217    await message.answer(f"Время начала работы бота: {started_at}")
218
219
220@router.message(Command("id"))
221@router.callback_query(F.data == "user_id_info")
222@bot_except
223async def start_bot(trigger: Union[Message, CallbackQuery]):
224    """Отправляет ID пользователя.
225
226    Args:
227        trigger (Union[Message, CallbackQuery]): Сообщение или событие обратного вызова, инициировавшее команду.
228
229    Отправляет сообщение с Telegram ID пользователя.
230    """
231    await getattr(trigger, "message", trigger).answer(
232        f'Ваш Telegram ID\n\n<span class="tg-spoiler">{trigger.from_user.id}</span>'
233    )
234
235
236@router.message(Command("cmd"))
237@router.message(Command("commands"))
238@router.message(F.text == "Команды")
239@router.callback_query(F.data == "cmd_help_info")
240@bot_except
241async def commands_list(trigger: Union[Message, CallbackQuery]):
242    """Отправляет список доступных команд.
243
244    Args:
245        trigger (Union[Message, CallbackQuery]): Сообщение или событие обратного вызова, инициировавшее команду.
246
247    Отправляет сообщение с перечислением доступных команд и их описанием.
248    """
249    help_t = as_list(
250        Bold("Запуск:"),
251        "/start - запуск (перезагрузка) бота",
252        Bold("Действия с аккаунтом:"),
253        as_marked_section(
254            "/account | /app - Основной функционал аккаунта",
255            "/reg - Регистрация в БД Бота",
256            "/freeze - Заморозить аккаунт",
257            "/recover - Разморозить аккаунт",
258            "/chat - Получить доступ в клиентский чат",
259            marker="~ ",
260        ),
261        Bold("Действия с конфигурациями:"),
262        as_marked_section(
263            "/me | /config - данные о моих конфигурациях wireguard",
264            "/create - создать конфигурацию",
265            marker="~ ",
266        ),
267        Bold("Действия с подпиской:"),
268        as_marked_section(
269            "/sub - Купить подписку",
270            "/refund - Сделать запрос на возврат средств",
271            "/history - История транзакций",
272            marker="~ ",
273        ),
274        Bold("Информация:"),
275        as_marked_section(
276            "/help - Помощь",
277            "/cmd - Список всех команд",
278            "/admin - функционал администратора",
279            "/bug - Доложить о баге",
280            "/id - Ваш Telegram ID",
281            "/time - время запуска бота",
282            marker="~ ",
283        ),
284        as_marked_section(
285            Bold("Расширенные возможности (тарифы от расширенного и выше):"),
286            "/server - Анализ работы сервера",
287            "/speed - Максимально доступная скорость VPN на данный момент",
288            "/mute - Отключить уведомления",
289        ),
290    )
291    await getattr(trigger, "message", trigger).answer(**help_t.as_kwargs())
292
293
294@router.callback_query(F.data == "freeze_info")
295@bot_except
296async def freeze_config_info(callback: CallbackQuery):
297    """Отправляет информацию о заморозке конфигураций.
298
299    Args:
300        callback (CallbackQuery): Событие обратного вызова, содержащее информацию о запросе.
301
302    Отправляет сообщение с информацией о том, как заморозить конфигурации и что делать в случае проблем.
303    """
304    await callback.message.answer(
305        "Конфигурации замораживаются, когда становятся недоступными. "
306        "\n\nПроверьте, достаточно ли средств на вашем счете? Соответствует ли количество ваших конфигураций вашему тарифу? "
307        "Возможно недавно произошли изменения в вашем тарифе, подождите несколько минут и попробуйте снова."
308        "\n\nЕсли вам все равно не понятно, почему ваша конфигурация заблокирована, воспользуйтесь командой /bug и сообщите о вашей проблеме."
309    )
310
311
312@router.callback_query(F.data == "freeze_account_info")
313@bot_except
314async def freeze_user_info(callback: CallbackQuery):
315    """Отправляет информацию о заморозке аккаунта.
316
317    Args:
318        callback (CallbackQuery): Событие обратного вызова, содержащее информацию о запросе.
319
320    Отправляет сообщение с информацией о том, как заморозить аккаунт и его последствиях.
321    """
322    await callback.message.answer(
323        "Заморозка аккаунта подразумевает приостановку ежедневных списаний "
324        "и, соответственно, блокировку всех созданных конфигураций."
325        "\nРазморозить свой аккаунт можно в меню /app. После разморозки восстановление конфигураций произойдет в течение 1 минуты."
326        "\n\n<b>Стоимость услуги равна одному ежедневному списанию вашего тарифа!</b>"
327        "\n<b>Разблокировка бесплатна.</b>",
328        reply_markup=kb.freeze_user_button,
329    )
330
331
332@router.message(Command("server"))
333@router.callback_query(F.data == "server_status")
334@bot_except
335async def server_status(trigger: Union[Message, CallbackQuery], bot: Bot):
336    """Отправляет информацию о состоянии сервера.
337
338    Args:
339        trigger (Union[Message, CallbackQuery]): Сообщение или событие обратного вызова, инициировавшее команду.
340        bot (Bot): Экземпляр бота для выполнения действий.
341
342    Отправляет сообщение с текущим состоянием сервера и его загрузкой.
343    """
344    await bot.send_chat_action(trigger.from_user.id, "typing")
345
346    user_data: UserData = await find_user(trigger)
347    if not user_data:
348        return
349    elif user_data.stage < 2:
350        await getattr(trigger, "message", trigger).answer(
351            "Команда заблокирована. Выберите тариф от 'Расширенного' или выше."
352        )
353        return
354
355    try:
356        wg = WgServerTools()
357
358        server_status = await wg.get_server_status()
359        cpu_usage = await wg.get_server_cpu_usage()
360
361    except WireguardError:
362        await getattr(trigger, "message", trigger).answer(text.WG_ERROR)
363
364    else:
365        server_data = (
366            "Текущие параметры сервера:\n\n"
367            f"🖥 Сервер:        <b>{server_status.capitalize()}</b>\n\n"
368            f"🦾 СPU usage:  <b>{cpu_usage}</b>"
369        )
370
371        await getattr(trigger, "message", trigger).answer(server_data)
372
373
374@router.message(Command("speed"))
375@router.callback_query(F.data == "server_speed")
376@bot_except
377async def server_speed(trigger: Union[Message, CallbackQuery], bot: Bot):
378    """Отправляет информацию о скорости сервера.
379
380    Args:
381        trigger (Union[Message, CallbackQuery]): Сообщение или событие обратного вызова, инициировавшее команду.
382        bot (Bot): Экземпляр бота для выполнения действий.
383
384    Отправляет сообщение с текущей доступной скоростью интернет-соединения по VPN.
385    """
386    await bot.send_chat_action(trigger.from_user.id, "typing")
387
388    user_data: UserData = await find_user(trigger)
389    if not user_data:
390        return
391    elif user_data.stage < 2:
392        await getattr(trigger, "message", trigger).answer(
393            "Команда заблокирована. Выберите тариф от 'Расширенного' или выше."
394        )
395        return
396
397    try:
398        server_speed_in, server_speed_out = await test_server_speed()
399
400    except WireguardError:
401        await getattr(trigger, "message", trigger).answer(
402            text.WG_ERROR, show_alert=True
403        )
404    except DatabaseError:
405        await trigger.answer(text=text.DB_ERROR, show_alert=True)
406    except BaseBotError:
407        await trigger.answer(
408            text="Ошибка измерения пропускной способности. Попробуйте позже.",
409            show_alert=True,
410        )
411
412    else:
413        server_data = (
414            "Текущая максимально доступная скорость интернет соединения по VPN:\n\n"
415            f"📥 Скачивание:  <b>{round(server_speed_in/1048576, 2)} Мбит/с</b>\n\n"
416            f"📤 Загрузка:        <b>{round(server_speed_out/1048576, 2)} Мбит/с</b>"
417        )
418
419        await getattr(trigger, "message", trigger).answer(server_data)
420
421
422@router.message(Command("chat"))
423@router.message(F.text == "Чат")
424@router.callback_query(F.data == "invite_to_chat")
425@bot_except
426async def get_chat_invite(trigger: Union[Message, CallbackQuery], bot: Bot):
427    """Отправляет приглашение в чат.
428
429    Args:
430        trigger (Union[Message, CallbackQuery]): Сообщение или событие обратного вызова, инициировавшее команду.
431        bot (Bot): Экземпляр бота для выполнения действий.
432
433    Отправляет сообщение с ссылкой на чат для зарегистрированных пользователей.
434    """
435    user_data = await get_user(trigger.from_user.id)
436
437    if user_data is None:
438        await getattr(trigger, "message", trigger).answer(
439            "Чат доступен только зарегистрированным пользователям. "
440            "Для начала выполните команду /reg"
441        )
442    else:
443        try:
444            link = await bot.create_chat_invite_link(
445                settings.BOT_CHAT, "chat", timedelta(hours=12), member_limit=1
446            )
447        except TelegramBadRequest:
448            logger.exception("Недоступен пользовательский чат")
449        else:
450            await getattr(trigger, "message", trigger).answer(
451                "🎉Поздравляем! Вам открыт доступ в клиентский чат.",
452                reply_markup=kb.get_chat_button(link.invite_link),
453            )