修改管理后台
This commit is contained in:
@@ -4,9 +4,9 @@
|
||||
* @description 定义用户模型,用于数据库操作
|
||||
*/
|
||||
const { DataTypes } = require('sequelize');
|
||||
const bcrypt = require('bcrypt');
|
||||
const bcrypt = require('bcryptjs');
|
||||
const BaseModel = require('./BaseModel');
|
||||
const { sequelize } = require('../config/database-pool');
|
||||
const { sequelize } = require('../config/database-simple');
|
||||
|
||||
class User extends BaseModel {
|
||||
/**
|
||||
@@ -23,7 +23,58 @@ class User extends BaseModel {
|
||||
* @returns {Promise<Array>} 用户角色列表
|
||||
*/
|
||||
async getRoles() {
|
||||
return await this.getRoles();
|
||||
// 简化实现,直接返回当前角色
|
||||
try {
|
||||
const { Role } = require('./index');
|
||||
const role = await Role.findByPk(this.roles);
|
||||
return role ? [role] : [];
|
||||
} catch (error) {
|
||||
console.error('获取用户角色失败:', error);
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取用户权限列表
|
||||
* @returns {Promise<Array>} 用户权限列表
|
||||
*/
|
||||
async getPermissions() {
|
||||
try {
|
||||
const { getRolePermissions } = require('../config/permissions');
|
||||
const roles = await this.getRoles();
|
||||
|
||||
if (roles.length === 0) {
|
||||
return [];
|
||||
}
|
||||
|
||||
// 获取角色的所有权限
|
||||
const allPermissions = new Set();
|
||||
for (const role of roles) {
|
||||
const rolePermissions = getRolePermissions(role.name);
|
||||
rolePermissions.forEach(permission => allPermissions.add(permission));
|
||||
}
|
||||
|
||||
return Array.from(allPermissions);
|
||||
} catch (error) {
|
||||
console.error('获取用户权限失败:', error);
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 检查用户是否具有指定权限
|
||||
* @param {string|Array} permissions 权限名称或权限数组
|
||||
* @returns {Promise<boolean>} 是否有权限
|
||||
*/
|
||||
async hasPermission(permissions) {
|
||||
try {
|
||||
const { hasPermission } = require('../config/permissions');
|
||||
const userPermissions = await this.getPermissions();
|
||||
return hasPermission(userPermissions, permissions);
|
||||
} catch (error) {
|
||||
console.error('检查用户权限失败:', error);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -44,16 +95,13 @@ class User extends BaseModel {
|
||||
|
||||
/**
|
||||
* 为用户分配角色
|
||||
* @param {Number|Array} roleId 角色ID或角色ID数组
|
||||
* @param {Number} roleId 角色ID
|
||||
* @returns {Promise<Boolean>} 分配结果
|
||||
*/
|
||||
async assignRole(roleId) {
|
||||
try {
|
||||
if (Array.isArray(roleId)) {
|
||||
await this.addRoles(roleId);
|
||||
} else {
|
||||
await this.addRole(roleId);
|
||||
}
|
||||
this.roles = roleId;
|
||||
await this.save();
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('分配角色失败:', error);
|
||||
@@ -63,16 +111,12 @@ class User extends BaseModel {
|
||||
|
||||
/**
|
||||
* 移除用户角色
|
||||
* @param {Number|Array} roleId 角色ID或角色ID数组
|
||||
* @returns {Promise<Boolean>} 移除结果
|
||||
*/
|
||||
async removeRole(roleId) {
|
||||
async removeRole() {
|
||||
try {
|
||||
if (Array.isArray(roleId)) {
|
||||
await this.removeRoles(roleId);
|
||||
} else {
|
||||
await this.removeRole(roleId);
|
||||
}
|
||||
this.roles = null;
|
||||
await this.save();
|
||||
return true;
|
||||
} catch (error) {
|
||||
console.error('移除角色失败:', error);
|
||||
@@ -122,6 +166,15 @@ User.init({
|
||||
type: DataTypes.STRING(255),
|
||||
allowNull: true
|
||||
},
|
||||
roles: {
|
||||
type: DataTypes.INTEGER,
|
||||
allowNull: true,
|
||||
defaultValue: 2, // 默认为user角色ID
|
||||
references: {
|
||||
model: 'roles',
|
||||
key: 'id'
|
||||
}
|
||||
},
|
||||
status: {
|
||||
type: DataTypes.ENUM('active', 'inactive', 'suspended'),
|
||||
defaultValue: 'active'
|
||||
|
||||
Reference in New Issue
Block a user