implement auth explicit initialization

Author: Feiyang1

common/api-review/app-exp.api.md

@@ -13,7 +13,7 @@ import { Name } from '@firebase/component';
 import { Provider } from '@firebase/component';
 
 // @internal (undocumented)
-export function _addComponent(app: FirebaseApp, component: Component): void;
+export function _addComponent<T extends Name>(app: FirebaseApp, component: Component<T>): void;
 
 // @internal (undocumented)
 export function _addOrOverwriteComponent(app: FirebaseApp, component: Component): void;
@@ -102,7 +102,7 @@ export function initializeApp(options: FirebaseOptions, config?: FirebaseAppConf
 export function onLog(logCallback: LogCallback | null, options?: LogOptions): void;
 
 // @internal (undocumented)
-export function _registerComponent(component: Component): boolean;
+export function _registerComponent<T extends Name>(component: Component<T>): boolean;
 
 // @public
 export function registerVersion(libraryKeyOrName: string, version: string, variant?: string): void;

packages-exp/app-exp/src/internal.ts

@@ -39,7 +39,10 @@ export const _components = new Map<string, Component<any>>();
  *
  * @internal
  */
-export function _addComponent(app: FirebaseApp, component: Component): void {
+export function _addComponent<T extends Name>(
+  app: FirebaseApp,
+  component: Component<T>
+): void {
   try {
     (app as FirebaseAppImpl).container.addComponent(component);
   } catch (e) {
@@ -68,7 +71,9 @@ export function _addOrOverwriteComponent(
  *
  * @internal
  */
-export function _registerComponent(component: Component): boolean {
+export function _registerComponent<T extends Name>(
+  component: Component<T>
+): boolean {
   const componentName = component.name;
   if (_components.has(componentName)) {
     logger.debug(

packages-exp/app-exp/src/platformLoggerService.ts

@@ -21,7 +21,7 @@ import {
   Provider,
   Name
 } from '@firebase/component';
-import { PlatformLoggerService } from './types';
+import { PlatformLoggerService, VersionService } from './types';
 
 export class PlatformLoggerServiceImpl implements PlatformLoggerService {
   constructor(private readonly container: ComponentContainer) {}
@@ -34,7 +34,7 @@ export class PlatformLoggerServiceImpl implements PlatformLoggerService {
     return providers
       .map(provider => {
         if (isVersionServiceProvider(provider)) {
-          const service = provider.getImmediate();
+          const service = provider.getImmediate() as VersionService;
           return `${service.library}/${service.version}`;
         } else {
           return null;
@@ -52,9 +52,7 @@ export class PlatformLoggerServiceImpl implements PlatformLoggerService {
  * provides VersionService. The provider is not necessarily a 'app-version'
  * provider.
  */
-function isVersionServiceProvider(
-  provider: Provider<Name>
-): provider is Provider<'app-version'> {
+function isVersionServiceProvider(provider: Provider<Name>): boolean {
   const component = provider.getComponent();
   return component?.type === ComponentType.VERSION;
 }

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

@@ -17,7 +17,11 @@
 
 import { _registerComponent, registerVersion } from '@firebase/app-exp';
 import { Config, Dependencies } from '../../model/public_types';
-import { Component, ComponentType } from '@firebase/component';
+import {
+  Component,
+  ComponentType,
+  InstantiationMode
+} from '@firebase/component';
 
 import { version } from '../../../package.json';
 import { AuthErrorCode } from '../errors';
@@ -76,6 +80,23 @@ export function registerAuth(clientPlatform: ClientPlatform): void {
       },
       ComponentType.PUBLIC
     )
+      /**
+       * Auth can only be initialized by explicitly calling getAuth() or initializeAuth()
+       * For why we do this, See https://docs.google.com/document/d/1vy9aI-fELpL2xlREoZt12o0JNw-JVNz4OwxYd9HcT0Y/edit?resourcekey=0-A714tRurcRonkj3lDuKVCw#heading=h.wcvc4xz0pwxf
+       */
+      .setInstantiationMode(InstantiationMode.EXPLICIT)
+      /**
+       * Because all firebase products that depend on auth depend on auth-internal directly,
+       * we need to initialize auth-internal after auth is initialized to make it available to other firebase products.
+       */
+      .setInstanceCreatedCallback(
+        (container, _instanceIdentifier, _instance) => {
+          const authInternalProvider = container.getProvider(
+            _ComponentName.AUTH_INTERNAL
+          );
+          authInternalProvider.initialize();
+        }
+      )
   );
 
   _registerComponent(
@@ -88,7 +109,7 @@ export function registerAuth(clientPlatform: ClientPlatform): void {
         return (auth => new AuthInterop(auth))(auth);
       },
       ComponentType.PRIVATE
-    )
+    ).setInstantiationMode(InstantiationMode.EXPLICIT)
   );
 
   registerVersion(

packages/component/src/component.ts

@@ -19,7 +19,8 @@ import {
   InstanceFactory,
   ComponentType,
   Dictionary,
-  Name
+  Name,
+  onInstanceCreatedCallback
 } from './types';
 
 /**
@@ -34,6 +35,8 @@ export class Component<T extends Name = Name> {
 
   instantiationMode = InstantiationMode.LAZY;
 
+  onInstanceCreated: onInstanceCreatedCallback<T> | null = null;
+
   /**
    *
    * @param name The public service name, e.g. app, auth, firestore, database
@@ -60,4 +63,9 @@ export class Component<T extends Name = Name> {
     this.serviceProps = props;
     return this;
   }
+
+  setInstanceCreatedCallback(callback: onInstanceCreatedCallback<T>): this {
+    this.onInstanceCreated = callback;
+    return this;
+  }
 }

packages/component/src/component_container.ts

@@ -66,12 +66,12 @@ export class ComponentContainer {
    */
   getProvider<T extends Name>(name: T): Provider<T> {
     if (this.providers.has(name)) {
-      return this.providers.get(name) as Provider<T>;
+      return (this.providers.get(name) as unknown) as Provider<T>;
     }
 
     // create a Provider for a service that hasn't registered with Firebase
     const provider = new Provider<T>(name, this);
-    this.providers.set(name, provider);
+    this.providers.set(name, (provider as unknown) as Provider<Name>);
 
     return provider as Provider<T>;
   }

packages/component/src/provider.ts

@@ -267,6 +267,23 @@ export class Provider<T extends Name> {
         options
       });
       this.instances.set(instanceIdentifier, instance);
+
+      /**
+       * Order is important
+       * onInstanceCreated() should be called after this.instances.set(instanceIdentifier, instance); which
+       * makes `isInitialized()` return true.
+       */
+      if (this.component.onInstanceCreated) {
+        try {
+          this.component.onInstanceCreated(
+            this.container,
+            instanceIdentifier,
+            instance
+          );
+        } catch {
+          // ignore errors in the onInstanceCreatedCallback
+        }
+      }
     }
 
     return instance || null;
@@ -293,6 +310,6 @@ function normalizeIdentifierForFactory(identifier: string): string | undefined {
   return identifier === DEFAULT_ENTRY_NAME ? undefined : identifier;
 }
 
-function isComponentEager(component: Component<Name>): boolean {
+function isComponentEager<T extends Name>(component: Component<T>): boolean {
   return component.instantiationMode === InstantiationMode.EAGER;
 }

packages/component/src/types.ts

@@ -57,6 +57,12 @@ export type InstanceFactory<T extends Name> = (
   options: InstanceFactoryOptions
 ) => NameServiceMapping[T];
 
+export type onInstanceCreatedCallback<T extends Name> = (
+  container: ComponentContainer,
+  instanceIdentifier: string,
+  instance: NameServiceMapping[T]
+) => void;
+
 export interface Dictionary {
   [key: string]: unknown;
 }