@squidcloud/client 1.0.128 → 1.0.130

package/dist/cjs/index.js

@@ -27050,6 +27050,7 @@ const DatabaseIntegrationTypes = [
     IntegrationType.built_in_db,
     IntegrationType.mongo,
     IntegrationType.mysql,
+    IntegrationType.bigquery,
     IntegrationType.mssql,
     IntegrationType.postgres,
     IntegrationType.cockroach,
@@ -27116,6 +27117,24 @@ const MysqlConnectionOptionsSchema = {
         connectionLimit: { type: 'number', nullable: false },
     },
 };
+const BigQueryConnectionOptionsSchema = {
+    type: 'object',
+    nullable: false,
+    required: ['secrets', 'projectId', 'datasetId', 'email'],
+    properties: {
+        projectId: { type: 'string', nullable: false },
+        datasetId: { type: 'string', nullable: false },
+        email: { type: 'string', nullable: false },
+        secrets: {
+            type: 'object',
+            required: ['privateKey'],
+            nullable: false,
+            properties: {
+                privateKey: { type: 'string', nullable: false, isSecret: {} },
+            },
+        },
+    },
+};
 const OracleConnectionOptionsSchema = {
     type: 'object',
     nullable: false,
@@ -27359,6 +27378,27 @@ const UpsertIntegrationRequestSchema = {
                 },
             },
         },
+        {
+            type: 'object',
+            properties: {
+                type: { const: IntegrationType.bigquery },
+                config: {
+                    type: 'object',
+                    required: ['configuration'],
+                    properties: {
+                        supportsExternalChanges: { type: 'boolean', nullable: true },
+                        configuration: {
+                            type: 'object',
+                            nullable: false,
+                            required: ['connectionOptions'],
+                            properties: {
+                                connectionOptions: BigQueryConnectionOptionsSchema,
+                            },
+                        },
+                    },
+                },
+            },
+        },
         {
             type: 'object',
             properties: {
@@ -27774,6 +27814,20 @@ const TestDataConnectionRequestSchema = {
                 },
             },
         },
+        {
+            type: 'object',
+            required: ['configuration'],
+            properties: {
+                type: { const: IntegrationType.bigquery },
+                configuration: {
+                    type: 'object',
+                    required: ['connectionOptions'],
+                    properties: {
+                        connectionOptions: BigQueryConnectionOptionsSchema,
+                    },
+                },
+            },
+        },
         {
             type: 'object',
             required: ['configuration'],
@@ -27901,6 +27955,14 @@ const DiscoverDataConnectionSchemaRequestSchema = {
                 connectionOptions: MysqlConnectionOptionsSchema,
             },
         },
