from datetime import date
from typing import Annotated
from fastapi import APIRouter, Depends, Query
import asyncpg

from app.core.pagination import pagination_params
from app.core.responses import ok, paginated
from app.dependencies import get_db, get_current_user, get_current_user_optional
from app.models.bookings import BookingCreate
from app.services import booking_service, slot_service

router = APIRouter(prefix="/bookings", tags=["Bookings"])


@router.get("/slots")
async def get_available_slots(
    from_date: date = Query(...),
    to_date: date = Query(...),
    db: asyncpg.Connection = Depends(get_db),
):
    slots = await slot_service.list_slots(db, from_date, to_date, available_only=True)
    return ok(slots)


@router.post("", status_code=201)
async def create_booking(
    body: BookingCreate,
    user: dict | None = Depends(get_current_user_optional),
    db: asyncpg.Connection = Depends(get_db),
):
    result = await booking_service.create_booking(db, body, user=user)
    return ok(result)


@router.get("")
async def list_my_bookings(
    pagination: Annotated[tuple, Depends(pagination_params)],
    status: str | None = Query(None),
    user: dict = Depends(get_current_user),
    db: asyncpg.Connection = Depends(get_db),
):
    page, per_page, offset = pagination
    bookings, total = await booking_service.list_bookings(
        db, page, per_page, offset, user_id=str(user["id"]), status=status
    )
    return paginated(bookings, total, page, per_page)


@router.get("/{booking_id}")
async def get_booking(
    booking_id: str,
    user: dict = Depends(get_current_user),
    db: asyncpg.Connection = Depends(get_db),
):
    booking = await booking_service.get_booking(db, booking_id, user_id=str(user["id"]))
    return ok(booking)


@router.delete("/{booking_id}", status_code=204)
async def cancel_booking(
    booking_id: str,
    user: dict = Depends(get_current_user),
    db: asyncpg.Connection = Depends(get_db),
):
    await booking_service.cancel_booking(db, booking_id, str(user["id"]))