rez_core 5.0.84 → 5.0.86

package/src/module/module/repository/menu.repository.ts

@@ -1,191 +1,184 @@
-import { Injectable } from '@nestjs/common';
-import { DataSource, Repository } from 'typeorm';
-import { MenuData } from '../entity/menu.entity';
-import { ModuleAccess } from '../entity/module-access.entity';
-import { UserRoleMapping } from 'src/module/user/entity/user-role-mapping.entity';
-import { Role } from 'src/module/user/entity/role.entity';
-import { InjectRepository } from '@nestjs/typeorm';
-import { get } from 'http';
-import { ReflectionHelper } from '../../../utils/service/reflection-helper.service';
-
-@Injectable()
-export class MenuRepository extends Repository<MenuData> {
-  constructor(
-    private readonly dataSource: DataSource,
-    @InjectRepository(MenuData) private readonly menuData: Repository<MenuData>,
-    private readonly reflectionHelper: ReflectionHelper,
-  ) {
-    super(MenuData, dataSource.createEntityManager());
-  }
-
-  async getUserRoles(userId: number, appcode: string): Promise<number[]> {
-    const roles = await this.dataSource
-      .getRepository(UserRoleMapping)
-      .createQueryBuilder('urm')
-      .innerJoin('sso_role', 'role', 'role.id = urm.role_id')
-      .select('urm.role_id')
-      .where('urm.user_id = :userId', { userId })
-      .andWhere('role.appcode = :appcode', { appcode })
-      .getRawMany();
-
-    return roles.map((r) => r.urm_role_id);
-  }
-
-  async getAccessibleModules(
-    roleIds: number[],
-    appCode: string,
-  ): Promise<string[]> {
-    const roleCodes = await this.dataSource
-      .getRepository(Role)
-      .createQueryBuilder('role')
-      .select('role.code')
-      .where('role.id IN (:...roleIds)', { roleIds })
-      .getMany();
-
-    const codes = roleCodes.map((role) => role.code);
-    if (codes.length === 0) return [];
-
-    const modules = await this.dataSource
-      .getRepository(ModuleAccess)
-      .createQueryBuilder('moduleAccess')
-      .select('moduleAccess.module_code')
-      .where('moduleAccess.role_code IN (:...codes)', { codes })
-      .andWhere('moduleAccess.access_flag > 0')
-      .andWhere('LOWER(moduleAccess.appcode) = LOWER(:appCode)', { appCode })
-      .getMany();
-
-    return Array.from(new Set(modules.map((module) => module.module_code)));
-  }
-
-  /**
-   * ✅ FIXED: Use this.createQueryBuilder() instead of this.menuData.createQueryBuilder()
-   */
-  async getMenuItems(
-    moduleCodes: string[],
-    appcode: string,
-    levelType: string,
-  ) {
-    return await this.menuData
-      .createQueryBuilder('menu')
-      .leftJoin(
-        'sso_module',
-        'module',
-        `
-          menu.module_code = module.module_code
-          AND LOWER(menu.appcode) = LOWER(module.appcode)
-          AND menu.level_type = module.level_type
-        `,
-      )
-      .where('menu.module_code IN (:...moduleCodes)', { moduleCodes })
-      .andWhere('LOWER(menu.appcode) = LOWER(:appcode)', { appcode })
-      .andWhere('menu.level_type = :levelType', { levelType })
-      .andWhere('menu.ui_visible = 1')
-      .select([
-        'menu.*',
-        'module.component_name AS component',
-        'module.title AS title',
-        'module.entity_type AS entity_type',
-        'module.type AS type',
-        'module.ui_config AS ui_config',
-      ])
-      .getRawMany();
-  }
-
-  async resolveUserRoles(
-    userId: number,
-    appcode: string,
-    levelType: string,
-    levelId: number,
-  ): Promise<number[]> {
-    const repo = this.reflectionHelper.getRepoService('UserRoleMapping');
-
-    if (userId) {
-      // Check if user has any role mappings for the given appcode
-      const getUserDetails = await this.dataSource.query(
-        `
-            SELECT *
-            from sso_user
-            where id = $1
-        `,
-        [userId],
-      );
-
-      // get user role from urm
-
-      const getUserUrm = await this.dataSource.query(
-        `
-            SELECT *
-            from sso_user_role_mapping
-            where user_id = $1
-              and appcode = $2
-        `,
-        [userId, appcode],
-      );
-
-      if (
-        getUserDetails.length > 0 &&
-        getUserDetails[0].organization_id === 1
-      ) {
-        return [Number(getUserUrm[0].role_id)];
-      }
-    }
-
-    // 1. Try exact level match
-    let roles = await repo
-      .createQueryBuilder('urm')
-      .innerJoin('sso_role', 'role', 'role.id = urm.role_id')
-      .select('urm.role_id')
-      .where('urm.user_id = :userId', { userId: userId })
-      .andWhere('urm.level_type = :levelType', { levelType })
-      .andWhere('urm.level_id = :levelId', { levelId: levelId })
-      .andWhere('role.appcode = :appcode', { appcode })
-      .getRawMany();
-
-    if (roles.length) return roles.map((r) => r.urm_role_id);
-
-    // 2. If SCH fallback → BRN → ORG
-    if (levelType === 'SCH') {
-      const [sch] = await this.dataSource.query(
-        `
-            SELECT s.brand_id, s.organization_id
-            FROM sso_school s
-            WHERE s.id = $1
-        `,
-        [levelId],
-      );
-
-      const brandId = sch?.brand_id;
-      const orgId = sch?.organization_id;
-
-      if (brandId) {
-        roles = await repo
-          .createQueryBuilder('urm')
-          .innerJoin('sso_role', 'role', 'role.id = urm.role_id')
-          .select('urm.role_id')
-          .where('urm.user_id = :userId', { userId })
-          .andWhere('urm.level_type = :levelType', { levelType: 'BRN' })
-          .andWhere('urm.level_id = :levelId', { levelId: brandId })
-          .andWhere('role.appcode = :appcode', { appcode })
-          .getRawMany();
-
-        if (roles.length) return roles.map((r) => r.urm_role_id);
-      }
-
-      if (orgId) {
-        roles = await repo
-          .createQueryBuilder('urm')
-          .innerJoin('sso_role', 'role', 'role.id = urm.role_id')
-          .select('urm.role_id')
-          .where('urm.user_id = :userId', { userId })
-          .andWhere('urm.level_type = :levelType', { levelType: 'ORG' })
-          .andWhere('urm.level_id = :levelId', { levelId: orgId })
-          .andWhere('role.appcode = :appcode', { appcode })
-          .getRawMany();
-
-        if (roles.length) return roles.map((r) => r.urm_role_id);
-      }
-    }
-
-    return [];
-  }
-}
+import { Injectable } from '@nestjs/common';
+import { DataSource, Repository } from 'typeorm';
+import { MenuData } from '../entity/menu.entity';
+import { ModuleAccess } from '../entity/module-access.entity';
+import { InjectRepository } from '@nestjs/typeorm';
+import { ReflectionHelper } from '../../../utils/service/reflection-helper.service';
+
+@Injectable()
+export class MenuRepository extends Repository<MenuData> {
+  constructor(
+    private readonly dataSource: DataSource,
+    @InjectRepository(MenuData) private readonly menuData: Repository<MenuData>,
+    private readonly reflectionHelper: ReflectionHelper,
+    @InjectRepository(ModuleAccess) private readonly moduleAccessRepo: Repository<ModuleAccess>,
+  ) {
+    super(MenuData, dataSource.createEntityManager());
+  }
+
+  async getUserRoles(userId: number, appcode: string): Promise<number[]> {
+    const userRoleMappingRepo = this.reflectionHelper.getRepoService('UserRoleMapping');
+
+    const roles = await userRoleMappingRepo
+      .createQueryBuilder('urm')
+      .innerJoin('sso_role', 'role', 'role.id = urm.role_id')
+      .select('urm.role_id', 'role_id')
+      .where('urm.user_id = :userId', { userId })
+      .andWhere('role.appcode = :appcode', { appcode })
+      .getRawMany();
+
+    return roles.map((r) => r.urm_role_id);
+  }
+
+  async getAccessibleModules(
+    roleIds: number[],
+    appCode: string,
+  ): Promise<string[]> {
+    const roleRepo = this.reflectionHelper.getRepoService('Role');
+
+    const roleCodes = await roleRepo
+      .createQueryBuilder('role')
+      .select('role.code', 'code')
+      .where('role.id IN (:...roleIds)', { roleIds })
+      .getRawMany();
+
+    const codes = roleCodes.map(r => r.code);
+
+    if (codes.length === 0) return [];
+
+
+    const modules = await this.moduleAccessRepo
+      .createQueryBuilder('moduleAccess')
+      .select('moduleAccess.module_code')
+      .where('moduleAccess.role_code IN (:...codes)', { codes })
+      .andWhere('moduleAccess.access_flag > 0')
+      .andWhere('LOWER(moduleAccess.appcode) = LOWER(:appCode)', { appCode })
+      .getMany();
+
+    return Array.from(new Set(modules.map((module) => module.module_code)));
+  }
+
+  /**
+   * ✅ FIXED: Use this.createQueryBuilder() instead of this.menuData.createQueryBuilder()
+   */
+  async getMenuItems(
+    moduleCodes: string[],
+    appcode: string,
+    levelType: string,
+  ) {
+    return await this.menuData
+      .createQueryBuilder('menu')
+      .leftJoin(
+        'sso_module',
+        'module',
+        `
+          menu.module_code = module.module_code
+          AND LOWER(menu.appcode) = LOWER(module.appcode)
+          AND menu.level_type = module.level_type
+        `,
+      )
+      .where('menu.module_code IN (:...moduleCodes)', { moduleCodes })
+      .andWhere('LOWER(menu.appcode) = LOWER(:appcode)', { appcode })
+      .andWhere('menu.level_type = :levelType', { levelType })
+      .andWhere('menu.ui_visible = 1')
+      .select([
+        'menu.*',
+        'module.component_name AS component',
+        'module.title AS title',
+        'module.entity_type AS entity_type',
+        'module.type AS type',
+        'module.ui_config AS ui_config',
+      ])
+      .getRawMany();
+  }
+
+  async resolveUserRoles(
+    userId: number,
+    appcode: string,
+    levelType: string,
+    levelId: number,
+  ): Promise<number[]> {
+    const repo = this.reflectionHelper.getRepoService('UserRoleMapping');
+
+    if (userId) {
+      // Check if user has any role mappings for the given appcode
+      const userRepo = this.reflectionHelper.getRepoService('User');
+      const getUserDetails = await userRepo.findOne({
+        where: {
+          id: userId,
+        },
+      });
+
+      // get user role from urm
+      const getUserUrm = await repo.findOne({
+        where: {
+          user_id: userId,
+          appcode: appcode,
+        },
+      });
+
+      if (
+        getUserDetails.length > 0 &&
+        getUserDetails[0].organization_id === 1
+      ) {
+        return [Number(getUserUrm.role_id)];
+      }
+    }
+
+    // 1. Try exact level match
+    let roles = await repo
+      .createQueryBuilder('urm')
+      .innerJoin('sso_role', 'role', 'role.id = urm.role_id')
+      .select('urm.role_id')
+      .where('urm.user_id = :userId', { userId: userId })
+      .andWhere('urm.level_type = :levelType', { levelType })
+      .andWhere('urm.level_id = :levelId', { levelId: levelId })
+      .andWhere('role.appcode = :appcode', { appcode })
+      .getRawMany();
+
+    if (roles.length) return roles.map((r) => r.urm_role_id);
+
+    // 2. If SCH fallback → BRN → ORG
+    if (levelType === 'SCH') {
+      const schoolRepo = this.reflectionHelper.getRepoService('SSOSchool');
+      const sch = await schoolRepo.findOne({
+        where: {
+          id: levelId,
+        },
+      });
+
+      const brandId = sch?.brand_id;
+      const orgId = sch?.organization_id;
+
+      if (brandId) {
+        roles = await repo
+          .createQueryBuilder('urm')
+          .innerJoin('sso_role', 'role', 'role.id = urm.role_id')
+          .select('urm.role_id')
+          .where('urm.user_id = :userId', { userId })
+          .andWhere('urm.level_type = :levelType', { levelType: 'BRN' })
+          .andWhere('urm.level_id = :levelId', { levelId: brandId })
+          .andWhere('role.appcode = :appcode', { appcode })
+          .getRawMany();
+
+        if (roles.length) return roles.map((r) => r.urm_role_id);
+      }
+
+      if (orgId) {
+        roles = await repo
+          .createQueryBuilder('urm')
+          .innerJoin('sso_role', 'role', 'role.id = urm.role_id')
+          .select('urm.role_id')
+          .where('urm.user_id = :userId', { userId })
+          .andWhere('urm.level_type = :levelType', { levelType: 'ORG' })
+          .andWhere('urm.level_id = :levelId', { levelId: orgId })
+          .andWhere('role.appcode = :appcode', { appcode })
+          .getRawMany();
+
+        if (roles.length) return roles.map((r) => r.urm_role_id);
+      }
+    }
+
+    return [];
+  }
+}