"""My bookings handler - view, edit, cancel bookings."""
from aiogram import Router, F
from aiogram.types import Message, CallbackQuery
from aiogram.fsm.context import FSMContext

from app.config import USE_POSTGRES
from app.services.booking_service import get_user_bookings, cancel_booking
from app.keyboards.my_booking import get_booking_actions
from app.keyboards.services import get_services_keyboard
from app.states.booking import BookingState
from app.logger import logger

router = Router()


@router.message(F.text == "\U0001f4c4 \u041c\u043e\u0438 \u0437\u0430\u043f\u0438\u0441\u0438")
async def my_bookings(message: Message):
    logger.info("User {uid} opened My Bookings", uid=message.from_user.id)
    try:
        bookings = await get_user_bookings(message.from_user.id)
        logger.debug("Found {n} bookings", n=len(bookings))

        if not bookings:
            await message.answer("\u0423 \u0432\u0430\u0441 \u043d\u0435\u0442 \u0430\u043a\u0442\u0438\u0432\u043d\u044b\u0445 \u0437\u0430\u043f\u0438\u0441\u0435\u0439")
            return

        for booking in bookings:
            svc = booking["service"]
            mst = booking["master"]
            dat = booking["date"]
            tim = booking["time"]
            text = (
                "\U0001f486 " + svc + "\n"
                "\U0001f464 " + mst + "\n"
                "\U0001f4c5 " + dat + "\n"
                "\U0001f552 " + tim
            )
            await message.answer(text, reply_markup=get_booking_actions(booking["id"]))
    except Exception as e:
        logger.exception("Error in my_bookings: {error}", error=e)
        await message.answer("\u041f\u0440\u043e\u0438\u0437\u043e\u0448\u043b\u0430 \u043e\u0448\u0438\u0431\u043a\u0430. \u041f\u043e\u043f\u0440\u043e\u0431\u0443\u0439\u0442\u0435 \u043f\u043e\u0437\u0436\u0435.")


@router.callback_query(F.data.startswith("cancel_booking_"))
async def cancel_booking_handler(callback: CallbackQuery):
    booking_id = int(callback.data.split("_")[2])
    logger.info("User {uid} cancelled booking #{bid}", uid=callback.from_user.id, bid=booking_id)
    await cancel_booking(booking_id)
    await callback.message.edit_text("\u2705 \u0417\u0430\u043f\u0438\u0441\u044c \u043e\u0442\u043c\u0435\u043d\u0435\u043d\u0430")
    await callback.answer()


@router.callback_query(F.data.startswith("edit_booking_"))
async def edit_booking_handler(callback: CallbackQuery, state: FSMContext):
    booking_id = int(callback.data.split("_")[2])

    if USE_POSTGRES:
        from app.db.postgres import PostgresDB
        booking = await PostgresDB.get_booking_by_id(booking_id)
    else:
        from app.db.memory import load_bookings
        bookings = load_bookings()
        booking = next((b for b in bookings if b["id"] == booking_id), None)

    if not booking:
        await callback.message.edit_text("\u0417\u0430\u043f\u0438\u0441\u044c \u043d\u0435 \u043d\u0430\u0439\u0434\u0435\u043d\u0430")
        await callback.answer()
        return

    await state.set_data({
        "editing_booking_id": booking_id,
        "service": booking.get("service", ""),
        "master": booking.get("master", ""),
        "date": booking.get("date", ""),
        "time": booking.get("time", ""),
    })

    await state.set_state(BookingState.service)
    await callback.message.edit_text(
        "\u270f\ufe0f \u0420\u0435\u0434\u0430\u043a\u0442\u0438\u0440\u043e\u0432\u0430\u043d\u0438\u0435 \u0437\u0430\u043f\u0438\u0441\u0438 #" + str(booking_id) + "\n\n\u0412\u044b\u0431\u0435\u0440\u0438\u0442\u0435 \u043d\u043e\u0432\u0443\u044e \u0443\u0441\u043b\u0443\u0433\u0443:",
        reply_markup=get_services_keyboard()
    )
    await callback.answer()