Add request initiating username to privileged carbon context when sta…

…rting tenant flow
wso2-extensions · Nov 19, 2023 · 1ef5e85 · 1ef5e85
1 parent 647fdaa
commit 1ef5e85
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/...n/java/org/wso2/carbon/identity/organization/management/handler/SharedRoleMgtHandler.java b/...n/java/org/wso2/carbon/identity/organization/management/handler/SharedRoleMgtHandler.java
@@ -202,10 +202,14 @@ private void createSharedRolesOnNewRoleCreation(Map<String, Object> eventPropert
                                 String sharedOrganizationId = sharedApplications.get(taskId).getOrganizationId();
                                 String shareAppTenantDomain =
                                         getOrganizationManager().resolveTenantDomain(sharedOrganizationId);
+                                String associatedUserName =
+                                        PrivilegedCarbonContext.getThreadLocalCarbonContext().getUsername();
                                 try {
                                     PrivilegedCarbonContext.startTenantFlow();
                                     PrivilegedCarbonContext.getThreadLocalCarbonContext()
                                             .setTenantDomain(shareAppTenantDomain, true);
+                                    PrivilegedCarbonContext.getThreadLocalCarbonContext()
+                                            .setUsername(associatedUserName);
                                     RoleBasicInfo sharedRoleInfo =
                                         getRoleManagementServiceV2().addRole(mainRoleName, Collections.emptyList(),
                                                 Collections.emptyList(),

diff --git a/...gement/organization/user/sharing/listener/SharingOrganizationCreatorUserEventHandler.java b/...gement/organization/user/sharing/listener/SharingOrganizationCreatorUserEventHandler.java
@@ -74,9 +74,11 @@ public void handleEvent(Event event) throws IdentityEventException {
                         return;
                     }
                     String associatedUserId = PrivilegedCarbonContext.getThreadLocalCarbonContext().getUserId();
+                    String associatedUserName = PrivilegedCarbonContext.getThreadLocalCarbonContext().getUsername();
                     try {
                         PrivilegedCarbonContext.startTenantFlow();
                         PrivilegedCarbonContext.getThreadLocalCarbonContext().setTenantDomain(tenantDomain, true);
+                        PrivilegedCarbonContext.getThreadLocalCarbonContext().setUsername(associatedUserName);
                         Role organizationCreatorRole = buildOrgCreatorRole(associatedUserId);
                         Role administratorRole = buildAdministratorRole(associatedUserId);
                         getRoleManager().createRole(orgId, organizationCreatorRole);
@@ -94,6 +96,7 @@ public void handleEvent(Event event) throws IdentityEventException {
                     if (!OrganizationManagementUtil.isOrganization(tenantDomain)) {
                         return;
                     }
+                    String associatedUserName = PrivilegedCarbonContext.getThreadLocalCarbonContext().getUsername();
                     String associatedUserId = PrivilegedCarbonContext.getThreadLocalCarbonContext().getUserId();
                     String associatedOrgId = PrivilegedCarbonContext.getThreadLocalCarbonContext()
                             .getUserResidentOrganizationId();
@@ -103,6 +106,7 @@ public void handleEvent(Event event) throws IdentityEventException {
                     try {
                         PrivilegedCarbonContext.startTenantFlow();
                         PrivilegedCarbonContext.getThreadLocalCarbonContext().setTenantDomain(tenantDomain, true);
+                        PrivilegedCarbonContext.getThreadLocalCarbonContext().setUsername(associatedUserName);
                         userSharingService.shareOrganizationUser(orgId, associatedUserId, associatedOrgId);
                         String userId = userSharingService
                                 .getUserAssociationOfAssociatedUserByOrgId(associatedUserId, orgId)