"""Router for Pemindahan Produk endpoints."""

from datetime import date

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

from app.core.database import get_db
from app.dependencies import get_current_user
from app.models.user import User
from app.schemas.pemindahan import PemindahanCreate
from app.services import pemindahan_service

router = APIRouter(
    prefix="/spbus/{spbu_id}/pemindahan-produk",
    tags=["pemindahan"],
)


@router.get("", response_model=dict)
async def list_pemindahan(
    spbu_id: int,
    tanggal_mulai: date | None = Query(default=None),
    tanggal_akhir: date | None = Query(default=None),
    skip: int = Query(default=0, ge=0),
    limit: int = Query(default=50, ge=1, le=200),
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user),
) -> dict:
    rows = await pemindahan_service.list_pemindahan(
        db, spbu_id, tanggal_mulai=tanggal_mulai, tanggal_akhir=tanggal_akhir, skip=skip, limit=limit
    )
    return {"data": [r.model_dump() for r in rows]}


@router.get("/{pemindahan_id}", response_model=dict)
async def get_pemindahan(
    spbu_id: int,
    pemindahan_id: int,
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user),
) -> dict:
    try:
        result = await pemindahan_service.get_pemindahan(db, spbu_id, pemindahan_id)
    except ValueError as e:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(e))
    return {"data": result.model_dump()}


@router.post("", response_model=dict, status_code=status.HTTP_201_CREATED)
async def create_pemindahan(
    spbu_id: int,
    data: PemindahanCreate,
    db: AsyncSession = Depends(get_db),
    current_user: User = Depends(get_current_user),
) -> dict:
    try:
        result = await pemindahan_service.create_pemindahan(db, spbu_id, data, current_user.id)
    except ValueError as e:
        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(e))
    return {"data": result.model_dump(), "message": "Pemindahan produk berhasil dicatat"}