import logging from datetime import datetime from typing import Optional from aiogram import F, Router, types from sqlalchemy.ext.asyncio import AsyncSession from bot.keyboards.inline.user_keyboards import get_payment_url_keyboard from bot.middlewares.i18n import JsonI18n from bot.services.freekassa_service import FreeKassaService from config.settings import Settings from db.dal import payment_dal router = Router(name="user_subscription_payments_freekassa_router") @router.callback_query(F.data.startswith("pay_fk:")) async def pay_fk_callback_handler( callback: types.CallbackQuery, settings: Settings, i18n_data: dict, freekassa_service: FreeKassaService, session: AsyncSession, ): current_lang = i18n_data.get("current_language", settings.DEFAULT_LANGUAGE) i18n: Optional[JsonI18n] = i18n_data.get("i18n_instance") get_text = lambda key, **kwargs: i18n.gettext(current_lang, key, **kwargs) if i18n else key if not i18n or not callback.message: try: await callback.answer(get_text("error_occurred_try_again"), show_alert=True) except Exception: pass return if not freekassa_service or not freekassa_service.configured: logging.error("FreeKassa service is not configured or unavailable.") try: await callback.answer(get_text("payment_service_unavailable_alert"), show_alert=True) except Exception: pass try: await callback.message.edit_text(get_text("payment_service_unavailable")) except Exception: pass return try: _, data_payload = callback.data.split(":", 1) parts = data_payload.split(":") months = float(parts[0]) price_rub = float(parts[1]) sale_mode = parts[2] if len(parts) > 2 else "subscription" except (ValueError, IndexError): logging.error(f"Invalid pay_fk data in callback: {callback.data}") try: await callback.answer(get_text("error_try_again"), show_alert=True) except Exception: pass return user_id = callback.from_user.id human_value = str(int(months)) if float(months).is_integer() else f"{months:g}" sale_base = sale_mode.split("@", 1)[0].split("|", 1)[0] payment_description = ( get_text("payment_description_traffic", traffic_gb=human_value) if sale_base in {"traffic", "traffic_package", "topup"} else (get_text("payment_description_hwid_devices", count=int(months)) if sale_base in {"hwid_device", "hwid_devices"} else get_text("payment_description_subscription", months=int(months))) ) currency_code = getattr(freekassa_service, "default_currency", None) or settings.DEFAULT_CURRENCY_SYMBOL or "RUB" payment_record_payload = { "user_id": user_id, "amount": price_rub, "currency": currency_code, "status": "pending_freekassa", "description": payment_description, "subscription_duration_months": int(months), "provider": "freekassa", "sale_mode": sale_mode, "tariff_key": sale_mode.split("@", 1)[1] if "@" in sale_mode else None, "purchased_gb": float(months) if sale_base in {"traffic", "traffic_package", "topup"} else None, "purchased_hwid_devices": int(months) if sale_base in {"hwid_device", "hwid_devices"} else None, } try: payment_record = await payment_dal.create_payment_record(session, payment_record_payload) await session.commit() except Exception as e_db_create: await session.rollback() logging.error( f"FreeKassa: failed to create payment record for user {user_id}: {e_db_create}", exc_info=True, ) try: await callback.message.edit_text(get_text("error_creating_payment_record")) except Exception: pass try: await callback.answer(get_text("error_try_again"), show_alert=True) except Exception: pass return success, response_data = await freekassa_service.create_order( payment_db_id=payment_record.payment_id, user_id=payment_record.user_id, months=months, amount=price_rub, currency=freekassa_service.default_currency, payment_method_id=freekassa_service.payment_method_id, ip_address=freekassa_service.server_ip, extra_params={ "us_method": freekassa_service.payment_method_id, }, ) if success: location = response_data.get("location") order_hash = response_data.get("orderHash") order_id_api = response_data.get("orderId") provider_identifier = order_hash or order_id_api if provider_identifier: try: await payment_dal.update_provider_payment_and_status( session, payment_record.payment_id, str(provider_identifier), payment_record.status, ) await session.commit() except Exception as e_status: await session.rollback() logging.error( f"FreeKassa: failed to store provider order id for payment {payment_record.payment_id}: {e_status}", exc_info=True, ) if location: order_identifier_display = str(order_id_api or provider_identifier or payment_record.payment_id) order_info_text = get_text( "free_kassa_order_info", order_id=order_identifier_display, date=datetime.now().strftime("%Y-%m-%d"), ) try: await callback.message.edit_text( f"{order_info_text}\n\n" + get_text( key="payment_link_message_traffic" if sale_mode == "traffic" else "payment_link_message", months=int(months), traffic_gb=human_value, ), reply_markup=get_payment_url_keyboard( location, current_lang, i18n, back_callback=f"subscribe_period:{human_value}", back_text_key="back_to_payment_methods_button", ), disable_web_page_preview=False, ) except Exception as e_edit: logging.warning(f"FreeKassa: failed to display payment link ({e_edit}), sending new message.") try: await callback.message.answer( f"{order_info_text}\n\n" + get_text( key="payment_link_message_traffic" if sale_mode == "traffic" else "payment_link_message", months=int(months), traffic_gb=human_value, ), reply_markup=get_payment_url_keyboard( location, current_lang, i18n, back_callback=f"subscribe_period:{human_value}", back_text_key="back_to_payment_methods_button", ), disable_web_page_preview=False, ) except Exception: pass try: await callback.answer() except Exception: pass return logging.error( "FreeKassa: create_order succeeded but no payment link returned for payment %s. Response: %s", payment_record.payment_id, response_data, ) else: logging.error( "FreeKassa: create_order failed for payment %s with response %s", payment_record.payment_id, response_data, ) try: await payment_dal.update_payment_status_by_db_id( session, payment_record.payment_id, "failed_creation", ) await session.commit() except Exception as e_status: await session.rollback() logging.error(f"FreeKassa: failed to mark payment {payment_record.payment_id} as failed_creation: {e_status}", exc_info=True) try: await callback.message.edit_text(get_text("error_payment_gateway")) except Exception: pass try: await callback.answer(get_text("error_payment_gateway"), show_alert=True) except Exception: pass