148 lines
3.6 KiB
Go
148 lines
3.6 KiB
Go
package handler
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"kra/internal/server/httpx"
|
|
"kra/internal/service"
|
|
"kra/internal/service/dto"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type Menu struct{ service *service.MenuService }
|
|
|
|
func NewMenu(service *service.MenuService) *Menu { return &Menu{service: service} }
|
|
|
|
func (h *Menu) List(c *gin.Context) {
|
|
items, err := h.service.Tree(c.Request.Context())
|
|
if err != nil {
|
|
httpx.Fail(c, "获取失败")
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, items, "获取成功")
|
|
}
|
|
|
|
func (h *Menu) Tree(c *gin.Context) {
|
|
items, err := h.service.Tree(c.Request.Context())
|
|
if err != nil {
|
|
httpx.Fail(c, "获取失败")
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{"menus": items}, "获取成功")
|
|
}
|
|
|
|
func (h *Menu) Create(c *gin.Context) {
|
|
var req dto.MenuRequest
|
|
if c.ShouldBindJSON(&req) != nil {
|
|
httpx.Fail(c, "参数错误")
|
|
return
|
|
}
|
|
if err := h.service.Create(c.Request.Context(), &req); err != nil {
|
|
httpx.Fail(c, "添加失败："+err.Error())
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{}, "添加成功")
|
|
}
|
|
|
|
func (h *Menu) Update(c *gin.Context) {
|
|
var req dto.MenuRequest
|
|
if c.ShouldBindJSON(&req) != nil {
|
|
httpx.Fail(c, "参数错误")
|
|
return
|
|
}
|
|
if err := h.service.Update(c.Request.Context(), &req); err != nil {
|
|
httpx.Fail(c, "更新失败")
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{}, "更新成功")
|
|
}
|
|
|
|
func (h *Menu) Delete(c *gin.Context) {
|
|
var req dto.DeleteMenuRequest
|
|
if c.ShouldBindJSON(&req) != nil {
|
|
httpx.Fail(c, "参数错误")
|
|
return
|
|
}
|
|
if err := h.service.Delete(c.Request.Context(), req.ID); err != nil {
|
|
httpx.Fail(c, "删除失败:"+err.Error())
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{}, "删除成功")
|
|
}
|
|
|
|
func (h *Menu) Find(c *gin.Context) {
|
|
var req dto.GetMenuRequest
|
|
if c.ShouldBindJSON(&req) != nil {
|
|
httpx.Fail(c, "参数错误")
|
|
return
|
|
}
|
|
item, err := h.service.Find(c.Request.Context(), req.ID)
|
|
if err != nil {
|
|
httpx.Fail(c, "获取失败")
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{"menu": item}, "获取成功")
|
|
}
|
|
|
|
func (h *Menu) SetAuthorityMenus(c *gin.Context) {
|
|
var req dto.SetAuthorityMenusRequest
|
|
if c.ShouldBindJSON(&req) != nil {
|
|
httpx.Fail(c, "参数错误")
|
|
return
|
|
}
|
|
if err := h.service.SetAuthorityMenus(c.Request.Context(), &req); err != nil {
|
|
httpx.Fail(c, "添加失败")
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{}, "添加成功")
|
|
}
|
|
|
|
func (h *Menu) AuthorityMenus(c *gin.Context) {
|
|
var req dto.GetAuthorityMenusRequest
|
|
if c.ShouldBindJSON(&req) != nil {
|
|
httpx.Fail(c, "参数错误")
|
|
return
|
|
}
|
|
menus, err := h.service.AuthorityMenus(c.Request.Context(), req.AuthorityID)
|
|
if err != nil {
|
|
httpx.Fail(c, "获取失败")
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{"menus": menus}, "获取成功")
|
|
}
|
|
|
|
func (h *Menu) RoleIDs(c *gin.Context) {
|
|
id, _ := strconv.ParseUint(c.Query("menuId"), 10, 64)
|
|
ids, err := h.service.RoleIDs(c.Request.Context(), uint(id))
|
|
if err != nil {
|
|
httpx.Fail(c, "获取失败")
|
|
return
|
|
}
|
|
if ids == nil {
|
|
ids = []uint{}
|
|
}
|
|
defaultIDs, err := h.service.DefaultRouterRoleIDs(c.Request.Context(), uint(id))
|
|
if err != nil {
|
|
httpx.Fail(c, "获取失败")
|
|
return
|
|
}
|
|
if defaultIDs == nil {
|
|
defaultIDs = []uint{}
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{"authorityIds": ids, "defaultRouterAuthorityIds": defaultIDs}, "获取成功")
|
|
}
|
|
|
|
func (h *Menu) SetRoles(c *gin.Context) {
|
|
var req dto.SetMenuRolesRequest
|
|
if c.ShouldBindJSON(&req) != nil {
|
|
httpx.Fail(c, "参数错误")
|
|
return
|
|
}
|
|
if err := h.service.SetRoles(c.Request.Context(), &req); err != nil {
|
|
httpx.Fail(c, "设置失败")
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{}, "设置成功")
|
|
}
|