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

from dto.checkout import CheckoutContext
from misc.kb.client import (
    order_confirmation,
    order_specs_confirmation,
    return_menu,
)
from misc.states import CheckoutStorage
from misc.texts import get_order_item_list
from misc.utils import is_valid_phone
from repositories.orders import OrderRepository
from services.orders import OrderService

router = Router()


@router.callback_query(F.data == "checkout")
async def checkout(
    cb: CallbackQuery,
    state: FSMContext,
    session: AsyncSession,
    order_service: OrderService,
):
    await state.clear()

    cart = await order_service.build_full_cart_dto(session, cb.from_user.id)
    await cb.message.edit_text(
        text=get_order_item_list(cart), reply_markup=order_confirmation(cart.order_id)
    )


@router.callback_query(F.data.startswith("confirm_order:"))
async def order_confirm(
    cb: CallbackQuery,
    state: FSMContext,
    session: AsyncSession,
    order_repo: OrderRepository,
):
    await state.clear()

    order_id = cb.data.split(":")[1]
    if not order_id.isdigit():
        await cb.message.edit_text("🍃 Что-то пошло не так.", reply_markup=return_menu)
        return

    order = await order_repo.get_order_by_id(session, int(order_id))
    if not order or order.customer != cb.from_user.id:
        await cb.message.edit_text("🍃 Что-то пошло не так.", reply_markup=return_menu)
        return

    await cb.message.edit_text("<b>👤 Укажите ваше имя:</b>", reply_markup=return_menu)

    ctx = CheckoutContext(orig_msg=cb.message, order_id=order.id)
    await state.set_state(CheckoutStorage.name)
    await state.set_data({"ctx": ctx})


@router.message(CheckoutStorage.name)
async def checkout_name(msg: Message, state: FSMContext):
    data = await state.get_data()
    ctx: CheckoutContext = data.get("ctx", CheckoutContext)
    await state.clear()

    ctx.name = msg.text
    await msg.delete()

    await ctx.orig_msg.edit_text(
        f"<i>👤 Имя: {ctx.name}</i>\n\n<b>📱 Укажите ваш номер телефона:</b>",
        reply_markup=return_menu,
    )

    await state.set_state(CheckoutStorage.phone)
    await state.set_data({"ctx": ctx})


@router.message(CheckoutStorage.phone)
async def checkout_phone(msg: Message, state: FSMContext):
    data = await state.get_data()
    ctx: CheckoutContext = data.get("ctx", CheckoutContext)
    await state.clear()

    await msg.delete()
    if not is_valid_phone(msg.text):
        await ctx.orig_msg.edit_text(
            f"❌ <i>{msg.text}</i> не является действительным номером, попробуйте ещё раз.",
            reply_markup=return_menu,
        )
        await state.set_state(CheckoutStorage.phone)
        await state.set_data({"ctx": ctx})
        return

    ctx.phone = msg.text
    await ctx.orig_msg.edit_text(
        f"<i>👤 Имя: {ctx.name}</i>\n<i>📱 Номер телефона: {ctx.phone}</i>\n\n<b>📍 Укажите ваш адрес (город, улица):</b>",
        reply_markup=return_menu,
    )
    await state.set_state(CheckoutStorage.address)
    await state.set_data({"ctx": ctx})


@router.message(CheckoutStorage.address)
async def checkout_address(msg: Message, state: FSMContext):
    data = await state.get_data()
    ctx: CheckoutContext = data.get("ctx", CheckoutContext)
    await state.clear()

    await msg.delete()

    ctx.address = msg.text
    await ctx.orig_msg.edit_text(
        "<b>‼️ Проверьте правильность ваших данных.</b>\n\n"
        "━━━━━━━━━━━━━━\n\n"
        f"<i>👤 Имя: {ctx.name}</i>\n"
        f"<i>📱 Номер телефона: {ctx.phone}</i>\n"
        f"<i>📍 Ваш адрес: {ctx.address}</i>\n\n"
        "━━━━━━━━━━━━━━",
        reply_markup=order_specs_confirmation(ctx.order_id),
    )  # TODO: State-driven Payment System.