csg-server/app/api/v1/module_common/file/controller.py

from pathlib import Path
from typing import Annotated, Literal

from fastapi import (
    APIRouter,
    BackgroundTasks,
    Body,
    Depends,
    Form,
    Path,
    Query,
    Request,
    UploadFile,
)
from fastapi.responses import FileResponse, JSONResponse

from app.common.response import ResponseSchema, SuccessResponse, UploadFileResponse
from app.core.base_params import PaginationQueryParam
from app.core.base_schema import AuthSchema, PageResultSchema
from app.core.dependencies import AuthPermission
from app.core.router_class import OperationLogRoute
from app.utils.upload_util import UploadUtil

from .schema import FileOutSchema, FileQueryParam, FileUploadOutSchema
from .service import FileService

FileRouter = APIRouter(route_class=OperationLogRoute, prefix="/file", tags=["文件管理"])


@FileRouter.get(
    "/list",
    summary="查询文件记录列表",
    response_model=ResponseSchema[PageResultSchema[FileOutSchema]],
)
async def get_file_list_controller(
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[FileQueryParam, Depends()],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_common:file:query"]))],
) -> JSONResponse:
    result = await FileService.page(
        auth=auth,
        page_no=page.page_no,
        page_size=page.page_size,
        search=search,
        order_by=page.order_by or [{"id": "desc"}],
    )
    return SuccessResponse(data=result, msg="查询文件记录列表成功")


@FileRouter.get(
    "/detail/{id}",
    summary="查询文件记录详情",
    response_model=ResponseSchema[FileOutSchema],
)
async def get_file_detail_controller(
    id: Annotated[int, Path(description="文件记录ID")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_common:file:detail"]))],
) -> JSONResponse:
    result = await FileService.detail_service(auth=auth, id=id)
    return SuccessResponse(data=result, msg="查询文件记录详情成功")


@FileRouter.post(
    "/upload",
    summary="上传文件",
    response_model=ResponseSchema[FileUploadOutSchema],
)
async def upload_controller(
    file: UploadFile,
    request: Request,
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_common:file:upload"]))],
    upload_type: Annotated[
        Literal["file", "avatar", "param", "resource"] | None,
        Query(description="上传类型: file=通用文件, avatar=头像, param=参数配置, resource=监控资源"),
    ] = "file",
    business_type: Annotated[str | None, Query(description="业务类型，如 rule、contract、avatar")] = None,
    business_id: Annotated[int | None, Query(ge=0, description="业务ID")] = None,
    target_path: Annotated[str | None, Form(description="目标目录路径（仅 resource 类型支持）")] = None,
) -> JSONResponse:
    result = await FileService.upload_service(
        auth=auth,
        base_url=str(request.base_url),
        file=file,
        upload_type=upload_type or "file",
        target_path=target_path,
        business_type=business_type,
        business_id=business_id,
    )
    return SuccessResponse(
        data=FileUploadOutSchema(
            id=result.id,
            original_name=result.original_name,
            file_name=result.file_name,
            file_ext=result.file_ext,
            file_size=result.file_size,
            file_url=result.file_url,
        ),
        msg="上传文件成功",
    )


@FileRouter.get(
    "/presigned/{id}",
    summary="生成文件预签名访问地址",
    response_model=ResponseSchema[str],
)
async def get_file_presigned_url_controller(
    id: Annotated[int, Path(description="文件记录ID")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_common:file:detail"]))],
) -> JSONResponse:
    result = await FileService.presigned_url_service(auth=auth, id=id)
    return SuccessResponse(data=result, msg="生成文件预签名访问地址成功")

@FileRouter.post(
    "/download",
    summary="下载文件",
    dependencies=[Depends(AuthPermission(["module_common:file:download"]))],
)
async def download_controller(
    background_tasks: BackgroundTasks,
    file_path: Annotated[str, Body(description="文件路径")],
    delete: Annotated[bool, Body(description="是否删除文件")] = False,
) -> FileResponse:
    result = await FileService.download_service(file_path=file_path)
    if delete:
        background_tasks.add_task(UploadUtil.delete_file, Path(result.file_path))
    return UploadFileResponse(file_path=result.file_path, filename=result.file_name)


@FileRouter.delete(
    "/delete",
    summary="删除文件记录",
    response_model=ResponseSchema[None],
)
async def delete_file_controller(
    ids: Annotated[list[int], Body(description="ID列表")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_common:file:delete"]))],
    delete_object: Annotated[bool, Query(description="是否同时删除MinIO/本地文件")] = False,
) -> JSONResponse:
    await FileService.delete_service(auth=auth, ids=ids, delete_object=delete_object)
    return SuccessResponse(msg="删除文件记录成功")