@firebase/data-connect 0.2.0 → 0.3.0-20250205191324

package/dist/index.node.cjs.js

@@ -107,6 +107,31 @@ function logError(msg) {
     logger.error(`DataConnect (${SDK_VERSION}): ${msg}`);
 }
 
+/**
+ * @license
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+const CallerSdkTypeEnum = {
+    Base: 'Base', // Core JS SDK
+    Generated: 'Generated', // Generated JS SDK
+    TanstackReactCore: 'TanstackReactCore', // Tanstack non-generated React SDK
+    GeneratedReact: 'GeneratedReact', // Tanstack non-generated Angular SDK
+    TanstackAngularCore: 'TanstackAngularCore', // Tanstack non-generated Angular SDK
+    GeneratedAngular: 'GeneratedAngular' // Generated Angular SDK
+};
+
 /**
  * @license
  * Copyright 2024 Google LLC
@@ -127,20 +152,24 @@ let connectFetch = globalThis.fetch;
 function initializeFetch(fetchImpl) {
     connectFetch = fetchImpl;
 }
-function getGoogApiClientValue(_isUsingGen) {
+function getGoogApiClientValue(_isUsingGen, _callerSdkType) {
     let str = 'gl-js/ fire/' + SDK_VERSION;
-    if (_isUsingGen) {
+    if (_callerSdkType !== CallerSdkTypeEnum.Base &&
+        _callerSdkType !== CallerSdkTypeEnum.Generated) {
+        str += ' js/' + _callerSdkType.toLowerCase();
+    }
+    else if (_isUsingGen || _callerSdkType === CallerSdkTypeEnum.Generated) {
         str += ' js/gen';
     }
     return str;
 }
-function dcFetch(url, body, { signal }, appId, accessToken, appCheckToken, _isUsingGen) {
+function dcFetch(url, body, { signal }, appId, accessToken, appCheckToken, _isUsingGen, _callerSdkType) {
     if (!connectFetch) {
         throw new DataConnectError(Code.OTHER, 'No Fetch Implementation detected!');
     }
     const headers = {
         'Content-Type': 'application/json',
-        'X-Goog-Api-Client': getGoogApiClientValue(_isUsingGen)
+        'X-Goog-Api-Client': getGoogApiClientValue(_isUsingGen, _callerSdkType)
     };
     if (accessToken) {
         headers['X-Firebase-Auth-Token'] = accessToken;
@@ -197,7 +226,7 @@ function getMessage(obj) {
 }
 
 const name = "@firebase/data-connect";
-const version = "0.2.0";
+const version = "0.3.0-20250205191324";
 
 /**
  * @license
@@ -220,15 +249,20 @@ const version = "0.2.0";
  * Abstraction around AppCheck's token fetching capabilities.
  */
 class AppCheckTokenProvider {
-    constructor(appName_, appCheckProvider) {
-        this.appName_ = appName_;
+    constructor(app$1, appCheckProvider) {
         this.appCheckProvider = appCheckProvider;
+        if (app._isFirebaseServerApp(app$1) && app$1.settings.appCheckToken) {
+            this.serverAppAppCheckToken = app$1.settings.appCheckToken;
+        }
         this.appCheck = appCheckProvider === null || appCheckProvider === void 0 ? void 0 : appCheckProvider.getImmediate({ optional: true });
         if (!this.appCheck) {
             void (appCheckProvider === null || appCheckProvider === void 0 ? void 0 : appCheckProvider.get().then(appCheck => (this.appCheck = appCheck)).catch());
         }
     }
-    getToken(forceRefresh) {
+    getToken() {
+        if (this.serverAppAppCheckToken) {
+            return Promise.resolve({ token: this.serverAppAppCheckToken });
+        }
         if (!this.appCheck) {
             return new Promise((resolve, reject) => {
                 // Support delayed initialization of FirebaseAppCheck. This allows our
@@ -237,7 +271,7 @@ class AppCheckTokenProvider {
                 // becomes available before the timoeout below expires.
                 setTimeout(() => {
                     if (this.appCheck) {
-                        this.getToken(forceRefresh).then(resolve, reject);
+                        this.getToken().then(resolve, reject);
                     }
                     else {
                         resolve(null);
@@ -245,7 +279,7 @@ class AppCheckTokenProvider {
                 }, 0);
             });
         }
-        return this.appCheck.getToken(forceRefresh);
+        return this.appCheck.getToken();
     }
     addTokenChangeListener(listener) {
         var _a;
@@ -584,13 +618,14 @@ function addToken(url, apiKey) {
  * limitations under the License.
  */
 class RESTTransport {
-    constructor(options, apiKey, appId, authProvider, appCheckProvider, transportOptions, _isUsingGen = false) {
+    constructor(options, apiKey, appId, authProvider, appCheckProvider, transportOptions, _isUsingGen = false, _callerSdkType = CallerSdkTypeEnum.Base) {
         var _a, _b;
         this.apiKey = apiKey;
         this.appId = appId;
         this.authProvider = authProvider;
         this.appCheckProvider = appCheckProvider;
         this._isUsingGen = _isUsingGen;
+        this._callerSdkType = _callerSdkType;
         this._host = '';
         this._location = 'l';
         this._connectorName = '';
@@ -607,7 +642,7 @@ class RESTTransport {
                 name: `projects/${this._project}/locations/${this._location}/services/${this._serviceName}/connectors/${this._connectorName}`,
                 operationName: queryName,
                 variables: body
-            }, abortController, this.appId, this._accessToken, this._appCheckToken, this._isUsingGen));
+            }, abortController, this.appId, this._accessToken, this._appCheckToken, this._isUsingGen, this._callerSdkType));
             return withAuth;
         };
         this.invokeMutation = (mutationName, body) => {
@@ -617,7 +652,7 @@ class RESTTransport {
                     name: `projects/${this._project}/locations/${this._location}/services/${this._serviceName}/connectors/${this._connectorName}`,
                     operationName: mutationName,
                     variables: body
-                }, abortController, this.appId, this._accessToken, this._appCheckToken, this._isUsingGen);
+                }, abortController, this.appId, this._accessToken, this._appCheckToken, this._isUsingGen, this._callerSdkType);
             });
             return taskResult;
         };
