新增获取用户按钮权限

api/controller/adminUser.go

@@ -1,7 +1,6 @@
 package controller
 
 import (
-	"fmt"
 	"github.com/gin-gonic/gin"
 	"hospital-admin-api/api/dao"
 	"hospital-admin-api/api/requests"
@@ -278,7 +277,6 @@ func (r *User) GetUserMenuList(c *gin.Context) {
 
 	// 获取用户角色身份
 	isAdmin, err := roleService.GetRoleIden(c)
-	fmt.Println(isAdmin)
 	if err != nil {
 		responses.FailWithMessage(err.Error(), c)
 		return
@@ -298,3 +296,38 @@ func (r *User) GetUserMenuList(c *gin.Context) {
 
 	responses.OkWithData(roleMenuList, c)
 }
+
+// GetUserButtonList 获取用户按钮列表
+func (r *User) GetUserButtonList(c *gin.Context) {
+	roleId := c.GetInt64("RoleId")
+	if roleId == 0 {
+		responses.Fail(c)
+		return
+	}
+
+	roleService := service.RoleService{}
+
+	// 获取用户角色身份
+	isAdmin, err := roleService.GetRoleIden(c)
+	if err != nil {
+		responses.FailWithMessage(err.Error(), c)
+		return
+	}
+
+	var buttons []string
+	if isAdmin {
+		// 管理员
+		buttons = append(buttons, "*")
+		responses.OkWithData(buttons, c)
+		return
+	}
+
+	// 获取角色菜单按钮
+	buttons, err = roleService.GetRoleMenuButtonList(roleId)
+	if err != nil {
+		responses.FailWithMessage(err.Error(), c)
+		return
+	}
+
+	responses.OkWithData(buttons, c)
+}