FastapiAdmin/backend/app/api/v1/module_monitor/resource/controller.py

from typing import Annotated

from fastapi import APIRouter, Body, Depends, Form, Query, Request, UploadFile
from fastapi.responses import FileResponse, JSONResponse, StreamingResponse

from app.api.v1.module_common.file.service import FileService
from app.common.request import PaginationService
from app.common.response import ResponseSchema, StreamResponse, SuccessResponse, UploadFileResponse
from app.core.base_params import PaginationQueryParam
from app.core.base_schema import UploadResponseSchema
from app.core.dependencies import AuthPermission
from app.core.router_class import OperationLogRoute
from app.utils.common_util import bytes2file_response

from .schema import (
    ResourceCopySchema,
    ResourceCreateDirSchema,
    ResourceItemSchema,
    ResourceMoveSchema,
    ResourceRenameSchema,
    ResourceSearchQueryParam,
)
from .service import ResourceService

ResourceRouter = APIRouter(route_class=OperationLogRoute, prefix="/resource", tags=["系统监控/资源管理"])


@ResourceRouter.get(
    "/list",
    summary="获取目录列表",
    response_model=ResponseSchema[list[ResourceItemSchema]],
    dependencies=[Depends(AuthPermission(["module_monitor:resource:query"]))],
)
async def get_directory_list_controller(
    request: Request,
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[ResourceSearchQueryParam, Depends()],
) -> JSONResponse:
    """
    获取目录列表

    参数:
    - request (Request): FastAPI请求对象，用于获取基础URL。
    - page (PaginationQueryParam): 分页查询参数模型。
    - search (ResourceSearchQueryParam): 资源查询参数模型。

    返回:
    - JSONResponse: 包含目录列表的JSON响应。
    """
    # 获取资源列表（与案例模块保持一致的分页实现）
    result_dict_list = await ResourceService.get_resources_list_service(
        search=search, base_url=str(request.base_url)
    )
    # 目录列表来自本地文件系统扫描，无 SQL 分页
    result_dict = await PaginationService.paginate(
        data_list=result_dict_list,
        page_no=page.page_no,
        page_size=page.page_size,
    )

    return SuccessResponse(data=result_dict, msg="获取目录列表成功")


@ResourceRouter.post(
    "/upload",
    summary="上传文件",
    response_model=ResponseSchema[UploadResponseSchema],
    dependencies=[Depends(AuthPermission(["module_monitor:resource:upload"]))],
)
async def upload_file_controller(
    file: UploadFile,
    request: Request,
    target_path: Annotated[str | None, Form(description="目标目录路径")] = None,
) -> JSONResponse:
    """
    上传文件

    参数:
    - file (UploadFile): 要上传的文件对象。
    - request (Request): FastAPI请求对象，用于获取基础URL。
    - target_path (str | None): 目标目录路径，默认为None。

    返回:
    - JSONResponse: 包含上传文件信息的JSON响应。
    """
    # 调用统一上传接口，使用 resource 类型
    result = await FileService.upload_service(
        base_url=str(request.base_url),
        file=file,
        upload_type="resource",
        target_path=target_path,
    )
    return SuccessResponse(data=result, msg="上传文件成功")


@ResourceRouter.get(
    "/download",
    summary="下载文件",
    dependencies=[Depends(AuthPermission(["module_monitor:resource:download"]))],
)
async def download_file_controller(
    path: Annotated[str, Query(description="文件路径")],
) -> FileResponse:
    """
    下载文件

    参数:
    - path (str): 文件路径。

    返回:
    - FileResponse: 包含文件内容的文件响应。
    """
    file_path = await ResourceService.download_file_service(file_path=path)

    # 获取文件名
    import os

    filename = os.path.basename(file_path)

    return UploadFileResponse(
        file_path=file_path,
        filename=filename,
        media_type="application/octet-stream",
    )


