EdgeAdmin/internal/web/actions/actionutils/menu_group.go

package actionutils

import (
	"github.com/iwind/TeaGo/actions"
	"github.com/iwind/TeaGo/lists"
)

// MenuGroup 菜单分组
type MenuGroup struct {
	Menus      []*Menu `json:"menus"`
	AlwaysMenu *Menu   `json:"alwaysMenu"`
}

// NewMenuGroup 获取新菜单分组对象
func NewMenuGroup() *MenuGroup {
	return &MenuGroup{
		Menus: []*Menu{},
	}
}

// FindMenu 查找菜单，如果找不到则自动创建
func (this *MenuGroup) FindMenu(menuId string, menuName string) *Menu {
	for _, m := range this.Menus {
		if m.Id == menuId {
			return m
		}
	}
	menu := NewMenu()
	menu.Id = menuId
	menu.Name = menuName
	menu.Items = []*MenuItem{}
	this.Menus = append(this.Menus, menu)
	return menu
}

// Sort 排序
func (this *MenuGroup) Sort() {
	lists.Sort(this.Menus, func(i int, j int) bool {
		menu1 := this.Menus[i]
		menu2 := this.Menus[j]
		return menu1.Index < menu2.Index
	})
}

// SetSubMenu 设置子菜单
func SetSubMenu(action actions.ActionWrapper, menu *MenuGroup) {
	action.Object().Data["teaSubMenus"] = menu
}