import logging

from aiogram import F, Router
from aiogram.fsm.context import FSMContext
from aiogram.types import CallbackQuery
from sqlalchemy.ext.asyncio import AsyncSession

from bot.keyboards.client import pay_url
from bot.keyboards.common import return_to_menu
from bot.states.buy import BillingStorage
from bot.texts import BILL_CREATED, BILL_CREATION, CALLBACK_FALLBACK
from schemas.billing import BillingContext, BillingProviders
from schemas.di import DependenciesDTO

router = Router()
logger = logging.getLogger(__name__)


@router.callback_query(BillingStorage.pending, F.data == "billing:pally")
async def pally_init(
    cb: CallbackQuery, state: FSMContext, session: AsyncSession, deps: DependenciesDTO
):
    data = await state.get_data()
    ctx: BillingContext | None = data.get("ctx")
    if not ctx:
        logger.warning("ctx not found on billing choice")
        await cb.message.edit_text(CALLBACK_FALLBACK, reply_markup=return_to_menu)
        return

    await state.clear()

    ctx.provider = BillingProviders.PALLY
    ctx.amount = int(ctx.amount)

    await cb.message.edit_text(
        BILL_CREATION.format(amount=ctx.amount, provider=ctx.provider.value.title())
    )

    bill = await deps.billing_service.pally_initiate_payment(
        session, deps.pally_client, user_id=cb.from_user.id, amount=ctx.amount
    )

    if not bill:
        await cb.message.edit_text(CALLBACK_FALLBACK, reply_markup=return_to_menu)
        logger.critical("bill creation prompted, but bill is None")
        return

    await cb.message.edit_text(
        BILL_CREATED.format(amount=ctx.amount, provider=ctx.provider.value.title()),
        reply_markup=pay_url(bill.link_page_url),
    )