+        {
+            type: 'object',
+            required: ['connectionOptions'],
+            properties: {
+                integrationType: { const: IntegrationType.bigquery },
+                connectionOptions: BigQueryConnectionOptionsSchema,
+            },
+        },
         {
             type: 'object',
             required: ['connectionOptions'],
@@ -28165,14 +28227,14 @@ class mutation_context_MutationContext {
 
 function sortKeys(json) {
     if (Array.isArray(json)) {
-        return json.map((o) => sortKeys(o));
+        return json.map(o => sortKeys(o));
     }
     if (typeof json !== 'object' || json === null || json instanceof Date) {
         return json;
     }
     const keys = Object.keys(json);
     const result = {};
-    keys.sort().forEach((key) => {
+    keys.sort().forEach(key => {
         result[key] = sortKeys(json[key]);
     });
     return result;
@@ -28183,14 +28245,14 @@ function serialization_normalizeJsonAsString(json) {
 function serialization_serializeObj(obj) {
     if (obj === undefined)
         return null;
-    const objWithReplacedDates = lodash.cloneDeepWith(obj, (value) => {
+    const objWithReplacedDates = lodash.cloneDeepWith(obj, value => {
         return lodash.isDate(value) ? { $date: value.toISOString() } : undefined;
     });
     return JSON.stringify(objWithReplacedDates);
 }
 function deserializeObj(str) {
     const deserializedObj = JSON.parse(str);
-    return lodash.cloneDeepWith(deserializedObj, (value) => {
+    return lodash.cloneDeepWith(deserializedObj, value => {
         return lodash.isObject(value) && lodash.has(value, '$date') && Object.keys(value).length === 1
             ? new Date(value['$date'])
             : undefined;
@@ -28551,8 +28613,8 @@ class query_context_QueryContext {
      * @returns Whether the query is a subquery of the parent condition.
      */
     isSubqueryOfCondition(condition) {
-        const conditions = this.parsedConditions.filter((c) => c.fieldName === condition.fieldName);
-        return !!conditions.find((c) => this.evaluateSubset(c, condition));
+        const conditions = this.parsedConditions.filter(c => c.fieldName === condition.fieldName);
+        return !!conditions.find(c => this.evaluateSubset(c, condition));
     }
     /**
      * Verifies that the query is a subquery of the specified conditions. A subquery is defined as a query that evaluates
@@ -28564,7 +28626,7 @@ class query_context_QueryContext {
      */
     isSubqueryOfConditions(conditions) {
         const parsedConditions = this.parseConditions(conditions);
-        return parsedConditions.every((c) => this.isSubqueryOfCondition(c));
+        return parsedConditions.every(c => this.isSubqueryOfCondition(c));
     }
     /**
      * Verifies that the query is a subquery of the specified query. A subquery is defined as a query that evaluates
@@ -28590,7 +28652,7 @@ class query_context_QueryContext {
      * @returns An array of conditions that involve any of the specified field names.
      */
     getConditionsFor(...fieldNames) {
-        return this.parsedConditions.filter((cond) => fieldNames.includes(cond.fieldName));
+        return this.parsedConditions.filter(cond => fieldNames.includes(cond.fieldName));
     }
     /**
      * Returns all conditions that apply to the specified field name. This method provides
@@ -28600,7 +28662,7 @@ class query_context_QueryContext {
      * @returns An array of conditions that involve the specified field name.
      */
     getConditionsForField(fieldName) {
-        return this.parsedConditions.filter((cond) => cond.fieldName === fieldName);
+        return this.parsedConditions.filter(cond => cond.fieldName === fieldName);
     }
     /**
      * Returns true if the given document can be a result of the query.
@@ -28658,7 +28720,7 @@ class query_context_QueryContext {
         const parsedConditions = [];
         const inMap = new Map();
         const notInMap = new Map();
-        conditions.forEach((c) => {
+        conditions.forEach(c => {
             switch (c.operator) {
                 case '==':
                 case 'in':
@@ -28905,7 +28967,7 @@ class Pagination {
          */
         this.navigatingToLastPage = false;
         this.lastElement = null;
-        this.snapshotSubject.pipe((0,external_rxjs_namespaceObject.switchAll)()).subscribe((data) => this.dataReceived(data));
+        this.snapshotSubject.pipe((0,external_rxjs_namespaceObject.switchAll)()).subscribe(data => this.dataReceived(data));
         this.templateSnapshotEmitter = snapshotEmitter.clone();
         this.options = Object.assign({ pageSize: 100, subscribe: true }, options);
         this.goToFirstPage();
@@ -28955,7 +29017,7 @@ class Pagination {
     async dataReceived(data) {
         // Because documents might be deleted by the time we need them (if we are not subscribed), we save the extracted
         // data here. This is the only place we're allowed to call extractData.
-        const extractedData = data.map((s) => this.templateSnapshotEmitter.extractData(s));
+        const extractedData = data.map(s => this.templateSnapshotEmitter.extractData(s));
         if (data.length === 0) {
             if (this.onFirstPage) {
                 this.internalStateObserver.next({ numBefore: 0, numAfter: 0, data, extractedData });
@@ -28969,7 +29031,7 @@ class Pagination {
             if (this.lastElement !== null) {
                 // We just executed a `prev` and we know what the last element is on the page but not the first.
                 // We need to find the first element on the page instead, because that's our anchor.
-                const numAfter = extractedData.filter((s) => this.compare(s, this.lastElement) === 1).length;
+                const numAfter = extractedData.filter(s => this.compare(s, this.lastElement) === 1).length;
                 this.firstElement = extractedData[data.length - numAfter - this.options.pageSize];
                 this.lastElement = null;
             }
@@ -28979,7 +29041,7 @@ class Pagination {
                 this.navigatingToLastPage = false;
             }
         }
-        const numBefore = extractedData.filter((s) => this.compare(s, this.firstElement) === -1).length;
+        const numBefore = extractedData.filter(s => this.compare(s, this.firstElement) === -1).length;
         const numAfter = Math.max(0, data.length - numBefore - this.options.pageSize);
         // Current page is empty, go to previous page
         if (numBefore === data.length) {
@@ -28996,24 +29058,24 @@ class Pagination {
                 .limit(this.options.pageSize * 3)
                 .flipSortOrder();
             if (startingDoc) {
-                newSnapshotEmitter.addCompositeCondition(this.templateSnapshotEmitter.getSortOrders().map((sortOrder) => {
+                newSnapshotEmitter.addCompositeCondition(this.templateSnapshotEmitter.getSortOrders().map(sortOrder => {
                     return {
                         fieldName: sortOrder.fieldName,
                         operator: sortOrder.asc ? '<=' : '>=',
-                        value: getInPath(startingDoc, sortOrder.fieldName),
+                        value: getInPath(startingDoc, sortOrder.fieldName) || null,
                     };
                 }));
             }
-            this.snapshotSubject.next(newSnapshotEmitter.snapshots(this.options.subscribe).pipe((0,external_rxjs_namespaceObject.map)((s) => s.reverse())));
+            this.snapshotSubject.next(newSnapshotEmitter.snapshots(this.options.subscribe).pipe((0,external_rxjs_namespaceObject.map)(s => s.reverse())));
         }
         else {
             const newSnapshotEmitter = this.templateSnapshotEmitter.clone().limit(this.options.pageSize * 3);
             if (startingDoc) {
-                newSnapshotEmitter.addCompositeCondition(this.templateSnapshotEmitter.getSortOrders().map((sortOrder) => {
+                newSnapshotEmitter.addCompositeCondition(this.templateSnapshotEmitter.getSortOrders().map(sortOrder => {
                     return {
                         fieldName: sortOrder.fieldName,
                         operator: sortOrder.asc ? '>=' : '<=',
-                        value: getInPath(startingDoc, sortOrder.fieldName),
+                        value: getInPath(startingDoc, sortOrder.fieldName) || null,
                     };
                 }));
             }
@@ -29029,7 +29091,7 @@ class Pagination {
         // It is possible that we get here and unsubscribe() was called. In that case we might not get any new state,
         // so to avoid stalling we return an empty page in that situation. (We can't return the last page we saw
         // because that has already been deleted, and also because it's possible that we've never seen any data.)
-        this.isDestroyed.pipe((0,external_rxjs_namespaceObject.filter)(Boolean), (0,external_rxjs_namespaceObject.map)((_) => ({
+        this.isDestroyed.pipe((0,external_rxjs_namespaceObject.filter)(Boolean), (0,external_rxjs_namespaceObject.map)(_ => ({
             data: [],
             extractedData: [],
             numBefore: 0,
@@ -29068,7 +29130,11 @@ class Pagination {
     /** Returns a promise that resolves when the next page of data is available. */
     async next() {
         const { numBefore, extractedData } = await this.waitForInternalState();
-        this.firstElement = extractedData[numBefore + this.options.pageSize];
+        // Setting the firstElement implicitly moves us to the next page. If we are on the last page, we don't change
+        // firstElement because we want next() to be a no-op.
+        if (numBefore + this.options.pageSize < extractedData.length) {
+            this.firstElement = extractedData[numBefore + this.options.pageSize];
+        }
         this.internalStateObserver.next(null);
         this.doNewQuery(extractedData[numBefore], false);
         return await this.waitForData();
@@ -29079,7 +29145,7 @@ class Pagination {
     }
     /** Returns an observable that emits the current state of the pagination. */
     observeState() {
-        return this.internalStateObserver.pipe((0,external_rxjs_namespaceObject.filter)((state) => state !== null), (0,external_rxjs_namespaceObject.map)((state) => {
+        return this.internalStateObserver.pipe((0,external_rxjs_namespaceObject.filter)((state) => state !== null), (0,external_rxjs_namespaceObject.map)(state => {
             return this.internalStateToState(state);
         }));
     }
@@ -29114,7 +29180,7 @@ class Pagination {
             .limit(this.options.pageSize * 3)
             .flipSortOrder()
             .snapshots(this.options.subscribe)
-            .pipe((0,external_rxjs_namespaceObject.map)((s) => s.reverse()));
+            .pipe((0,external_rxjs_namespaceObject.map)(s => s.reverse()));
         this.snapshotSubject.next(lastPageSnapshot);
         return await this.waitForData();
     }
@@ -29137,12 +29203,14 @@ class Pagination {
 
 
 
+
 /** @internal */
 const ExecuteFunctionSecureAnnotations = (/* unused pure expression or super */ null && ([
     'secureDistributedLock',
     'secureQuery',
     'secureMutation',
     'secureNamedQuery',
+    'secureNativeQuery',
     'secureGraphQL',
     'secureApi',
     'secureAiAssistantChat',
@@ -29159,6 +29227,8 @@ function transformParams(params, executeFunctionAnnotationType) {
             return [new MutationContext(params[0].mutation, params[0].beforeAndAfterDocs, params[0].serverTimestamp)];
         case 'secureNamedQuery':
             return [new NamedQueryContext(params[0])];
+        case 'secureNativeQuery':
+            return [new NativeQueryContext(params[0], params[1], params[2])];
         case 'secureDistributedLock':
             return [new DistributedLockContext(params[0].mutex, params[0].exclusive)];
         case 'secureGraphQL':
@@ -29687,12 +29757,12 @@ function convertInsertToUpdate(insertMutation) {
 function reduceMutations(mutations) {
     let result = [];
     (0,external_rxjs_namespaceObject.from)(mutations)
-        .pipe((0,external_rxjs_namespaceObject.groupBy)((mutation) => {
+        .pipe((0,external_rxjs_namespaceObject.groupBy)(mutation => {
         return `${mutation.squidDocIdObj.integrationId}${mutation.squidDocIdObj.collectionName}/${mutation.squidDocIdObj.docId}`;
-    }), (0,external_rxjs_namespaceObject.mergeMap)((mutationsGroup) => mutationsGroup.pipe((0,external_rxjs_namespaceObject.reduce)((mutationA, mutationB) => {
+    }), (0,external_rxjs_namespaceObject.mergeMap)(mutationsGroup => mutationsGroup.pipe((0,external_rxjs_namespaceObject.reduce)((mutationA, mutationB) => {
         return mergeMutations(mutationA, mutationB);
     }))), (0,external_rxjs_namespaceObject.toArray)())
-        .subscribe((value) => {
+        .subscribe(value => {
         result = value;
     });
     return result;
@@ -29975,13 +30045,13 @@ function validateCorrectStringType(value) {
 function isRightType(value, type) {
     // TODO: the method is ambiguous when the value is an empty array and will return 'true' for any type.
     if (Array.isArray(value)) {
-        return value.every((element) => typeof element === type);
+        return value.every(element => typeof element === type);
     }
     return typeof value === type;
 }
 /** Returns true if 'obj' has only keys listed in the 'keys'. Object can't be an array. */
 function hasOnlyKeys(obj, keys) {
-    return !Array.isArray(obj) && [...Object.keys(obj)].every((key) => keys.includes(key));
+    return !Array.isArray(obj) && [...Object.keys(obj)].every(key => keys.includes(key));
 }
 
 ;// CONCATENATED MODULE: ../common/src/schema/schema.types.ts
@@ -30084,7 +30154,7 @@ function validateSchema(schema, data, updatedPaths = [], dataBefore = {}) {
                 fieldPath = `${fieldPath}/${schemaError.params['missingProperty']}`;
             }
             fieldPath = fieldPath.slice(1).replace(/\//g, '.') + '.';
-            if (updatedPaths.some((updatedPath) => fieldPath.startsWith(updatedPath + '.'))) {
+            if (updatedPaths.some(updatedPath => fieldPath.startsWith(updatedPath + '.'))) {
                 throw new ValidationError(`${fieldPath} does not conform with the collection schema.`, HttpStatus.BAD_REQUEST);
             }
         }
@@ -30321,7 +30391,17 @@ function validateTruthy(value, message, statusCode = HttpStatus.BAD_REQUEST, det
 }
 
 ;// CONCATENATED MODULE: ../common/src/utils/http.ts
-const kotlinControllers = ['query', 'aiData', 'api', 'backend-function', 'webhooks', 'ws'];
+const kotlinControllers = [
+    'query',
+    'aiData',
+    'api',
+    'backend-function',
+    'webhooks',
+    'ws',
+    'quota',
+    'named-query',
+    'native-query',
+];
 function getApplicationUrl(regionPrefix, appId, path) {
     const baseUrl = 'https://squid.cloud';
     const parsedBaseUrl = new URL(baseUrl);
@@ -30408,7 +30488,7 @@ class LockManager {
         }
     }
     canGetLock(...mutexes) {
-        return !mutexes.some((mutex) => { var _a; return (_a = this.locks[mutex]) === null || _a === void 0 ? void 0 : _a.value; });
+        return !mutexes.some(mutex => { var _a; return (_a = this.locks[mutex]) === null || _a === void 0 ? void 0 : _a.value; });
     }
     lockSync(...mutexes) {
         (0,dist.assertTruthy)(this.canGetLock(...mutexes), 'Cannot acquire lock sync');
@@ -30545,10 +30625,13 @@ function createWebSocketWrapper(url, opts = {}) {
             ws.send(x);
         },
         close(code = 4999, message) {
-            $.connected = false;
-            clearTimeout(timer);
-            timer = undefined;
-            ws.close(code, message);
+            try {
+                $.connected = false;
+                clearTimeout(timer);
+                timer = undefined;
+                ws.close(code, message);
+            }
+            catch (e) { }
         },
     };
     $.open();
@@ -30608,6 +30691,7 @@ function createWebSocketWrapper(url, opts = {}) {
 
 
 
+
 
 
 
@@ -31549,7 +31633,7 @@ class JoinQueryBuilder extends BaseQueryBuilder {
         }
         return this.querySubscriptionManager
             .processQuery(this.build(), this.rootAlias, lodash.cloneDeep(this.joins), lodash.cloneDeep(this.joinConditions), subscribe, false)
-            .pipe(map_map((docs) => docs.map((docRecord) => {
+            .pipe(map_map(docs => docs.map(docRecord => {
             const result = {};
             for (const [alias, doc] of Object.entries(docRecord)) {
                 const collectionName = alias === this.rootAlias ? this.collectionName : this.joins[alias].collectionName;
@@ -31633,9 +31717,9 @@ class DereferencedJoin {
     }
     /** @inheritDoc */
     snapshots(subscribe) {
-        return this.joinQueryBuilder.snapshots(subscribe).pipe(map_map((docs) => {
-            return docs.map((doc) => {
-                return lodash.mapValues(doc, (doc1) => {
+        return this.joinQueryBuilder.snapshots(subscribe).pipe(map_map(docs => {
+            return docs.map(doc => {
+                return lodash.mapValues(doc, doc1 => {
                     return doc1 === null || doc1 === void 0 ? void 0 : doc1.data;
                 });
             });
@@ -31688,8 +31772,8 @@ class DereferencedGroupedJoin {
     }
     /** @inheritDoc */
     snapshots(subscribe) {
-        return this.groupedJoin.snapshots(subscribe).pipe(map_map((docs) => {
-            return docs.map((doc) => {
+        return this.groupedJoin.snapshots(subscribe).pipe(map_map(docs => {
+            return docs.map(doc => {
                 return this.dereference(doc, this.groupedJoin.joinQueryBuilder.rootAlias);
             });
         }));
@@ -31762,7 +31846,7 @@ class GroupedJoin {
     }
     /** @inheritDoc */
     snapshots(subscribe) {
-        return this.joinQueryBuilder.snapshots(subscribe).pipe(map_map((docs) => {
+        return this.joinQueryBuilder.snapshots(subscribe).pipe(map_map(docs => {
             return this.groupData(docs, this.joinQueryBuilder.rootAlias);
         }));
     }
@@ -31777,12 +31861,12 @@ class GroupedJoin {
         return new DereferencedGroupedJoin(this);
     }
     groupData(input, rootAlias) {
-        const oneLevelGroup = lodash.groupBy(input, (inputRow) => { var _a; return (_a = inputRow[rootAlias]) === null || _a === void 0 ? void 0 : _a.squidDocId; });
+        const oneLevelGroup = lodash.groupBy(input, inputRow => { var _a; return (_a = inputRow[rootAlias]) === null || _a === void 0 ? void 0 : _a.squidDocId; });
         return Object.values(oneLevelGroup)
-            .filter((value) => {
+            .filter(value => {
             return value[0][rootAlias] !== undefined;
         })
-            .map((value) => {
+            .map(value => {
             const rights = this.joinQueryBuilder.leftToRight[rootAlias];
             const actualValue = value[0][rootAlias];
             if (rights.length === 0) {
@@ -31957,12 +32041,12 @@ class MergedQueryBuilder {
     constructor(...snapshotEmitters) {
         if (snapshotEmitters.length === 0)
             throw new Error('At least one query builder must be provided');
-        this.snapshotEmitters = snapshotEmitters.map((builder) => builder.clone());
-        const maxLimit = Math.max(...this.snapshotEmitters.map((builder) => {
+        this.snapshotEmitters = snapshotEmitters.map(builder => builder.clone());
+        const maxLimit = Math.max(...this.snapshotEmitters.map(builder => {
             const limit = builder.getLimit();
             return limit === -1 ? 1000 : limit;
         }));
-        this.snapshotEmitters.forEach((builder) => builder.limit(maxLimit));
+        this.snapshotEmitters.forEach(builder => builder.limit(maxLimit));
         const sortOrder = this.snapshotEmitters[0].getSortOrders();
         for (const builder of this.snapshotEmitters) {
             const builderSortOrder = builder.getSortOrders();
@@ -31986,7 +32070,7 @@ class MergedQueryBuilder {
      * @inheritDoc
      */
     snapshots(subscribe = true) {
-        const observables = this.snapshotEmitters.map((builder) => builder.snapshots(subscribe));
+        const observables = this.snapshotEmitters.map(builder => builder.snapshots(subscribe));
         return this.or(this.snapshotEmitters[0].getSortOrders(), ...observables);
     }
     /**
@@ -31996,7 +32080,7 @@ class MergedQueryBuilder {
         throw new Error('peek is not currently supported for merged queries');
     }
     or(sort, ...observables) {
-        return (0,external_rxjs_namespaceObject.combineLatest)([...observables]).pipe((0,external_rxjs_namespaceObject.map)((results) => {
+        return (0,external_rxjs_namespaceObject.combineLatest)([...observables]).pipe((0,external_rxjs_namespaceObject.map)(results => {
             const seenData = new Set();
             const flatResult = results.flat();
             const result = [];
@@ -32027,7 +32111,7 @@ class MergedQueryBuilder {
         }));
     }
     clone() {
-        return new MergedQueryBuilder(...this.snapshotEmitters.map((builder) => builder.clone()));
+        return new MergedQueryBuilder(...this.snapshotEmitters.map(builder => builder.clone()));
     }
     getSortOrders() {
         return this.snapshotEmitters[0].getSortOrders();
@@ -32039,20 +32123,20 @@ class MergedQueryBuilder {
         return this;
     }
     limit(limit) {
-        this.snapshotEmitters.forEach((builder) => builder.limit(limit));
+        this.snapshotEmitters.forEach(builder => builder.limit(limit));
         return this;
     }
     getLimit() {
         return this.snapshotEmitters[0].getLimit();
     }
     flipSortOrder() {
-        this.snapshotEmitters.forEach((builder) => builder.flipSortOrder());
+        this.snapshotEmitters.forEach(builder => builder.flipSortOrder());
         return this;
     }
     serialize() {
         return {
             type: 'merged',
-            queries: this.snapshotEmitters.map((s) => s.serialize()),
+            queries: this.snapshotEmitters.map(s => s.serialize()),
         };
     }
     extractData(data) {
@@ -32147,7 +32231,7 @@ class DocumentReference {
             .getForDocument(this.squidDocId)
             .dereference()
             .snapshots()
-            .pipe((0,external_rxjs_namespaceObject.map)((results) => {
+            .pipe((0,external_rxjs_namespaceObject.map)(results => {
             (0,dist.truthy)(results.length <= 1, 'Got more than one doc for the same id:' + this.squidDocId);
             return results.length ? results[0] : undefined;
         }));
@@ -32360,7 +32444,7 @@ class DereferenceEmitter {
     }
     /** @inheritDoc */
     peek() {
-        return this.queryBuilder.peek().map((ref) => ref.data);
+        return this.queryBuilder.peek().map(ref => ref.data);
     }
     /** @inheritDoc */
     snapshot() {
@@ -32368,8 +32452,8 @@ class DereferenceEmitter {
     }
     /** @inheritDoc */
     snapshots(subscribe) {
-        return this.queryBuilder.snapshots(subscribe).pipe(map_map((refs) => {
-            return refs.map((ref) => ref.data);
+        return this.queryBuilder.snapshots(subscribe).pipe(map_map(refs => {
+            return refs.map(ref => ref.data);
         }));
     }
     getSortOrders() {
@@ -32459,7 +32543,7 @@ class QueryBuilder extends BaseQueryBuilder {
         return this.query.limit;
     }
     limitBy(limit, ...fields) {
-        const sorts = this.query.sortOrder.map((s) => {
+        const sorts = this.query.sortOrder.map(s => {
             return s.fieldName;
         });
         (0,dist.assertTruthy)(lodash.isEqual(fields.sort(), sorts.slice(0, fields.length).sort()), 'All fields in limitBy must be appear in the first fields in the sortBy list.');
@@ -32470,7 +32554,7 @@ class QueryBuilder extends BaseQueryBuilder {
     sortBy(fieldName, asc = true) {
         const fieldSort = { asc, fieldName };
         validateFieldSort(fieldSort);
-        (0,dist.assertTruthy)(!this.query.sortOrder.some((so) => so.fieldName === fieldName), `${fieldName} already in the sort list.`);
+        (0,dist.assertTruthy)(!this.query.sortOrder.some(so => so.fieldName === fieldName), `${fieldName} already in the sort list.`);
         this.query.sortOrder.push(fieldSort);
         return this;
     }
@@ -32484,15 +32568,15 @@ class QueryBuilder extends BaseQueryBuilder {
     mergeConditions() {
         const simpleConditions = this.query.conditions.filter(isSimpleCondition);
         const result = [];
-        const groupByFieldName = lodash.groupBy(simpleConditions || [], (condition) => condition.fieldName);
+        const groupByFieldName = lodash.groupBy(simpleConditions || [], condition => condition.fieldName);
         for (const fieldNameGroup of Object.values(groupByFieldName)) {
-            const groupByOperator = lodash.groupBy(fieldNameGroup, (operator) => operator.operator);
+            const groupByOperator = lodash.groupBy(fieldNameGroup, operator => operator.operator);
             for (const [operator, operatorGroup] of Object.entries(groupByOperator)) {
                 if (operator === '==' || operator === '!=') {
                     result.push(...operatorGroup);
                     continue;
                 }
-                const sorted = lodash.sortBy(operatorGroup, (o) => o.value);
+                const sorted = lodash.sortBy(operatorGroup, o => o.value);
                 if (operator === '>' || operator === '>=') {
                     result.push(sorted[sorted.length - 1]);
                 }
@@ -32504,7 +32588,7 @@ class QueryBuilder extends BaseQueryBuilder {
                 }
             }
         }
-        return [...this.query.conditions.filter((c) => !isSimpleCondition(c)), ...result];
+        return [...this.query.conditions.filter(c => !isSimpleCondition(c)), ...result];
     }
     // noinspection JSUnusedGlobalSymbols
     getSortOrder() {
@@ -32540,8 +32624,8 @@ class QueryBuilder extends BaseQueryBuilder {
         const query = this.build();
         return this.querySubscriptionManager
             .processQuery(query, this.collectionName, {}, {}, subscribe, this.forceFetchFromServer)
-            .pipe(map_map((docs) => {
-            return docs.map((docRecord) => {
+            .pipe(map_map(docs => {
+            return docs.map(docRecord => {
                 (0,dist.assertTruthy)(Object.keys(docRecord).length === 1);
                 const doc = docRecord[this.collectionName];
                 const squidDocId = getSquidDocId((0,dist.truthy)(doc).__docId__, this.collectionName, this.integrationId);
@@ -32555,7 +32639,7 @@ class QueryBuilder extends BaseQueryBuilder {
     changes() {
         let beforeDocMap = undefined;
         let beforeDocSet = new Set();
-        return this.snapshots().pipe((0,external_rxjs_namespaceObject.combineLatestWith)(this.documentIdentityService.observeChanges().pipe((0,external_rxjs_namespaceObject.switchMap)((idResolutionMap) => {
+        return this.snapshots().pipe((0,external_rxjs_namespaceObject.combineLatestWith)(this.documentIdentityService.observeChanges().pipe((0,external_rxjs_namespaceObject.switchMap)(idResolutionMap => {
             Object.entries(idResolutionMap).forEach(([squidDocId, newSquidDocId]) => {
                 replaceKeyInRecord(beforeDocMap || {}, squidDocId, newSquidDocId);
             });
@@ -32632,7 +32716,7 @@ class QueryBuilder extends BaseQueryBuilder {
         return this;
     }
     flipSortOrder() {
-        this.query.sortOrder = this.query.sortOrder.map((sort) => {
+        this.query.sortOrder = this.query.sortOrder.map(sort => {
             return Object.assign(Object.assign({}, sort), { asc: !sort.asc });
         });
         if (this.query.limitBy) {
@@ -32693,7 +32777,7 @@ function deserializeSimpleQuery(squid, serializedQuery) {
 function deserializeMergedQuery(squid, serializedQuery) {
     const { queries } = serializedQuery;
     const { collectionName, integrationId } = getCollection(queries[0]);
-    const builders = queries.map((q) => deserializeQuery(squid, q));
+    const builders = queries.map(q => deserializeQuery(squid, q));
     return squid.collection(collectionName, integrationId).or(...builders);
 }
 function deserializeJoinQuery(squid, serializedQuery) {
@@ -32758,8 +32842,8 @@ class AiAssistantClient {
         this.ongoingChatSequences = {};
         this.socketManager
             .observeNotifications()
-            .pipe((0,external_rxjs_namespaceObject.filter)((notification) => notification.type === 'aiAssistant'), map_map((n) => n))
-            .subscribe((notification) => {
+            .pipe((0,external_rxjs_namespaceObject.filter)((notification) => notification.type === 'aiAssistant'), map_map(n => n))
+            .subscribe(notification => {
             this.handleChatResponse(notification).then();
         });
     }
@@ -32798,14 +32882,14 @@ class AiAssistantClient {
             if (complete) {
                 return (0,external_rxjs_namespaceObject.of)({ value, complete });
             }
-            return (0,external_rxjs_namespaceObject.of)(value).pipe((0,external_rxjs_namespaceObject.delay)(5), map_map((char) => ({ value: char, complete: false })));
+            return (0,external_rxjs_namespaceObject.of)(value).pipe((0,external_rxjs_namespaceObject.delay)(5), map_map(char => ({ value: char, complete: false })));
         }), (0,external_rxjs_namespaceObject.takeWhile)(({ complete }) => !complete, true))
             .subscribe({
             next: ({ value }) => {
                 accumulatedValue += value;
                 subject.next(accumulatedValue);
             },
-            error: (e) => {
+            error: e => {
                 console.error(e);
             },
             complete: () => {
@@ -32819,7 +32903,7 @@ class AiAssistantClient {
             integrationId: this.integrationId,
             clientRequestId,
         };
-        this.rpcManager.post('ai/assistant/chat', request).catch((e) => {
+        this.rpcManager.post('ai/assistant/chat', request).catch(e => {
             subject.error(e);
             subject.complete();
         });
@@ -33128,7 +33212,7 @@ class ApiManager {
             request,
             serverUrlOverride: this.apiServerUrlOverrideMapping[integrationId],
         };
-        return (0,external_rxjs_namespaceObject.race)((0,external_rxjs_namespaceObject.from)(this.rpcManager.post('api/call', callApiRequest)).pipe(map_map((response) => {
+        return (0,external_rxjs_namespaceObject.race)((0,external_rxjs_namespaceObject.from)(this.rpcManager.post('api/call', callApiRequest)).pipe(map_map(response => {
             if (response.success) {
                 return deserializeObj(response.payload);
             }
@@ -33143,45 +33227,87 @@ class ApiManager {
 
 ;// CONCATENATED MODULE: ./src/auth.manager.ts
 
+
 class AuthManager {
-    constructor(destructManager, apiKey) {
-        this.destructManager = destructManager;
+    constructor(apiKey, authTokenProvider) {
         this.apiKey = apiKey;
-        this.authIdTokenSet = new external_rxjs_namespaceObject.BehaviorSubject(false);
-        this.authIdTokenObservableSubject = new external_rxjs_namespaceObject.Subject();
-        this.shouldWaitForAuth = false;
-        this.observeAuthIdToken().subscribe((authData) => {
-            this.idToken = authData.token;
-            this.integrationId = authData.integrationId;
-            this.authIdTokenSet.next(true);
-        });
+        // By default, the auth provider resolves to 'undefined' token.
+        this.authDataProvider = async () => ({ token: undefined });
+        if (authTokenProvider) {
+            this.setAuthTokenProvider(authTokenProvider, undefined);
+        }
+    }
+    /**
+     * Sets a new ID token or an ID token provider.
+     * Deprecated. Pass the provider via constructor parameters.
+     */
+    setAuthIdToken(tokenArg, integrationId) {
+        // Using this warning in DD we can track how many active clients use the deprecated API.
+        console.warn('Using a deprecated setAuthIdToken method. Update code to pass AuthTokenProvider in Squid constructor.');
+        // Convert the input into AuthTokenProvider.
+        if (typeof tokenArg === 'string' || tokenArg === undefined) {
+            this.setAuthTokenProvider(async () => tokenArg, integrationId);
+            return;
+        }
+        if (typeof tokenArg === 'function') {
+            this.setAuthTokenProvider(tokenArg, integrationId);
+            return;
+        }
+        (0,dist.assertTruthy)(typeof tokenArg === 'object');
+        if ((0,external_rxjs_namespaceObject.isObservable)(tokenArg)) {
+            this.setAuthTokenProvider(() => (0,external_rxjs_namespaceObject.firstValueFrom)(tokenArg), integrationId);
+            return;
+        }
+        // tokenArg is a Promise.
+        this.setAuthTokenProvider(() => tokenArg, integrationId);
+    }
+    setAuthTokenProvider(tokenProvider, integrationId) {
+        this.authDataProvider = async () => {
+            const authData = { token: await tokenProvider(integrationId), integrationId };
+            this.lastUsedJwtTokenForDeprecatedCode = authData.token;
+            this.lastUsedIntegrationIdForDeprecatedCode = integrationId;
+            return authData;
+        };
     }
-    setAuthIdToken(idToken, integrationId) {
-        this.shouldWaitForAuth = true;
-        this.authIdTokenSet.next(false);
-        const authDataObservable = !idToken || typeof idToken === 'string'
-            ? (0,external_rxjs_namespaceObject.of)({ token: idToken, integrationId })
-            : (0,external_rxjs_namespaceObject.from)(idToken).pipe((0,external_rxjs_namespaceObject.map)((idToken) => ({ token: idToken, integrationId })));
-        this.authIdTokenObservableSubject.next(authDataObservable);
+    async getAuthData() {
+        return this.authDataProvider();
     }
+    /**
+     * Returns an observable over AuthData.
+     * Deprecated: the method will be removed soon. Use getAuthData().
+     */
     observeAuthIdToken() {
-        return this.authIdTokenObservableSubject.pipe((0,external_rxjs_namespaceObject.switchMap)((o) => o), (0,external_rxjs_namespaceObject.takeUntil)(this.destructManager.observeIsDestructing()));
-    }
-    async waitForReadyState() {
-        if (this.shouldWaitForAuth && !this.destructManager.isDestructing) {
-            await (0,external_rxjs_namespaceObject.firstValueFrom)((0,external_rxjs_namespaceObject.race)(this.authIdTokenSet.asObservable().pipe((0,external_rxjs_namespaceObject.filter)(Boolean)), this.destructManager.observeIsDestructing()));
-        }
+        const authData$ = new external_rxjs_namespaceObject.ReplaySubject();
+        this.getAuthData().then(value => authData$.next(value));
+        return authData$;
     }
     getApiKey() {
         return this.apiKey;
     }
+    /**
+     * Returns the last used Auth token.
+     * Deprecated. Use: `getToken()` instead.
+     */
     getAuthToken() {
         if (this.apiKey) {
             return { type: 'ApiKey', token: this.apiKey };
         }
-        if (!this.idToken)
+        if (!this.lastUsedJwtTokenForDeprecatedCode)
             return undefined;
-        return { type: 'Bearer', token: this.idToken, integrationId: this.integrationId };
+        return {
+            type: 'Bearer',
+            token: this.lastUsedJwtTokenForDeprecatedCode,
+            integrationId: this.lastUsedIntegrationIdForDeprecatedCode,
+        };
+    }
+    async getToken() {
+        if (this.apiKey) {
+            return { type: 'ApiKey', token: this.apiKey };
+        }
+        const { token, integrationId } = await this.getAuthData();
+        if (!token)
+            return undefined;
+        return { type: 'Bearer', token, integrationId };
     }
 }
 
@@ -33199,7 +33325,7 @@ class BackendFunctionManager {
             functionName,
             paramsArrayStr: serialization_serializeObj(params),
         };
-        return (0,external_rxjs_namespaceObject.race)((0,external_rxjs_namespaceObject.from)(this.rpcManager.post('backend-function/execute', request)).pipe(map_map((response) => {
+        return (0,external_rxjs_namespaceObject.race)((0,external_rxjs_namespaceObject.from)(this.rpcManager.post('backend-function/execute', request)).pipe(map_map(response => {
             if (!response.success) {
                 throw new Error(response.payload);
             }
@@ -33236,7 +33362,7 @@ class ClientIdService {
         return this.clientIdSubject;
     }
     observeClientTooOld() {
-        return this.clientTooOldSubject.pipe((0,external_rxjs_namespaceObject.filter)((v) => v), (0,external_rxjs_namespaceObject.map)(() => undefined));
+        return this.clientTooOldSubject.pipe((0,external_rxjs_namespaceObject.filter)(v => v), (0,external_rxjs_namespaceObject.map)(() => undefined));
     }
     /**  there was a long-term disconnection of the socket */
     notifyClientTooOld() {
@@ -33249,7 +33375,7 @@ class ClientIdService {
     observeClientReadyToBeRegenerated() {
         return this.clientTooOldSubject.pipe(
         // skip the initial connection
-        (0,external_rxjs_namespaceObject.skip)(1), (0,external_rxjs_namespaceObject.filter)((v) => !v), (0,external_rxjs_namespaceObject.map)((v) => undefined));
+        (0,external_rxjs_namespaceObject.skip)(1), (0,external_rxjs_namespaceObject.filter)(v => !v), (0,external_rxjs_namespaceObject.map)(v => undefined));
     }
     getClientId() {
         return this.clientIdSubject.value;
@@ -33296,7 +33422,7 @@ class ConnectionDetails {
         this.clientIdService = clientIdService;
         this.socketManager = socketManager;
         this.isConnected = false;
-        this.socketManager.observeConnectionReady().subscribe((isConnected) => {
+        this.socketManager.observeConnectionReady().subscribe(isConnected => {
             this.isConnected = isConnected;
         });
     }
@@ -33415,7 +33541,7 @@ class DataManager {
         this.handleNotifications();
         this.startDeleteExpiredTimestampsJob();
         this.handleOrphanDocs();
-        this.outgoingMutationsEmpty.subscribe((isEmpty) => {
+        this.outgoingMutationsEmpty.subscribe(isEmpty => {
             /**
              * We  should not send queries to the server before we know that all outgoing updates were
              * applied on the server.
@@ -33528,7 +33654,7 @@ class DataManager {
                 this.docIdToLocalTimestamp.set(squidDocId, new Date().getTime());
             }
             const allClientRequestIds = this.querySubscriptionManager.setClientRequestIdsForLocalDoc(squidDocId, docAfter);
-            allClientRequestIds.forEach((clientRequestId) => this.batchClientRequestIds.add(clientRequestId));
+            allClientRequestIds.forEach(clientRequestId => this.batchClientRequestIds.add(clientRequestId));
         }, transactionId);
     }
     /** Same as runInTransaction with the exception that the passed function runs synchronously. */
@@ -33571,13 +33697,13 @@ class DataManager {
     handleNotifications() {
         this.socketManager
             .observeNotifications()
-            .pipe((0,external_rxjs_namespaceObject.filter)((notification) => notification.type === 'mutations'), map_map((n) => n))
-            .subscribe((notification) => {
+            .pipe((0,external_rxjs_namespaceObject.filter)((notification) => notification.type === 'mutations'), map_map(n => n))
+            .subscribe(notification => {
             this.outgoingMutationsEmpty.pipe((0,external_rxjs_namespaceObject.filter)(Boolean), (0,external_rxjs_namespaceObject.take)(1)).subscribe(() => {
                 this.handleIncomingMutations(notification.payload);
             });
         });
-        this.querySubscriptionManager.observeQueryResults().subscribe((queryResult) => {
+        this.querySubscriptionManager.observeQueryResults().subscribe(queryResult => {
             this.outgoingMutationsEmpty.pipe((0,external_rxjs_namespaceObject.filter)(Boolean), (0,external_rxjs_namespaceObject.take)(1)).subscribe(() => {
                 this.handleIncomingQuerySnapshots(queryResult);
             });
@@ -33612,7 +33738,7 @@ class DataManager {
             const squidDocId = getSquidDocId(doc.__docId__, query.collectionName, query.integrationId);
             squidDocIdToNewData[squidDocId] = { properties: doc, timestamp: doc.__ts__ };
         }
-        this.runInTransactionSync((transactionId) => {
+        this.runInTransactionSync(transactionId => {
             this.querySubscriptionManager.setGotInitialResult(queryResult.clientRequestId);
             this.batchClientRequestIds.add(queryResult.clientRequestId);
             const someDocumentsWereOutdated = this.applyIncomingUpdates(squidDocIdToNewData, transactionId);
@@ -33672,7 +33798,7 @@ class DataManager {
         /** The incomingUpdate was applied locally - remove it from pendingIncomingUpdates. */
         this.pendingIncomingUpdates.delete(squidDocId);
         const allClientRequestIds = this.querySubscriptionManager.setClientRequestIdsForLocalDoc(squidDocId, incomingUpdate.properties);
-        allClientRequestIds.forEach((clientRequestId) => {
+        allClientRequestIds.forEach(clientRequestId => {
             this.batchClientRequestIds.add(clientRequestId);
         });
         /**
@@ -33782,7 +33908,7 @@ class DataManager {
         try {
             await promise_pool_dist.PromisePool.for(outgoingMutationsByIntegrationId)
                 .withConcurrency(outgoingMutationsByIntegrationId.length || 1)
-                .handleError((e) => {
+                .handleError(e => {
                 throw e;
             })
                 .process(async ([integrationId, outgoingMutations]) => {
@@ -33811,9 +33937,9 @@ class DataManager {
     }
     async sendMutationsForIntegration(outgoingMutations, integrationId) {
         try {
-            const { timestamp, idResolutionMap = {}, refreshList = [], } = await this.mutationSender.sendMutations(outgoingMutations.map((outgoingMutation) => outgoingMutation.mutation), integrationId);
+            const { timestamp, idResolutionMap = {}, refreshList = [], } = await this.mutationSender.sendMutations(outgoingMutations.map(outgoingMutation => outgoingMutation.mutation), integrationId);
             this.documentIdentityService.migrate(idResolutionMap);
-            refreshList.forEach((docId) => {
+            refreshList.forEach(docId => {
                 this.refreshDocIdToTimestamp.set(idResolutionMap[docId] || docId, timestamp);
             });
             for (const outgoingMutation of outgoingMutations) {
@@ -33915,7 +34041,7 @@ class DataManager {
      * An orphan document should not stay locally since it may be stale after some time.
      */
     handleOrphanDocs() {
-        this.querySubscriptionManager.onOrphanDocuments.subscribe((orphanDocs) => {
+        this.querySubscriptionManager.onOrphanDocuments.subscribe(orphanDocs => {
             for (const squidDocId of orphanDocs) {
                 if (!this.isTracked(squidDocId)) {
                     this.forgetDocument(squidDocId);
@@ -33942,8 +34068,8 @@ class DataManager {
         this.setExpiration(squidDocId, true);
     }
     migrateDocIds(idResolutionMap) {
-        this.pendingOutgoingMutations.forEach((outgoingMutations) => {
-            outgoingMutations.forEach((outgoingMutation) => {
+        this.pendingOutgoingMutations.forEach(outgoingMutations => {
+            outgoingMutations.forEach(outgoingMutation => {
                 const squidDocId = getSquidDocId(outgoingMutation.mutation.squidDocIdObj);
                 const resolvedId = idResolutionMap[squidDocId];
                 if (resolvedId) {
@@ -34023,21 +34149,21 @@ class DistributedLockManager {
         this.ongoingLocks = {};
         this.acquireLockMessagesFromServer = this.socketManager
             .observeNotifications()
-            .pipe((0,external_rxjs_namespaceObject.filter)((message) => message.type === 'lockAcquired'));
+            .pipe((0,external_rxjs_namespaceObject.filter)(message => message.type === 'lockAcquired'));
         this.releaseLockMessagesFromServer = this.socketManager
             .observeNotifications()
-            .pipe((0,external_rxjs_namespaceObject.filter)((message) => message.type === 'lockReleased'));
+            .pipe((0,external_rxjs_namespaceObject.filter)(message => message.type === 'lockReleased'));
         // we may override this value in tests
         this.lockWaitForConnectionThreshold = 2000;
         destructManager.onPreDestruct(() => {
             this.releaseAllLocks();
         });
-        this.socketManager.observeConnectionReady().subscribe((ready) => {
+        this.socketManager.observeConnectionReady().subscribe(ready => {
             if (ready)
                 return;
             this.releaseAllLocks();
         });
-        this.releaseLockMessagesFromServer.subscribe((message) => {
+        this.releaseLockMessagesFromServer.subscribe(message => {
             const lock = this.ongoingLocks[message.payload.clientRequestId];
             if (lock === undefined)
                 return;
@@ -34072,7 +34198,7 @@ class DistributedLockManager {
                     lockId: undefined,
                 },
             };
-        })), this.acquireLockMessagesFromServer.pipe((0,external_rxjs_namespaceObject.filter)((message) => message.payload.clientRequestId === clientRequestId))));
+        })), this.acquireLockMessagesFromServer.pipe((0,external_rxjs_namespaceObject.filter)(message => message.payload.clientRequestId === clientRequestId))));
         if (!result.payload.lockId) {
             throw new Error(`Failed to acquire lock: ${result.payload.error}`);
         }
@@ -34179,7 +34305,7 @@ class DocumentReferenceFactory {
      */
     getDocumentsForCollection(integrationId, collectionName) {
         const collectionKey = this.getCollectionKey(integrationId, collectionName);
-        return (this.documentsForCollection.get(collectionKey) || []).filter((d) => d.hasData);
+        return (this.documentsForCollection.get(collectionKey) || []).filter(d => d.hasData);
     }
     /**
      * @internal
@@ -34258,18 +34384,18 @@ class DocumentStore {
         return 0;
     }
     group(sortedDocs, sortFieldNames) {
-        return Object.values(lodash_default().groupBy(sortedDocs, (doc) => {
-            return serialization_normalizeJsonAsString(sortFieldNames.map((fieldName) => getInPath(doc, fieldName)));
+        return Object.values(lodash_default().groupBy(sortedDocs, doc => {
+            return serialization_normalizeJsonAsString(sortFieldNames.map(fieldName => getInPath(doc, fieldName)));
         }));
     }
     sortAndLimitDocs(docIdSet, query) {
         if (docIdSet.size === 0) {
             return [];
         }
-        const docs = [...docIdSet].map((id) => this.squidDocIdToDoc.get(id)).filter(dist.isNonNullable);
+        const docs = [...docIdSet].map(id => this.squidDocIdToDoc.get(id)).filter(dist.isNonNullable);
         const { sortOrder, limitBy } = query;
-        const sortFieldNames = sortOrder.map((s) => s.fieldName);
-        const sortOrders = sortOrder.map((s) => (s.asc ? 'asc' : 'desc'));
+        const sortFieldNames = sortOrder.map(s => s.fieldName);
+        const sortOrders = sortOrder.map(s => (s.asc ? 'asc' : 'desc'));
         const sortedDocs = docs.sort((a, b) => {
             return this.compareSquidDocs(a, b, sortFieldNames, sortOrders);
         });
@@ -34281,10 +34407,10 @@ class DocumentStore {
         const sortedGroups = this.group(sortedDocs, fields);
         let limitedGroups;
         if (reverseSort) {
-            limitedGroups = sortedGroups.map((group) => group.slice(-internalLimit));
+            limitedGroups = sortedGroups.map(group => group.slice(-internalLimit));
         }
         else {
-            limitedGroups = sortedGroups.map((group) => group.slice(0, internalLimit));
+            limitedGroups = sortedGroups.map(group => group.slice(0, internalLimit));
         }
         return limitedGroups.flat().slice(0, mainLimit);
     }
@@ -47572,7 +47698,7 @@ class MutationSender {
     }
     async sendMutations(mutations, integrationId) {
         const reducedMutations = reduceMutations(mutations);
-        const mutexes = reducedMutations.map((mutation) => `sendMutation_${getSquidDocId(mutation.squidDocIdObj)}`);
+        const mutexes = reducedMutations.map(mutation => `sendMutation_${getSquidDocId(mutation.squidDocIdObj)}`);
         await this.lockManager.lock(...mutexes);
         await this.querySender.waitForAllQueriesToFinish();
         try {
@@ -47602,12 +47728,11 @@ class MutationSender {
 class NamedQueryManager {
     constructor(rpcManager, socketManager) {
         this.rpcManager = rpcManager;
-        this.socketManager = socketManager;
         this.ongoingNamedQueryExecutions = {};
         socketManager
             .observeNotifications()
-            .pipe((0,external_rxjs_namespaceObject.filter)((notification) => notification.type === 'namedQuery'), map_map((n) => n))
-            .subscribe((notification) => {
+            .pipe((0,external_rxjs_namespaceObject.filter)((notification) => notification.type === 'namedQuery'), map_map(n => n))
+            .subscribe(notification => {
             this.handleNamedQueryResponse(notification.clientRequestId, notification.payload);
         });
     }
@@ -47621,7 +47746,7 @@ class NamedQueryManager {
             paramsRecordStr: serialization_serializeObj(params),
             clientRequestId,
         };
-        this.rpcManager.post('named-query/execute', request).catch((e) => {
+        this.rpcManager.post('named-query/execute', request).catch(e => {
             console.error('Got error while executing named query', queryName, 'for integration', integrationId, e);
             subject.error(e);
             subject.complete();
@@ -47662,13 +47787,13 @@ class LocalQueryManager {
         const queryContext = new query_context_QueryContext(query);
         const references = this.documentReferenceFactory
             .getDocumentsForCollection(integrationId, collectionName)
-            .filter((doc) => queryContext.documentMatchesQuery(doc.data));
+            .filter(doc => queryContext.documentMatchesQuery(doc.data));
         const refMap = {};
-        references.forEach((reference) => {
+        references.forEach(reference => {
             refMap[reference.squidDocId] = reference;
         });
         const documents = this.documentStore.sortAndLimitDocs(new Set(Object.keys(refMap)), query);
-        return documents.map((doc) => refMap[getSquidDocId(doc.__docId__, collectionName, integrationId)]);
+        return documents.map(doc => refMap[getSquidDocId(doc.__docId__, collectionName, integrationId)]);
     }
 }
 
@@ -48025,7 +48150,7 @@ class StateService {
             const subscription = subject
                 .pipe(
             // If there are path to exclude - remove them
-            filter((dataAndAction) => {
+            filter(dataAndAction => {
                 if (!pathsToExcludeFn)
                     return true;
                 const pathToExcludeTrie = new PathTrie();
@@ -48033,7 +48158,7 @@ class StateService {
                 for (const pathToExclude of pathsToExclude) {
                     pathToExcludeTrie.getOrCreatePathTrieNode(pathToExclude, true);
                 }
-                const pathsInAction = extractAllPathActions(dataAndAction.action).map((a) => a.path);
+                const pathsInAction = extractAllPathActions(dataAndAction.action).map(a => a.path);
                 let affectsNonExcludedPaths = false;
                 for (const pathInAction of pathsInAction) {
                     if (pathToExcludeTrie.getNodeList(pathInAction).length === 0) {
@@ -48044,7 +48169,7 @@ class StateService {
                 return affectsNonExcludedPaths;
             }), 
             // Extract the data
-            map_map((dataAndAction) => dataAndAction.data))
+            map_map(dataAndAction => dataAndAction.data))
                 .subscribe(observer);
             return () => {
                 subscription.unsubscribe();
@@ -48258,7 +48383,7 @@ class QuerySubscriptionManager {
     setClientRequestIdsForLocalDoc(squidDocId, properties) {
         const clientRequestIdsBefore = this.localDocumentToClientRequestIds.get(squidDocId) || new Set();
         const clientRequestIdsAfter = new Set(properties
-            ? this.findQueriesForDocument(properties, squidDocId).map((querySubscriptionId) => parseQuerySubscriptionId(querySubscriptionId).clientRequestId)
+            ? this.findQueriesForDocument(properties, squidDocId).map(querySubscriptionId => parseQuerySubscriptionId(querySubscriptionId).clientRequestId)
             : []);
         const allClientRequestIds = new Set([...clientRequestIdsBefore, ...clientRequestIdsAfter]);
         for (const clientRequestIdBefore of [...clientRequestIdsBefore]) {
@@ -48302,7 +48427,12 @@ class QuerySubscriptionManager {
             const ongoingQuery = this.ongoingQueries.get(clientRequestId);
             if (!ongoingQuery)
                 continue;
-            ongoingQuery.dataSubject.error(err);
+            if (!this.destructManager.isDestructing) {
+                ongoingQuery.dataSubject.error(err);
+            }
+            else {
+                ongoingQuery.dataSubject.complete();
+            }
             ongoingQuery.done = true;
             // TODO handle joins
         }
@@ -48314,12 +48444,12 @@ class QuerySubscriptionManager {
             const ongoingQuery = this.ongoingQueries.get(clientRequestId);
             if (!ongoingQuery)
                 continue;
-            if (!ongoingQuery.gotInitialResponse || !ongoingQuery.activated)
+            if (!ongoingQuery.gotInitialResponse || !ongoingQuery.activated || ongoingQuery.isInFlight)
                 continue;
             const docIdSet = this.clientRequestIdToLocalDocuments.get(clientRequestId) || new Set();
             const result = this.documentStore.sortAndLimitDocs(docIdSet, ongoingQuery.query);
             const observablesUpdated = ongoingQuery.supportedQueries
-                .map((supportedOngoingQuery) => this.updateOngoingQueryWithNewDataFromSupportingQuery(result, supportedOngoingQuery))
+                .map(supportedOngoingQuery => this.updateOngoingQueryWithNewDataFromSupportingQuery(result, supportedOngoingQuery))
                 .some(Boolean);
             let rootOngoingQuery = ongoingQuery;
             while (!rootOngoingQuery.allObservables) {
@@ -48414,8 +48544,8 @@ class QuerySubscriptionManager {
             }
             this.sendQueryToServerOrUseParentQuery(rootOngoingQuery);
             rootOngoingQuery.allObservables = new external_rxjs_namespaceObject.ReplaySubject(1);
-            const result = rootOngoingQuery.allObservables.pipe((0,external_rxjs_namespaceObject.switchMap)((allObservables) => {
-                return (0,external_rxjs_namespaceObject.combineLatest)(allObservables).pipe(map_map((allResults) => {
+            const result = rootOngoingQuery.allObservables.pipe((0,external_rxjs_namespaceObject.switchMap)(allObservables => {
+                return (0,external_rxjs_namespaceObject.combineLatest)(allObservables).pipe(map_map(allResults => {
                     return this.joinResults(allResults, joinConditions, rootOngoingQuery);
                 }));
             }), filter(() => {
@@ -48472,7 +48602,7 @@ class QuerySubscriptionManager {
         ongoingQuery.dataSubject
             .pipe((0,external_rxjs_namespaceObject.finalize)(async () => {
             if (ongoingQuery.unsubscribeBlockerCount.value > 0) {
-                await (0,external_rxjs_namespaceObject.firstValueFrom)((0,external_rxjs_namespaceObject.race)(this.destructManager.observeIsDestructing(), ongoingQuery.unsubscribeBlockerCount.pipe(filter((count) => count === 0))));
+                await (0,external_rxjs_namespaceObject.firstValueFrom)((0,external_rxjs_namespaceObject.race)(this.destructManager.observeIsDestructing(), ongoingQuery.unsubscribeBlockerCount.pipe(filter(count => count === 0))));
             }
             this.queryMappingManager.removeQuery(querySubscriptionId).then();
             this.ongoingQueries.delete(clientRequestId);
@@ -48481,7 +48611,9 @@ class QuerySubscriptionManager {
                     const unsubscribeRequest = {
                         clientRequestId,
                     };
-                    this.rpcManager.post('query/unsubscribe', unsubscribeRequest).catch((e) => {
+                    this.rpcManager.post('query/unsubscribe', unsubscribeRequest).catch(e => {
+                        if (this.destructManager.isDestructing)
+                            return;
                         console.error('Got error while unsubscribing from query', ongoingQuery.query, e);
                     });
                 }
@@ -48544,7 +48676,7 @@ class QuerySubscriptionManager {
             return result;
         }
         const alias = ongoingQuery.alias;
-        result.push(ongoingQuery.dataSubject.pipe(filter(Boolean), map_map((docs) => {
+        result.push(ongoingQuery.dataSubject.pipe(filter(Boolean), map_map(docs => {
             return { docs, alias };
         })));
         for (const supportedQuery of ongoingQuery.supportedQueries) {
@@ -48562,7 +48694,7 @@ class QuerySubscriptionManager {
             }
             return accum;
         }, {});
-        let result = aliasToDocs[rootOngoingQuery.alias].map((doc) => ({
+        let result = aliasToDocs[rootOngoingQuery.alias].map(doc => ({
             [rootOngoingQuery.alias]: doc,
         }));
         const ongoingQueriesInOrder = this.getOngoingQueriesBfs(rootOngoingQuery);
@@ -48586,17 +48718,17 @@ class QuerySubscriptionManager {
             throw new Error('No join condition found for alias ' + rightAlias);
         }
         const rightAsMap = new Map();
-        (rightDocs || []).forEach((doc) => {
+        (rightDocs || []).forEach(doc => {
             const val = doc[joinCondition.right];
             if (!rightAsMap.has(val))
                 rightAsMap.set(val, []);
             (0,dist.truthy)(rightAsMap.get(val)).push(doc);
         });
-        return left.flatMap((leftElement) => {
+        return left.flatMap(leftElement => {
             var _a;
             const rightDocsWithSameValue = rightAsMap.get((_a = leftElement[joinCondition.leftAlias]) === null || _a === void 0 ? void 0 : _a[joinCondition.left]) || [];
             if (rightDocsWithSameValue.length) {
-                return rightDocsWithSameValue.map((rightDoc) => (Object.assign(Object.assign({}, leftElement), { [rightAlias]: rightDoc })));
+                return rightDocsWithSameValue.map(rightDoc => (Object.assign(Object.assign({}, leftElement), { [rightAlias]: rightDoc })));
             }
             else if (joinCondition.isInner) {
                 return [];
@@ -48623,7 +48755,7 @@ class QuerySubscriptionManager {
         const joinCondition = (0,dist.truthy)(supportedOngoingQuery.joinCondition);
         const query = supportedOngoingQuery.query;
         if (!supportedOngoingQuery.activated) {
-            const newConditions = supportingQueryResult.map((supportingDoc) => {
+            const newConditions = supportingQueryResult.map(supportingDoc => {
                 var _a;
                 return {
                     fieldName: joinCondition.right,
@@ -48643,13 +48775,13 @@ class QuerySubscriptionManager {
             return true;
         }
         else {
-            const supportedQueriesWithSameAlias = (0,dist.truthy)((_a = supportedOngoingQuery.supportingOngoingQuery) === null || _a === void 0 ? void 0 : _a.supportedQueries).filter((q) => q.alias === supportedOngoingQuery.alias);
-            const allNeededValues = new Set(supportingQueryResult.map((resultDoc) => { var _a; return (_a = resultDoc[joinCondition.left]) !== null && _a !== void 0 ? _a : null; }));
+            const supportedQueriesWithSameAlias = (0,dist.truthy)((_a = supportedOngoingQuery.supportingOngoingQuery) === null || _a === void 0 ? void 0 : _a.supportedQueries).filter(q => q.alias === supportedOngoingQuery.alias);
+            const allNeededValues = new Set(supportingQueryResult.map(resultDoc => { var _a; return (_a = resultDoc[joinCondition.left]) !== null && _a !== void 0 ? _a : null; }));
             for (const supportedQuery of supportedQueriesWithSameAlias) {
                 supportedQuery.query.conditions
                     .filter(isSimpleCondition)
-                    .filter((cond) => cond.fieldName === joinCondition.right)
-                    .forEach((cond) => {
+                    .filter(cond => cond.fieldName === joinCondition.right)
+                    .forEach(cond => {
                     allNeededValues.delete(cond.value);
                 });
             }
@@ -48657,8 +48789,8 @@ class QuerySubscriptionManager {
                 return false;
             }
             const newQuery = lodash.cloneDeep(query);
-            newQuery.conditions = newQuery.conditions.filter((cond) => !isSimpleCondition(cond) || cond.fieldName !== joinCondition.right);
-            [...allNeededValues].forEach((value) => {
+            newQuery.conditions = newQuery.conditions.filter(cond => !isSimpleCondition(cond) || cond.fieldName !== joinCondition.right);
+            [...allNeededValues].forEach(value => {
                 newQuery.conditions.push({
                     fieldName: joinCondition.right,
                     operator: '==',
@@ -48680,14 +48812,14 @@ class QuerySubscriptionManager {
             return false;
         if (!rootOngoingQuery.supportedQueries.length)
             return true;
-        return rootOngoingQuery.supportedQueries.every((ongoingQuery) => this.allOngoingQueriesGotInitialResult(ongoingQuery));
+        return rootOngoingQuery.supportedQueries.every(ongoingQuery => this.allOngoingQueriesGotInitialResult(ongoingQuery));
     }
     async completeAllSupportedQueries(rootOngoingQuery) {
         const supportedQueries = [...(rootOngoingQuery.supportedQueries || [])];
         while (supportedQueries.length) {
             const supportedQuery = (0,dist.truthy)(supportedQueries.shift());
             supportedQueries.push(...(supportedQuery.supportedQueries || []));
-            await (0,external_rxjs_namespaceObject.firstValueFrom)(supportedQuery.unsubscribeBlockerCount.pipe(filter((count) => count === 0)));
+            await (0,external_rxjs_namespaceObject.firstValueFrom)(supportedQuery.unsubscribeBlockerCount.pipe(filter(count => count === 0)));
             supportedQuery.dataSubject.complete();
         }
     }
@@ -48744,8 +48876,14 @@ class QuerySubscriptionManager {
             await (0,external_rxjs_namespaceObject.firstValueFrom)(parentOngoingQuery.queryRegistered.pipe(filter(Boolean)));
         }
         catch (e) {
-            ongoingQuery.dataSubject.error(e);
-            ongoingQuery.queryRegistered.error(e);
+            if (!this.destructManager.isDestructing) {
+                ongoingQuery.dataSubject.error(e);
+                ongoingQuery.queryRegistered.error(e);
+            }
+            else {
+                ongoingQuery.dataSubject.complete();
+                ongoingQuery.queryRegistered.complete();
+            }
             ongoingQuery.done = true;
             return;
         }
@@ -48757,16 +48895,22 @@ class QuerySubscriptionManager {
         this.rpcManager
             .post('query/register', queryRegisterRequest)
             .then(() => {
+            ongoingQuery.isInFlight = false;
             ongoingQuery.queryRegistered.next(true);
         })
-            .catch((e) => {
-            console.error('Query error', ongoingQuery.query, parentOngoingQuery.query, e);
-            ongoingQuery.dataSubject.error(e);
+            .catch(e => {
+            ongoingQuery.isInFlight = false;
+            if (!this.destructManager.isDestructing) {
+                console.error('Query error', ongoingQuery.query, parentOngoingQuery.query, e);
+                ongoingQuery.dataSubject.error(e);
+            }
+            else {
+                ongoingQuery.dataSubject.complete();
+            }
             ongoingQuery.done = true;
         })
             .finally(() => {
             parentOngoingQuery.unsubscribeBlockerCount.next(parentOngoingQuery.unsubscribeBlockerCount.value - 1);
-            ongoingQuery.isInFlight = false;
         });
         const takeUntilNotifier = (0,external_rxjs_namespaceObject.race)(
         /**
@@ -48782,15 +48926,22 @@ class QuerySubscriptionManager {
             if (ongoingQuery.gotInitialResponse)
                 return;
             this.setGotInitialResult(ongoingQuery.clientRequestId);
-        }), map_map((documents) => documents.filter((doc) => queryContext.documentMatchesQuery(doc))))
+        }), map_map(documents => documents.filter(doc => queryContext.documentMatchesQuery(doc))))
             .subscribe({
-            next: (results) => {
+            next: results => {
                 for (const result of results) {
                     this.setClientRequestIdsForLocalDoc(getSquidDocId(result.__docId__, ongoingQuery.query.collectionName, ongoingQuery.query.integrationId), result);
                 }
                 this.notifyAllSubscriptions([ongoingQuery.clientRequestId]);
             },
-            error: (e) => ongoingQuery.dataSubject.error(e),
+            error: e => {
+                if (!this.destructManager.isDestructing) {
+                    ongoingQuery.dataSubject.error(e);
+                }
+                else {
+                    ongoingQuery.dataSubject.complete();
+                }
+            },
         });
     }
     /**
@@ -48811,18 +48962,23 @@ class QuerySubscriptionManager {
         ongoingQuery.isInFlight = true;
         this.querySender
             .sendQuery(queryRequest)
-            .then((queryResult) => {
+            .then(queryResult => {
+            ongoingQuery.isInFlight = false;
             ongoingQuery.queryRegistered.next(true);
             this.queryResultsSubject.next(queryResult);
         })
-            .catch((e) => {
-            DebugLogger.debug('Query error', ongoingQuery.query, e);
-            ongoingQuery.dataSubject.error(e);
-            ongoingQuery.done = true;
-            ongoingQuery.queryRegistered.error('query failed');
-        })
-            .finally(() => {
+            .catch(e => {
             ongoingQuery.isInFlight = false;
+            if (!this.destructManager.isDestructing) {
+                DebugLogger.debug('Query error', ongoingQuery.query, e);
+                ongoingQuery.dataSubject.error(e);
+                ongoingQuery.queryRegistered.error('query failed');
+            }
+            else {
+                ongoingQuery.dataSubject.complete();
+                ongoingQuery.queryRegistered.complete();
+            }
+            ongoingQuery.done = true;
         });
     }
     /** naive way to refresh queries/subscriptions when we have a new client id */
@@ -48835,14 +48991,14 @@ class QuerySubscriptionManager {
     migrateDocIds(idResolutionMap) {
         const squidDocIds = Object.keys(idResolutionMap);
         for (const set of this.clientRequestIdToLocalDocuments.values()) {
-            squidDocIds.forEach((key) => {
+            squidDocIds.forEach(key => {
                 if (set.has(key)) {
                     set.delete(key);
                     set.add(idResolutionMap[key]);
                 }
             });
         }
-        squidDocIds.forEach((key) => {
+        squidDocIds.forEach(key => {
             replaceKeyInMap(this.localDocumentToClientRequestIds, key, idResolutionMap[key]);
         });
     }
@@ -49061,7 +49217,7 @@ class RpcManager {
         if (apiKey) {
             this.setStaticHeader('Authorization', `ApiKey ${apiKey}`);
         }
-        this.clientIdService.observeClientId().subscribe((clientId) => {
+        this.clientIdService.observeClientId().subscribe(clientId => {
             if (clientId) {
                 this.setStaticHeader('x-squid-clientid', clientId);
             }
@@ -49069,18 +49225,6 @@ class RpcManager {
                 this.deleteStaticHeader('x-squid-clientid');
             }
         });
-        this.authManager.observeAuthIdToken().subscribe((authData) => {
-            if (authData.token) {
-                let header = `Bearer ${authData.token}`;
-                if (authData.integrationId) {
-                    header += `; IntegrationId ${authData.integrationId}`;
-                }
-                this.setStaticHeader('Authorization', header);
-            }
-            else {
-                this.deleteStaticHeader('Authorization');
-            }
-        });
         destructManager.onDestruct(async () => {
             await this.awaitAllSettled();
         });
@@ -49091,8 +49235,18 @@ class RpcManager {
             secret: new RateLimiter(20 * rateLimiterMultiplier, 5),
         };
     }
+    async getAuthHeaders() {
+        const { token, integrationId } = await this.authManager.getAuthData();
+        if (!token)
+            return {};
+        let header = `Bearer ${token}`;
+        if (integrationId) {
+            header += `; IntegrationId ${integrationId}`;
+        }
+        return { Authorization: header };
+    }
     async awaitAllSettled() {
-        await (0,external_rxjs_namespaceObject.firstValueFrom)(this.onGoingRpcCounter.pipe((0,external_rxjs_namespaceObject.filter)((value) => value === 0)));
+        await (0,external_rxjs_namespaceObject.firstValueFrom)(this.onGoingRpcCounter.pipe((0,external_rxjs_namespaceObject.filter)(value => value === 0)));
     }
     setStaticHeader(key, value) {
         this.staticHeaders[key] = value;
@@ -49103,24 +49257,22 @@ class RpcManager {
     getStaticHeaders() {
         return this.staticHeaders;
     }
-    async ready() {
-        await this.authManager.waitForReadyState();
-    }
     async post(path, message, files = []) {
         this.onGoingRpcCounter.next(this.onGoingRpcCounter.value + 1);
         try {
             await this.getRateLimiterBucket(path).consume();
-            await (0,external_rxjs_namespaceObject.firstValueFrom)((0,external_rxjs_namespaceObject.from)(this.ready()).pipe((0,external_rxjs_namespaceObject.timeout)(20000)));
             let headers = {};
             if (Object.keys(this.staticHeaders)) {
                 headers = Object.assign(Object.assign({}, headers), this.staticHeaders);
             }
+            const authHeaders = await this.getAuthHeaders();
+            headers = Object.assign(Object.assign({}, headers), authHeaders);
             DebugLogger.debug(`sending request: path: ${path} message: ${JSON.stringify(message)}`);
             const url = getApplicationUrl(this.region, this.appId, path);
             let response;
             if (files.length) {
                 const formData = new FormData();
-                files.forEach((file) => {
+                files.forEach(file => {
                     formData.append('files', file);
                 });
                 formData.append('body', serialization_serializeObj(message));
@@ -49202,7 +49354,7 @@ class SecretClient {
         return this.rpcManager.post('/secret/getAll', {});
     }
     upsert(key, value) {
-        return this.upsertMany([{ key, value }]).then((entries) => entries[0]);
+        return this.upsertMany([{ key, value }]).then(entries => entries[0]);
     }
     upsertMany(entries) {
         const request = { entries };
@@ -49279,25 +49431,47 @@ class SocketManager {
         });
         this.setupMessageAcknowledgments();
         this.connect();
-        this.connectionReady
-            .pipe((0,external_rxjs_namespaceObject.skip)(1), (0,external_rxjs_namespaceObject.filter)((v) => !v), (0,external_rxjs_namespaceObject.switchMap)(() => {
+        this.lastTick = new Date();
+        this.tickInterval = setInterval(() => this.tick(), 5000);
+        this.observeConnectionReady()
+            .pipe((0,external_rxjs_namespaceObject.skip)(1), (0,external_rxjs_namespaceObject.filter)(v => !v), (0,external_rxjs_namespaceObject.switchMap)(() => {
             return (0,external_rxjs_namespaceObject.race)((0,external_rxjs_namespaceObject.timer)(this.clientTooOldThreshold), this.connectionReady.pipe((0,external_rxjs_namespaceObject.filter)(Boolean)), this.destructManager.observeIsDestructing());
         }))
             .subscribe(() => {
-            if (this.connectionReady.value || this.destructManager.isDestructing) {
-                DebugLogger.debug('Client too old but already connected or the client is destructing. Ignoring...');
+            if (this.connectionReady.value) {
+                DebugLogger.debug('Client reconnected before becoming too old. Ignoring...');
                 return;
             }
-            this.clientIdService.notifyClientTooOld();
-            DebugLogger.debug('Client too old. Reconnecting...');
-            this.connect();
+            this.refreshClient();
         });
-        this.connectionReady.pipe((0,external_rxjs_namespaceObject.filter)(Boolean)).subscribe(() => {
+        this.observeConnectionReady()
+            .pipe((0,external_rxjs_namespaceObject.filter)(Boolean))
+            .subscribe(() => {
             if (this.clientIdService.isClientTooOld()) {
                 this.clientIdService.notifyClientReadyToBeRegenerated();
             }
         });
     }
+    refreshClient() {
+        if (this.destructManager.isDestructing) {
+            DebugLogger.debug('Client too old but is destructed. Ignoring...');
+            return;
+        }
+        else if (this.clientIdService.isClientTooOld()) {
+            DebugLogger.debug('Client is already marked as too old. Ignoring...');
+            return;
+        }
+        this.clientIdService.notifyClientTooOld();
+        DebugLogger.debug('Client too old. Reconnecting...');
+        this.connect();
+    }
+    tick() {
+        const diff = Math.abs(Date.now() - this.lastTick.getTime());
+        if (diff > this.clientTooOldThreshold) {
+            this.refreshClient();
+        }
+        this.lastTick = new Date();
+    }
     observeNotifications() {
         return this.webSocketObserver.asObservable();
     }
@@ -49305,12 +49479,39 @@ class SocketManager {
         return this.connectionReady.asObservable().pipe((0,external_rxjs_namespaceObject.distinctUntilChanged)());
     }
     sendMessage(message) {
-        this.authManager.waitForReadyState().then(() => {
-            (0,external_rxjs_namespaceObject.firstValueFrom)(this.connectionReady.pipe((0,external_rxjs_namespaceObject.filter)(Boolean))).then(() => {
-                const authToken = this.authManager.getAuthToken();
-                (0,dist.truthy)(this.socket).send(serialization_serializeObj({ message, authToken }));
-            });
-        });
+        this.sendMessageAsync(message).then();
+    }
+    async sendMessageAsync(message) {
+        var _a;
+        await (0,external_rxjs_namespaceObject.firstValueFrom)(this.connectionReady.pipe((0,external_rxjs_namespaceObject.filter)(Boolean)));
+        const authToken = await this.authManager.getToken();
+        if (!this.connectionReady.value) {
+            // Connection state was changed during `await this.authManager.getToken()` call. Retry.
+            await this.sendMessageAsync(message);
+            return;
+        }
+        try {
+            (0,dist.assertTruthy)(this.socket, 'Socket is undefined in sendMessageAsync');
+            this.socket.send(serialization_serializeObj({ message, authToken }));
+        }
+        catch (e) {
+            if (!((_a = this.socket) === null || _a === void 0 ? void 0 : _a.connected)) {
+                // Message sending is failed due to the websocket IO error. Retry.
+                this.connectionReady.next(false);
+                await this.sendMessageAsync(message);
+            }
+            else {
+                console.error('Websocket message is ignored due to a non-recoverable error', e);
+            }
+        }
+    }
+    /**  Sends 'kill' message ignoring 'connectionReady' observable. */
+    sendKillMessage() {
+        var _a;
+        if (!((_a = this.socket) === null || _a === void 0 ? void 0 : _a.connected))
+            return;
+        const message = { type: 'kill' };
+        this.socket.send(serialization_serializeObj({ message }));
     }
     connect() {
         var _a;
@@ -49337,6 +49538,11 @@ class SocketManager {
             onerror: (e) => console.error('WebSocket error:', e),
         });
     }
+    disconnect() {
+        var _a;
+        this.connectionReady.next(false);
+        (_a = this.socket) === null || _a === void 0 ? void 0 : _a.close(4998);
+    }
     onConnectionReady() {
         this.connectionReady.next(true);
         this.sendMessage({ type: 'catchup' });
@@ -49369,7 +49575,7 @@ class SocketManager {
         });
         const collectedMessageIds = [];
         ackSubject
-            .pipe((0,external_rxjs_namespaceObject.tap)((messageId) => collectedMessageIds.push(messageId)), (0,external_rxjs_namespaceObject.debounceTime)(100))
+            .pipe((0,external_rxjs_namespaceObject.tap)(messageId => collectedMessageIds.push(messageId)), (0,external_rxjs_namespaceObject.debounceTime)(100))
             .subscribe(async () => {
             const messageIds = [...collectedMessageIds.splice(0)];
             this.sendMessage({ type: 'acknowledge', payload: messageIds });
@@ -49377,10 +49583,15 @@ class SocketManager {
     }
     async destruct() {
         var _a;
-        this.sendMessage({ type: 'kill' });
+        this.sendKillMessage();
+        // Wait until VM processes socket message before closing the socket.
+        // Without the sleep below, there is a race condition between socket closing
+        // and 'kill' message sending, so 'kill' may not be delivered at all.
+        await (0,external_rxjs_namespaceObject.firstValueFrom)((0,external_rxjs_namespaceObject.timer)(0));
         this.connectionReady.next(false);
         // Allow the message to be sent by waiting on a 0 timer.
         await (0,external_rxjs_namespaceObject.firstValueFrom)((0,external_rxjs_namespaceObject.timer)(0));
+        clearInterval(this.tickInterval);
         (_a = this.socket) === null || _a === void 0 ? void 0 : _a.close();
         this.webSocketObserver.complete();
         this.allMessagesObserver.complete();
@@ -49454,7 +49665,7 @@ class QuerySender {
             }
         }
         catch (e) {
-            responseSubjects.forEach((responseSubject) => responseSubject.error(e));
+            responseSubjects.forEach(responseSubject => responseSubject.error(e));
         }
         finally {
             this.inflightQueriesCount.next(this.inflightQueriesCount.value - pendingQueryRequests.length);
@@ -49462,7 +49673,7 @@ class QuerySender {
     }
     /** Will resolve once all the in-flight queries are done. */
     async waitForAllQueriesToFinish() {
-        return (0,external_rxjs_namespaceObject.firstValueFrom)(this.inflightQueriesCount.pipe(filter((count) => count === 0))).then(() => {
+        return (0,external_rxjs_namespaceObject.firstValueFrom)(this.inflightQueriesCount.pipe(filter(count => count === 0))).then(() => {
             return undefined;
         });
     }
@@ -49472,6 +49683,17 @@ class QuerySender {
     }
 }
 
+;// CONCATENATED MODULE: ./src/native-query-manager.ts
+class NativeQueryManager {
+    constructor(rpcManager) {
+        this.rpcManager = rpcManager;
+    }
+    async executeNativeQuery(integrationId, request) {
+        const executeRequest = Object.assign({ integrationId }, request);
+        return this.rpcManager.post('native-query/execute', executeRequest);
+    }
+}
+
 ;// CONCATENATED MODULE: ./src/squid.ts
 
 
@@ -49497,6 +49719,7 @@ class QuerySender {
 
 
 
+
 
 
 /**
@@ -49517,16 +49740,19 @@ class Squid {
         // Note: The following methods are bound using arrow functions to ensure that if a user accesses the methods via
         // destructuring (i.e. `{ setAuthIdToken } = squid`) then `this` will still be bound to the Squid class.
         /**
-         * Sets the auth id token (OpenId) that will be sent to the server and will be used for providing the `auth` object
-         * to the security rules.
+         * Sets the authorization access token (OAuth2.0) that will be sent to the server and will be used for providing the
+         * `auth` object to the security rules.
+         *
+         * @deprecated: pass `AuthTokenProvider` in the constructor of the class.
          *
-         * @param idToken The auth id token, a promise that resolves with the id token, or an observable that emits the id
-         * token. Undefined if the user is not authenticated.
+         * @param accessToken The OAuth2.0 access token or a promise that resolves with the access token.
+         * When undefined, no authorization information is sent.
+         * The Observable variant of the parameter is deprecated and will be removed soon.
          * @param integrationId The id of the integration.
          * @returns void
          */
-        this.setAuthIdToken = (idToken, integrationId) => {
-            this.authManager.setAuthIdToken(idToken, integrationId);
+        this.setAuthIdToken = (accessToken, integrationId) => {
+            this.authManager.setAuthIdToken(accessToken, integrationId);
         };
         /**
          * Returns a reference to the collection in the provided integration.
@@ -49592,6 +49818,22 @@ class Squid {
             this.validateNotDestructed();
             return (0,external_rxjs_namespaceObject.firstValueFrom)(this.namedQueryManager.executeNamedQueryAndSubscribe(integrationId, queryName, params));
         };
+        /**
+         * Executes a native relational query with the given parameters and returns a promise with the result.
+         *
+         * Native queries allow you to execute raw SQL or other database-specific queries directly against the database.
+         * This can be useful when you need to perform operations that are not easily accomplished with named queries or
+         * other high-level abstractions.
+         *
+         * @param integrationId The id of the integration that the query is associated with.
+         * @param query The raw SQL or other database-specific query to execute.
+         * @param params (Optional) The parameters to pass to the query. Defaults to an empty object.
+         * @returns A promise that resolves with the result of the query.
+         * @type {Promise<Array<SquidDocument>>}
+         */
+        this.executeNativeRelationalQuery = (integrationId, query, params = {}) => {
+            return this.nativeQueryManager.executeNativeQuery(integrationId, { type: 'relational', query, params });
+        };
         /**
          * Invokes the given HTTP API (defined by the integration ID and the endpoint ID) with the given request parameters
          * and returns a promise with the response. The structure of the request and the response is defined in the
@@ -49632,6 +49874,10 @@ class Squid {
             this.validateNotDestructed();
             return {
                 assistant: (integrationId) => this.aiClientFactory.getAssistant(integrationId),
+                executeAiQuery: (integrationId, prompt) => {
+                    const req = { integrationId, prompt };
+                    return this.rpcManager.post('aiData/executeAiQuery', req);
+                },
             };
         };
         /**
@@ -49679,7 +49925,7 @@ class Squid {
         const appId = appIdWithEnvironmentIdAndDevId(options.appId, options.environmentId, shouldAddDeveloperId ? options.squidDeveloperId : undefined);
         const httpHeaders = getApplicationHttpHeaders(options.region, appId);
         this.clientIdService = new ClientIdService(this.destructManager);
-        this.authManager = new AuthManager(this.destructManager, options.apiKey);
+        this.authManager = new AuthManager(options.apiKey, options.authTokenProvider);
         this.socketManager = new SocketManager(this.clientIdService, options.region, appId, options.messageNotificationWrapper, this.destructManager, this.authManager);
         this.rpcManager = new RpcManager(options.region, appId, this.destructManager, httpHeaders, this.authManager, this.clientIdService);
         this.documentStore = new DocumentStore();
@@ -49697,6 +49943,7 @@ class Squid {
         this.documentReferenceFactory.setDataManager(this.dataManager);
         this.backendFunctionManager = new BackendFunctionManager(this.clientIdService, this.rpcManager);
         this.namedQueryManager = new NamedQueryManager(this.rpcManager, this.socketManager);
+        this.nativeQueryManager = new NativeQueryManager(this.rpcManager);
         this.apiManager = new ApiManager(this.clientIdService, this.rpcManager, options.apiServerUrlOverrideMapping);
         this.graphqlClientFactory = new GraphQLClientFactory(this.rpcManager, options.region, appId);
         this.aiClientFactory = new AiClientFactory(this.rpcManager, this.socketManager);