365 lines
10 KiB
Go
365 lines
10 KiB
Go
package controller
|
||
|
||
import (
|
||
"ems.agt/src/framework/config"
|
||
"ems.agt/src/framework/constants/common"
|
||
"ems.agt/src/framework/constants/menu"
|
||
"ems.agt/src/framework/i18n"
|
||
"ems.agt/src/framework/utils/ctx"
|
||
"ems.agt/src/framework/utils/regular"
|
||
"ems.agt/src/framework/vo"
|
||
"ems.agt/src/framework/vo/result"
|
||
"ems.agt/src/modules/system/model"
|
||
"ems.agt/src/modules/system/service"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/gin-gonic/gin/binding"
|
||
)
|
||
|
||
// 实例化控制层 SysMenuController 结构体
|
||
var NewSysMenu = &SysMenuController{
|
||
sysMenuService: service.NewSysMenuImpl,
|
||
}
|
||
|
||
// 菜单信息
|
||
//
|
||
// PATH /system/menu
|
||
type SysMenuController struct {
|
||
// 菜单服务
|
||
sysMenuService service.ISysMenu
|
||
}
|
||
|
||
// 菜单列表
|
||
//
|
||
// GET /list
|
||
func (s *SysMenuController) List(c *gin.Context) {
|
||
query := model.SysMenu{}
|
||
if v, ok := c.GetQuery("menuName"); ok && v != "" {
|
||
query.MenuName = v
|
||
}
|
||
if v, ok := c.GetQuery("status"); ok && v != "" {
|
||
query.Status = v
|
||
}
|
||
|
||
userId := ctx.LoginUserToUserID(c)
|
||
if config.IsAdmin(userId) {
|
||
userId = "*"
|
||
}
|
||
data := s.sysMenuService.SelectMenuList(query, userId)
|
||
|
||
// 闭包函数处理多语言
|
||
language := ctx.AcceptLanguage(c)
|
||
var converI18n func(language string, arr *[]model.SysMenu)
|
||
converI18n = func(language string, arr *[]model.SysMenu) {
|
||
for i := range *arr {
|
||
(*arr)[i].MenuName = i18n.TKey(language, (*arr)[i].MenuName)
|
||
(*arr)[i].Remark = i18n.TKey(language, (*arr)[i].Remark)
|
||
if len((*arr)[i].Children) > 0 {
|
||
converI18n(language, &(*arr)[i].Children)
|
||
}
|
||
}
|
||
}
|
||
converI18n(language, &data)
|
||
|
||
c.JSON(200, result.OkData(data))
|
||
}
|
||
|
||
// 菜单信息
|
||
//
|
||
// GET /:menuId
|
||
func (s *SysMenuController) Info(c *gin.Context) {
|
||
language := ctx.AcceptLanguage(c)
|
||
menuId := c.Param("menuId")
|
||
if menuId == "" {
|
||
c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
|
||
return
|
||
}
|
||
data := s.sysMenuService.SelectMenuById(menuId)
|
||
if data.MenuID == menuId {
|
||
// 处理多语言
|
||
data.MenuName = i18n.TKey(language, data.MenuName)
|
||
data.Remark = i18n.TKey(language, data.Remark)
|
||
c.JSON(200, result.OkData(data))
|
||
return
|
||
}
|
||
c.JSON(200, result.Err(nil))
|
||
}
|
||
|
||
// 菜单新增
|
||
//
|
||
// POST /
|
||
func (s *SysMenuController) Add(c *gin.Context) {
|
||
language := ctx.AcceptLanguage(c)
|
||
var body model.SysMenu
|
||
err := c.ShouldBindBodyWith(&body, binding.JSON)
|
||
if err != nil || body.MenuID != "" {
|
||
c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
|
||
return
|
||
}
|
||
|
||
// 目录和菜单检查地址唯一
|
||
if menu.TYPE_DIR == body.MenuType || menu.TYPE_MENU == body.MenuType {
|
||
uniqueNenuPath := s.sysMenuService.CheckUniqueMenuPath(body.Path, body.ParentID, "")
|
||
if !uniqueNenuPath {
|
||
// 菜单新增【%s】失败，菜单路由地址已存在
|
||
msg := i18n.TTemplate(language, "menu.errPathExists", map[string]any{"name": body.MenuName})
|
||
c.JSON(200, result.ErrMsg(msg))
|
||
return
|
||
}
|
||
}
|
||
|
||
// 检查名称唯一
|
||
uniqueNenuName := s.sysMenuService.CheckUniqueMenuName(body.MenuName, body.ParentID, "")
|
||
if !uniqueNenuName {
|
||
// 菜单新增【%s】失败，菜单名称已存在
|
||
msg := i18n.TTemplate(language, "menu.errNameExists", map[string]any{"name": body.MenuName})
|
||
c.JSON(200, result.ErrMsg(msg))
|
||
return
|
||
}
|
||
|
||
// 外链菜单需要符合网站http(s)开头
|
||
if body.IsFrame == common.STATUS_NO && !regular.ValidHttp(body.Path) {
|
||
// 操作菜单【{name}】失败，非内部地址请以http(s)://开头
|
||
msg := i18n.TTemplate(language, "menu.errFramePath", map[string]any{"name": body.MenuName})
|
||
c.JSON(200, result.ErrMsg(msg))
|
||
return
|
||
}
|
||
|
||
body.CreateBy = ctx.LoginUserToUserName(c)
|
||
insertId := s.sysMenuService.InsertMenu(body)
|
||
if insertId != "" {
|
||
c.JSON(200, result.Ok(nil))
|
||
return
|
||
}
|
||
c.JSON(200, result.Err(nil))
|
||
}
|
||
|
||
// 菜单修改
|
||
//
|
||
// PUT /
|
||
func (s *SysMenuController) Edit(c *gin.Context) {
|
||
language := ctx.AcceptLanguage(c)
|
||
var body model.SysMenu
|
||
err := c.ShouldBindBodyWith(&body, binding.JSON)
|
||
if err != nil || body.MenuID == "" {
|
||
c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
|
||
return
|
||
}
|
||
|
||
// 上级菜单不能选自己
|
||
if body.MenuID == body.ParentID {
|
||
// 菜单修改【%s】失败，上级菜单不能选择自己
|
||
msg := i18n.TTemplate(language, "menu.errPathExists", map[string]any{"name": body.MenuName})
|
||
c.JSON(200, result.ErrMsg(msg))
|
||
return
|
||
}
|
||
|
||
// 检查数据是否存在
|
||
menuInfo := s.sysMenuService.SelectMenuById(body.MenuID)
|
||
if menuInfo.MenuID != body.MenuID {
|
||
// 没有可访问菜单数据！
|
||
c.JSON(200, result.ErrMsg(i18n.TKey(language, "menu.noData")))
|
||
return
|
||
}
|
||
// 父级ID不为0是要检查
|
||
if body.ParentID != "0" {
|
||
menuParent := s.sysMenuService.SelectMenuById(body.ParentID)
|
||
if menuParent.MenuID != body.ParentID {
|
||
// 没有可访问菜单数据！
|
||
c.JSON(200, result.ErrMsg(i18n.TKey(language, "menu.noData")))
|
||
return
|
||
}
|
||
// 禁用菜单时检查父菜单是否使用
|
||
if body.Status == common.STATUS_YES && menuParent.Status == common.STATUS_NO {
|
||
// 上级菜单未启用！
|
||
c.JSON(200, result.ErrMsg(i18n.TKey(language, "menu.errParentStatus")))
|
||
return
|
||
}
|
||
}
|
||
|
||
// 目录和菜单检查地址唯一
|
||
if menu.TYPE_DIR == body.MenuType || menu.TYPE_MENU == body.MenuType {
|
||
uniqueNenuPath := s.sysMenuService.CheckUniqueMenuPath(body.Path, body.ParentID, body.MenuID)
|
||
if !uniqueNenuPath {
|
||
// 菜单修改【%s】失败，菜单路由地址已存在
|
||
msg := i18n.TTemplate(language, "menu.errPathExists", map[string]any{"name": body.MenuName})
|
||
c.JSON(200, result.ErrMsg(msg))
|
||
return
|
||
}
|
||
}
|
||
|
||
// 检查名称唯一
|
||
uniqueNenuName := s.sysMenuService.CheckUniqueMenuName(body.MenuName, body.ParentID, body.MenuID)
|
||
if !uniqueNenuName {
|
||
// 菜单修改【%s】失败，菜单名称已存在
|
||
msg := i18n.TTemplate(language, "menu.errNameExists", map[string]any{"name": body.MenuName})
|
||
c.JSON(200, result.ErrMsg(msg))
|
||
return
|
||
}
|
||
|
||
// 外链菜单需要符合网站http(s)开头
|
||
if body.IsFrame == common.STATUS_NO && !regular.ValidHttp(body.Path) {
|
||
// 菜单修改【%s】失败，非内部地址必须以http(s)://开头
|
||
msg := i18n.TTemplate(language, "menu.errFramePath", map[string]any{"name": body.MenuName})
|
||
c.JSON(200, result.ErrMsg(msg))
|
||
return
|
||
}
|
||
|
||
// 禁用菜单时检查子菜单是否使用
|
||
if body.Status == common.STATUS_NO {
|
||
hasStatus := s.sysMenuService.HasChildByMenuIdAndStatus(body.MenuID, common.STATUS_YES)
|
||
if hasStatus > 0 {
|
||
// 操作菜单【%s】失败，存在使用子菜单数：%d
|
||
msg := i18n.TTemplate(language, "menu.errHasChildUse", map[string]any{"name": body.MenuName, "num": hasStatus})
|
||
c.JSON(200, result.ErrMsg(msg))
|
||
return
|
||
}
|
||
}
|
||
|
||
// 多语言非原始值
|
||
i18nValue := i18n.TKey(language, menuInfo.MenuName)
|
||
if i18nValue != menuInfo.MenuName {
|
||
i18n.UpdateKeyValue(language, menuInfo.MenuName, body.MenuName)
|
||
body.MenuName = menuInfo.MenuName
|
||
}
|
||
// 多语言非原始值
|
||
i18nValue2 := i18n.TKey(language, menuInfo.Remark)
|
||
if i18nValue2 != menuInfo.Remark {
|
||
i18n.UpdateKeyValue(language, menuInfo.Remark, body.Remark)
|
||
body.Remark = menuInfo.Remark
|
||
}
|
||
|
||
body.UpdateBy = ctx.LoginUserToUserName(c)
|
||
rows := s.sysMenuService.UpdateMenu(body)
|
||
if rows > 0 {
|
||
c.JSON(200, result.Ok(nil))
|
||
return
|
||
}
|
||
c.JSON(200, result.Err(nil))
|
||
}
|
||
|
||
// 菜单删除
|
||
//
|
||
// DELETE /:menuId
|
||
func (s *SysMenuController) Remove(c *gin.Context) {
|
||
language := ctx.AcceptLanguage(c)
|
||
menuId := c.Param("menuId")
|
||
if menuId == "" {
|
||
c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
|
||
return
|
||
}
|
||
|
||
// 检查数据是否存在
|
||
menu := s.sysMenuService.SelectMenuById(menuId)
|
||
if menu.MenuID != menuId {
|
||
// 没有可访问菜单数据！
|
||
c.JSON(200, result.ErrMsg(i18n.TKey(language, "menu.noData")))
|
||
return
|
||
}
|
||
|
||
// 检查是否存在子菜单
|
||
hasChild := s.sysMenuService.HasChildByMenuIdAndStatus(menuId, "")
|
||
if hasChild > 0 {
|
||
// 不允许删除，存在子菜单数：%d
|
||
msg := i18n.TTemplate(language, "menu.errHasChildUse", map[string]any{"name": menu.MenuName, "num": hasChild})
|
||
c.JSON(200, result.ErrMsg(msg))
|
||
return
|
||
}
|
||
|
||
// 检查是否分配给角色
|
||
existRole := s.sysMenuService.CheckMenuExistRole(menuId)
|
||
if existRole > 0 {
|
||
// 不允许删除，菜单已分配给角色数：%d
|
||
msg := i18n.TTemplate(language, "menu.errHasRoleUse", map[string]any{"name": menu.MenuName, "num": existRole})
|
||
c.JSON(200, result.ErrMsg(msg))
|
||
return
|
||
}
|
||
|
||
rows := s.sysMenuService.DeleteMenuById(menuId)
|
||
if rows > 0 {
|
||
msg := i18n.TTemplate(language, "app.common.deleteSuccess", map[string]any{"num": rows})
|
||
c.JSON(200, result.OkMsg(msg))
|
||
return
|
||
}
|
||
c.JSON(200, result.Err(nil))
|
||
}
|
||
|
||
// 菜单树结构列表
|
||
//
|
||
// GET /treeSelect
|
||
func (s *SysMenuController) TreeSelect(c *gin.Context) {
|
||
query := model.SysMenu{}
|
||
if v, ok := c.GetQuery("menuName"); ok && v != "" {
|
||
query.MenuName = v
|
||
}
|
||
if v, ok := c.GetQuery("status"); ok && v != "" {
|
||
query.Status = v
|
||
}
|
||
|
||
userId := ctx.LoginUserToUserID(c)
|
||
if config.IsAdmin(userId) {
|
||
userId = "*"
|
||
}
|
||
data := s.sysMenuService.SelectMenuTreeSelectByUserId(query, userId)
|
||
|
||
// 闭包函数处理多语言
|
||
language := ctx.AcceptLanguage(c)
|
||
var converI18n func(language string, arr *[]vo.TreeSelect)
|
||
converI18n = func(language string, arr *[]vo.TreeSelect) {
|
||
for i := range *arr {
|
||
(*arr)[i].Label = i18n.TKey(language, (*arr)[i].Label)
|
||
if len((*arr)[i].Children) > 0 {
|
||
converI18n(language, &(*arr)[i].Children)
|
||
}
|
||
}
|
||
}
|
||
converI18n(language, &data)
|
||
|
||
c.JSON(200, result.OkData(data))
|
||
|
||
}
|
||
|
||
// 菜单树结构列表（指定角色）
|
||
//
|
||
// GET /roleMenuTreeSelect/:roleId
|
||
func (s *SysMenuController) RoleMenuTreeSelect(c *gin.Context) {
|
||
language := ctx.AcceptLanguage(c)
|
||
roleId := c.Param("roleId")
|
||
if roleId == "" {
|
||
c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
|
||
return
|
||
}
|
||
|
||
query := model.SysMenu{}
|
||
if v, ok := c.GetQuery("menuName"); ok && v != "" {
|
||
query.MenuName = v
|
||
}
|
||
if v, ok := c.GetQuery("status"); ok && v != "" {
|
||
query.Status = v
|
||
}
|
||
|
||
userId := ctx.LoginUserToUserID(c)
|
||
if config.IsAdmin(userId) {
|
||
userId = "*"
|
||
}
|
||
menuTreeSelect := s.sysMenuService.SelectMenuTreeSelectByUserId(query, userId)
|
||
checkedKeys := s.sysMenuService.SelectMenuListByRoleId(roleId)
|
||
|
||
// 闭包函数处理多语言
|
||
var converI18n func(language string, arr *[]vo.TreeSelect)
|
||
converI18n = func(language string, arr *[]vo.TreeSelect) {
|
||
for i := range *arr {
|
||
(*arr)[i].Label = i18n.TKey(language, (*arr)[i].Label)
|
||
if len((*arr)[i].Children) > 0 {
|
||
converI18n(language, &(*arr)[i].Children)
|
||
}
|
||
}
|
||
}
|
||
converI18n(language, &menuTreeSelect)
|
||
|
||
c.JSON(200, result.OkData(map[string]any{
|
||
"menus": menuTreeSelect,
|
||
"checkedKeys": checkedKeys,
|
||
}))
|
||
}
|