from fastapi import APIRouter, Depends, Query
from sqlalchemy.orm import Session

from app.core.database import get_db
from app.schemas.box import BoxInfoResponse, BoxSaveRequest, BoxSaveResponse
from app.services.box_service import get_box_info, save_box_record

router = APIRouter(tags=["boxing"])


@router.get("/box/info", response_model=BoxInfoResponse)
def box_info(
    zongpai_no: str = Query(..., description="总排号"),
    db: Session = Depends(get_db),
):
    """查询装箱信息：根据总排号查询排产号、数量和已有箱号明细。"""
    return get_box_info(db, zongpai_no)


@router.post("/box", response_model=BoxSaveResponse)
def create_box(req: BoxSaveRequest, db: Session = Depends(get_db)):
    """保存装箱记录：将总排号记录到指定箱号中。"""
    return save_box_record(db, req)