from fastapi import APIRouter, Request

from modules.job.schemas import JobForm, JobStatusForm
from core.middleware import check_demo
from core.middleware import permission_required
from modules.job import service as job

from core import R

# 创建路由
job_router = APIRouter()


# 查询分页数据
@job_router.get('/page', summary='查询分页数据')
@permission_required("sys:job:page")
def page(request: Request):
    result = job.get_job_page(request)
    return result


# 查询任务详情
@job_router.get('/detail/{id}', summary='查询任务详情')
@permission_required("sys:job:detail")
def detail(request: Request, id):
    data = job.get_job_detail(id)
    return R.ok(data=data)


# 添加定时任务
@job_router.post('/add', summary='添加定时任务')
@permission_required("sys:job:add")
@check_demo
def add(request: Request, data: JobForm):
    result = job.add_job(request, data)
    return result


# 更新定时任务
@job_router.put('/update', summary='更新定时任务')
@permission_required("sys:job:update")
@check_demo
def update(request: Request, data: JobForm):
    result = job.update_job(request, data)
    return result


# 删除定时任务
@job_router.delete('/delete/{id}', summary='删除定时任务')
@permission_required("sys:job:delete")
@check_demo
def delete(request: Request, id):
    result = job.delete_job(id)
    return result


# 设置任务状态
@job_router.put('/status', summary='设置任务状态')
@permission_required("sys:job:status")
@check_demo
def status(request: Request, data: JobStatusForm):
    result = job.update_job_status(request, data)
    return result


# 暂停任务
@job_router.put('/pause/{id}', summary='暂停任务')
@permission_required("sys:job:pause")
@check_demo
def pause(request: Request, id):
    result = job.pause_job_by_id(id)
    return result


# 恢复任务
@job_router.put('/resume/{id}', summary='恢复任务')
@permission_required("sys:job:resume")
@check_demo
def resume(request: Request, id):
    result = job.resume_job_by_id(id)
    return result


# 立即执行一次
@job_router.post('/runOnce/{id}', summary='立即执行一次')
@permission_required("sys:job:run_once")
@check_demo
def run_once(request: Request, id):
    result = job.run_job_once_api(id)
    return result


# 重新加载调度器
@job_router.post('/reload', summary='重新加载调度器')
@permission_required("sys:job:reload")
@check_demo
def reload(request: Request):
    result = job.reload_all_jobs()
    return result


import asyncio

from fastapi import APIRouter, Request

from core.middleware import check_demo
from core.middleware import permission_required
from modules.job import service as job_log

from core import R
from utils.request import parse_batch_ids

# 创建路由
job_log_router = APIRouter()


# 查询分页数据
@job_log_router.get('/page', summary='查询分页数据')
@permission_required("sys:job_log:page")
def page(request: Request):
    result = job_log.get_job_log_page(request)
    return result


# 查询日志详情
@job_log_router.get('/detail/{id}', summary='查询日志详情')
@permission_required("sys:job_log:detail")
def detail(request: Request, id):
    data = job_log.get_job_log_detail(id)
    return R.ok(data=data)


# 删除任务日志
@job_log_router.delete('/delete/{id}', summary='删除任务日志')
@permission_required("sys:job_log:delete")
@check_demo
def delete(request: Request, id):
    result = job_log.delete_job_log(id)
    return result


# 批量删除任务日志
@job_log_router.delete('/batchDelete', summary='批量删除任务日志')
@permission_required("sys:job_log:batch_delete")
@check_demo
async def batch_delete(request: Request):
    """
    批量删除任务日志

    Request Body:
        [1, 2, 3]  # 日志ID数组
    """
    id_str, err = await parse_batch_ids(request)
    if err:
        return err
    return await asyncio.to_thread(job_log.delete_job_log, id_str)