Add LiteAppCheckTokenProvider.

Author: ehsannas

packages/firestore/lite/register.ts

@@ -24,7 +24,7 @@ import { Component, ComponentType } from '@firebase/component';
 
 import { version } from '../package.json';
 import {
-  FirebaseAppCheckTokenProvider,
+  LiteAppCheckTokenProvider,
   LiteCredentialsProvider
 } from '../src/api/credentials';
 import { setSDKVersion } from '../src/core/version';
@@ -47,7 +47,7 @@ export function registerFirestore(): void {
         const firestoreInstance = new Firestore(
           app,
           new LiteCredentialsProvider(container.getProvider('auth-internal')),
-          new FirebaseAppCheckTokenProvider(
+          new LiteAppCheckTokenProvider(
             container.getProvider('app-check-internal')
           )
         );

packages/firestore/src/api/credentials.ts

@@ -579,6 +579,46 @@ export class EmptyAppCheckTokenProvider implements CredentialsProvider<string> {
   shutdown(): void {}
 }
 
+/** AppCheck token provider for the Lite SDK. */
+export class LiteAppCheckTokenProvider implements CredentialsProvider<String> {
+  private appCheck: FirebaseAppCheckInternal | null = null;
+
+  constructor(
+    private appCheckProvider: Provider<AppCheckInternalComponentName>
+  ) {
+    appCheckProvider.onInit(appCheck => {
+      this.appCheck = appCheck;
+    });
+  }
+
+  getToken(): Promise<Token | null> {
+    if (!this.appCheck) {
+      return Promise.resolve(null);
+    }
+
+    return this.appCheck.getToken().then(tokenResult => {
+      if (tokenResult) {
+        hardAssert(
+          typeof tokenResult.token === 'string',
+          'Invalid tokenResult returned from getToken():' + tokenResult
+        );
+        return new AppCheckToken(tokenResult.token);
+      } else {
+        return null;
+      }
+    });
+  }
+
+  invalidateToken(): void {}
+
+  start(
+    asyncQueue: AsyncQueue,
+    changeListener: CredentialChangeListener<string>
+  ): void {}
+
+  shutdown(): void {}
+}
+
 /**
  * Builds a CredentialsProvider depending on the type of
  * the credentials passed in.