chore: src/modules/base/service/sys/role.ts

2026-07-01 17:56:18 +08:00
parent a2b5fa2a2b
commit 6202dfe387
1 changed files with 136 additions and 0 deletions
@@ -0,0 +1,136 @@
+import { Inject, Provide } from '@midwayjs/core';
+import { BaseService } from '@cool-midway/core';
+import { InjectEntityModel } from '@midwayjs/typeorm';
+import { Repository } from 'typeorm';
+import { BaseSysRoleEntity } from '../../entity/sys/role';
+import { BaseSysUserRoleEntity } from '../../entity/sys/user_role';
+import * as _ from 'lodash';
+import { BaseSysRoleMenuEntity } from '../../entity/sys/role_menu';
+import { BaseSysRoleDepartmentEntity } from '../../entity/sys/role_department';
+import { BaseSysPermsService } from './perms';
+import { Brackets } from 'typeorm';
+
+/**
+ * 角色
+ */
+@Provide()
+export class BaseSysRoleService extends BaseService {
+  @InjectEntityModel(BaseSysRoleEntity)
+  baseSysRoleEntity: Repository<BaseSysRoleEntity>;
+
+  @InjectEntityModel(BaseSysUserRoleEntity)
+  baseSysUserRoleEntity: Repository<BaseSysUserRoleEntity>;
+
+  @InjectEntityModel(BaseSysRoleMenuEntity)
+  baseSysRoleMenuEntity: Repository<BaseSysRoleMenuEntity>;
+
+  @InjectEntityModel(BaseSysRoleDepartmentEntity)
+  baseSysRoleDepartmentEntity: Repository<BaseSysRoleDepartmentEntity>;
+
+  @Inject()
+  baseSysPermsService: BaseSysPermsService;
+
+  @Inject()
+  ctx;
+
+  /**
+   * 根据用户ID获得所有用户角色
+   * @param userId
+   */
+  async getByUser(userId: number): Promise<number[]> {
+    const userRole = await this.baseSysUserRoleEntity.findBy({ userId });
+    if (!_.isEmpty(userRole)) {
+      return userRole.map(e => {
+        return e.roleId;
+      });
+    }
+    return [];
+  }
+
+  /**
+   *
+   * @param param
+   */
+  async modifyAfter(param) {
+    if (param.id) {
+      this.updatePerms(param.id, param.menuIdList, param.departmentIdList);
+    }
+  }
+
+  /**
+   * 更新权限
+   * @param roleId
+   * @param menuIdList
+   * @param departmentIds
+   */
+  async updatePerms(roleId, menuIdList?, departmentIds = []) {
+    // 更新菜单权限
+    await this.baseSysRoleMenuEntity.delete({ roleId });
+    await Promise.all(
+      menuIdList.map(async e => {
+        return await this.baseSysRoleMenuEntity.save({ roleId, menuId: e });
+      })
+    );
+    // 更新部门权限
+    await this.baseSysRoleDepartmentEntity.delete({ roleId });
+    await Promise.all(
+      departmentIds.map(async e => {
+        return await this.baseSysRoleDepartmentEntity.save({
+          roleId,
+          departmentId: e,
+        });
+      })
+    );
+    // 刷新权限
+    const userRoles = await this.baseSysUserRoleEntity.findBy({ roleId });
+    for (const userRole of userRoles) {
+      await this.baseSysPermsService.refreshPerms(userRole.userId);
+    }
+  }
+
+  /**
+   * 角色信息
+   * @param id
+   */
+  async info(id) {
+    const info = await this.baseSysRoleEntity.findOneBy({ id });
+    if (info) {
+      const menus = await this.baseSysRoleMenuEntity.findBy(
+        id !== 1 ? { roleId: id } : {}
+      );
+      const menuIdList = menus.map(e => {
+        return parseInt(e.menuId + '');
+      });
+      const departments = await this.baseSysRoleDepartmentEntity.findBy(
+        id !== 1 ? { roleId: id } : {}
+      );
+      const departmentIdList = departments.map(e => {
+        return parseInt(e.departmentId + '');
+      });
+      return {
+        ...info,
+        menuIdList,
+        departmentIdList,
+      };
+    }
+    return {};
+  }
+
+  async list() {
+    return this.baseSysRoleEntity
+      .createQueryBuilder('a')
+      .where(
+        new Brackets(qb => {
+          qb.where('a.id !=:id', { id: 1 }); // 超级管理员的角色不展示
+          // 如果不是超管，只能看到自己新建的或者自己有的角色
+          if (this.ctx.admin.username !== 'admin') {
+            qb.andWhere('(a.userId=:userId or a.id in (:...roleId))', {
+              userId: this.ctx.admin.userId,
+              roleId: this.ctx.admin.roleIds,
+            });
+          }
+        })
+      )
+      .getMany();
+  }
+}