Add instance ID support for multi-instance functionality in Asgardeo components

kavindadimuthu · kavindadimuthu · commit 2af8495347ba · 2026-02-16T18:17:50.000+05:30
diff --git a/packages/react/src/AsgardeoReactClient.ts b/packages/react/src/AsgardeoReactClient.ts
@@ -189,8 +189,8 @@ class AsgardeoReactClient<T extends AsgardeoReactConfig = AsgardeoReactConfig> e
           baseUrl = configData?.baseUrl;
         }
 
-        const profile: User = await getScim2Me({baseUrl});
-        const schemas: any = await getSchemas({baseUrl});
+        const profile: User = await getScim2Me({baseUrl, instanceId: this.getInstanceId()});
+        const schemas: any = await getSchemas({baseUrl, instanceId: this.getInstanceId()});
 
         const processedSchemas: any = flattenUserSchema(schemas);
 
@@ -220,7 +220,7 @@ class AsgardeoReactClient<T extends AsgardeoReactConfig = AsgardeoReactConfig> e
         baseUrl = configData?.baseUrl;
       }
 
-      return await getMeOrganizations({baseUrl});
+      return await getMeOrganizations({baseUrl, instanceId: this.getInstanceId()});
     } catch (error) {
       throw new AsgardeoRuntimeError(
         `Failed to fetch the user's associated organizations: ${
@@ -242,7 +242,7 @@ class AsgardeoReactClient<T extends AsgardeoReactConfig = AsgardeoReactConfig> e
         baseUrl = configData?.baseUrl;
       }
 
-      return await getAllOrganizations({baseUrl});
+      return await getAllOrganizations({baseUrl, instanceId: this.getInstanceId()});
     } catch (error) {
       throw new AsgardeoRuntimeError(
         `Failed to fetch all organizations: ${error instanceof Error ? error.message : String(error)}`,
diff --git a/packages/react/src/components/presentation/CreateOrganization/CreateOrganization.tsx b/packages/react/src/components/presentation/CreateOrganization/CreateOrganization.tsx
@@ -76,7 +76,7 @@ export const CreateOrganization: FC<CreateOrganizationProps> = ({
   defaultParentId,
   ...props
 }: CreateOrganizationProps): ReactElement => {
-  const {isSignedIn, baseUrl} = useAsgardeo();
+  const {isSignedIn, baseUrl, instanceId} = useAsgardeo();
   const {currentOrganization, revalidateMyOrganizations} = useOrganization();
   const [loading, setLoading] = useState(false);
   const [error, setError] = useState<string | null>(null);
@@ -112,6 +112,7 @@ export const CreateOrganization: FC<CreateOrganizationProps> = ({
             ...payload,
             parentId,
           },
+          instanceId,
         });
       }
 
diff --git a/packages/react/src/components/presentation/OrganizationProfile/OrganizationProfile.tsx b/packages/react/src/components/presentation/OrganizationProfile/OrganizationProfile.tsx
@@ -146,7 +146,7 @@ const OrganizationProfile: FC<OrganizationProfileProps> = ({
   errorFallback,
   ...rest
 }: OrganizationProfileProps): ReactElement => {
-  const {baseUrl} = useAsgardeo();
+  const {baseUrl, instanceId} = useAsgardeo();
   const {t} = useTranslation();
   const [organization, setOrganization] = useState<OrganizationDetails | null>(null);
 
@@ -159,6 +159,7 @@ const OrganizationProfile: FC<OrganizationProfileProps> = ({
       const orgData: any = await getOrganization({
         baseUrl,
         organizationId,
+        instanceId,
       });
       setOrganization(orgData);
     } catch (err) {
@@ -182,6 +183,7 @@ const OrganizationProfile: FC<OrganizationProfileProps> = ({
         baseUrl,
         operations,
         organizationId,
+        instanceId,
       });
       // Refetch organization data after update
       await fetchOrganization();
diff --git a/packages/react/src/components/presentation/UserProfile/UserProfile.tsx b/packages/react/src/components/presentation/UserProfile/UserProfile.tsx
@@ -65,7 +65,7 @@ export type UserProfileProps = Omit<BaseUserProfileProps, 'user' | 'profile' | '
  * ```
  */
 const UserProfile: FC<UserProfileProps> = ({...rest}: UserProfileProps): ReactElement => {
-  const {baseUrl} = useAsgardeo();
+  const {baseUrl, instanceId} = useAsgardeo();
   const {profile, flattenedProfile, schemas, onUpdateProfile} = useUser();
   const {t} = useTranslation();
 
@@ -75,7 +75,7 @@ const UserProfile: FC<UserProfileProps> = ({...rest}: UserProfileProps): ReactEl
     setError(null);
 
     try {
-      const response: User = await updateMeProfile({baseUrl, payload});
+      const response: User = await updateMeProfile({baseUrl, payload, instanceId});
       onUpdateProfile(response);
     } catch (caughtError: unknown) {
       let message: string = t('user.profile.update.generic.error');
diff --git a/packages/react/src/contexts/Asgardeo/AsgardeoContext.ts b/packages/react/src/contexts/Asgardeo/AsgardeoContext.ts
@@ -146,6 +146,10 @@ export type AsgardeoContextProps = {
   signUpUrl: string | undefined;
 
   user: any;
+  /**
+   * Instance ID for multi-instance support.
+   */
+  instanceId: number;
 } & Pick<AsgardeoReactConfig, 'storage' | 'platform'> &
   Pick<AsgardeoReactClient, 'clearSession' | 'switchOrganization'>;
 
@@ -182,6 +186,7 @@ const AsgardeoContext: Context<AsgardeoContextProps | null> = createContext<null
   storage: 'sessionStorage',
   switchOrganization: null,
   user: null,
+  instanceId: 0,
 });
 
 AsgardeoContext.displayName = 'AsgardeoContext';
diff --git a/packages/react/src/contexts/Asgardeo/AsgardeoProvider.tsx b/packages/react/src/contexts/Asgardeo/AsgardeoProvider.tsx
@@ -587,6 +587,7 @@ const AsgardeoProvider: FC<PropsWithChildren<AsgardeoProviderProps>> = ({
       switchOrganization,
       syncSession,
       user,
+      instanceId,
     }),
     [
       applicationId,
@@ -615,6 +616,7 @@ const AsgardeoProvider: FC<PropsWithChildren<AsgardeoProviderProps>> = ({
       signUp,
       clearSession,
       reInitialize,
+      instanceId,
     ],
   );
 
