384 lines
9.7 KiB
JavaScript
384 lines
9.7 KiB
JavaScript
/**
|
||
* 菜单权限模型
|
||
* @file MenuPermission.js
|
||
* @description 菜单权限配置数据模型
|
||
*/
|
||
const { DataTypes } = require('sequelize');
|
||
const BaseModel = require('./BaseModel');
|
||
const { sequelize } = require('../config/database-simple');
|
||
|
||
class MenuPermission extends BaseModel {
|
||
/**
|
||
* 获取用户可访问的菜单
|
||
* @param {Array} userRoles 用户角色数组
|
||
* @param {Array} userPermissions 用户权限数组
|
||
* @returns {Array} 菜单树结构
|
||
*/
|
||
static async getUserMenus(userRoles = [], userPermissions = []) {
|
||
try {
|
||
const allMenus = await this.findAll({
|
||
where: {
|
||
is_visible: true,
|
||
is_enabled: true
|
||
},
|
||
order: [['sort_order', 'ASC']]
|
||
});
|
||
|
||
// 过滤用户有权限的菜单
|
||
const accessibleMenus = allMenus.filter(menu => {
|
||
return this.checkMenuAccess(menu, userRoles, userPermissions);
|
||
});
|
||
|
||
// 构建菜单树
|
||
return this.buildMenuTree(accessibleMenus);
|
||
} catch (error) {
|
||
console.error('获取用户菜单失败:', error);
|
||
return [];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 检查菜单访问权限
|
||
* @param {Object} menu 菜单对象
|
||
* @param {Array} userRoles 用户角色
|
||
* @param {Array} userPermissions 用户权限
|
||
* @returns {boolean} 是否有权限
|
||
*/
|
||
static checkMenuAccess(menu, userRoles, userPermissions) {
|
||
// 解析所需角色
|
||
let requiredRoles = [];
|
||
if (menu.required_roles) {
|
||
try {
|
||
requiredRoles = JSON.parse(menu.required_roles);
|
||
} catch (error) {
|
||
console.error('解析菜单所需角色失败:', error);
|
||
}
|
||
}
|
||
|
||
// 解析所需权限
|
||
let requiredPermissions = [];
|
||
if (menu.required_permissions) {
|
||
try {
|
||
requiredPermissions = JSON.parse(menu.required_permissions);
|
||
} catch (error) {
|
||
console.error('解析菜单所需权限失败:', error);
|
||
}
|
||
}
|
||
|
||
// 如果没有配置权限要求，则默认允许访问
|
||
if (requiredRoles.length === 0 && requiredPermissions.length === 0) {
|
||
return true;
|
||
}
|
||
|
||
// 检查角色权限
|
||
if (requiredRoles.length > 0) {
|
||
const hasRole = requiredRoles.some(role => userRoles.includes(role));
|
||
if (!hasRole) {
|
||
return false;
|
||
}
|
||
}
|
||
|
||
// 检查具体权限
|
||
if (requiredPermissions.length > 0) {
|
||
const hasPermission = requiredPermissions.some(permission =>
|
||
userPermissions.includes(permission)
|
||
);
|
||
if (!hasPermission) {
|
||
return false;
|
||
}
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 构建菜单树结构
|
||
* @param {Array} menus 菜单数组
|
||
* @returns {Array} 菜单树
|
||
*/
|
||
static buildMenuTree(menus) {
|
||
const menuMap = new Map();
|
||
const roots = [];
|
||
|
||
// 创建菜单映射
|
||
menus.forEach(menu => {
|
||
menuMap.set(menu.id, {
|
||
...menu.dataValues,
|
||
children: []
|
||
});
|
||
});
|
||
|
||
// 构建树结构
|
||
menus.forEach(menu => {
|
||
const menuNode = menuMap.get(menu.id);
|
||
|
||
if (menu.parent_id && menuMap.has(menu.parent_id)) {
|
||
const parent = menuMap.get(menu.parent_id);
|
||
parent.children.push(menuNode);
|
||
} else {
|
||
roots.push(menuNode);
|
||
}
|
||
});
|
||
|
||
return roots;
|
||
}
|
||
|
||
/**
|
||
* 初始化默认菜单权限
|
||
*/
|
||
static async initDefaultMenus() {
|
||
try {
|
||
const defaultMenus = [
|
||
// 主要功能模块
|
||
{
|
||
menu_key: 'home',
|
||
menu_name: '首页',
|
||
menu_path: '/',
|
||
menu_type: 'page',
|
||
icon: 'home-outlined',
|
||
sort_order: 1,
|
||
required_roles: '["user", "admin", "manager"]'
|
||
},
|
||
{
|
||
menu_key: 'dashboard',
|
||
menu_name: '系统概览',
|
||
menu_path: '/dashboard',
|
||
menu_type: 'page',
|
||
icon: 'dashboard-outlined',
|
||
sort_order: 2,
|
||
required_roles: '["user", "admin", "manager"]'
|
||
},
|
||
{
|
||
menu_key: 'monitor',
|
||
menu_name: '实时监控',
|
||
menu_path: '/monitor',
|
||
menu_type: 'page',
|
||
icon: 'line-chart-outlined',
|
||
sort_order: 3,
|
||
required_roles: '["user", "admin", "manager"]'
|
||
},
|
||
{
|
||
menu_key: 'analytics',
|
||
menu_name: '数据分析',
|
||
menu_path: '/analytics',
|
||
menu_type: 'page',
|
||
icon: 'bar-chart-outlined',
|
||
sort_order: 4,
|
||
required_roles: '["user", "admin", "manager"]'
|
||
},
|
||
|
||
// 管理功能模块
|
||
{
|
||
menu_key: 'farms',
|
||
menu_name: '养殖场管理',
|
||
menu_path: '/farms',
|
||
menu_type: 'page',
|
||
icon: 'home-outlined',
|
||
sort_order: 10,
|
||
required_roles: '["admin", "manager"]'
|
||
},
|
||
{
|
||
menu_key: 'animals',
|
||
menu_name: '动物管理',
|
||
menu_path: '/animals',
|
||
menu_type: 'page',
|
||
icon: 'bug-outlined',
|
||
sort_order: 11,
|
||
required_roles: '["admin", "manager"]'
|
||
},
|
||
{
|
||
menu_key: 'devices',
|
||
menu_name: '设备管理',
|
||
menu_path: '/devices',
|
||
menu_type: 'page',
|
||
icon: 'desktop-outlined',
|
||
sort_order: 12,
|
||
required_roles: '["admin", "manager"]'
|
||
},
|
||
{
|
||
menu_key: 'alerts',
|
||
menu_name: '预警管理',
|
||
menu_path: '/alerts',
|
||
menu_type: 'page',
|
||
icon: 'alert-outlined',
|
||
sort_order: 13,
|
||
required_roles: '["admin", "manager"]'
|
||
},
|
||
|
||
// 业务功能模块
|
||
{
|
||
menu_key: 'products',
|
||
menu_name: '产品管理',
|
||
menu_path: '/products',
|
||
menu_type: 'page',
|
||
icon: 'shopping-outlined',
|
||
sort_order: 20,
|
||
required_roles: '["admin", "manager"]'
|
||
},
|
||
{
|
||
menu_key: 'orders',
|
||
menu_name: '订单管理',
|
||
menu_path: '/orders',
|
||
menu_type: 'page',
|
||
icon: 'shopping-cart-outlined',
|
||
sort_order: 21,
|
||
required_roles: '["admin", "manager"]'
|
||
},
|
||
{
|
||
menu_key: 'reports',
|
||
menu_name: '报表管理',
|
||
menu_path: '/reports',
|
||
menu_type: 'page',
|
||
icon: 'file-text-outlined',
|
||
sort_order: 22,
|
||
required_roles: '["admin", "manager"]'
|
||
},
|
||
|
||
// 系统管理模块
|
||
{
|
||
menu_key: 'users',
|
||
menu_name: '用户管理',
|
||
menu_path: '/users',
|
||
menu_type: 'page',
|
||
icon: 'user-outlined',
|
||
sort_order: 30,
|
||
required_roles: '["admin"]'
|
||
},
|
||
{
|
||
menu_key: 'system',
|
||
menu_name: '系统管理',
|
||
menu_path: '/system',
|
||
menu_type: 'page',
|
||
icon: 'setting-outlined',
|
||
sort_order: 31,
|
||
required_roles: '["admin"]'
|
||
}
|
||
];
|
||
|
||
for (const menuData of defaultMenus) {
|
||
const existing = await this.findOne({
|
||
where: { menu_key: menuData.menu_key }
|
||
});
|
||
|
||
if (!existing) {
|
||
await this.create(menuData);
|
||
}
|
||
}
|
||
|
||
console.log('默认菜单权限初始化完成');
|
||
} catch (error) {
|
||
console.error('初始化默认菜单权限失败:', error);
|
||
throw error;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 初始化MenuPermission模型
|
||
MenuPermission.init({
|
||
id: {
|
||
type: DataTypes.INTEGER,
|
||
primaryKey: true,
|
||
autoIncrement: true,
|
||
comment: '权限ID'
|
||
},
|
||
menu_key: {
|
||
type: DataTypes.STRING(100),
|
||
allowNull: false,
|
||
comment: '菜单标识'
|
||
},
|
||
menu_name: {
|
||
type: DataTypes.STRING(100),
|
||
allowNull: false,
|
||
comment: '菜单名称'
|
||
},
|
||
menu_path: {
|
||
type: DataTypes.STRING(200),
|
||
allowNull: true,
|
||
comment: '菜单路径'
|
||
},
|
||
parent_id: {
|
||
type: DataTypes.INTEGER,
|
||
allowNull: true,
|
||
comment: '父菜单ID'
|
||
},
|
||
menu_type: {
|
||
type: DataTypes.ENUM('page', 'button', 'api'),
|
||
allowNull: false,
|
||
defaultValue: 'page',
|
||
comment: '菜单类型'
|
||
},
|
||
required_roles: {
|
||
type: DataTypes.TEXT,
|
||
allowNull: true,
|
||
comment: '所需角色（JSON数组）'
|
||
},
|
||
required_permissions: {
|
||
type: DataTypes.TEXT,
|
||
allowNull: true,
|
||
comment: '所需权限（JSON数组）'
|
||
},
|
||
icon: {
|
||
type: DataTypes.STRING(50),
|
||
allowNull: true,
|
||
comment: '菜单图标'
|
||
},
|
||
sort_order: {
|
||
type: DataTypes.INTEGER,
|
||
allowNull: false,
|
||
defaultValue: 0,
|
||
comment: '排序顺序'
|
||
},
|
||
is_visible: {
|
||
type: DataTypes.BOOLEAN,
|
||
allowNull: false,
|
||
defaultValue: true,
|
||
comment: '是否可见'
|
||
},
|
||
is_enabled: {
|
||
type: DataTypes.BOOLEAN,
|
||
allowNull: false,
|
||
defaultValue: true,
|
||
comment: '是否启用'
|
||
},
|
||
description: {
|
||
type: DataTypes.STRING(255),
|
||
allowNull: true,
|
||
comment: '菜单描述'
|
||
}
|
||
}, {
|
||
sequelize,
|
||
tableName: 'menu_permissions',
|
||
modelName: 'MenuPermission',
|
||
comment: '菜单权限表',
|
||
indexes: [
|
||
{
|
||
fields: ['menu_key'],
|
||
unique: true
|
||
},
|
||
{
|
||
fields: ['parent_id']
|
||
},
|
||
{
|
||
fields: ['menu_type']
|
||
},
|
||
{
|
||
fields: ['sort_order']
|
||
}
|
||
]
|
||
});
|
||
|
||
// 定义自关联关系
|
||
MenuPermission.associate = function(models) {
|
||
// 自关联：父子菜单关系
|
||
MenuPermission.hasMany(MenuPermission, {
|
||
as: 'children',
|
||
foreignKey: 'parent_id'
|
||
});
|
||
|
||
MenuPermission.belongsTo(MenuPermission, {
|
||
as: 'parent',
|
||
foreignKey: 'parent_id'
|
||
});
|
||
};
|
||
|
||
module.exports = MenuPermission; |