from fastapi import Depends, Query
from starlette.requests import Request

from core.security import check_token, get_password_hash
from dbhelper.user import (
    del_user,
    get_user,
    get_user_info,
    get_users,
    insert_user,
    put_user,
    select_role,
)
from schemas import Response, UserAdd, UserInfo, UserPut, UserQuery, UserRead
from schemas.common import ListAll


async def user_add(data: UserAdd) -> Response[UserRead]:
    """新增用户并分配角色 一步到位"""
    if await get_user({"username": data.username}) is not None:
        return Response(code=400, msg="用户名已存在")
    rids = data.roles
    del data.roles
    data.password = get_password_hash(data.password)
    result = await insert_user(data, rids)
    if isinstance(result, int):
        return Response(code=400, msg=f"角色{result}不存在")
    return Response(data=result)


async def user_info(pk: int) -> Response[UserInfo]:
    """获取用户信息"""
    obj = await get_user({"id": pk})
    if obj is None:
        return Response(code=400, msg="用户不存在")
    return Response(data=await get_user_info(obj))


async def user_arr(
    offset: int = Query(default=1, description="偏移量-页码"),
    limit: int = Query(default=10, description="数据量"),
) -> Response[ListAll[list[UserRead]]]:
    """分页列表数据"""
    skip = (offset - 1) * limit
    users, count = await get_users(skip, limit)
    return Response(data=ListAll(total=count, items=users))


async def user_list(query: UserQuery) -> Response[ListAll[list[UserRead]]]:
    """post查询用户列表"""
    size = query.limit
    skip = (query.offset - 1) * size
    del query.offset, query.limit
    users, count = await get_users(skip, size, query.dict())
    return Response(data=ListAll(total=count, items=users))


async def user_del(pk: int) -> Response:
    """删除用户"""
    if await del_user(pk) == 0:
        return Response(code=400, msg="用户不存在")
    return Response()


async def user_put(pk: int, data: UserPut) -> Response:
    """更新用户"""
    if await get_user({"id": pk}) is None:
        return Response(code=400, msg="用户不存在")

    result = await put_user(pk, data)
    if isinstance(result, int):
        return Response(code=400, msg=f"角色不存在{result}")
    return Response()


async def user_select_role(rid: int, user=Depends(check_token)):
    """用户切换角色"""
    res = await select_role(user.id, rid)
    if res == 0:
        return Response(code=400, msg=f"角色不存在{res}")
    return Response()