guzhi/app/api/v1/invoice/invoice.py

from fastapi import APIRouter, Query, Depends, Header, HTTPException
from typing import Optional

from app.schemas.base import Success, SuccessExtra, BasicResponse, PageResponse, MessageOut
from app.schemas.invoice import (
    InvoiceCreate,
    InvoiceUpdate,
    UpdateStatus,
    UpdateType,
    InvoiceHeaderCreate,
    InvoiceHeaderUpdate,
    PaymentReceiptCreate,
    AppCreateInvoiceWithReceipt,
    InvoiceOut,
    InvoiceList,
    InvoiceHeaderOut,
    PaymentReceiptOut,
)
from app.controllers.invoice import invoice_controller
from app.utils.app_user_jwt import get_current_app_user
from app.core.dependency import DependAuth, DependPermission
from app.models.user import AppUser
from app.models.invoice import InvoiceHeader


invoice_router = APIRouter(tags=["发票管理"])


@invoice_router.get("/list", summary="获取发票列表", response_model=PageResponse[InvoiceOut], dependencies=[DependAuth, DependPermission])
async def list_invoices(
    phone: Optional[str] = Query(None),
    company_name: Optional[str] = Query(None),
    tax_number: Optional[str] = Query(None),
    status: Optional[str] = Query(None),
    ticket_type: Optional[str] = Query(None),
    invoice_type: Optional[str] = Query(None),
    user_id: Optional[int] = Query(None, description="按App用户ID过滤"),
    page: int = Query(1, ge=1),
    page_size: int = Query(10, ge=1, le=100),
):
    """
    发票列表查询
    参数支持按手机号、公司名称、税号、状态、发票类型进行筛选
    返回分页结构
    """
    result = await invoice_controller.list(
        page=page,
        page_size=page_size,
        phone=phone,
        company_name=company_name,
        tax_number=tax_number,
        status=status,
        ticket_type=ticket_type,
        invoice_type=invoice_type,
        app_user_id=user_id,
    )
    return SuccessExtra(
        data=result.items, total=result.total, page=result.page, page_size=result.page_size, msg="获取成功"
    )


@invoice_router.get("/detail", summary="发票详情", response_model=BasicResponse[InvoiceOut], dependencies=[DependAuth, DependPermission])
async def invoice_detail(id: int = Query(...)):
    """
    根据ID获取发票详情
    """
    out = await invoice_controller.get_out(id)
    if not out:
        return Success(data={}, msg="未找到")
    return Success(data=out, msg="获取成功")


@invoice_router.post("/create", summary="创建发票", response_model=BasicResponse[InvoiceOut], dependencies=[DependAuth, DependPermission])
async def create_invoice(data: InvoiceCreate):
    """
    创建发票记录
    """
    inv = await invoice_controller.create(data)
    out = await invoice_controller.get_out(inv.id)
    return Success(data=out, msg="创建成功")


@invoice_router.post("/update", summary="更新发票", response_model=BasicResponse[InvoiceOut], dependencies=[DependAuth, DependPermission])
async def update_invoice(data: InvoiceUpdate, id: int = Query(...)):
    """
    更新发票记录
    """
    updated = await invoice_controller.update(id, data)
    out = await invoice_controller.get_out(id) if updated else None
    return Success(data=out or {}, msg="更新成功" if updated else "未找到")


@invoice_router.delete("/delete", summary="删除发票", response_model=BasicResponse[MessageOut], dependencies=[DependAuth, DependPermission])
async def delete_invoice(id: int = Query(...)):
    """
    删除发票记录
    """
    try:
        await invoice_controller.remove(id)
        ok = True
    except Exception:
        ok = False
    return Success(data={"deleted": ok}, msg="删除成功" if ok else "未找到")


@invoice_router.post("/update-status", summary="更新发票状态", response_model=BasicResponse[InvoiceOut], dependencies=[DependAuth, DependPermission])
async def update_invoice_status(data: UpdateStatus):
    """
    更新发票状态（pending|invoiced|rejected|refunded）
    """
    out = await invoice_controller.update_status(data)
    return Success(data=out or {}, msg="更新成功" if out else "未找到")




@invoice_router.post("/{id}/receipt", summary="上传付款凭证", response_model=BasicResponse[dict], dependencies=[DependAuth, DependPermission])
async def upload_payment_receipt(id: int, data: PaymentReceiptCreate):
    """
    上传对公转账付款凭证
    """
    receipt = await invoice_controller.create_receipt(id, data)
    detail = await invoice_controller.get_receipt_by_id(receipt.id)
    return Success(data=detail, msg="上传成功")


@invoice_router.get("/headers", summary="发票抬头列表", response_model=PageResponse[InvoiceHeaderOut], dependencies=[DependAuth, DependPermission])
async def get_invoice_headers(
    app_user_id: Optional[int] = Query(None, description="按App用户ID过滤"),
    user_id: Optional[int] = Query(None, description="按App用户ID过滤(兼容参数)"),
    page: int = Query(1, ge=1),
    page_size: int = Query(10, ge=1, le=100),
):
    """
    管理端抬头列表（管理员token）：支持按 App 用户过滤与分页。
    """
    uid = app_user_id if app_user_id is not None else user_id
    qs = InvoiceHeader.all()
    if uid is not None:
        qs = qs.filter(app_user_id=uid)
    total = await qs.count()
    rows = await qs.order_by("-created_at").offset((page - 1) * page_size).limit(page_size)
    items = [InvoiceHeaderOut.model_validate(r) for r in rows]
    return SuccessExtra(data=[i.model_dump() for i in items], total=total, page=page, page_size=page_size, msg="获取成功")


@invoice_router.get("/headers/{id}", summary="发票抬头详情", response_model=BasicResponse[InvoiceHeaderOut], dependencies=[DependAuth, DependPermission])
async def get_invoice_header_by_id(id: int):
    """
    获取发票抬头详情
    """
    header = await invoice_controller.get_header_by_id(id)
    return Success(data=header or {}, msg="获取成功" if header else "未找到")


@invoice_router.post("/headers", summary="新增发票抬头", response_model=BasicResponse[InvoiceHeaderOut], dependencies=[DependAuth, DependPermission])
async def create_invoice_header(data: InvoiceHeaderCreate, app_user_id: Optional[int] = Query(None)):
    """
    新增发票抬头
    """
    header = await invoice_controller.create_header(user_id=app_user_id, data=data)
    return Success(data=header, msg="创建成功")


@invoice_router.put("/{id}/type", summary="更新发票类型", response_model=BasicResponse[InvoiceOut], dependencies=[DependAuth, DependPermission])
async def update_invoice_type(id: int, data: UpdateType):
    """
    更新发票的电子/纸质与专票/普票类型
    """
    out = await invoice_controller.update_type(id, data)
    return Success(data=out or {}, msg="更新成功" if out else "未找到")


@invoice_router.delete("/headers/{id}", summary="删除发票抬头", response_model=BasicResponse[MessageOut], dependencies=[DependAuth, DependPermission])
async def delete_invoice_header(id: int):
    ok = await invoice_controller.delete_header(id)
    return Success(msg="删除成功" if ok else "未找到")


@invoice_router.put("/headers/{id}", summary="更新发票抬头", response_model=BasicResponse[InvoiceHeaderOut], dependencies=[DependAuth, DependPermission])
async def update_invoice_header(id: int, data: InvoiceHeaderUpdate):
    header = await invoice_controller.update_header(id, data)
    return Success(data=header or {}, msg="更新成功" if header else "未找到")