from pathlib import Path
from typing import Annotated, Literal

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

from app.common.response import ResponseSchema, SuccessResponse, UploadFileResponse
from app.core.dependencies import AuthPermission
from app.core.logger import log
from app.core.router_class import OperationLogRoute
from app.utils.upload_util import UploadUtil

from .service import FileService

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


@FileRouter.post(
    "/upload",
    summary="上传文件",
    description="统一文件上传入口，支持多种上传类型，可指定目标目录",
    response_model=ResponseSchema[dict],
    dependencies=[Depends(AuthPermission(["module_common:file:upload"]))],
)
async def upload_controller(
    file: UploadFile,
    request: Request,
    upload_type: Annotated[
        Literal["file", "avatar", "param", "resource"] | None,
        Query(description="上传类型: file=通用文件, avatar=头像, param=参数配置, resource=监控资源")
    ] = "file",
    target_path: Annotated[str | None, Form(description="目标目录路径（仅 resource 类型支持）")] = None,
) -> JSONResponse:
    """
    统一文件上传接口

    参数:
    - file (UploadFile): 上传的文件
    - request (Request): 请求对象
    - upload_type (str): 上传类型，默认 "file"
    - target_path (str | None): 目标目录路径，仅 resource 类型支持

    返回:
    - JSONResponse: 包含上传文件详情的JSON响应
    """
    result_dict = await FileService.upload_service(
        base_url=str(request.base_url),
        file=file,
        upload_type=upload_type or "file",
        target_path=target_path,
    )
    log.info(f"上传文件成功 {result_dict}")
    return SuccessResponse(data=result_dict, msg="上传文件成功")


@FileRouter.post(
    "/download",
    summary="下载文件",
    description="下载文件",
    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:
    """
    下载文件

    参数:
    - background_tasks (BackgroundTasks): 后台任务对象
    - file_path (str): 文件路径
    - delete (bool): 是否删除文件

    返回:
    - FileResponse: 包含下载文件的响应
    """
    result = await FileService.download_service(file_path=file_path)
    if delete:
        background_tasks.add_task(UploadUtil.delete_file, Path(result.file_path))
    log.info("下载文件成功")
    return UploadFileResponse(file_path=result.file_path, filename=result.file_name)