from aiogram import F, Router, types
from aiogram.fsm.context import FSMContext
from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession

from database import get_temporary_data
from database.models import User
from handlers.buttons import MAIN_MENU, PAY_2
from handlers.payments.currency_rates import format_for_user
from handlers.texts import DEFAULT_PAYMENT_MESSAGE
from handlers.utils import edit_or_send_message
from logger import logger
from ..constants import ALLOWED_TEMP_PAYMENT_STATES

from .service import (
    HELEKET_METHODS,
    generate_heleket_payment_link,
    process_callback_pay_heleket,
    router as service_router,
)

router = Router(name="heleket_router")
router.include_router(service_router)


@router.callback_query(F.data == "pay_heleket_crypto")
async def handle_pay_heleket_crypto(
    callback_query: types.CallbackQuery,
    state: FSMContext,
    session: AsyncSession,
):
    """Обработчик оплаты через Heleket криптовалютой."""
    await process_callback_pay_heleket(callback_query, state, session, method_name="crypto")


async def handle_custom_amount_input_heleket(
    event,
    session: AsyncSession,
    pay_button_text: str = PAY_2,
    main_menu_text: str = MAIN_MENU,
):
    """Функция быстрого потока для Heleket.

    Принимает недостающую сумму и формирует платеж.
    Работает с временными данными из fast_payment_flow для
    создания/продления/подарка.
    """
    message = event.message
    from_user = event.from_user
    tg_id = from_user.id

    temp_data = await get_temporary_data(session, tg_id)
    if not temp_data or temp_data["state"] not in ALLOWED_TEMP_PAYMENT_STATES:
        await edit_or_send_message(
            target_message=message,
            text="❌ Не удалось получить данные для оплаты.",
        )
        return

    amount = int(temp_data["data"].get("required_amount", 0))
    if amount <= 0:
        await edit_or_send_message(
            target_message=message,
            text="❌ Не удалось определить сумму оплаты.",
        )
        return

    if amount < 10:
        await edit_or_send_message(
            target_message=message,
            text="❌ Минимальная сумма для оплаты криптовалютой — 10₽ (≈0.1$).",
        )
        return

    enabled_methods = [m for m in HELEKET_METHODS.values() if m["enable"]]
    if not enabled_methods:
        await edit_or_send_message(
            target_message=message,
            text="❌ Способ оплаты Heleket временно недоступен.",
        )
        return
    method = enabled_methods[0]

    try:
        payment_url = await generate_heleket_payment_link(amount, tg_id, method)

        if not payment_url or payment_url == "https://heleket.com/":
            await edit_or_send_message(
                target_message=message,
                text=("❌ Произошла ошибка при создании платежа. Попробуйте позже или выберите другой способ оплаты."),
            )
            return

        markup = InlineKeyboardMarkup(
            inline_keyboard=[
                [InlineKeyboardButton(text=pay_button_text, url=payment_url)],
                [InlineKeyboardButton(text=main_menu_text, callback_data="profile")],
            ]
        )

        result = await session.execute(select(User.language_code).where(User.tg_id == tg_id))
        language_code = result.scalar_one_or_none()
        amount_text = await format_for_user(
            session,
            tg_id,
            float(amount),
            language_code,
            force_currency="USD",
        )
        text_out = DEFAULT_PAYMENT_MESSAGE.format(amount=amount_text)

        await edit_or_send_message(target_message=message, text=text_out, reply_markup=markup)
    except Exception as e:
        logger.error(f"Ошибка при создании платежа Heleket для пользователя {tg_id}: {e}")
        await edit_or_send_message(
            target_message=message,
            text="Произошла ошибка при создании платежа. Попробуйте позже.",
            reply_markup=InlineKeyboardMarkup(inline_keyboard=[]),
        )