Make installation provider optional

Author: tonybaroneee

packages/telemetry/src/logging/installation-id-provider.test.ts

@@ -29,7 +29,12 @@ describe('InstallationIdProvider', () => {
       }
     } as unknown as _FirebaseInstallationsInternal;
 
-    const provider = new InstallationIdProvider(mockInstallations);
+    const mockProvider = {
+      getImmediate: () => mockInstallations,
+      get: async () => mockInstallations
+    } as any;
+
+    const provider = new InstallationIdProvider(mockProvider);
 
     const attr1 = await provider.getAttribute();
     expect(attr1).to.deep.equal(['user.id', 'iid-123']);
@@ -50,7 +55,12 @@ describe('InstallationIdProvider', () => {
       }
     } as unknown as _FirebaseInstallationsInternal;
 
-    const provider = new InstallationIdProvider(mockInstallations);
+    const mockProvider = {
+      getImmediate: () => mockInstallations,
+      get: async () => mockInstallations
+    } as any;
+
+    const provider = new InstallationIdProvider(mockProvider);
 
     const attr1 = await provider.getAttribute();
     expect(attr1).to.be.null;

packages/telemetry/src/logging/installation-id-provider.ts

@@ -15,6 +15,7 @@
  * limitations under the License.
  */
 
+import { Provider } from '@firebase/component';
 import { DynamicLogAttributeProvider, LogEntryAttribute } from '../types';
 import { _FirebaseInstallationsInternal } from '@firebase/installations';
 
@@ -24,16 +25,30 @@ import { _FirebaseInstallationsInternal } from '@firebase/installations';
  * @internal
  */
 export class InstallationIdProvider implements DynamicLogAttributeProvider {
+  private installations: _FirebaseInstallationsInternal | null;
   private _iid: string | undefined;
 
-  constructor(private installationsProvider: _FirebaseInstallationsInternal) {}
+  constructor(installationsProvider: Provider<'installations-internal'>) {
+    this.installations = installationsProvider?.getImmediate({
+      optional: true
+    });
+    if (!this.installations) {
+      void installationsProvider
+        ?.get()
+        .then(installations => (this.installations = installations))
+        .catch();
+    }
+  }
 
   async getAttribute(): Promise<LogEntryAttribute | null> {
+    if (!this.installations) {
+      return null;
+    }
     if (this._iid) {
       return ['user.id', this._iid];
     }
 
-    const iid = await this.installationsProvider.getId();
+    const iid = await this.installations.getId();
     if (!iid) {
       return null;
     }

packages/telemetry/src/register.ts

@@ -43,9 +43,9 @@ export function registerTelemetry(): void {
         // getImmediate for FirebaseApp will always succeed
         const app = container.getProvider('app').getImmediate();
         const appCheckProvider = container.getProvider('app-check-internal');
-        const installationsProvider = container
-          .getProvider('installations-internal')
-          .getImmediate();
+        const installationsProvider = container.getProvider(
+          'installations-internal'
+        );
         const dynamicHeaderProviders = [new AppCheckProvider(appCheckProvider)];
         const dynamicLogAttributeProviders = [
           new InstallationIdProvider(installationsProvider)