@ResourceRouter.delete(
    "/delete",
    summary="删除文件",
    response_model=ResponseSchema[None],
    dependencies=[Depends(AuthPermission(["module_monitor:resource:delete"]))],
)
async def delete_files_controller(
    paths: Annotated[list[str], Body(description="文件路径列表")],
) -> JSONResponse:
    """
    删除文件

    参数:
    - paths (list[str]): 文件路径列表。

    返回:
    - JSONResponse: 包含删除结果的JSON响应。
    """
    await ResourceService.delete_file_service(paths=paths)
    return SuccessResponse(msg="删除文件成功")


@ResourceRouter.post(
    "/move",
    summary="移动文件",
    response_model=ResponseSchema[None],
    dependencies=[Depends(AuthPermission(["module_monitor:resource:move"]))],
)
async def move_file_controller(data: ResourceMoveSchema) -> JSONResponse:
    """
    移动文件

    参数:
    - data (ResourceMoveSchema): 移动文件参数模型。

    返回:
    - JSONResponse: 包含移动结果的JSON响应。
    """
    await ResourceService.move_file_service(data=data)
    return SuccessResponse(msg="移动文件成功")


@ResourceRouter.post(
    "/copy",
    summary="复制文件",
    response_model=ResponseSchema[None],
    dependencies=[Depends(AuthPermission(["module_monitor:resource:copy"]))],
)
async def copy_file_controller(data: ResourceCopySchema) -> JSONResponse:
    """
    复制文件

    参数:
    - data (ResourceCopySchema): 复制文件参数模型。

    返回:
    - JSONResponse: 包含复制结果的JSON响应。
    """
    await ResourceService.copy_file_service(data=data)
    return SuccessResponse(msg="复制文件成功")


@ResourceRouter.post(
    "/rename",
    summary="重命名文件",
    response_model=ResponseSchema[None],
    dependencies=[Depends(AuthPermission(["module_monitor:resource:rename"]))],
)
async def rename_file_controller(data: ResourceRenameSchema) -> JSONResponse:
    """
    重命名文件

    参数:
    - data (ResourceRenameSchema): 重命名文件参数模型。

    返回:
    - JSONResponse: 包含重命名结果的JSON响应。
    """
    await ResourceService.rename_file_service(data=data)
    return SuccessResponse(msg="重命名文件成功")


@ResourceRouter.post(
    "/create-dir",
    summary="创建目录",
    response_model=ResponseSchema[None],
    dependencies=[Depends(AuthPermission(["module_monitor:resource:create_dir"]))],
)
async def create_directory_controller(
    data: ResourceCreateDirSchema,
) -> JSONResponse:
    """
    创建目录

    参数:
    - data (ResourceCreateDirSchema): 创建目录参数模型。

    返回:
    - JSONResponse: 包含创建目录结果的JSON响应。
    """
    await ResourceService.create_directory_service(data=data)
    return SuccessResponse(msg="创建目录成功")


@ResourceRouter.post(
    "/export",
    summary="导出资源列表",
    response_model=ResponseSchema[None],
    dependencies=[Depends(AuthPermission(["module_monitor:resource:export"]))],
)
async def export_resource_list_controller(
    request: Request, search: Annotated[ResourceSearchQueryParam, Depends()]
) -> StreamingResponse:
    """
    导出资源列表

    参数:
    - request (Request): FastAPI请求对象，用于获取基础URL。
    - search (ResourceSearchQueryParam): 资源查询参数模型。

    返回:
    - StreamingResponse: 包含导出资源列表的流式响应。
    """
    # 获取搜索结果
    result_dict_list = await ResourceService.get_resources_list_service(
        search=search, base_url=str(request.base_url)
    )
    export_result = await ResourceService.export_resource_service(data_list=result_dict_list)

    return StreamResponse(
        data=bytes2file_response(export_result),
        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
        headers={"Content-Disposition": "attachment; filename=resource_list.xlsx"},
    )