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

from backend.app.api.deps import get_supplier_service, require_roles
from backend.app.core.error_codes import ErrorCode
from backend.app.core.exceptions import AppException
from backend.app.db import get_db_session
from backend.app.schemas.common import success_payload
from backend.app.schemas.suppliers import CreateSupplierRequest, UpdateSupplierRequest
from backend.app.services.supplier_service import SupplierService

router = APIRouter(prefix="/api/suppliers", tags=["suppliers"])


@router.get("")
def list_suppliers(
    supplier_name: str | None = Query(default=None),
    supplier_type: str | None = Query(default=None),
    status: int | None = Query(default=None),
    page_no: int = Query(default=1),
    page_size: int = Query(default=20),
    supplier_service: SupplierService = Depends(get_supplier_service),
    session: Session = Depends(get_db_session),
    current_user: dict = Depends(require_roles("salesman", "manager", "admin")),
) -> dict:
    result = supplier_service.list_suppliers(
        {"supplier_name": supplier_name, "supplier_type": supplier_type, "status": status},
        session,
    )
    start = max(page_no - 1, 0) * page_size
    result["list"] = result["list"][start : start + page_size]
    result["page_no"] = page_no
    result["page_size"] = page_size
    return success_payload(result)


@router.post("")
def create_supplier(
    payload: CreateSupplierRequest,
    supplier_service: SupplierService = Depends(get_supplier_service),
    session: Session = Depends(get_db_session),
    current_user: dict = Depends(require_roles("manager", "admin")),
) -> dict:
    return success_payload(supplier_service.create_supplier(payload.model_dump(), session))


@router.get("/{supplier_id}")
def get_supplier(
    supplier_id: int,
    supplier_service: SupplierService = Depends(get_supplier_service),
    session: Session = Depends(get_db_session),
    current_user: dict = Depends(require_roles("salesman", "manager", "admin")),
) -> dict:
    supplier = supplier_service.get_supplier(supplier_id, session)
    if not supplier:
        raise AppException(code=ErrorCode.NOT_FOUND, message="供应商不存在", status_code=404)
    return success_payload(supplier)


@router.put("/{supplier_id}")
def update_supplier(
    supplier_id: int,
    payload: UpdateSupplierRequest,
    supplier_service: SupplierService = Depends(get_supplier_service),
    session: Session = Depends(get_db_session),
    current_user: dict = Depends(require_roles("manager", "admin")),
) -> dict:
    result = supplier_service.update_supplier(supplier_id, payload.model_dump(), session)
    if not result:
        raise AppException(code=ErrorCode.NOT_FOUND, message="供应商不存在", status_code=404)
    return success_payload(result)