@flusys/nestjs-iam 6.1.0 → 6.1.2

package/cjs/services/permission.service.js

@@ -18,10 +18,8 @@ const _permissionwithcompanyentity = require("../entities/permission-with-compan
 const _rolewithcompanyentity = require("../entities/role-with-company.entity");
 const _roleentity = require("../entities/role.entity");
 const _useriampermissionentity = require("../entities/user-iam-permission.entity");
-const _actiontypeenum = require("../enums/action-type.enum");
 const _iamentitytypeenum = require("../enums/iam-entity-type.enum");
 const _iampermissiontypeenum = require("../enums/iam-permission-type.enum");
-const _permissiontypeenum = require("../enums/permission-type.enum");
 const _iamconfigservice = require("./iam-config.service");
 const _iamdatasourceservice = require("./iam-datasource.service");
 const _permissioncacheservice = require("./permission-cache.service");
@@ -120,7 +118,6 @@ let PermissionService = class PermissionService {
                     sourceId: dto.userId,
                     targetType: _iamentitytypeenum.IamEntityType.ACTION,
                     targetId: item.id,
-                    userId: dto.userId,
                     companyId: enableCompanyFeature ? companyId : null,
                     branchId: enableCompanyFeature ? branchId : null
                 }));
@@ -155,12 +152,12 @@ let PermissionService = class PermissionService {
             const result = await permissionRepo.delete(whereDelete);
             removed = result.affected || 0;
         }
-        await this.permissionCacheService.invalidateUser(dto.userId, enableCompanyFeature ? companyId : null, [
+        await this.permissionCacheService.invalidateUser(dto.userId, enableCompanyFeature ? companyId : null, enableCompanyFeature ? [
             branchId
-        ]);
+        ] : []);
         return this.buildOperationResult(dto.items.length, added, removed);
     }
