@firebase/data-connect 0.0.3-dataconnect-preview.d986d4bf2 → 0.1.0

package/dist/index.node.cjs.js

@@ -138,7 +138,7 @@ function getGoogApiClientValue(_isUsingGen) {
     }
     return str;
 }
-function dcFetch(url, body, _a, accessToken, _isUsingGen) {
+function dcFetch(url, body, _a, appId, accessToken, appCheckToken, _isUsingGen) {
     var _this = this;
     var signal = _a.signal;
     if (!connectFetch) {
@@ -151,6 +151,12 @@ function dcFetch(url, body, _a, accessToken, _isUsingGen) {
     if (accessToken) {
         headers['X-Firebase-Auth-Token'] = accessToken;
     }
+    if (appId) {
+        headers['x-firebase-gmpid'] = appId;
+    }
+    if (appCheckToken) {
+        headers['X-Firebase-AppCheck'] = appCheckToken;
+    }
     var bodyStr = JSON.stringify(body);
     logDebug("Making request out to ".concat(url, " with body: ").concat(bodyStr));
     return connectFetch(url, {
@@ -208,7 +214,64 @@ function getMessage(obj) {
 }
 
 var name = "@firebase/data-connect";
-var version = "0.0.3-dataconnect-preview.d986d4bf2";
+var version = "0.1.0";
+
+/**
+ * @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.
+ */
+/**
+ * @internal
+ * Abstraction around AppCheck's token fetching capabilities.
+ */
+var AppCheckTokenProvider = /** @class */ (function () {
+    function AppCheckTokenProvider(appName_, appCheckProvider) {
+        var _this = this;
+        this.appName_ = appName_;
+        this.appCheckProvider = appCheckProvider;
+        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(function (appCheck) { return (_this.appCheck = appCheck); }).catch());
+        }
+    }
+    AppCheckTokenProvider.prototype.getToken = function (forceRefresh) {
+        var _this = this;
+        if (!this.appCheck) {
+            return new Promise(function (resolve, reject) {
+                // Support delayed initialization of FirebaseAppCheck. This allows our
+                // customers to initialize the RTDB SDK before initializing Firebase
+                // AppCheck and ensures that all requests are authenticated if a token
+                // becomes available before the timoeout below expires.
+                setTimeout(function () {
+                    if (_this.appCheck) {
+                        _this.getToken(forceRefresh).then(resolve, reject);
+                    }
+                    else {
+                        resolve(null);
+                    }
+                }, 0);
+            });
+        }
+        return this.appCheck.getToken(forceRefresh);
+    };
+    AppCheckTokenProvider.prototype.addTokenChangeListener = function (listener) {
+        var _a;
+        void ((_a = this.appCheckProvider) === null || _a === void 0 ? void 0 : _a.get().then(function (appCheck) { return appCheck.addTokenListener(listener); }));
+    };
+    return AppCheckTokenProvider;
+}());
 
 /**
  * @license
@@ -226,6 +289,7 @@ var version = "0.0.3-dataconnect-preview.d986d4bf2";
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+// @internal
 var FirebaseAuthProvider = /** @class */ (function () {
     function FirebaseAuthProvider(_appName, _options, _authProvider) {
         var _this = this;
@@ -514,7 +578,7 @@ function urlBuilder(projectConfig, transportOptions) {
         logError('Port type is of an invalid type');
         throw new DataConnectError(Code.INVALID_ARGUMENT, 'Incorrect type for port passed in!');
     }
-    return "".concat(baseUrl, "/v1alpha/projects/").concat(project, "/locations/").concat(location, "/services/").concat(service, "/connectors/").concat(connector);
+    return "".concat(baseUrl, "/v1beta/projects/").concat(project, "/locations/").concat(location, "/services/").concat(service, "/connectors/").concat(connector);
 }
 function addToken(url, apiKey) {
     if (!apiKey) {
@@ -542,12 +606,14 @@ function addToken(url, apiKey) {
  * limitations under the License.
  */
 var RESTTransport = /** @class */ (function () {
-    function RESTTransport(options, apiKey, authProvider, transportOptions, _isUsingGen) {
+    function RESTTransport(options, apiKey, appId, authProvider, appCheckProvider, transportOptions, _isUsingGen) {
         if (_isUsingGen === void 0) { _isUsingGen = false; }
         var _this = this;
-        var _a;
+        var _a, _b;
         this.apiKey = apiKey;
+        this.appId = appId;
         this.authProvider = authProvider;
+        this.appCheckProvider = appCheckProvider;
         this._isUsingGen = _isUsingGen;
         this._host = '';
         this._location = 'l';
@@ -555,7 +621,7 @@ var RESTTransport = /** @class */ (function () {
         this._secure = true;
         this._project = 'p';
         this._accessToken = null;
-        this._authInitialized = false;
+        this._appCheckToken = null;
         this._lastToken = null;
         // TODO(mtewani): Update U to include shape of body defined in line 13.
         this.invokeQuery = function (queryName, body) {
@@ -567,7 +633,7 @@ var RESTTransport = /** @class */ (function () {
                     operationName: queryName,
                     variables: body
                 }, // TODO(mtewani): This is a patch, fix this.
-                abortController, _this._accessToken, _this._isUsingGen);
+                abortController, _this.appId, _this._accessToken, _this._appCheckToken, _this._isUsingGen);
             });
             return {
                 then: withAuth.then.bind(withAuth),
@@ -581,7 +647,7 @@ var RESTTransport = /** @class */ (function () {
                     name: "projects/".concat(_this._project, "/locations/").concat(_this._location, "/services/").concat(_this._serviceName, "/connectors/").concat(_this._connectorName),
                     operationName: mutationName,
                     variables: body
-                }, abortController, _this._accessToken, _this._isUsingGen);
+                }, abortController, _this.appId, _this._accessToken, _this._appCheckToken, _this._isUsingGen);
             });
             return {
                 then: taskResult.then.bind(taskResult),
@@ -615,6 +681,11 @@ var RESTTransport = /** @class */ (function () {
             logDebug("New Token Available: ".concat(token));
             _this._accessToken = token;
         });
+        (_b = this.appCheckProvider) === null || _b === void 0 ? void 0 : _b.addTokenChangeListener(function (result) {
+            var token = result.token;
+            logDebug("New App Check Token Available: ".concat(token));
+            _this._appCheckToken = token;
+        });
     }
     Object.defineProperty(RESTTransport.prototype, "endpointUrl", {
         get: function () {
@@ -641,28 +712,42 @@ var RESTTransport = /** @class */ (function () {
         this._accessToken = newToken;
     };
     RESTTransport.prototype.getWithAuth = function (forceToken) {
-        var _this = this;
+        var _a;
         if (forceToken === void 0) { forceToken = false; }
-        var starterPromise = new Promise(function (resolve) {
-            return resolve(_this._accessToken);
+        return tslib.__awaiter(this, void 0, void 0, function () {
+            var starterPromise, _b;
+            var _this = this;
+            return tslib.__generator(this, function (_c) {
+                switch (_c.label) {
+                    case 0:
+                        starterPromise = new Promise(function (resolve) {
+                            return resolve(_this._accessToken);
+                        });
+                        if (!this.appCheckProvider) return [3 /*break*/, 2];
+                        _b = this;
+                        return [4 /*yield*/, this.appCheckProvider.getToken()];
+                    case 1:
+                        _b._appCheckToken = (_a = (_c.sent())) === null || _a === void 0 ? void 0 : _a.token;
+                        _c.label = 2;
+                    case 2:
+                        if (this.authProvider) {
+                            starterPromise = this.authProvider
+                                .getToken(/*forceToken=*/ forceToken)
+                                .then(function (data) {
+                                if (!data) {
+                                    return null;
+                                }
+                                _this._accessToken = data.accessToken;
+                                return _this._accessToken;
+                            });
+                        }
+                        else {
+                            starterPromise = new Promise(function (resolve) { return resolve(''); });
+                        }
+                        return [2 /*return*/, starterPromise];
+                }
+            });
         });
-        if (!this._authInitialized) {
-            if (this.authProvider) {
-                starterPromise = this.authProvider
-                    .getToken(/*forceToken=*/ forceToken)
-                    .then(function (data) {
-                    if (!data) {
-                        return null;
-                    }
-                    _this._accessToken = data.accessToken;
-                    return _this._accessToken;
-                });
-            }
-            else {
-                starterPromise = new Promise(function (resolve) { return resolve(''); });
-            }
-        }
-        return starterPromise;
     };
     RESTTransport.prototype._setLastToken = function (lastToken) {
         this._lastToken = lastToken;
@@ -793,14 +878,16 @@ function parseOptions(fullHost) {
  * Class representing Firebase Data Connect
  */
 var DataConnect = /** @class */ (function () {
+    // @internal
     function DataConnect(app, 
     // TODO(mtewani): Replace with _dataConnectOptions in the future
-    dataConnectOptions, _authProvider) {
+    dataConnectOptions, _authProvider, _appCheckProvider) {
         this.app = app;
         this.dataConnectOptions = dataConnectOptions;
         this._authProvider = _authProvider;
+        this._appCheckProvider = _appCheckProvider;
         this.isEmulator = false;
-        this.initialized = false;
+        this._initialized = false;
         this._isUsingGeneratedSdk = false;
         if (typeof process !== 'undefined' && process.env) {
             var host = process.env[FIREBASE_DATA_CONNECT_EMULATOR_HOST_VAR];
@@ -811,9 +898,7 @@ var DataConnect = /** @class */ (function () {
             }
         }
     }
-    /*
-      @internal
-    */
+    // @internal
     DataConnect.prototype._useGeneratedSdk = function () {
         if (!this._isUsingGeneratedSdk) {
             this._isUsingGeneratedSdk = true;
@@ -823,13 +908,15 @@ var DataConnect = /** @class */ (function () {
         app._removeServiceInstance(this.app, 'data-connect', JSON.stringify(this.getSettings()));
         return Promise.resolve();
     };
+    // @internal
     DataConnect.prototype.getSettings = function () {
         var copy = JSON.parse(JSON.stringify(this.dataConnectOptions));
         delete copy.projectId;
         return copy;
     };
+    // @internal
     DataConnect.prototype.setInitialized = function () {
-        if (this.initialized) {
+        if (this._initialized) {
             return;
         }
         if (this._transportClass === undefined) {
@@ -839,16 +926,20 @@ var DataConnect = /** @class */ (function () {
         if (this._authProvider) {
             this._authTokenProvider = new FirebaseAuthProvider(this.app.name, this.app.options, this._authProvider);
         }
-        this.initialized = true;
-        this._transport = new this._transportClass(this.dataConnectOptions, this.app.options.apiKey, this._authTokenProvider, undefined, this._isUsingGeneratedSdk);
+        if (this._appCheckProvider) {
+            this._appCheckTokenProvider = new AppCheckTokenProvider(this.app.name, 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);
         if (this._transportOptions) {
             this._transport.useEmulator(this._transportOptions.host, this._transportOptions.port, this._transportOptions.sslEnabled);
         }
         this._queryManager = new QueryManager(this._transport);
         this._mutationManager = new MutationManager(this._transport);
     };
+    // @internal
     DataConnect.prototype.enableEmulator = function (transportOptions) {
-        if (this.initialized) {
+        if (this._initialized) {
             logError('enableEmulator called after initialization');
             throw new DataConnectError(Code.ALREADY_INITIALIZED, 'DataConnect instance already initialized!');
         }
@@ -935,6 +1026,7 @@ function registerDataConnect(variant) {
         var settings = _a.instanceIdentifier, options = _a.options;
         var app = container.getProvider('app').getImmediate();
         var authProvider = container.getProvider('auth-internal');
+        var appCheckProvider = container.getProvider('app-check-internal');
         var newOpts = options;
         if (settings) {
             newOpts = JSON.parse(settings);
@@ -942,7 +1034,7 @@ function registerDataConnect(variant) {
         if (!app.options.projectId) {
             throw new DataConnectError(Code.INVALID_ARGUMENT, 'Project ID must be provided. Did you pass in a proper projectId to initializeApp?');
         }
-        return new DataConnect(app, tslib.__assign(tslib.__assign({}, newOpts), { projectId: app.options.projectId }), authProvider);
+        return new DataConnect(app, tslib.__assign(tslib.__assign({}, newOpts), { projectId: app.options.projectId }), authProvider, appCheckProvider);
     }, "PUBLIC" /* ComponentType.PUBLIC */).setMultipleInstances(true));
     app.registerVersion(name, version, variant);
     // BUILD_TARGET will be replaced by values like esm5, esm2017, cjs5, etc during the compilation
@@ -1119,8 +1211,6 @@ initializeFetch(fetch);
 registerDataConnect('node');
 
 exports.DataConnect = DataConnect;
-exports.FIREBASE_DATA_CONNECT_EMULATOR_HOST_VAR = FIREBASE_DATA_CONNECT_EMULATOR_HOST_VAR;
-exports.FirebaseAuthProvider = FirebaseAuthProvider;
 exports.MUTATION_STR = MUTATION_STR;
 exports.MutationManager = MutationManager;
 exports.QUERY_STR = QUERY_STR;