from core.utils import list_to_tree
from dbhelper.menu import del_menu, get_menu, get_tree_menu, insert_menu, put_menu
from schemas import MenuIn, MenuRead, Response


async def menu_add(data: MenuIn) -> Response[MenuRead]:
    return Response(data=await insert_menu(data))


async def menu_arr() -> Response:
    menus = await get_tree_menu()
    try:
        data = list_to_tree(menus)
    except KeyError:
        return Response(code=400, msg="菜单根节点丢失")
    return Response(data=data)


async def menu_del(pk: int) -> Response:
    if await get_menu({"pid": pk}) is not None:
        return Response(code=400, msg="请先删除子节点")
    if await del_menu(pk) == 0:
        return Response(code=400, msg="菜单不存在")
    return Response()


async def menu_put(pk: int, data: MenuIn) -> Response:
    """更新菜单"""
    if await put_menu(pk, data) == 0:
        return Response(code=400, msg="菜单不存在")
    return Response()