Adding onFirebaseTokenChanged method for persistence

Author: mansisampat

common/api-review/auth.api.md

@@ -92,6 +92,8 @@ export interface Auth {
     languageCode: string | null;
     readonly name: string;
     onAuthStateChanged(nextOrObserver: NextOrObserver<User | null>, error?: ErrorFn, completed?: CompleteFn): Unsubscribe;
+    // (undocumented)
+    onFirebaseTokenChanged(nextOrObserver: NextOrObserver<FirebaseToken | null>, error?: ErrorFn, completed?: CompleteFn): Unsubscribe;
     onIdTokenChanged(nextOrObserver: NextOrObserver<User | null>, error?: ErrorFn, completed?: CompleteFn): Unsubscribe;
     setPersistence(persistence: Persistence): Promise<void>;
     readonly settings: AuthSettings;

packages/auth/demo/public/index.html

@@ -852,14 +852,14 @@
               </form>
             </div>
             <div class="tab-pane" id="tab-byo-ciam-content">
+              <div id="firebase-token-status">No CIAM token found. User not logged in.</div>
               <h2>Sign in with your CIAM token</h2>
               <input type="text" id="byo-ciam-token"
                      class="form-control" placeholder="Enter CIAM token" />
               <button class="btn btn-block btn-primary"
                   id="exchange-token">
                 Exchange Token
               </button>
-              <pre id="byo-ciam-result"></pre>
             </div>
             <div class="tab-pane" id="logs-section">
               <pre class="well logs"></pre>

packages/auth/demo/src/index.js

@@ -1520,16 +1520,15 @@ async function exchangeCIAMToken(token) {
 function onExchangeToken(event) {
   event.preventDefault();
   const byoCiamInput = document.getElementById('byo-ciam-token');
-  const byoCiamResult = document.getElementById('byo-ciam-result');
-
-  byoCiamResult.textContent = 'Exchanging token...';
+  const firebaseTokenStatus = document.getElementById('firebase-token-status');
 
   exchangeCIAMToken(byoCiamInput.value)
     .then(response => {
-      byoCiamResult.textContent = response;
+      firebaseTokenStatus.textContent = "✅ Firebase token is set: " + response;
       console.log('Token:', response);
     })
     .catch(error => {
+      firebaseTokenStatus.textContent = "Error exchanging token: ", error;
       console.error('Error exchanging token:', error);
     });
 }
@@ -2091,6 +2090,18 @@ function initApp() {
     tenantConfig: tenantConfig
   });
 
+  if (regionalAuth.onFirebaseTokenChanged) {
+    regionalAuth.onFirebaseTokenChanged((token) => {
+      if (token) {
+        firebaseTokenStatus.textContent = "✅ Firebase token is set: " + token.token;
+        console.log("Token here:", token.token);
+      } else {
+        firebaseTokenStatus.textContent = "No CIAM token found. User not logged in.";
+        console.log("Token not available");
+      }
+    });
+  }
+
   tempApp = initializeApp(
     {
       apiKey: config.apiKey,

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

@@ -107,6 +107,7 @@ export class AuthImpl implements AuthInternal, _FirebaseService {
   private redirectPersistenceManager?: PersistenceUserManager;
   private authStateSubscription = new Subscription<User>(this);
   private idTokenSubscription = new Subscription<User>(this);
+  private firebaseTokenSubscription = new Subscription<FirebaseToken>(this);
   private readonly beforeStateQueue = new AuthMiddlewareQueue(this);
   private redirectUser: UserInternal | null = null;
   private isProactiveRefreshEnabled = false;
@@ -407,6 +408,7 @@ export class AuthImpl implements AuthInternal, _FirebaseService {
   private async initializeFirebaseToken(): Promise<void> {
     this.firebaseToken =
       (await this.persistenceManager?.getFirebaseToken()) ?? null;
+      this.firebaseTokenSubscription.next(this.firebaseToken);
   }
 
   useDeviceLanguage(): void {
@@ -467,6 +469,7 @@ export class AuthImpl implements AuthInternal, _FirebaseService {
     firebaseToken: FirebaseToken | null
   ): Promise<void> {
     this.firebaseToken = firebaseToken;
+    this.firebaseTokenSubscription.next(firebaseToken);
     if (firebaseToken) {
       await this.assertedPersistence.setFirebaseToken(firebaseToken);
     } else {
@@ -588,6 +591,28 @@ export class AuthImpl implements AuthInternal, _FirebaseService {
     );
   }
 
+  onFirebaseTokenChanged(
+    nextOrObserver: NextOrObserver<FirebaseToken | null>,
+    error?: ErrorFn,
+    completed?: CompleteFn
+  ): Unsubscribe {
+    if (typeof nextOrObserver === 'function') {
+      const unsubscribe = this.firebaseTokenSubscription.addObserver(
+        nextOrObserver,
+        error,
+        completed
+      );
+      return () => {
+        unsubscribe();
+      };
+    } else {
+      const unsubscribe = this.firebaseTokenSubscription.addObserver(nextOrObserver);
+      return () => {
+        unsubscribe();
+      };
+    }
+  }
+
   beforeAuthStateChanged(
     callback: (user: User | null) => void | Promise<void>,
     onAbort?: () => void

packages/auth/src/model/public_types.ts

@@ -280,6 +280,12 @@ export interface Auth {
     callback: (user: User | null) => void | Promise<void>,
     onAbort?: () => void
   ): Unsubscribe;
+
+  onFirebaseTokenChanged(
+    nextOrObserver: NextOrObserver<FirebaseToken | null>,
+    error?: ErrorFn,
+    completed?: CompleteFn
+  ): Unsubscribe;
   /**
    * Adds an observer for changes to the signed-in user's ID token.
    *