send cohortids for learner in auth API

tekdi · Feb 17, 2025 · 30ca3ca · 30ca3ca
1 parent 4dc96df
commit 30ca3ca
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/src/adapters/postgres/user-adapter.ts b/src/adapters/postgres/user-adapter.ts
@@ -677,15 +677,24 @@ export class PostgresUserService implements IServicelocator {
 
         const privilegeData = await this.postgresRoleService.findPrivilegeByRoleId(roleArray);
         const privileges = privilegeData.map(priv => priv.name);
-
-        combinedResult.push({
+        let cohortIds = [];
+        if(roleName === 'learner') {
+          const mappedCohorts = await this.cohortMemberRepository.find({
+            where: { userId },
+            select: ['cohortId']
+          })
+          cohortIds = mappedCohorts.map(id => id.cohortId)
+        }
+        const result = {
           tenantName: data.tenantname,
           tenantId: data.tenantId,
           userTenantMappingId: data.usertenantmappingid,
           roleId: roleId,
           roleName: roleName,
-          privileges: privileges
-        });
+          privileges: privileges,
+          ...(roleName === 'learner' && { mappedCohorts: cohortIds })
+        }
+        combinedResult.push(result);
       }
     }