import asyncio

from fastapi import APIRouter

from starlette.requests import Request

from modules.system.user.schemas import UserForm, UserStatusForm, ResetPwdForm
from core.middleware import check_demo
from core.middleware import permission_required
from modules.system.user import service as user
from core import R
from utils.request import parse_batch_ids

# 创建路由
router = APIRouter()


# 查询分页数据
@router.get('/page', summary='查询分页数据')
@permission_required("sys:user:page")
def page(request: Request):
    # 调用查询分页数据服务方法
    result = user.get_user_page(request)
    # 返回结果
    return result


# 查询用户详情
@router.get('/detail/{id}', summary='查询用户详情')
@permission_required("sys:user:detail")
def detail(request: Request, id):
    # 调用查询用户详情服务方法
    data = user.get_user_detail(id)
    # 返回结果
    return R.ok(data=data)


# 添加用户
@router.post('/add', summary='添加用户')
@permission_required("sys:user:add")
@check_demo
async def add(request: Request, data: UserForm):
    # 调用添加用户服务
    result = await user.add_user(request, data)
    # 返回结果
    return result


# 更新用户
@router.put('/update', summary='更新用户')
@permission_required("sys:user:update")
@check_demo
async def update(request: Request, data: UserForm):
    # 调用更新用户服务方法
    result = await user.update_user(request, data)
    # 返回结果
    return result


# 删除用户
@router.delete('/delete/{id}', summary='删除用户')
@permission_required("sys:user:delete")
@check_demo
def delete(request: Request, id):
    # 调用删除用户服务方法
    result = user.delete_user(id)
    # 返回结果
    return result


# 设置用户状态
@router.put('/status', summary='设置用户状态')
@permission_required("sys:user:status")
@check_demo
def status(request: Request, data: UserStatusForm):
    # 调用设置用户状态服务方法
    result = user.update_user_status(request, data)
    # 返回结果
    return result


# 重置密码
@router.post('/resetPwd', summary='重置密码')
@permission_required("sys:user:resetPwd")
@check_demo
def reset_pwd(request: Request, data: ResetPwdForm):
    # 调用重置密码服务方法
    result = user.reset_user_pwd(request, data)
    # 返回结果
    return result


# 批量删除用户
@router.delete('/batchDelete', summary='批量删除用户')
@permission_required("sys:user: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(user.delete_user, id_str)