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 (
        CrossCombinationCreateSchema,
        CrossCombinationOutSchema,
        CrossCombinationQueryParam,
        CrossCombinationUpdateSchema,
)
from .service import CrossCombinationService

CrossCombinationRouter = APIRouter(route_class=OperationLogRoute, prefix="/cross_combination", tags=["杂交组合"])


@CrossCombinationRouter.get("/detail/{id}", summary="获取杂交组合详情", response_model=ResponseSchema[CrossCombinationOutSchema])
async def get_cross_combination__detail_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:cross_combination:detail"]))],
    id: Annotated[int, Path(description="杂交组合ID")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = CrossCombinationService(auth, db)
    result_dict = await service.detail(id=id)
    return SuccessResponse(data=result_dict, msg="获取杂交组合详情成功")


@CrossCombinationRouter.get("/list", summary="分页查询杂交组合", response_model=ResponseSchema[PageResultSchema[CrossCombinationOutSchema]])
async def get_cross_combination__list_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:cross_combination:query"]))],
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[CrossCombinationQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = CrossCombinationService(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="查询杂交组合列表成功")


@CrossCombinationRouter.get("/options", summary="杂交组合下拉选项")
async def get_cross_combination__options_controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:cross_combination:query"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = CrossCombinationService(auth, db)
    options = await service.list_options()
    return SuccessResponse(data=options, msg="获取杂交组合选项成功")


@CrossCombinationRouter.post("/create", summary="创建杂交组合", response_model=ResponseSchema[CrossCombinationOutSchema])
async def create_cross_combination__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:cross_combination:create"]))],
    data: Annotated[CrossCombinationCreateSchema, Body(description="创建参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = CrossCombinationService(auth, db)
    result_dict = await service.create(data=data)
    return SuccessResponse(data=result_dict, msg="创建杂交组合成功")


@CrossCombinationRouter.put("/update/{id}", summary="修改杂交组合", response_model=ResponseSchema[CrossCombinationOutSchema])
async def update_cross_combination__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:cross_combination:update"]))],
    id: Annotated[int, Path(description="杂交组合ID")],
    data: Annotated[CrossCombinationUpdateSchema, Body(description="修改参数")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = CrossCombinationService(auth, db)
    result_dict = await service.update(id=id, data=data)
    return SuccessResponse(data=result_dict, msg="修改杂交组合成功")


@CrossCombinationRouter.delete("/delete", summary="删除杂交组合", response_model=ResponseSchema[None])
async def delete_cross_combination__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:cross_combination:delete"]))],
    ids: Annotated[list[int], Body(description="ID列表")],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = CrossCombinationService(auth, db)
    await service.delete(ids=ids)
    return SuccessResponse(msg="删除杂交组合成功")


@CrossCombinationRouter.post("/export", summary="导出杂交组合")
async def export_cross_combination__controller(
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:cross_combination:export"]))],
    search: Annotated[CrossCombinationQueryParam, Query()],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> StreamingResponse:
    service = CrossCombinationService(auth, db)
    result_dict_list = await service.get_list(search=search)
    export_result = CrossCombinationService.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')}"},
    )


@CrossCombinationRouter.post("/import", summary="导入杂交组合", response_model=ResponseSchema[ImportResultSchema])
async def import_cross_combination__controller(
    file: Annotated[UploadFile, File(description="导入文件")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(["module_bre:cross_combination:import"]))],
    db: Annotated[AsyncSession, Depends(db_getter)],
) -> JSONResponse:
    service = CrossCombinationService(auth, db)
    batch_import_result = await service.batch_import(file=file, update_support=True)
    return SuccessResponse(data=batch_import_result, msg="导入杂交组合成功")


@CrossCombinationRouter.post("/download/template", summary="获取杂交组合导入模板", dependencies=[Depends(AuthPermission(["module_bre:cross_combination:download"]))])
async def download_cross_combination__template_controller() -> StreamingResponse:
    import_template_result = CrossCombinationService.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",
        },
    )