import uuid

from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy.ext.asyncio import AsyncSession

from backend.app.core.database import get_session
from backend.app.models.rider import Rider
from backend.app.schemas.rider import RiderCreate, RiderUpdate, RiderResponse

router = APIRouter()


@router.post("/profile", response_model=RiderResponse)
async def create_rider(
    data: RiderCreate,
    session: AsyncSession = Depends(get_session),
):
    rider = Rider(**data.model_dump())
    session.add(rider)
    await session.commit()
    await session.refresh(rider)
    return rider


@router.get("/profile/{rider_id}", response_model=RiderResponse)
async def get_rider(
    rider_id: uuid.UUID,
    session: AsyncSession = Depends(get_session),
):
    rider = await session.get(Rider, rider_id)
    if not rider:
        raise HTTPException(status_code=404, detail="Rider not found")
    return rider


@router.put("/profile/{rider_id}", response_model=RiderResponse)
async def update_rider(
    rider_id: uuid.UUID,
    data: RiderUpdate,
    session: AsyncSession = Depends(get_session),
):
    rider = await session.get(Rider, rider_id)
    if not rider:
        raise HTTPException(status_code=404, detail="Rider not found")

    update_data = data.model_dump(exclude_unset=True)
    for key, value in update_data.items():
        setattr(rider, key, value)

    await session.commit()
    await session.refresh(rider)
    return rider