from fastapi import APIRouter, Request

from apps.forms.role_menu import RoleMenuForm
from apps.middleware.check_demo import check_demo
from apps.middleware.permission_required import permission_required
from apps.services import role_menu
from utils import R

# 创建路由
router = APIRouter()


# 查询角色菜单列表
@router.get('/index/{role_id}', summary='查询角色菜单列表')
@permission_required("sys:rolemenu:index")
def index(request: Request, role_id):
    menu_list = role_menu.get_role_menu_list(role_id)
    return R.ok(data=menu_list)


# 保存角色菜单数据
@router.post('/save', summary='保存角色菜单数据')
@permission_required("sys:rolemenu:save")
@check_demo
async def save(request: Request, data: RoleMenuForm):
    result = await role_menu.save_role_menu(request, data)
    return result