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 (
    EnvironmentConditionCreateSchema,
    EnvironmentConditionOutSchema,
    EnvironmentConditionQueryParam,
    EnvironmentConditionUpdateSchema,
)
from .service import EnvironmentConditionService

EnvironmentConditionRouter = APIRouter(route_class=OperationLogRoute, prefix="/environment_condition", tags=["环境因子"])


@EnvironmentConditionRouter.get("/detail/{id}", summary="获取环境因子详情", response_model=ResponseSchema[EnvironmentConditionOutSchema])
async def get_environment_condition__detail_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:environment_condition:detail"]))],
    id: Annotated[int, Path(description="环境因子ID")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = EnvironmentConditionService(auth, db)
    result_dict = await service.detail(id=id)
    return SuccessResponse(data=result_dict, msg="获取环境因子详情成功")


@EnvironmentConditionRouter.get("/list", summary="分页查询环境因子", response_model=ResponseSchema[PageResultSchema[EnvironmentConditionOutSchema]])
async def get_environment_condition__list_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:environment_condition:query"]))],
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[EnvironmentConditionQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = EnvironmentConditionService(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="查询环境因子列表成功")


@EnvironmentConditionRouter.get("/options", summary="环境因子下拉选项")
async def get_environment_condition__options_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:environment_condition:query"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = EnvironmentConditionService(auth, db)
    options = await service.list_options()
    return SuccessResponse(data=options, msg="获取环境因子选项成功")


@EnvironmentConditionRouter.post("/create", summary="创建环境因子", response_model=ResponseSchema[EnvironmentConditionOutSchema])
async def create_environment_condition__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:environment_condition:create"]))],
    data: Annotated[EnvironmentConditionCreateSchema, Body(description="创建参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = EnvironmentConditionService(auth, db)
    result_dict = await service.create(data=data)
    return SuccessResponse(data=result_dict, msg="创建环境因子成功")


@EnvironmentConditionRouter.put("/update/{id}", summary="修改环境因子", response_model=ResponseSchema[EnvironmentConditionOutSchema])
async def update_environment_condition__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:environment_condition:update"]))],
    id: Annotated[int, Path(description="环境因子ID")],
    data: Annotated[EnvironmentConditionUpdateSchema, Body(description="修改参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = EnvironmentConditionService(auth, db)
    result_dict = await service.update(id=id, data=data)
    return SuccessResponse(data=result_dict, msg="修改环境因子成功")


@EnvironmentConditionRouter.delete("/delete", summary="删除环境因子", response_model=ResponseSchema[None])
async def delete_environment_condition__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:environment_condition:delete"]))],
    ids: Annotated[list[int], Body(description="ID列表")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = EnvironmentConditionService(auth, db)
    await service.delete(ids=ids)
    return SuccessResponse(msg="删除环境因子成功")


@EnvironmentConditionRouter.post("/export", summary="导出环境因子")
async def export_environment_condition__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:environment_condition:export"]))],
    search: Annotated[EnvironmentConditionQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> StreamingResponse:
    service = EnvironmentConditionService(auth, db)
    result_dict_list = await service.get_list(search=search)
    export_result = EnvironmentConditionService.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')}"},
    )


@EnvironmentConditionRouter.post("/import", summary="导入环境因子", response_model=ResponseSchema[ImportResultSchema])
async def import_environment_condition__controller(
    file: Annotated[UploadFile, File(description="导入文件")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:environment_condition:import"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = EnvironmentConditionService(auth, db)
    batch_import_result = await service.batch_import(file=file, update_support=True)
    return SuccessResponse(data=batch_import_result, msg="导入环境因子成功")


@EnvironmentConditionRouter.post("/download/template", summary="获取环境因子导入模板", dependencies=[Depends(AuthPermission(["module_bre:environment_condition:download"]))])
async def download_environment_condition__template_controller() -> StreamingResponse:
    import_template_result = EnvironmentConditionService.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",
        },
    )