-    async getUserActions(userId, branchId, companyId) {
+    async getUserActions(userId, companyId, branchId) {
         const permissionRepo = await this.getPermissionRepository();
         const enableCompanyFeature = this.iamConfigService.isCompanyFeatureEnabled();
         const where = {
@@ -170,7 +167,7 @@ let PermissionService = class PermissionService {
         };
         if (enableCompanyFeature) {
             if (companyId) where.companyId = companyId;
-            where.branchId = branchId ?? null;
+            if (branchId) where.branchId = branchId;
         }
         const permissions = await permissionRepo.find({
             where
@@ -184,10 +181,11 @@ let PermissionService = class PermissionService {
             const action = actionMap.get(p.targetId);
             return {
                 id: p.id,
-                userId: p.userId,
+                userId: p.sourceId,
                 actionId: action.id,
                 actionCode: action.code ?? '',
                 actionName: action.name,
+                companyId: ('companyId' in p ? p.companyId : null) ?? null,
                 branchId: ('branchId' in p ? p.branchId : null) ?? null,
                 createdAt: p.createdAt
             };
@@ -241,7 +239,6 @@ let PermissionService = class PermissionService {
                     sourceId: dto.roleId,
                     targetType: _iamentitytypeenum.IamEntityType.ACTION,
                     targetId: item.id,
-                    userId: null,
                     companyId: enableCompanyFeature ? roleCompanyId : null,
                     branchId: null
                 }));
@@ -271,7 +268,7 @@ let PermissionService = class PermissionService {
             });
             removed = result.affected || 0;
         }
-        await this.invalidateRoleMembersCache(dto.roleId);
+        await this.permissionCacheService.invalidateRoleMembersCache(dto.roleId);
         return this.buildOperationResult(dto.items.length, added, removed);
     }
     async getRoleActions(roleId) {
@@ -303,94 +300,88 @@ let PermissionService = class PermissionService {
     // Company-Action Permissions (Whitelist)
     /** Assign or remove actions to/from a company (whitelist) */ async assignCompanyActions(dto) {
         const permissionRepo = await this.getPermissionRepository();
-        const dataSource = permissionRepo.manager.connection;
+        const dataSource = await this.dataSourceProvider.getDataSource();
+        const enableCompanyFeature = this.iamConfigService.isCompanyFeatureEnabled();
         const { toAdd: itemsToAdd, toRemove: itemsToRemove } = this.splitItemsByAction(dto.items);
         let added = 0;
         let removed = 0;
         await dataSource.transaction(async (manager)=>{
             const transactionalPermissionRepo = manager.getRepository(permissionRepo.target);
             if (itemsToAdd.length > 0) {
-                added = await this.addCompanyActions(transactionalPermissionRepo, dto.companyId, itemsToAdd.map((item)=>item.id));
+                const actionIdsToAdd = itemsToAdd.map((item)=>item.id);
+                const existingPermissions = await transactionalPermissionRepo.find({
+                    where: {
+                        permissionType: _iampermissiontypeenum.IamPermissionType.COMPANY_ACTION,
+                        sourceType: _iamentitytypeenum.IamEntityType.COMPANY,
+                        sourceId: dto.companyId,
+                        targetType: _iamentitytypeenum.IamEntityType.ACTION,
+                        targetId: (0, _typeorm.In)(actionIdsToAdd)
+                    },
+                    select: [
+                        'targetId'
+                    ]
+                });
+                const existingActionIds = new Set(existingPermissions.map((p)=>p.targetId));
+                const newActionIds = actionIdsToAdd.filter((id)=>!existingActionIds.has(id));
+                if (newActionIds.length > 0) {
+                    const newPermissions = newActionIds.map((actionId)=>transactionalPermissionRepo.create({
+                            permissionType: _iampermissiontypeenum.IamPermissionType.COMPANY_ACTION,
+                            sourceType: _iamentitytypeenum.IamEntityType.COMPANY,
+                            sourceId: dto.companyId,
+                            targetType: _iamentitytypeenum.IamEntityType.ACTION,
+                            targetId: actionId
+                        }));
+                    await transactionalPermissionRepo.save(newPermissions);
+                    added = newPermissions.length;
+                }
             }
             if (itemsToRemove.length > 0) {
                 const actionIdsToRemove = itemsToRemove.map((item)=>item.id);
-                removed = await this.removeCompanyActionsWithCascade(manager, dto.companyId, actionIdsToRemove);
+                const companyResult = await transactionalPermissionRepo.delete({
+                    permissionType: _iampermissiontypeenum.IamPermissionType.COMPANY_ACTION,
+                    sourceType: _iamentitytypeenum.IamEntityType.COMPANY,
+                    sourceId: dto.companyId,
+                    targetType: _iamentitytypeenum.IamEntityType.ACTION,
+                    targetId: (0, _typeorm.In)(actionIdsToRemove)
+                });
+                removed = companyResult.affected || 0;
+                // Cascade: an action no longer on the company whitelist must not remain
+                // reachable via role grants or direct user grants scoped to this company.
+                const roleEntity = enableCompanyFeature ? _rolewithcompanyentity.RoleWithCompany : _roleentity.Role;
+                const roleRepo = manager.getRepository(roleEntity);
+                const companyRoles = await roleRepo.find({
+                    where: {
+                        companyId: dto.companyId,
+                        deletedAt: (0, _typeorm.IsNull)()
+                    },
+                    select: [
+                        'id'
+                    ]
+                });
+                if (companyRoles.length > 0) {
+                    const roleIds = companyRoles.map((role)=>role.id);
+                    await transactionalPermissionRepo.delete({
+                        permissionType: _iampermissiontypeenum.IamPermissionType.ROLE_ACTION,
+                        sourceType: _iamentitytypeenum.IamEntityType.ROLE,
+                        sourceId: (0, _typeorm.In)(roleIds),
+                        targetType: _iamentitytypeenum.IamEntityType.ACTION,
+                        targetId: (0, _typeorm.In)(actionIdsToRemove)
+                    });
+                }
+                if (enableCompanyFeature) {
+                    await transactionalPermissionRepo.delete({
+                        permissionType: _iampermissiontypeenum.IamPermissionType.USER_ACTION,
+                        companyId: dto.companyId,
+                        targetType: _iamentitytypeenum.IamEntityType.ACTION,
+                        targetId: (0, _typeorm.In)(actionIdsToRemove)
+                    });
+                }
             }
         });
-        await this.invalidateCompanyMembersCache(dto.companyId);
+        await this.permissionCacheService.invalidateCompanyMembersCache(dto.companyId);
         return this.buildOperationResult(dto.items.length, added, removed);
     }
-    async addCompanyActions(permissionRepo, companyId, actionIds) {
-        const existingPermissions = await permissionRepo.find({
-            where: {
-                permissionType: _iampermissiontypeenum.IamPermissionType.COMPANY_ACTION,
-                sourceType: _iamentitytypeenum.IamEntityType.COMPANY,
-                sourceId: companyId,
-                targetType: _iamentitytypeenum.IamEntityType.ACTION,
-                targetId: (0, _typeorm.In)(actionIds)
-            },
-            select: [
-                'targetId'
-            ]
-        });
-        const existingActionIds = new Set(existingPermissions.map((p)=>p.targetId));
-        const newActionIds = actionIds.filter((id)=>!existingActionIds.has(id));
-        if (newActionIds.length === 0) {
-            return 0;
-        }
-        const newPermissions = newActionIds.map((actionId)=>permissionRepo.create({
-                permissionType: _iampermissiontypeenum.IamPermissionType.COMPANY_ACTION,
-                sourceType: _iamentitytypeenum.IamEntityType.COMPANY,
-                sourceId: companyId,
-                targetType: _iamentitytypeenum.IamEntityType.ACTION,
-                targetId: actionId,
-                userId: null
-            }));
-        await permissionRepo.save(newPermissions);
-        return newPermissions.length;
-    }
-    async removeCompanyActionsWithCascade(manager, companyId, actionIds) {
-        const permissionEntity = this.iamConfigService.isCompanyFeatureEnabled() ? _permissionwithcompanyentity.UserIamPermissionWithCompany : _useriampermissionentity.UserIamPermission;
-        const permissionRepo = manager.getRepository(permissionEntity);
-        const companyResult = await permissionRepo.delete({
-            permissionType: _iampermissiontypeenum.IamPermissionType.COMPANY_ACTION,
-            sourceType: _iamentitytypeenum.IamEntityType.COMPANY,
-            sourceId: companyId,
-            targetType: _iamentitytypeenum.IamEntityType.ACTION,
-            targetId: (0, _typeorm.In)(actionIds)
-        });
-        const roleEntity = this.iamConfigService.isCompanyFeatureEnabled() ? _rolewithcompanyentity.RoleWithCompany : _roleentity.Role;
-        const roleRepo = manager.getRepository(roleEntity);
-        const companyRoles = await roleRepo.find({
-            where: {
-                companyId,
-                deletedAt: (0, _typeorm.IsNull)()
-            },
-            select: [
-                'id'
-            ]
-        });
-        if (companyRoles.length > 0) {
-            const roleIds = companyRoles.map((role)=>role.id);
-            await permissionRepo.delete({
-                permissionType: _iampermissiontypeenum.IamPermissionType.ROLE_ACTION,
-                sourceType: _iamentitytypeenum.IamEntityType.ROLE,
-                sourceId: (0, _typeorm.In)(roleIds),
-                targetType: _iamentitytypeenum.IamEntityType.ACTION,
-                targetId: (0, _typeorm.In)(actionIds)
-            });
-        }
-        if (this.iamConfigService.isCompanyFeatureEnabled()) {
-            await permissionRepo.delete({
-                permissionType: _iampermissiontypeenum.IamPermissionType.USER_ACTION,
-                companyId,
-                targetType: _iamentitytypeenum.IamEntityType.ACTION,
-                targetId: (0, _typeorm.In)(actionIds)
-            });
-        }
-        return companyResult.affected || 0;
-    }
-    /** Get all actions assigned to a company (whitelist) */ async getCompanyActions(companyId) {
+    /** Get all actions assigned to a company (whitelist) */ async getCompanyActions(companyId, isOnlyId = false) {
         const permissionRepo = await this.getPermissionRepository();
         const permissions = await permissionRepo.find({
             where: {
@@ -403,6 +394,7 @@ let PermissionService = class PermissionService {
             return [];
         }
         const actionIds = permissions.map((p)=>p.targetId);
+        if (isOnlyId) return actionIds;
         const actionMap = await this.getActionsMap(actionIds);
         return permissions.filter((p)=>actionMap.has(p.targetId)).map((p)=>{
             const action = actionMap.get(p.targetId);
@@ -416,20 +408,6 @@ let PermissionService = class PermissionService {
             };
         });
     }
-    /** Get action IDs whitelisted for a company */ async getCompanyActionIds(companyId) {
-        const permissionRepo = await this.getPermissionRepository();
-        const permissions = await permissionRepo.find({
-            where: {
-                permissionType: _iampermissiontypeenum.IamPermissionType.COMPANY_ACTION,
-                sourceType: _iamentitytypeenum.IamEntityType.COMPANY,
-                sourceId: companyId
-            },
-            select: [
-                'targetId'
-            ]
-        });
-        return permissions.map((p)=>p.targetId);
-    }
     // User-Role Permissions
     /** Assign user to roles (branch-scoped when company feature is enabled) */ async assignUserRoles(dto) {
         if (!this.iamConfigService.isRbacEnabled()) {
@@ -471,7 +449,6 @@ let PermissionService = class PermissionService {
                     sourceId: dto.userId,
                     targetType: _iamentitytypeenum.IamEntityType.ROLE,
                     targetId: item.id,
-                    userId: dto.userId,
                     companyId: enableCompanyFeature ? companyId : null,
                     branchId: enableCompanyFeature ? branchId : null
                 }));
@@ -522,7 +499,7 @@ let PermissionService = class PermissionService {
         };
         if (enableCompanyFeature) {
             if (companyId) where.companyId = companyId;
-            where.branchId = branchId ?? null;
+            if (branchId) where.branchId = branchId;
         }
         const permissions = await permissionRepo.find({
             where
@@ -546,7 +523,7 @@ let PermissionService = class PermissionService {
             const permissionEntity = p;
             return {
                 id: p.id,
-                userId: p.userId,
+                userId: p.sourceId,
                 roleId: role.id,
                 roleName: role.name,
                 branchId: enableCompanyFeature ? permissionEntity.branchId ?? null : null,
@@ -556,129 +533,7 @@ let PermissionService = class PermissionService {
     }
     // My Permissions
     /** Get user's effective permissions (cache-first approach) */ async getMyPermissions(userId, branchId, companyId, parentCodes) {
-        const enableCompanyFeature = this.iamConfigService.isCompanyFeatureEnabled();
-        const cacheOptions = {
-            userId,
-            companyId: enableCompanyFeature ? companyId : null,
-            branchId
-        };
-        const cachedData = await this.permissionCacheService.getMyPermissions(cacheOptions);
-        if (cachedData) {
-            return this.buildResponseFromCache(cachedData, parentCodes);
-        }
-        const freshData = await this.fetchAndCachePermissions(userId, branchId, companyId);
-        return this.buildResponseFromCache(freshData, parentCodes);
-    }
-    /** Build response from cached data, applying optional parentCodes filter */ async buildResponseFromCache(cachedData, parentCodes) {
-        let frontendActions = cachedData.frontendActions;
-        if (parentCodes?.length) {
-            const parentIds = await this.getParentIdsByCodesWithCache(parentCodes);
-            if (parentIds.size > 0) {
-                frontendActions = frontendActions.filter((a)=>a.parentId && parentIds.has(a.parentId));
-            } else {
-                frontendActions = [];
-            }
-        }
-        return {
-            frontendActions: frontendActions.map((a)=>({
-                    id: a.id,
-                    code: a.code,
-                    name: a.name,
-                    description: a.description
-                })),
-            cachedEndpoints: cachedData.backendCodes.length
-        };
-    }
-    /** Resolve action IDs for the given codes */ async getParentIdsByCodesWithCache(codes) {
-        const actionRepo = await this.getActionRepository();
-        const actions = await actionRepo.find({
-            where: {
-                code: (0, _typeorm.In)(codes)
-            },
-            select: [
-                'id',
-                'code'
-            ]
-        });
-        return new Set(actions.map((action)=>action.id));
-    }
-    /** Fetch permissions from DB and cache them (empty permissions are also cached) */ async fetchAndCachePermissions(userId, branchId, companyId) {
-        const enableCompanyFeature = this.iamConfigService.isCompanyFeatureEnabled();
-        const cacheOptions = {
-            userId,
-            companyId: enableCompanyFeature ? companyId : null,
-            branchId
-        };
-        const emptyData = {
-            frontendActions: [],
-            backendCodes: []
-        };
-        const allActionIds = await this.collectAllActionIds(userId, branchId, companyId);
-        if (allActionIds.size === 0) {
-            await this.permissionCacheService.setMyPermissions(cacheOptions, emptyData);
-            return emptyData;
-        }
-        await this.applyCompanyWhitelist(allActionIds, companyId);
-        if (allActionIds.size === 0) {
-            await this.permissionCacheService.setMyPermissions(cacheOptions, emptyData);
-            return emptyData;
-        }
-        return this.buildAndCachePermissionData(allActionIds, cacheOptions);
-    }
-    /** Collect all action IDs based on permission mode (RBAC, DIRECT, or FULL) */ async collectAllActionIds(userId, branchId, companyId) {
-        const permissionMode = this.iamConfigService.getPermissionMode();
-        const allActionIds = new Set();
-        if (permissionMode === _permissiontypeenum.IAMPermissionMode.RBAC || permissionMode === _permissiontypeenum.IAMPermissionMode.FULL) {
-            const userRoleIds = await this.getUserRoleIds(userId, branchId, companyId);
-            if (userRoleIds.length > 0) {
-                const roleActionIds = await this.getRoleActionIds(userRoleIds);
-                roleActionIds.forEach((id)=>allActionIds.add(id));
-            }
-        }
-        if (permissionMode === _permissiontypeenum.IAMPermissionMode.DIRECT || permissionMode === _permissiontypeenum.IAMPermissionMode.FULL) {
-            const userActionIds = await this.getUserActionIds(userId, branchId, companyId);
-            userActionIds.forEach((id)=>allActionIds.add(id));
-        }
-        return allActionIds;
-    }
-    /** Apply company whitelist filter to action IDs (mutates the set) */ async applyCompanyWhitelist(actionIds, companyId) {
-        if (!this.iamConfigService.isCompanyFeatureEnabled() || !companyId) {
-            return;
-        }
-        const companyActionIds = await this.getCompanyActionIds(companyId);
-        if (companyActionIds.length === 0) {
-            return;
-        }
-        const allowedActionIds = new Set(companyActionIds);
-        for (const actionId of actionIds){
-            if (!allowedActionIds.has(actionId)) {
-                actionIds.delete(actionId);
-            }
-        }
-    }
-    /** Fetch actions from DB, separate by type, build cache data, and cache it */ async buildAndCachePermissionData(actionIds, cacheOptions) {
-        const actionRepo = await this.getActionRepository();
-        const actions = await actionRepo.find({
-            where: {
-                id: (0, _typeorm.In)(Array.from(actionIds)),
-                isActive: true
-            }
-        });
-        const backendActions = actions.filter((a)=>a.actionType === _actiontypeenum.ActionType.BACKEND || a.actionType === _actiontypeenum.ActionType.BOTH);
-        const frontendActions = actions.filter((a)=>a.actionType === _actiontypeenum.ActionType.FRONTEND || a.actionType === _actiontypeenum.ActionType.BOTH);
-        const backendCodes = backendActions.map((a)=>a.code).filter((c)=>!!c);
-        const cacheData = {
-            frontendActions: frontendActions.map((a)=>({
-                    id: a.id,
-                    code: a.code ?? '',
-                    name: a.name,
-                    description: a.description,
-                    parentId: a.parentId
-                })),
-            backendCodes
-        };
-        await this.permissionCacheService.setMyPermissions(cacheOptions, cacheData);
-        return cacheData;
+        return this.permissionCacheService.getMyPermissionsResponse(userId, branchId, companyId, parentCodes);
     }
     // Helper Methods
     /** Split permission items into add and remove arrays */ splitItemsByAction(items) {
@@ -694,176 +549,13 @@ let PermissionService = class PermissionService {
             total: _totalItems
         };
     }
-    /** Get role IDs assigned to a user (merges company-wide + branch-specific roles) */ async getUserRoleIds(userId, branchId, companyId) {
-        const permissionRepo = await this.getPermissionRepository();
-        const enableCompanyFeature = this.iamConfigService.isCompanyFeatureEnabled();
-        if (!enableCompanyFeature) {
-            const permissions = await permissionRepo.find({
-                where: {
-                    permissionType: _iampermissiontypeenum.IamPermissionType.USER_ROLE,
-                    sourceType: _iamentitytypeenum.IamEntityType.USER,
-                    sourceId: userId
-                }
-            });
-            return permissions.filter((p)=>p.isValid()).map((p)=>p.targetId);
-        }
-        const roleIds = new Set();
-        const baseWhere = {
-            permissionType: _iampermissiontypeenum.IamPermissionType.USER_ROLE,
-            sourceType: _iamentitytypeenum.IamEntityType.USER,
-            sourceId: userId
-        };
-        if (companyId) {
-            baseWhere.companyId = companyId;
-        }
-        if (branchId) {
-            // Get company-wide + branch-specific roles
-            const companyWidePermissions = await permissionRepo.find({
-                where: {
-                    ...baseWhere,
-                    branchId: (0, _typeorm.IsNull)()
-                }
-            });
-            companyWidePermissions.filter((p)=>p.isValid()).forEach((p)=>roleIds.add(p.targetId));
-            const branchPermissions = await permissionRepo.find({
-                where: {
-                    ...baseWhere,
-                    branchId
-                }
-            });
-            branchPermissions.filter((p)=>p.isValid()).forEach((p)=>roleIds.add(p.targetId));
-        } else {
-            // No branch context: get all roles for this company
-            const allPermissions = await permissionRepo.find({
-                where: baseWhere
-            });
-            allPermissions.filter((p)=>p.isValid()).forEach((p)=>roleIds.add(p.targetId));
-        }
-        return Array.from(roleIds);
-    }
-    async getRoleActionIds(roleIds) {
-        const permissionRepo = await this.getPermissionRepository();
-        const permissions = await permissionRepo.find({
-            where: {
-                permissionType: _iampermissiontypeenum.IamPermissionType.ROLE_ACTION,
-                sourceType: _iamentitytypeenum.IamEntityType.ROLE,
-                sourceId: (0, _typeorm.In)(roleIds)
-            }
-        });
-        return permissions.filter((p)=>p.isValid()).map((p)=>p.targetId);
-    }
-    /** Get action IDs directly assigned to a user (merges company-wide + branch-specific actions) */ async getUserActionIds(userId, branchId, companyId) {
-        const permissionRepo = await this.getPermissionRepository();
-        const enableCompanyFeature = this.iamConfigService.isCompanyFeatureEnabled();
-        if (!enableCompanyFeature) {
-            const permissions = await permissionRepo.find({
-                where: {
-                    permissionType: _iampermissiontypeenum.IamPermissionType.USER_ACTION,
-                    sourceType: _iamentitytypeenum.IamEntityType.USER,
-                    sourceId: userId
-                }
-            });
-            return permissions.filter((p)=>p.isValid()).map((p)=>p.targetId);
-        }
-        const actionIds = new Set();
-        const baseWhere = {
-            permissionType: _iampermissiontypeenum.IamPermissionType.USER_ACTION,
-            sourceType: _iamentitytypeenum.IamEntityType.USER,
-            sourceId: userId
-        };
-        if (companyId) {
-            baseWhere.companyId = companyId;
-        }
-        if (branchId) {
-            // Get company-wide + branch-specific actions
-            const companyWidePermissions = await permissionRepo.find({
-                where: {
-                    ...baseWhere,
-                    branchId: (0, _typeorm.IsNull)()
-                }
-            });
-            companyWidePermissions.filter((p)=>p.isValid()).forEach((p)=>actionIds.add(p.targetId));
-            const branchPermissions = await permissionRepo.find({
-                where: {
-                    ...baseWhere,
-                    branchId
-                }
-            });
-            branchPermissions.filter((p)=>p.isValid()).forEach((p)=>actionIds.add(p.targetId));
-        } else {
-            // No branch context: get all actions for this company
-            const allPermissions = await permissionRepo.find({
-                where: baseWhere
-            });
-            allPermissions.filter((p)=>p.isValid()).forEach((p)=>actionIds.add(p.targetId));
-        }
-        return Array.from(actionIds);
-    }
-    async invalidateRoleMembersCache(roleId) {
-        const permissionRepo = await this.getPermissionRepository();
-        const roleRepo = await this.getRoleRepository();
-        const enableCompanyFeature = this.iamConfigService.isCompanyFeatureEnabled();
-        const userRoles = await permissionRepo.find({
-            where: {
-                permissionType: _iampermissiontypeenum.IamPermissionType.USER_ROLE,
-                sourceType: _iamentitytypeenum.IamEntityType.USER,
-                targetType: _iamentitytypeenum.IamEntityType.ROLE,
-                targetId: roleId
-            }
-        });
-        const userIds = [
-            ...new Set(userRoles.map((ur)=>ur.sourceId))
-        ];
-        if (userIds.length === 0) {
-            return 0;
-        }
-        const role = await roleRepo.findOne({
-            where: {
-                id: roleId
-            }
-        });
-        const companyId = role?.companyId || null;
-        let branchIds = [
-            null
-        ];
-        if (enableCompanyFeature && companyId) {
-            const userBranches = await permissionRepo.createQueryBuilder('p').select('DISTINCT p.branch_id', 'branchId').where('p.user_id IN (:...userIds)', {
-                userIds
-            }).andWhere('p.company_id = :companyId', {
-                companyId
-            }).getRawMany();
-            branchIds = [
-                ...new Set(userBranches.map((p)=>p.branchId))
-            ];
-        }
-        return this.permissionCacheService.invalidateUsers(userIds, companyId, branchIds);
-    }
-    /** Invalidate permission cache for all users in a company */ async invalidateCompanyMembersCache(companyId) {
-        if (!this.iamConfigService.isCompanyFeatureEnabled()) {
-            return 0;
-        }
-        const permissionRepo = await this.getPermissionRepository();
-        const userPermissions = await permissionRepo.createQueryBuilder('p').select('DISTINCT p.user_id', 'userId').addSelect('p.branch_id', 'branchId').where('p.company_id = :companyId', {
-            companyId
-        }).andWhere('p.user_id IS NOT NULL').getRawMany();
-        const userIds = [
-            ...new Set(userPermissions.map((p)=>p.userId).filter(Boolean))
-        ];
-        const branchIds = [
-            ...new Set(userPermissions.map((p)=>p.branchId))
-        ]; // includes null
-        if (userIds.length === 0) {
-            return 0;
-        }
-        return await this.permissionCacheService.invalidateUsers(userIds, companyId, branchIds);
-    }
     async revokeCompanyPermissions(companyId) {
         if (!this.iamConfigService.isCompanyFeatureEnabled()) {
             return;
         }
         const permissionRepo = await this.getPermissionRepository();
         const roleRepo = await this.getRoleRepository();
-        const dataSource = permissionRepo.manager.connection;
+        const dataSource = await this.dataSourceProvider.getDataSource();
         const companyRoles = await roleRepo.find({
             where: {
                 companyId,
@@ -880,14 +572,15 @@ let PermissionService = class PermissionService {
                     _iampermissiontypeenum.IamPermissionType.USER_ROLE,
                     _iampermissiontypeenum.IamPermissionType.USER_ACTION
                 ]),
+                sourceType: _iamentitytypeenum.IamEntityType.USER,
                 companyId
             },
             select: [
-                'userId'
+                'sourceId'
             ]
         });
         const affectedUserIds = [
-            ...new Set(affectedUserRows.map((row)=>row.userId).filter((id)=>!!id))
+            ...new Set(affectedUserRows.map((row)=>row.sourceId))
         ];
         await dataSource.transaction(async (manager)=>{
             const txRepo = manager.getRepository(permissionRepo.target);
@@ -929,17 +622,20 @@ let PermissionService = class PermissionService {
                 _iampermissiontypeenum.IamPermissionType.USER_ROLE,
                 _iampermissiontypeenum.IamPermissionType.USER_ACTION
             ]),
-            branchId,
-            companyId
+            companyId,
+            branchId
         };
         const affectedUserRows = await permissionRepo.find({
-            where,
+            where: {
+                ...where,
+                sourceType: _iamentitytypeenum.IamEntityType.USER
+            },
             select: [
-                'userId'
+                'sourceId'
             ]
         });
         const affectedUserIds = [
-            ...new Set(affectedUserRows.map((row)=>row.userId).filter((id)=>!!id))
+            ...new Set(affectedUserRows.map((row)=>row.sourceId))
         ];
         await permissionRepo.delete(where);
         if (affectedUserIds.length > 0) {