import asyncio

from fastapi import APIRouter, Request, UploadFile, File

from modules.system.level.schemas import LevelForm, LevelStatusForm
from core.middleware import check_demo
from core.middleware import permission_required
from modules.system.level import service as level

from core import R
from utils.request import parse_batch_ids

# 创建路由
router = APIRouter()


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


# 查询职级详情
@router.get('/detail/{id}', summary='查询职级详情')
@permission_required("sys:level:detail")
def detail(request: Request, id):
    data = level.get_level_detail(id)
    return R.ok(data=data)


# 添加职级
@router.post('/add', summary='添加职级')
@permission_required("sys:level:add")
@check_demo
def add(request: Request, data: LevelForm):
    result = level.add_level(request, data)
    return result


# 更新职级
@router.put('/update', summary='更新职级')
@permission_required("sys:level:update")
@check_demo
def update(request: Request, data: LevelForm):
    result = level.update_level(request, data)
    return result


# 删除职级
@router.delete('/delete/{id}', summary='删除职级')
@permission_required("sys:level:delete")
@check_demo
def delete(request: Request, id):
    result = level.delete_level(id)
    return result


# 设置职级状态
@router.put('/status', summary='设置职级状态')
@permission_required("sys:level:status")
@check_demo
def status(request: Request, data: LevelStatusForm):
    result = level.update_level_status(request, data)
    return result


# 导入Excel
@router.post('/import', summary='导入Excel')
@permission_required("sys:level:import")
@check_demo
async def imports(request: Request, file: UploadFile = File(...)):
    result = await level.import_level(request, file)
    return result


# 导出Excel
@router.post('/export', summary='导出Excel')
@permission_required("sys:level:export")
@check_demo
def export(request: Request):
    result = level.export_level(request)
    return result


# 获取职级列表
@router.get('/getLevelList', summary='获取职级列表')
def get_level_list(request: Request):
    result = level.get_level_list()
    return R.ok(data=result)


# 批量删除职级
@router.delete('/batchDelete', summary='批量删除职级')
@permission_required("sys:level: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(level.delete_level, id_str)