@firebase/data-connect 0.6.0-20260408221811 → 0.6.0

package/dist/index.esm.js

@@ -4,7 +4,7 @@ import { FirebaseError, generateSHA256Hash, isCloudWorkstation, pingServer } fro
 import { Logger } from '@firebase/logger';
 
 const name = "@firebase/data-connect";
-const version = "0.6.0-20260408221811";
+const version = "0.6.0";
 
 /**
  * @license
@@ -1701,6 +1701,8 @@ class AbstractDataConnectStreamTransport extends AbstractDataConnectTransport {
         this.closeTimeout = null;
         /** has the close timeout finished? */
         this.closeTimeoutFinished = false;
+        /** Flag to ensure we wait for the initial auth state once per connection attempt. */
+        this.hasWaitedForInitialAuth = false;
         /**
          * Tracks if the next message to be sent is the first message of the stream.
          */
@@ -1841,6 +1843,7 @@ class AbstractDataConnectStreamTransport extends AbstractDataConnectTransport {
     onConnectionReady() {
         this.isFirstStreamMessage = true;
         this.lastSentAuthToken = null;
+        this.hasWaitedForInitialAuth = false;
     }
     /**
      * Attempt to close the connection. Will only close if there are no active requests preventing it
@@ -1921,12 +1924,12 @@ class AbstractDataConnectStreamTransport extends AbstractDataConnectTransport {
         }
         headers['X-Goog-Api-Client'] = getGoogApiClientValue$1(this._isUsingGen, this._callerSdkType);
         if (this.shouldIncludeAuth && this._authToken) {
-            headers.authToken = this._authToken;
+            headers['X-Firebase-Auth-Token'] = this._authToken;
             this.lastSentAuthToken = this._authToken;
         }
         if (this.isFirstStreamMessage) {
             if (this._appCheckToken) {
-                headers.appCheckToken = this._appCheckToken;
+                headers['X-Firebase-App-Check'] = this._appCheckToken;
             }
             preparedRequestBody.name = this._connectorResourcePath;
         }
@@ -1934,12 +1937,17 @@ class AbstractDataConnectStreamTransport extends AbstractDataConnectTransport {
         this.isFirstStreamMessage = false;
         return preparedRequestBody;
     }
+    // TODO(stephenarosaj): just make this async
     /**
      * Sends a request message to the server via the concrete implementation.
      * Ensures the connection is ready and prepares the message before sending.
      * @returns A promise that resolves when the request message has been sent.
      */
-    sendRequestMessage(requestBody) {
+    async sendRequestMessage(requestBody) {
+        if (!this.hasWaitedForInitialAuth && this.authProvider) {
+            await this.getWithAuth();
+            this.hasWaitedForInitialAuth = true;
+        }
         if (this.streamIsReady) {
             const prepared = this.prepareMessage(requestBody);
             return this.sendMessage(prepared);