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

# -*- coding: utf-8 -*-

from fastapi import APIRouter, Body, Depends, Path, Query, Request, UploadFile, Form
from fastapi.responses import JSONResponse, StreamingResponse, FileResponse
import urllib.parse
import os
from typing import List, Optional

from app.common.response import StreamResponse, SuccessResponse, ErrorResponse
from app.common.request import PaginationService
from app.utils.common_util import bytes2file_response
from app.core.base_params import PaginationQueryParam
from app.core.dependencies import AuthPermission
from app.core.router_class import OperationLogRoute
from app.core.logger import logger
from ...module_system.auth.schema import AuthSchema
from .param import ResourceSearchQueryParam
from .schema import (
    ResourceMoveSchema,
    ResourceCopySchema,
    ResourceRenameSchema,
    ResourceCreateDirSchema
)
from .service import ResourceService

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

@ResourceRouter.get("/list", summary="获取目录列表", description="获取指定目录下的文件和子目录列表")
async def get_directory_list_controller(
    request: Request,
    page: PaginationQueryParam = Depends(),
    search: ResourceSearchQueryParam = Depends(),
    auth: AuthSchema = Depends(AuthPermission(permissions=["monitor:resource:query"]))
) -> JSONResponse:
    """获取目录列表"""
    # 获取资源列表（与案例模块保持一致的分页实现）
    result_dict_list = await ResourceService.get_resources_list_service(
        search=search, 
        order_by=page.order_by,
        base_url=str(request.base_url)
    )
    # 使用分页服务进行分页处理（与案例模块保持一致）
    result_dict = await PaginationService.paginate(
        data_list=result_dict_list, 
        page_no=page.page_no, 
        page_size=page.page_size
    )
    
    logger.info(f"获取目录列表成功: {getattr(search, 'name', None) or ''}")
    return SuccessResponse(data=result_dict, msg="获取目录列表成功")


@ResourceRouter.post("/upload", summary="上传文件", description="上传文件到指定目录")
async def upload_file_controller(
    file: UploadFile,
    request: Request,
    target_path: Optional[str] = Form(None, description="目标目录路径"),
    auth: AuthSchema = Depends(AuthPermission(permissions=["monitor:resource:upload"]))
) -> JSONResponse:
    """上传文件"""
    result_dict = await ResourceService.upload_file_service(
        auth=auth,
        file=file,
        target_path=target_path,
        base_url=str(request.base_url)
    )
    logger.info(f"上传文件成功: {result_dict['filename']}")
    return SuccessResponse(data=result_dict, msg="上传文件成功")


@ResourceRouter.get("/download", summary="下载文件", description="下载指定文件")
async def download_file_controller(
    request: Request,
    path: str = Query(..., description="文件路径"),
    auth: AuthSchema = Depends(AuthPermission(permissions=["monitor:resource:download"]))
) -> FileResponse:
    """下载文件"""
    file_path = await ResourceService.download_file_service(
        auth=auth, 
        file_path=path,
        base_url=str(request.base_url)
    )
    
    # 获取文件名
    import os
    filename = os.path.basename(file_path)
    
    logger.info(f"下载文件成功: {filename}")
    return FileResponse(
        path=file_path,
        filename=filename,
        media_type='application/octet-stream'
    )


@ResourceRouter.delete("/delete", summary="删除文件", description="删除指定文件或目录")
async def delete_files_controller(
    paths: List[str] = Body(..., description="文件路径列表"),
    auth: AuthSchema = Depends(AuthPermission(permissions=["monitor:resource:delete"]))
) -> JSONResponse:
    """删除文件"""
    await ResourceService.delete_file_service(auth=auth, paths=paths)
    logger.info(f"删除文件成功: {paths}")
    return SuccessResponse(msg="删除文件成功")


@ResourceRouter.post("/move", summary="移动文件", description="移动文件或目录")
async def move_file_controller(
    data: ResourceMoveSchema,
    auth: AuthSchema = Depends(AuthPermission(permissions=["monitor:resource:move"]))
) -> JSONResponse:
    """移动文件"""
    await ResourceService.move_file_service(auth=auth, data=data)
    logger.info(f"移动文件成功: {data.source_path} -> {data.target_path}")
    return SuccessResponse(msg="移动文件成功")


@ResourceRouter.post("/copy", summary="复制文件", description="复制文件或目录")
async def copy_file_controller(
    data: ResourceCopySchema,
    auth: AuthSchema = Depends(AuthPermission(permissions=["monitor:resource:copy"]))
) -> JSONResponse:
    """复制文件"""
    await ResourceService.copy_file_service(auth=auth, data=data)
    logger.info(f"复制文件成功: {data.source_path} -> {data.target_path}")
    return SuccessResponse(msg="复制文件成功")


@ResourceRouter.post("/rename", summary="重命名文件", description="重命名文件或目录")
async def rename_file_controller(
    data: ResourceRenameSchema,
    auth: AuthSchema = Depends(AuthPermission(permissions=["monitor:resource:rename"]))
) -> JSONResponse:
    """重命名文件"""
    await ResourceService.rename_file_service(auth=auth, data=data)
    logger.info(f"重命名文件成功: {data.old_path} -> {data.new_name}")
    return SuccessResponse(msg="重命名文件成功")


@ResourceRouter.post("/create-dir", summary="创建目录", description="在指定路径创建新目录")
async def create_directory_controller(
    data: ResourceCreateDirSchema,
    auth: AuthSchema = Depends(AuthPermission(permissions=["monitor:resource:create_dir"]))
) -> JSONResponse:
    """创建目录"""
    await ResourceService.create_directory_service(auth=auth, data=data)
    logger.info(f"创建目录成功: {data.parent_path}/{data.dir_name}")
    return SuccessResponse(msg="创建目录成功")


@ResourceRouter.post("/export", summary="导出资源列表", description="导出资源列表")
async def export_resource_list_controller(
    request: Request,
    search: ResourceSearchQueryParam = Depends(),
    auth: AuthSchema = Depends(AuthPermission(permissions=["monitor:resource:export"]))
) -> StreamingResponse:
    """导出资源列表"""
    # 获取搜索结果
    result_dict_list = await ResourceService.search_resources_service(
        search=search,
        base_url=str(request.base_url)
    )
    export_result = await ResourceService.export_resource_service(data_list=result_dict_list)
    
    logger.info("导出资源列表成功")
    return StreamResponse(
        data=bytes2file_response(export_result),
        media_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
        headers={
            'Content-Disposition': 'attachment; filename=resource_list.xlsx'
        }
    )