ff-admin-server/src/command/rolePermissionCommandHandler.ts

import { EntityManager } from "typeorm";
import { dataSource } from "../data-source";
import { rolePermission } from "../entity/role_permission";
import InternalException from "../exceptions/internalException";
import RoleService from "../service/roleService";
import {
  CreateRolePermissionCommand,
  DeleteRolePermissionCommand,
  UpdateRolePermissionsCommand,
} from "./rolePermissionCommand";
import PermissionHelper from "../helpers/permissionHelper";
import RolePermissionService from "../service/rolePermissionService";
import { role } from "../entity/role";

export default abstract class RolePermissionCommandHandler {
  /**
   * @description update role permissions
   * @param UpdateRolePermissionsCommand
   * @returns {Promise<void>}
   */
  static async updatePermissions(updateRolePermissions: UpdateRolePermissionsCommand): Promise<void> {
    let currentPermissions = (await RolePermissionService.getByRole(updateRolePermissions.roleId)).map(
      (r) => r.permission
    );
    return await dataSource.manager
      .transaction(async (manager) => {
        let newPermissions = PermissionHelper.getWhatToAdd(currentPermissions, updateRolePermissions.permissions);
        let removePermissions = PermissionHelper.getWhatToRemove(currentPermissions, updateRolePermissions.permissions);

        for (let permission of newPermissions) {
          await this.updatePermissionsAdd(manager, updateRolePermissions.roleId, permission);
        }

        for (let permission of removePermissions) {
          await this.updatePermissionsRemove(manager, updateRolePermissions.roleId, permission);
        }
      })
      .then(() => {})
      .catch((err) => {
        throw new InternalException("Failed saving role permissions");
      });
  }

  private static async updatePermissionsAdd(manager: EntityManager, userId: number, permission: string): Promise<void> {
    return await manager.createQueryBuilder().relation(role, "permissions").of(userId).add(permission);
  }

  private static async updatePermissionsRemove(
    manager: EntityManager,
    userId: number,
    permission: string
  ): Promise<void> {
    return await manager.createQueryBuilder().relation(role, "permissions").of(userId).remove(permission);
  }

  /**
   * @description grant permission to user
   * @param CreateRolePermissionCommand
   * @returns {Promise<number>}
   */
  static async create(createPermission: CreateRolePermissionCommand): Promise<number> {
    return await dataSource
      .createQueryBuilder()
      .insert()
      .into(rolePermission)
      .values({
        permission: createPermission.permission,
        role: await RoleService.getById(createPermission.roleId),
      })
      .execute()
      .then((result) => {
        return result.identifiers[0].id;
      })
      .catch((err) => {
        throw new InternalException("Failed saving role permission");
      });
  }

  /**
   * @description remove permission from role
   * @param DeleteRolePermissionCommand
   * @returns {Promise<any>}
   */
  static async delete(deletePermission: DeleteRolePermissionCommand): Promise<any> {
    return await dataSource
      .createQueryBuilder()
      .delete()
      .from(rolePermission)
      .where("roleId = :id", { id: deletePermission.roleId })
      .andWhere("permission = :permission", { permission: deletePermission.permission })
      .execute()
      .then((res) => {})
      .catch((err) => {
        throw new InternalException("failed role permission removal");
      });
  }
}