344 lines
12 KiB
Python
344 lines
12 KiB
Python
import logging
|
|
from aiogram import Router, F, types, Bot
|
|
from typing import Optional
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from datetime import datetime
|
|
|
|
from config.settings import Settings
|
|
from bot.services.subscription_service import SubscriptionService
|
|
from bot.services.panel_api_service import PanelApiService
|
|
from bot.services.notification_service import NotificationService
|
|
from bot.keyboards.inline.user_keyboards import (
|
|
get_trial_confirmation_keyboard,
|
|
get_main_menu_inline_keyboard,
|
|
get_connect_and_main_keyboard,
|
|
)
|
|
from bot.utils.config_link import prepare_config_links
|
|
from bot.middlewares.i18n import JsonI18n
|
|
from .start import send_main_menu
|
|
|
|
router = Router(name="user_trial_router")
|
|
|
|
|
|
async def request_trial_confirmation_handler(
|
|
callback: types.CallbackQuery,
|
|
settings: Settings,
|
|
i18n_data: dict,
|
|
subscription_service: SubscriptionService,
|
|
session: AsyncSession,
|
|
):
|
|
user_id = callback.from_user.id
|
|
current_lang = i18n_data.get("current_language", settings.DEFAULT_LANGUAGE)
|
|
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
|
_ = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs) if i18n else key
|
|
|
|
if not i18n or not callback.message:
|
|
try:
|
|
await callback.answer(_("error_occurred_try_again"), show_alert=True)
|
|
except Exception:
|
|
pass
|
|
return
|
|
|
|
show_trial_btn_in_menu_if_fail = False
|
|
if settings.TRIAL_ENABLED:
|
|
if not await subscription_service.has_had_any_subscription(session, user_id):
|
|
show_trial_btn_in_menu_if_fail = True
|
|
|
|
if not settings.TRIAL_ENABLED:
|
|
await callback.message.edit_text(
|
|
_("trial_feature_disabled"),
|
|
reply_markup=get_main_menu_inline_keyboard(
|
|
current_lang, i18n, settings, False
|
|
),
|
|
)
|
|
try:
|
|
await callback.answer()
|
|
except Exception:
|
|
pass
|
|
return
|
|
|
|
if await subscription_service.has_had_any_subscription(session, user_id):
|
|
await callback.message.edit_text(
|
|
_("trial_already_had_subscription_or_trial"),
|
|
reply_markup=get_main_menu_inline_keyboard(
|
|
current_lang, i18n, settings, False
|
|
),
|
|
)
|
|
try:
|
|
await callback.answer()
|
|
except Exception:
|
|
pass
|
|
return
|
|
|
|
# Directly activate trial without confirmation
|
|
activation_result = await subscription_service.activate_trial_subscription(
|
|
session, user_id
|
|
)
|
|
|
|
final_message_text_in_chat = ""
|
|
show_trial_button_after_action = False
|
|
config_link_display_for_trial = None
|
|
config_link_for_trial = None
|
|
connect_button_url_for_trial = None
|
|
|
|
if activation_result and activation_result.get("activated"):
|
|
try:
|
|
await callback.answer(_("trial_activated_alert"), show_alert=True)
|
|
except Exception:
|
|
pass
|
|
|
|
end_date_obj = activation_result.get("end_date")
|
|
config_link_display_for_trial, connect_button_url_for_trial = await prepare_config_links(
|
|
settings, activation_result.get("subscription_url")
|
|
)
|
|
config_link_for_trial = config_link_display_for_trial or _("config_link_not_available")
|
|
|
|
traffic_gb_val = activation_result.get(
|
|
"traffic_gb", settings.TRIAL_TRAFFIC_LIMIT_GB
|
|
)
|
|
traffic_display = (
|
|
f"{traffic_gb_val} GB"
|
|
if traffic_gb_val and traffic_gb_val > 0
|
|
else _("traffic_unlimited")
|
|
)
|
|
|
|
final_message_text_in_chat = _(
|
|
"trial_activated_details_message",
|
|
days=activation_result.get("days", settings.TRIAL_DURATION_DAYS),
|
|
end_date=(
|
|
end_date_obj.strftime("%Y-%m-%d")
|
|
if isinstance(end_date_obj, datetime)
|
|
else "N/A"
|
|
),
|
|
config_link=config_link_for_trial,
|
|
traffic_gb=traffic_display,
|
|
)
|
|
|
|
# Send notification to admin about new trial
|
|
notification_service = NotificationService(callback.bot, settings, i18n)
|
|
await notification_service.notify_trial_activation(user_id, end_date_obj)
|
|
# Mark ad attribution trial if exists
|
|
try:
|
|
from db.dal import ad_dal as _ad_dal
|
|
await _ad_dal.mark_trial_activated(session, user_id)
|
|
await session.commit()
|
|
except Exception as e_mark:
|
|
await session.rollback()
|
|
logging.error(f"Failed to mark trial for ad attribution for user {user_id}: {e_mark}")
|
|
else:
|
|
message_key_from_service = (
|
|
activation_result.get("message_key", "trial_activation_failed")
|
|
if activation_result
|
|
else "trial_activation_failed"
|
|
)
|
|
final_message_text_in_chat = _(message_key_from_service)
|
|
try:
|
|
await callback.answer(final_message_text_in_chat, show_alert=True)
|
|
except Exception:
|
|
pass
|
|
if (
|
|
settings.TRIAL_ENABLED
|
|
and not await subscription_service.has_had_any_subscription(
|
|
session, user_id
|
|
)
|
|
):
|
|
show_trial_button_after_action = True
|
|
|
|
reply_markup = (
|
|
get_connect_and_main_keyboard(
|
|
current_lang,
|
|
i18n,
|
|
settings,
|
|
config_link_display_for_trial,
|
|
connect_button_url=connect_button_url_for_trial,
|
|
)
|
|
if activation_result and activation_result.get("activated")
|
|
else get_main_menu_inline_keyboard(
|
|
current_lang, i18n, settings, show_trial_button_after_action
|
|
)
|
|
)
|
|
|
|
try:
|
|
await callback.message.edit_text(
|
|
final_message_text_in_chat,
|
|
parse_mode="HTML",
|
|
reply_markup=reply_markup,
|
|
disable_web_page_preview=True,
|
|
)
|
|
except Exception as e_edit:
|
|
logging.warning(
|
|
f"Could not edit trial result message: {e_edit}. Sending new one."
|
|
)
|
|
|
|
if callback.message:
|
|
await callback.message.answer(
|
|
final_message_text_in_chat,
|
|
parse_mode="HTML",
|
|
reply_markup=reply_markup,
|
|
disable_web_page_preview=True,
|
|
)
|
|
|
|
|
|
@router.callback_query(F.data == "trial_action:confirm_activate")
|
|
async def confirm_activate_trial_handler(
|
|
callback: types.CallbackQuery,
|
|
settings: Settings,
|
|
i18n_data: dict,
|
|
subscription_service: SubscriptionService,
|
|
panel_service: PanelApiService,
|
|
session: AsyncSession,
|
|
):
|
|
user_id = callback.from_user.id
|
|
|
|
current_lang = i18n_data.get("current_language", settings.DEFAULT_LANGUAGE)
|
|
i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance")
|
|
_ = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs) if i18n else key
|
|
|
|
if not i18n or not callback.message:
|
|
try:
|
|
await callback.answer(_("error_occurred_try_again"), show_alert=True)
|
|
except Exception:
|
|
pass
|
|
return
|
|
|
|
if not settings.TRIAL_ENABLED:
|
|
try:
|
|
await callback.answer(_("trial_feature_disabled"), show_alert=True)
|
|
except Exception:
|
|
pass
|
|
|
|
await send_main_menu(
|
|
callback, settings, i18n_data, subscription_service, session, is_edit=True
|
|
)
|
|
return
|
|
if await subscription_service.has_had_any_subscription(session, user_id):
|
|
try:
|
|
await callback.answer(
|
|
_("trial_already_had_subscription_or_trial"), show_alert=True
|
|
)
|
|
except Exception:
|
|
pass
|
|
await send_main_menu(
|
|
callback, settings, i18n_data, subscription_service, session, is_edit=True
|
|
)
|
|
return
|
|
|
|
activation_result = await subscription_service.activate_trial_subscription(
|
|
session, user_id
|
|
)
|
|
|
|
final_message_text_in_chat = ""
|
|
show_trial_button_after_action = False
|
|
config_link_display_for_trial = None
|
|
config_link_for_trial = None
|
|
connect_button_url_for_trial = None
|
|
|
|
if activation_result and activation_result.get("activated"):
|
|
try:
|
|
await callback.answer(_("trial_activated_alert"), show_alert=True)
|
|
except Exception:
|
|
pass
|
|
|
|
end_date_obj = activation_result.get("end_date")
|
|
config_link_display_for_trial, connect_button_url_for_trial = await prepare_config_links(
|
|
settings, activation_result.get("subscription_url")
|
|
)
|
|
config_link_for_trial = config_link_display_for_trial or _("config_link_not_available")
|
|
|
|
traffic_gb_val = activation_result.get(
|
|
"traffic_gb", settings.TRIAL_TRAFFIC_LIMIT_GB
|
|
)
|
|
traffic_display = (
|
|
f"{traffic_gb_val} GB"
|
|
if traffic_gb_val and traffic_gb_val > 0
|
|
else _("traffic_unlimited")
|
|
)
|
|
|
|
final_message_text_in_chat = _(
|
|
"trial_activated_details_message",
|
|
days=activation_result.get("days", settings.TRIAL_DURATION_DAYS),
|
|
end_date=(
|
|
end_date_obj.strftime("%Y-%m-%d")
|
|
if isinstance(end_date_obj, datetime)
|
|
else "N/A"
|
|
),
|
|
config_link=config_link_for_trial,
|
|
traffic_gb=traffic_display,
|
|
)
|
|
else:
|
|
message_key_from_service = (
|
|
activation_result.get("message_key", "trial_activation_failed")
|
|
if activation_result
|
|
else "trial_activation_failed"
|
|
)
|
|
final_message_text_in_chat = _(message_key_from_service)
|
|
try:
|
|
await callback.answer(final_message_text_in_chat, show_alert=True)
|
|
except Exception:
|
|
pass
|
|
if (
|
|
settings.TRIAL_ENABLED
|
|
and not await subscription_service.has_had_any_subscription(
|
|
session, user_id
|
|
)
|
|
):
|
|
show_trial_button_after_action = True
|
|
|
|
reply_markup = (
|
|
get_connect_and_main_keyboard(
|
|
current_lang,
|
|
i18n,
|
|
settings,
|
|
config_link_display_for_trial,
|
|
connect_button_url=connect_button_url_for_trial,
|
|
)
|
|
if activation_result and activation_result.get("activated")
|
|
else get_main_menu_inline_keyboard(
|
|
current_lang, i18n, settings, show_trial_button_after_action
|
|
)
|
|
)
|
|
|
|
try:
|
|
await callback.message.edit_text(
|
|
final_message_text_in_chat,
|
|
parse_mode="HTML",
|
|
reply_markup=reply_markup,
|
|
disable_web_page_preview=True,
|
|
)
|
|
except Exception as e_edit:
|
|
logging.warning(
|
|
f"Could not edit trial result message: {e_edit}. Sending new one."
|
|
)
|
|
|
|
if callback.message:
|
|
await callback.message.answer(
|
|
final_message_text_in_chat,
|
|
parse_mode="HTML",
|
|
reply_markup=reply_markup,
|
|
disable_web_page_preview=True,
|
|
)
|
|
|
|
if activation_result and activation_result.get("activated") and end_date_obj:
|
|
notification_service = NotificationService(callback.bot, settings, i18n)
|
|
await notification_service.notify_trial_activation(user_id, end_date_obj)
|
|
try:
|
|
from db.dal import ad_dal as _ad_dal
|
|
await _ad_dal.mark_trial_activated(session, user_id)
|
|
await session.commit()
|
|
except Exception as e_mark:
|
|
await session.rollback()
|
|
logging.error(f"Failed to mark trial for ad attribution for user {user_id}: {e_mark}")
|
|
|
|
|
|
@router.callback_query(F.data == "main_action:cancel_trial")
|
|
async def cancel_trial_activation(
|
|
callback: types.CallbackQuery,
|
|
settings: Settings,
|
|
i18n_data: dict,
|
|
subscription_service: SubscriptionService,
|
|
session: AsyncSession,
|
|
):
|
|
await send_main_menu(
|
|
callback, settings, i18n_data, subscription_service, session, is_edit=True
|
|
)
|