yunzer_go/server/models/menu.go

package models

import (
	"strings"
	"time"

	"github.com/beego/beego/v2/client/orm"
)

// Menu 菜单模型
type Menu struct {
	Id            int        `orm:"auto"`
	Name          string     `orm:"size(100)"`
	Path          string     `orm:"size(255)"`
	ParentId      int        `orm:"default(0)"`
	Icon          string     `orm:"size(100)"`
	Order         int        `orm:"default(0)"`
	Status        int8       `orm:"default(1)"`
	ComponentPath string     `orm:"size(500);null"`
	IsExternal    int8       `orm:"default(0)"`
	ExternalUrl   string     `orm:"size(1000);null"`
	MenuType      int8       `orm:"default(1)"`
	Permission    string     `orm:"size(200);null"`
	CreateTime    time.Time  `orm:"auto_now_add;type(datetime)"`
	UpdateTime    time.Time  `orm:"auto_now;type(datetime)"`
	DeleteTime    *time.Time `orm:"null;type(datetime)"`
}

// TableName 设置表名
func (m *Menu) TableName() string {
	return "yz_menus"
}

// GetAllMenus 获取所有菜单（未删除的）
func GetAllMenus() ([]map[string]interface{}, error) {
	o := orm.NewOrm()
	var menus []*Menu
	_, err := o.QueryTable("yz_menus").Filter("delete_time__isnull", true).All(&menus)
	if err != nil {
		return nil, err
	}
	result := make([]map[string]interface{}, 0, len(menus))
	for _, m := range menus {
		item := map[string]interface{}{
			"id":            m.Id,
			"name":          m.Name,
			"path":          m.Path,
			"parentId":      m.ParentId,
			"icon":          m.Icon,
			"order":         m.Order,
			"status":        m.Status,
			"componentPath": m.ComponentPath,
			"isExternal":    m.IsExternal,
			"externalUrl":   m.ExternalUrl,
			"menuType":      m.MenuType,
		}
		result = append(result, item)
	}
	return result, nil
}

// AddMenu 添加新菜单
func AddMenu(menu *Menu) (int64, error) {
	o := orm.NewOrm()
	id, err := o.Insert(menu)
	return id, err
}

// UpdateMenu 更新菜单
func UpdateMenu(menu *Menu) error {
	o := orm.NewOrm()
	_, err := o.Update(menu)
	return err
}

// UpdateMenuStatus 更新菜单状态
func UpdateMenuStatus(id int, status int8) error {
	o := orm.NewOrm()
	menu := Menu{Id: id}
	if err := o.Read(&menu); err != nil {
		return err
	}
	menu.Status = status
	_, err := o.Update(&menu, "Status")
	return err
}

// DeleteMenu 删除菜单（软删除）
func DeleteMenu(id int) error {
	o := orm.NewOrm()
	menu := Menu{Id: id}
	if err := o.Read(&menu); err != nil {
		return err
	}
	now := time.Now()
	menu.DeleteTime = &now
	_, err := o.Update(&menu, "DeleteTime")
	return err
}

// GetTenantMenus 根据角色ID获取租户菜单（只返回该角色有权限的菜单，且只返回页面菜单menu_type=1）
func GetTenantMenus(roleId int) ([]map[string]interface{}, error) {
	o := orm.NewOrm()
	
	// 如果角色ID为0或无效，返回空列表
	if roleId <= 0 {
		return []map[string]interface{}{}, nil
	}
	
	// 1. 从yz_role_menus表获取该角色的所有菜单ID
	var menuIds []int
	_, err := o.Raw("SELECT DISTINCT menu_id FROM yz_role_menus WHERE role_id = ?", roleId).QueryRows(&menuIds)
	if err != nil {
		return nil, err
	}
	
	// 如果没有权限，返回空列表
	if len(menuIds) == 0 {
		return []map[string]interface{}{}, nil
	}
	
	// 2. 构建IN查询的占位符
	placeholders := make([]string, len(menuIds))
	args := make([]interface{}, len(menuIds)+1)
	for i, id := range menuIds {
		placeholders[i] = "?"
		args[i] = id
	}
	args[len(menuIds)] = 1 // menu_type=1 表示页面菜单
	
	// 3. 查询菜单（只返回menu_type=1的页面菜单，且未删除的）
	query := "SELECT id, name, path, parent_id, icon, `order`, status, component_path, is_external, external_url, menu_type, permission " +
		"FROM yz_menus " +
		"WHERE id IN (" + strings.Join(placeholders, ",") + ") " +
		"AND delete_time IS NULL " +
		"AND menu_type = ? " +
		"ORDER BY `order`, id"
	
	var menus []*Menu
	_, err = o.Raw(query, args...).QueryRows(&menus)
	if err != nil {
		return nil, err
	}
	
	// 4. 转换为map格式
	result := make([]map[string]interface{}, 0, len(menus))
	for _, m := range menus {
		item := map[string]interface{}{
			"id":            m.Id,
			"name":          m.Name,
			"path":          m.Path,
			"parentId":      m.ParentId,
			"icon":          m.Icon,
			"order":         m.Order,
			"status":        m.Status,
			"componentPath": m.ComponentPath,
			"isExternal":    m.IsExternal,
			"externalUrl":   m.ExternalUrl,
			"menuType":      m.MenuType,
			"permission":    m.Permission,
		}
		result = append(result, item)
	}
	
	return result, nil
}