minio-manager/app/api/endpoints.py

# app/api/endpoints.py
from fastapi import APIRouter, UploadFile, File, Form, HTTPException, Depends
from fastapi.responses import StreamingResponse
from app.api.deps import get_request_context
from typing import List, Optional
import urllib.parse

from app.models.schemas import FileItem, CreateFolderRequest
from app.services.minio_service import minio_service
from app.services.img_minio_service import img_minio_service
from app.utils.path_utils import validate_path, sanitize_filename
from app.utils.logger_utils import logger
from app.core.config import settings
from app.core.exceptions import (
    NotFoundException,
    ValidationException,
    InternalServerException,
    ErrorMessage
)
from minio.error import S3Error

router = APIRouter(dependencies=[Depends(get_request_context)])


@router.get("/files", response_model=List[FileItem])
async def list_files(
    path: str = "/",
):
    # 验证路径
    if path != "/":
        path = validate_path(path, allow_root=True)
    return minio_service.list_files(path)


@router.post("/upload")
async def upload_file(
        file: UploadFile = File(...),
        path: str = Form(...),
):
    """
    上传文件到MinIO存储
    
    path: 完整的文件路径 (例如: folder/subfolder/image.png)
    """
    # 验证和清理路径
    path = validate_path(path)
    
    # 验证文件名
    if not file.filename:
        raise ValidationException(detail="文件名不能为空")
    filename = sanitize_filename(file.filename)
    
    # 获取文件大小
    file.file.seek(0, 2)
    size = file.file.tell()
    file.file.seek(0)
    
    # 检查文件大小限制
    if size > settings.MAX_UPLOAD_SIZE:
        max_size_mb = settings.MAX_UPLOAD_SIZE / (1024 * 1024)
        raise HTTPException(
            status_code=413,
            detail=f"文件大小超过限制 ({max_size_mb}MB)"
        )
    
    # 构建完整路径
    full_path = f"{path}/{filename}" if path else filename
    
    logger.info(f"上传文件: {filename}, 目标路径: {full_path}, 大小: {size} bytes")
    
    minio_service.upload_file(
        file.file,
        full_path,
        file.content_type or "application/octet-stream",
        size,
    )
    return {"message": "Upload successful", "filename": filename, "path": full_path}

@router.post("/folder")
async def create_folder(
    req: CreateFolderRequest,
):
    # 验证路径
    path = validate_path(req.path)
    minio_service.create_folder(path)
    logger.info(f"创建文件夹: {path}")
    return {"message": "Folder created"}

@router.get("/download")
async def download_file(
    path: str,
):
    # 验证路径
    path = validate_path(path)
    
    try:
        data_stream = minio_service.get_file_stream(path)
    except S3Error as e:
        if e.code == "NoSuchKey":
            logger.warning(f"文件不存在: {path}")
            raise NotFoundException(detail=ErrorMessage.FILE_NOT_FOUND)
        raise
    
    filename = path.split("/")[-1]
    quoted_filename = urllib.parse.quote(filename)
    
    logger.info(f"下载文件: {path}")

    return StreamingResponse(
        data_stream,
        media_type="application/octet-stream",
        headers={
            "Content-Disposition": f"attachment; filename*=UTF-8''{quoted_filename}"
        }
    )


@router.delete("/delete")
async def delete_item(
    path: str,
):
    # 验证路径
    path = validate_path(path)
    
    is_dir = path.endswith("/")
    minio_service.delete_file(path, is_dir)
    logger.info(f"删除{'文件夹' if is_dir else '文件'}: {path}")
    return {"message": "Deleted successfully"}


@router.get("/preview")
async def preview_file(
    path: str,
):
    # 验证路径
    path = validate_path(path)
    
    url = minio_service.get_presigned_url(path)
    return {"url": url}

@router.get("/storage-info")
async def get_storage_info():
    return minio_service.get_storage_info()



@router.get("/img")
async def preview_image(path: str):
    # 验证路径
    path = validate_path(path)
    
    url = img_minio_service.get_presigned_url(path)
    return {"url": url}


@router.post("/img/upload")
async def upload_image(
        file: UploadFile = File(..., description="上传的图片文件"),
        path: Optional[str] = Form(None, description="指定存储路径，不指定则使用文件名")
):
    """
    上传图片到MinIO存储
    """
    # 验证文件名
    if not file.filename:
        raise ValidationException(detail="文件名不能为空")
    filename = sanitize_filename(file.filename)
    
    # 读取文件内容
    file_content = await file.read()
    
    # 检查文件大小限制
    if len(file_content) > settings.MAX_UPLOAD_SIZE:
        max_size_mb = settings.MAX_UPLOAD_SIZE / (1024 * 1024)
        raise HTTPException(
            status_code=413,
            detail=f"文件大小超过限制 ({max_size_mb}MB)"
        )

    # 生成对象名称：路径 + 文件名
    if path:
        # 验证和清理路径
        path = validate_path(path)
        if not path.endswith('/'):
            path = path + '/'
        object_name = f"{path}{filename}"
    else:
        # 直接使用文件名
        object_name = filename

    logger.info(f"上传图片: {filename}, 目标路径: {object_name}, 大小: {len(file_content)} bytes")

    # 上传到MinIO
    stored_name = img_minio_service.upload_image_data(
        image_data=file_content,
        object_name=object_name,
        content_type=file.content_type or "image/jpeg"
    )

    return {
        "message": "图片上传成功",
        "object_name": stored_name,
        "file_size": len(file_content),
        "content_type": file.content_type
    }


@router.put("/img/rename")
async def rename_image(
        old_path: str = Form(..., description="原文件路径"),
        new_path: str = Form(..., description="新文件路径"),
        overwrite: bool = Form(False, description="是否覆盖已存在的新路径文件")
):
    """
    重命名或移动图片文件
    """
    # 验证路径
    old_path = validate_path(old_path)
    new_path = validate_path(new_path)
    
    # 检查原文件是否存在
    if not img_minio_service.image_exists(old_path):
        raise NotFoundException(detail="原文件不存在")

    # 检查新文件是否已存在
    if not overwrite and img_minio_service.image_exists(new_path):
        raise ValidationException(
            detail="目标文件已存在，如需覆盖请设置 overwrite=true"
        )

    # 执行重命名操作
    success = img_minio_service.rename_image(old_path, new_path)

    if success:
        logger.info(f"重命名图片: {old_path} -> {new_path}")
        return {
            "message": "文件重命名成功",
            "old_path": old_path,
            "new_path": new_path
        }
    else:
        raise InternalServerException(detail="文件重命名失败")