@@ -718,6 +753,9 @@ class RESTTransport {
             throw err;
         });
     }
+    _setCallerSdkType(callerSdkType) {
+        this._callerSdkType = callerSdkType;
+    }
 }
 
 /**
@@ -827,6 +865,7 @@ class DataConnect {
         this.isEmulator = false;
         this._initialized = false;
         this._isUsingGeneratedSdk = false;
+        this._callerSdkType = CallerSdkTypeEnum.Base;
         if (typeof process !== 'undefined' && process.env) {
             const host = process.env[FIREBASE_DATA_CONNECT_EMULATOR_HOST_VAR];
             if (host) {
@@ -842,6 +881,12 @@ class DataConnect {
             this._isUsingGeneratedSdk = true;
         }
     }
+    _setCallerSdkType(callerSdkType) {
+        this._callerSdkType = callerSdkType;
+        if (this._initialized) {
+            this._transport._setCallerSdkType(callerSdkType);
+        }
+    }
     _delete() {
         app._removeServiceInstance(this.app, 'data-connect', JSON.stringify(this.getSettings()));
         return Promise.resolve();
@@ -865,10 +910,10 @@ class DataConnect {
             this._authTokenProvider = new FirebaseAuthProvider(this.app.name, this.app.options, this._authProvider);
         }
         if (this._appCheckProvider) {
-            this._appCheckTokenProvider = new AppCheckTokenProvider(this.app.name, this._appCheckProvider);
+            this._appCheckTokenProvider = new AppCheckTokenProvider(this.app, this._appCheckProvider);
         }
         this._initialized = true;
-        this._transport = new this._transportClass(this.dataConnectOptions, this.app.options.apiKey, this.app.options.appId, this._authTokenProvider, this._appCheckTokenProvider, undefined, this._isUsingGeneratedSdk);
+        this._transport = new this._transportClass(this.dataConnectOptions, this.app.options.apiKey, this.app.options.appId, this._authTokenProvider, this._appCheckTokenProvider, undefined, this._isUsingGeneratedSdk, this._callerSdkType);
         if (this._transportOptions) {
             this._transport.useEmulator(this._transportOptions.host, this._transportOptions.port, this._transportOptions.sslEnabled);
         }
@@ -1043,7 +1088,7 @@ function queryRef(dcInstance, queryName, variables, initialCache) {
         dataConnect: dcInstance,
         refType: QUERY_STR,
         name: queryName,
-        variables: variables
+        variables
     };
 }
 /**
@@ -1173,6 +1218,7 @@ function subscribe(queryRefOrSerializedResult, observerOrOnNext, onError, onComp
 initializeFetch(fetch);
 registerDataConnect('node');
 
+exports.CallerSdkTypeEnum = CallerSdkTypeEnum;
 exports.DataConnect = DataConnect;
 exports.MUTATION_STR = MUTATION_STR;
 exports.MutationManager = MutationManager;