#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from typing import Annotated

from fastapi import APIRouter, Depends, Path, Query

from backend.app.task.schema.result import DeleteTaskResultParam, GetTaskResultDetail
from backend.app.task.service.result_service import task_result_service
from backend.common.pagination import DependsPagination, PageData, paging_data
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
from backend.common.security.jwt import DependsJwtAuth
from backend.common.security.permission import RequestPermission
from backend.common.security.rbac import DependsRBAC
from backend.database.db import CurrentSession

router = APIRouter()


@router.get('/{pk}', summary='获取任务结果详情', dependencies=[DependsJwtAuth])
async def get_task_result(
    pk: Annotated[int, Path(description='任务结果 ID')],
) -> ResponseSchemaModel[GetTaskResultDetail]:
    result = await task_result_service.get(pk=pk)
    return response_base.success(data=result)


@router.get(
    '',
    summary='分页获取所有任务结果',
    dependencies=[
        DependsJwtAuth,
        DependsPagination,
    ],
)
async def get_task_results_paged(
    db: CurrentSession,
    name: Annotated[str | None, Query(description='任务名称')] = None,
    task_id: Annotated[str | None, Query(description='任务 ID')] = None,
) -> ResponseSchemaModel[PageData[GetTaskResultDetail]]:
    result_select = await task_result_service.get_select(name=name, task_id=task_id)
    page_data = await paging_data(db, result_select)
    return response_base.success(data=page_data)


@router.delete(
    '',
    summary='批量删除任务结果',
    dependencies=[
        Depends(RequestPermission('sys:task:del')),
        DependsRBAC,
    ],
)
async def delete_task_result(obj: DeleteTaskResultParam) -> ResponseModel:
    count = await task_result_service.delete(obj=obj)
    if count > 0:
        return response_base.success()
    return response_base.fail()