import urllib.parse
from typing import Annotated

from fastapi import APIRouter, Body, Depends, File, Path, Query, UploadFile
from fastapi.responses import JSONResponse, StreamingResponse
from sqlalchemy.ext.asyncio import AsyncSession

from app.common.response import ResponseSchema, StreamResponse, SuccessResponse
from app.core.base_schema import AuthSchema, PageResultSchema, PaginationQueryParam, ImportResultSchema
from app.core.dependencies import AuthPermission, db_getter
from app.core.router_class import OperationLogRoute
from app.utils.common_util import bytes2file_response

from .schema import (
    AnalysisDatasetCreateSchema,
    AnalysisDatasetOutSchema,
    AnalysisDatasetQueryParam,
    AnalysisDatasetUpdateSchema,
)
from .service import AnalysisDatasetService

AnalysisDatasetRouter = APIRouter(route_class=OperationLogRoute, prefix="/analysis_dataset", tags=["分析数据集"])


@AnalysisDatasetRouter.get("/detail/{id}", summary="获取分析数据集详情", response_model=ResponseSchema[AnalysisDatasetOutSchema])
async def get_analysis_dataset__detail_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:analysis_dataset:detail"]))],
    id: Annotated[int, Path(description="分析数据集ID")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = AnalysisDatasetService(auth, db)
    result_dict = await service.detail(id=id)
    return SuccessResponse(data=result_dict, msg="获取分析数据集详情成功")


@AnalysisDatasetRouter.get("/list", summary="分页查询分析数据集", response_model=ResponseSchema[PageResultSchema[AnalysisDatasetOutSchema]])
async def get_analysis_dataset__list_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:analysis_dataset:query"]))],
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[AnalysisDatasetQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = AnalysisDatasetService(auth, db)
    result_dict = await service.page(
        page_no=page.page_no,
        page_size=page.page_size,
        search=search,
        order_by=page.order_by,
    )
    return SuccessResponse(data=result_dict, msg="查询分析数据集列表成功")


@AnalysisDatasetRouter.get("/options", summary="分析数据集下拉选项")
async def get_analysis_dataset__options_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:analysis_dataset:query"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = AnalysisDatasetService(auth, db)
    options = await service.list_options()
    return SuccessResponse(data=options, msg="获取分析数据集选项成功")


@AnalysisDatasetRouter.post("/create", summary="创建分析数据集", response_model=ResponseSchema[AnalysisDatasetOutSchema])
async def create_analysis_dataset__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:analysis_dataset:create"]))],
    data: Annotated[AnalysisDatasetCreateSchema, Body(description="创建参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = AnalysisDatasetService(auth, db)
    result_dict = await service.create(data=data)
    return SuccessResponse(data=result_dict, msg="创建分析数据集成功")


@AnalysisDatasetRouter.put("/update/{id}", summary="修改分析数据集", response_model=ResponseSchema[AnalysisDatasetOutSchema])
async def update_analysis_dataset__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:analysis_dataset:update"]))],
    id: Annotated[int, Path(description="分析数据集ID")],
    data: Annotated[AnalysisDatasetUpdateSchema, Body(description="修改参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = AnalysisDatasetService(auth, db)
    result_dict = await service.update(id=id, data=data)
    return SuccessResponse(data=result_dict, msg="修改分析数据集成功")


@AnalysisDatasetRouter.delete("/delete", summary="删除分析数据集", response_model=ResponseSchema[None])
async def delete_analysis_dataset__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:analysis_dataset:delete"]))],
    ids: Annotated[list[int], Body(description="ID列表")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = AnalysisDatasetService(auth, db)
    await service.delete(ids=ids)
    return SuccessResponse(msg="删除分析数据集成功")


@AnalysisDatasetRouter.post("/export", summary="导出分析数据集")
async def export_analysis_dataset__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:analysis_dataset:export"]))],
    search: Annotated[AnalysisDatasetQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> StreamingResponse:
    service = AnalysisDatasetService(auth, db)
    result_dict_list = await service.get_list(search=search)
    export_result = AnalysisDatasetService.batch_export(obj_list=[item.model_dump() for item in result_dict_list])
    return StreamResponse(
        data=bytes2file_response(export_result),
        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
        headers={"Content-Disposition": f"attachment; filename={urllib.parse.quote('分析数据集.xlsx')}"},
    )


@AnalysisDatasetRouter.post("/import", summary="导入分析数据集", response_model=ResponseSchema[ImportResultSchema])
async def import_analysis_dataset__controller(
    file: Annotated[UploadFile, File(description="导入文件")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:analysis_dataset:import"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = AnalysisDatasetService(auth, db)
    batch_import_result = await service.batch_import(file=file, update_support=True)
    return SuccessResponse(data=batch_import_result, msg="导入分析数据集成功")


@AnalysisDatasetRouter.post("/download/template", summary="获取分析数据集导入模板", dependencies=[Depends(AuthPermission(["module_bre:analysis_dataset:download"]))])
async def download_analysis_dataset__template_controller() -> StreamingResponse:
    import_template_result = AnalysisDatasetService.import_template_download()
    return StreamResponse(
        data=bytes2file_response(import_template_result),
        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
        headers={
            "Content-Disposition": f"attachment; filename={urllib.parse.quote('分析数据集导入模板.xlsx')}",
            "Access-Control-Expose-Headers": "Content-Disposition",
        },
    )