Changes to initialize regional getAuth instance

Author: mansisampat

packages/auth/src/core/auth/auth_impl.ts

@@ -30,6 +30,7 @@ import {
   NextOrObserver,
   Persistence,
   PopupRedirectResolver,
+  RegionData,
   User,
   UserCredential,
   CompleteFn,
@@ -91,7 +92,8 @@ interface AsyncAction {
 export const enum DefaultConfig {
   TOKEN_API_HOST = 'securetoken.googleapis.com',
   API_HOST = 'identitytoolkit.googleapis.com',
-  API_SCHEME = 'https'
+  API_SCHEME = 'https',
+  REGIONAL_API_HOST = 'identityplatform.googleapis.com'
 }
 
 export class AuthImpl implements AuthInternal, _FirebaseService {
@@ -125,6 +127,7 @@ export class AuthImpl implements AuthInternal, _FirebaseService {
     | undefined = undefined;
   _persistenceManagerAvailable: Promise<void>;
   readonly name: string;
+  readonly regionData: RegionData;
 
   // Tracks the last notified UID for state change listeners to prevent
   // repeated calls to the callbacks. Undefined means it's never been
@@ -139,7 +142,8 @@ export class AuthImpl implements AuthInternal, _FirebaseService {
     public readonly app: FirebaseApp,
     private readonly heartbeatServiceProvider: Provider<'heartbeat'>,
     private readonly appCheckServiceProvider: Provider<AppCheckInternalComponentName>,
-    public readonly config: ConfigInternal
+    public readonly config: ConfigInternal,
+    regionData?: RegionData
   ) {
     this.name = app.name;
     this.clientVersion = config.sdkClientVersion;
@@ -148,6 +152,7 @@ export class AuthImpl implements AuthInternal, _FirebaseService {
     this._persistenceManagerAvailable = new Promise<void>(
       resolve => (this._resolvePersistenceManagerAvailable = resolve)
     );
+    this.regionData = regionData ?? {};
   }
 
   _initializeWithPersistence(

packages/auth/src/core/auth/register.ts

@@ -58,6 +58,7 @@ function getVersionForPlatform(
 
 /** @internal */
 export function registerAuth(clientPlatform: ClientPlatform): void {
+  console.log("inside register auth");
   _registerComponent(
     new Component(
       _ComponentName.AUTH,
@@ -68,7 +69,8 @@ export function registerAuth(clientPlatform: ClientPlatform): void {
         const appCheckServiceProvider =
           container.getProvider<'app-check-internal'>('app-check-internal');
         const { apiKey, authDomain } = app.options;
-
+        const regionData = deps?.regionData ?? { location: '', tenantId: '' };
+        console.log("inside register auth new component");
         _assert(
           apiKey && !apiKey.includes(':'),
           AuthErrorCode.INVALID_API_KEY,
@@ -79,7 +81,7 @@ export function registerAuth(clientPlatform: ClientPlatform): void {
           apiKey,
           authDomain,
           clientPlatform,
-          apiHost: DefaultConfig.API_HOST,
+          apiHost: regionData.location ? DefaultConfig.REGIONAL_API_HOST : DefaultConfig.API_HOST,
           tokenApiHost: DefaultConfig.TOKEN_API_HOST,
           apiScheme: DefaultConfig.API_SCHEME,
           sdkClientVersion: _getClientVersion(clientPlatform)
@@ -89,7 +91,8 @@ export function registerAuth(clientPlatform: ClientPlatform): void {
           app,
           heartbeatServiceProvider,
           appCheckServiceProvider,
-          config
+          config,
+          regionData
         );
         _initializeAuthInstance(authInstance, deps);
 

packages/auth/src/model/public_types.ts

@@ -1260,6 +1260,12 @@ export interface Dependencies {
    * Which {@link AuthErrorMap} to use.
    */
   errorMap?: AuthErrorMap;
+  regionData?: RegionData
+}
+
+export interface RegionData {
+  location?: string;
+  tenantId?: string;
 }
 
 /**

packages/auth/src/platform_browser/index.ts

@@ -29,11 +29,12 @@ import { browserLocalPersistence } from './persistence/local_storage';
 import { browserSessionPersistence } from './persistence/session_storage';
 import { indexedDBLocalPersistence } from './persistence/indexed_db';
 import { browserPopupRedirectResolver } from './popup_redirect';
-import { Auth, User } from '../model/public_types';
-import { getDefaultEmulatorHost, getExperimentalSetting } from '@firebase/util';
+import { Auth, Dependencies, RegionData, User } from '../model/public_types';
+import { deepEqual, getDefaultEmulatorHost, getExperimentalSetting } from '@firebase/util';
 import { _setExternalJSProvider } from './load_js';
-import { _createError } from '../core/util/assert';
+import { _createError, _fail } from '../core/util/assert';
 import { AuthErrorCode } from '../core/errors';
+import { AuthImpl } from '../core/auth/auth_impl';
 
 const DEFAULT_ID_TOKEN_MAX_AGE = 5 * 60;
 const authIdTokenMaxAge =
@@ -73,10 +74,22 @@ const mintCookieFactory = (url: string) => async (user: User | null) => {
  *
  * @public
  */
-export function getAuth(app: FirebaseApp = getApp()): Auth {
+export function getAuth(app: FirebaseApp = getApp(), regionData?: RegionData): Auth {
   const provider = _getProvider(app, 'auth');
 
   if (provider.isInitialized()) {
+    const auth = provider.getImmediate() as AuthImpl;
+    // Extract previously used regionData from initialization
+    const initialOptions = provider.getOptions() as Dependencies;
+
+    const previousRegionData = initialOptions?.regionData ?? {};
+    const currentRegionData = regionData ?? {};
+    if (deepEqual(previousRegionData, currentRegionData)) {
+      return auth;
+    } else {
+      _fail(auth, AuthErrorCode.ALREADY_INITIALIZED);
+    }
+
     return provider.getImmediate();
   }
 
@@ -86,7 +99,8 @@ export function getAuth(app: FirebaseApp = getApp()): Auth {
       indexedDBLocalPersistence,
       browserLocalPersistence,
       browserSessionPersistence
-    ]
+    ],
+    regionData
   });
 
   const authTokenSyncPath = getExperimentalSetting('authTokenSyncURL');
@@ -111,6 +125,8 @@ export function getAuth(app: FirebaseApp = getApp()): Auth {
   if (authEmulatorHost) {
     connectAuthEmulator(auth, `http://${authEmulatorHost}`);
   }
+  console.log("here");
+  console.log(auth.config.apiHost);
 
   return auth;
 }