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

Author: kavindadimuthu

packages/react/src/AsgardeoReactClient.ts

@@ -189,8 +189,8 @@ class AsgardeoReactClient<T extends AsgardeoReactConfig = AsgardeoReactConfig> e
           baseUrl = configData?.baseUrl;
         }
 
-        const profile = await getScim2Me({baseUrl});
-        const schemas = await getSchemas({baseUrl});
+        const profile = await getScim2Me({baseUrl, instanceId: this._instanceId});
+        const schemas = await getSchemas({baseUrl, instanceId: this._instanceId});
 
         const processedSchemas = flattenUserSchema(schemas);
 
@@ -220,7 +220,7 @@ class AsgardeoReactClient<T extends AsgardeoReactConfig = AsgardeoReactConfig> e
         baseUrl = configData?.baseUrl;
       }
 
-      return getMeOrganizations({baseUrl});
+      return getMeOrganizations({baseUrl, instanceId: this._instanceId});
     } 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 getAllOrganizations({baseUrl});
+      return getAllOrganizations({baseUrl, instanceId: this._instanceId});
     } catch (error) {
       throw new AsgardeoRuntimeError(
         `Failed to fetch all organizations: ${error instanceof Error ? error.message : String(error)}`,

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,
         });
       }
 

packages/react/src/components/presentation/OrganizationProfile/OrganizationProfile.tsx

@@ -141,7 +141,7 @@ const OrganizationProfile: FC<OrganizationProfileProps> = ({
   errorFallback = <div>Failed to load organization data</div>,
   ...rest
 }: OrganizationProfileProps): ReactElement => {
-  const {baseUrl} = useAsgardeo();
+  const {baseUrl, instanceId} = useAsgardeo();
   const {t} = useTranslation();
   const [organization, setOrganization] = useState<OrganizationDetails | null>(null);
   const [loading, setLoading] = useState<boolean>(true);
@@ -160,6 +160,7 @@ const OrganizationProfile: FC<OrganizationProfileProps> = ({
       const orgData = await getOrganization({
         baseUrl,
         organizationId,
+        instanceId,
       });
       setOrganization(orgData);
     } catch (err) {
@@ -186,6 +187,7 @@ const OrganizationProfile: FC<OrganizationProfileProps> = ({
         baseUrl,
         organizationId,
         operations,
+        instanceId,
       });
       // Refetch organization data after update
       await fetchOrganization();

packages/react/src/components/presentation/UserProfile/UserProfile.tsx

@@ -64,7 +64,7 @@ export type UserProfileProps = Omit<BaseUserProfileProps, 'user' | 'profile' | '
  * ```
  */
 const UserProfile: FC<UserProfileProps> = ({...rest}: UserProfileProps): ReactElement => {
-  const {baseUrl, isLoading} = useAsgardeo();
+  const {baseUrl, isLoading, instanceId} = useAsgardeo();
   const {profile, flattenedProfile, schemas, onUpdateProfile} = useUser();
   const {t} = useTranslation();
 
@@ -74,7 +74,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 (error: unknown) {
       let message: string = t('user.profile.update.generic.error');

packages/react/src/contexts/Asgardeo/AsgardeoContext.ts

@@ -146,6 +146,10 @@ export type AsgardeoContextProps = {
    * @returns Promise resolving to boolean indicating success.
    */
   reInitialize: (config: Partial<AsgardeoReactConfig>) => Promise<boolean>;
+  /**
+   * 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
   switchOrganization: null,
   reInitialize: null,
   platform: undefined,
+  instanceId: 0,
 });
 
 AsgardeoContext.displayName = 'AsgardeoContext';

packages/react/src/contexts/Asgardeo/AsgardeoProvider.tsx

@@ -586,6 +586,7 @@ const AsgardeoProvider: FC<PropsWithChildren<AsgardeoProviderProps>> = ({
       syncSession,
       platform: config?.platform,
       switchOrganization,
+      instanceId,
     }),
     [
       applicationId,
@@ -614,6 +615,7 @@ const AsgardeoProvider: FC<PropsWithChildren<AsgardeoProviderProps>> = ({
       signUp,
       clearSession,
       reInitialize,
+      instanceId,
     ],
   );