@azure/cosmos 4.9.1 → 4.9.3

package/dist/commonjs/PartitionKeyRangeFailoverInfo.js

@@ -1,138 +1,153 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.PartitionKeyRangeFailoverInfo = void 0;
-const tslib_1 = require("tslib");
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT License.
-const index_js_1 = require("./index.js");
-const semaphore_1 = tslib_1.__importDefault(require("semaphore"));
-/**
- * @hidden
- * This class manages the failover information for partition key ranges in Cosmos DB.
- * It tracks the current endpoint, failed endpoints, and the number of consecutive read/write request failures.
- */
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var PartitionKeyRangeFailoverInfo_exports = {};
+__export(PartitionKeyRangeFailoverInfo_exports, {
+  PartitionKeyRangeFailoverInfo: () => PartitionKeyRangeFailoverInfo
+});
+module.exports = __toCommonJS(PartitionKeyRangeFailoverInfo_exports);
+var import_index = require("./index.js");
+var import_semaphore = __toESM(require("semaphore"));
 class PartitionKeyRangeFailoverInfo {
-    failedEndPoints = [];
-    currentEndPoint;
-    consecutiveReadRequestFailureCount = 0;
-    consecutiveWriteRequestFailureCount = 0;
-    firstRequestFailureTime = Date.now();
-    lastRequestFailureTime = Date.now();
-    failureCountSemaphore;
-    tryMoveNextLocationSemaphore;
-    /**
-     * @internal
-     */
-    constructor(currentEndpoint) {
-        this.currentEndPoint = currentEndpoint;
-        this.failureCountSemaphore = (0, semaphore_1.default)(1);
-        this.tryMoveNextLocationSemaphore = (0, semaphore_1.default)(1);
-    }
-    /**
-     * Checks if the circuit breaker can trigger a partition failover based on the failure counts.
-     * Returns true if the number of consecutive failures exceeds the defined thresholds for read or write requests.
-     */
-    async canCircuitBreakerTriggerPartitionFailOver(isReadOnlyRequest) {
-        const { consecutiveReadRequestFailureCount, consecutiveWriteRequestFailureCount } = await this.snapshotConsecutiveRequestFailureCount();
-        return isReadOnlyRequest
-            ? consecutiveReadRequestFailureCount > index_js_1.Constants.ReadRequestFailureCountThreshold
-            : consecutiveWriteRequestFailureCount > index_js_1.Constants.WriteRequestFailureCountThreshold;
-    }
-    /**
-     * Increments the failure counts for read or write requests and updates the timestamps.
-     * If the time since the last failure exceeds the reset window, it resets the failure counts.
-     */
-    async incrementRequestFailureCounts(isReadOnlyRequest, currentTimeInMilliseconds) {
-        return new Promise((resolve, reject) => {
-            this.failureCountSemaphore.take(async () => {
-                try {
-                    const { lastRequestFailureTime } = await this.snapshotPartitionFailoverTimestamps();
-                    if (currentTimeInMilliseconds - lastRequestFailureTime >
-                        index_js_1.Constants.ConsecutiveFailureCountResetIntervalInMS) {
-                        this.consecutiveReadRequestFailureCount = 0;
-                        this.consecutiveWriteRequestFailureCount = 0;
-                    }
-                    if (isReadOnlyRequest) {
-                        this.consecutiveReadRequestFailureCount++;
-                    }
-                    else {
-                        this.consecutiveWriteRequestFailureCount++;
-                    }
-                    this.lastRequestFailureTime = currentTimeInMilliseconds;
-                    return resolve();
-                }
-                catch (error) {
-                    reject(error);
-                }
-                finally {
-                    // Release the semaphore lock
-                    this.failureCountSemaphore.leave();
-                }
-            });
-        });
-    }
-    /**
-     * Returns a snapshot of the first and last request failure timestamps.
-     * This method is used to retrieve the current state of failure timestamps without modifying them.
-     */
-    async snapshotPartitionFailoverTimestamps() {
-        return {
-            firstRequestFailureTime: this.firstRequestFailureTime,
-            lastRequestFailureTime: this.lastRequestFailureTime,
-        };
-    }
-    /**
-     * Attempts to move to the next available location for the partition key range.
-     * If the current endpoint is the same as the failed endpoint, it tries to find a new endpoint
-     * from the provided list of endpoints. If a new endpoint is found, it updates the current endpoint
-     * and returns true. If no new endpoint is found, it returns false.
-     */
-    async tryMoveNextLocation(endPoints, failedEndPoint, diagnosticNode, partitionKeyRangeId) {
-        if (failedEndPoint !== this.currentEndPoint) {
-            return true;
+  failedEndPoints = [];
+  currentEndPoint;
+  consecutiveReadRequestFailureCount = 0;
+  consecutiveWriteRequestFailureCount = 0;
+  firstRequestFailureTime = Date.now();
+  lastRequestFailureTime = Date.now();
+  failureCountSemaphore;
+  tryMoveNextLocationSemaphore;
+  /**
+   * @internal
+   */
+  constructor(currentEndpoint) {
+    this.currentEndPoint = currentEndpoint;
+    this.failureCountSemaphore = (0, import_semaphore.default)(1);
+    this.tryMoveNextLocationSemaphore = (0, import_semaphore.default)(1);
+  }
+  /**
+   * Checks if the circuit breaker can trigger a partition failover based on the failure counts.
+   * Returns true if the number of consecutive failures exceeds the defined thresholds for read or write requests.
+   */
+  async canCircuitBreakerTriggerPartitionFailOver(isReadOnlyRequest) {
+    const { consecutiveReadRequestFailureCount, consecutiveWriteRequestFailureCount } = await this.snapshotConsecutiveRequestFailureCount();
+    return isReadOnlyRequest ? consecutiveReadRequestFailureCount > import_index.Constants.ReadRequestFailureCountThreshold : consecutiveWriteRequestFailureCount > import_index.Constants.WriteRequestFailureCountThreshold;
+  }
+  /**
+   * Increments the failure counts for read or write requests and updates the timestamps.
+   * If the time since the last failure exceeds the reset window, it resets the failure counts.
+   */
+  async incrementRequestFailureCounts(isReadOnlyRequest, currentTimeInMilliseconds) {
+    return new Promise((resolve, reject) => {
+      this.failureCountSemaphore.take(async () => {
+        try {
+          const { lastRequestFailureTime } = await this.snapshotPartitionFailoverTimestamps();
+          if (currentTimeInMilliseconds - lastRequestFailureTime > import_index.Constants.ConsecutiveFailureCountResetIntervalInMS) {
+            this.consecutiveReadRequestFailureCount = 0;
+            this.consecutiveWriteRequestFailureCount = 0;
+          }
+          if (isReadOnlyRequest) {
+            this.consecutiveReadRequestFailureCount++;
+          } else {
+            this.consecutiveWriteRequestFailureCount++;
+          }
+          this.lastRequestFailureTime = currentTimeInMilliseconds;
+          return resolve();
+        } catch (error) {
+          reject(error);
+        } finally {
+          this.failureCountSemaphore.leave();
         }
-        return new Promise((resolve, reject) => {
-            this.tryMoveNextLocationSemaphore.take(() => {
-                try {
-                    for (const endpoint of endPoints) {
-                        if (this.currentEndPoint === endpoint) {
-                            continue;
-                        }
-                        if (this.failedEndPoints.includes(endpoint)) {
-                            continue;
-                        }
-                        this.failedEndPoints.push(failedEndPoint);
-                        this.currentEndPoint = endpoint;
-                        return resolve(true);
-                    }
-                    diagnosticNode.addData({
-                        partitionKeyRangeFailoverInfo: `PartitionKeyRangeId: ${partitionKeyRangeId}, failedLocations: ${this.failedEndPoints}, newLocation: ${this.currentEndPoint}`,
-                    });
-                    return resolve(false);
-                }
-                catch (err) {
-                    reject(err);
-                }
-                finally {
-                    this.tryMoveNextLocationSemaphore.leave();
-                }
-            });
-        });
-    }
-    /** Returns the current endpoint being used for partition key range operations.*/
-    getCurrentEndPoint() {
-        return this.currentEndPoint;
-    }
-    /**
-     * Returns a snapshot of the current consecutive request failure counts for read and write requests.
-     * This method is used to retrieve the current state of failure counts without modifying them.
-     */
-    async snapshotConsecutiveRequestFailureCount() {
-        return {
-            consecutiveReadRequestFailureCount: this.consecutiveReadRequestFailureCount,
-            consecutiveWriteRequestFailureCount: this.consecutiveWriteRequestFailureCount,
-        };
+      });
+    });
+  }
+  /**
+   * Returns a snapshot of the first and last request failure timestamps.
+   * This method is used to retrieve the current state of failure timestamps without modifying them.
+   */
+  async snapshotPartitionFailoverTimestamps() {
+    return {
+      firstRequestFailureTime: this.firstRequestFailureTime,
+      lastRequestFailureTime: this.lastRequestFailureTime
+    };
+  }
+  /**
+   * Attempts to move to the next available location for the partition key range.
+   * If the current endpoint is the same as the failed endpoint, it tries to find a new endpoint
+   * from the provided list of endpoints. If a new endpoint is found, it updates the current endpoint
+   * and returns true. If no new endpoint is found, it returns false.
+   */
+  async tryMoveNextLocation(endPoints, failedEndPoint, diagnosticNode, partitionKeyRangeId) {
+    if (failedEndPoint !== this.currentEndPoint) {
+      return true;
     }
+    return new Promise((resolve, reject) => {
+      this.tryMoveNextLocationSemaphore.take(() => {
+        try {
+          for (const endpoint of endPoints) {
+            if (this.currentEndPoint === endpoint) {
+              continue;
+            }
+            if (this.failedEndPoints.includes(endpoint)) {
+              continue;
+            }
+            this.failedEndPoints.push(failedEndPoint);
+            this.currentEndPoint = endpoint;
+            return resolve(true);
+          }
+          diagnosticNode.addData({
+            partitionKeyRangeFailoverInfo: `PartitionKeyRangeId: ${partitionKeyRangeId}, failedLocations: ${this.failedEndPoints}, newLocation: ${this.currentEndPoint}`
+          });
+          return resolve(false);
+        } catch (err) {
+          reject(err);
+        } finally {
+          this.tryMoveNextLocationSemaphore.leave();
+        }
+      });
+    });
+  }
+  /** Returns the current endpoint being used for partition key range operations.*/
+  getCurrentEndPoint() {
+    return this.currentEndPoint;
+  }
+  /**
+   * Returns a snapshot of the current consecutive request failure counts for read and write requests.
+   * This method is used to retrieve the current state of failure counts without modifying them.
+   */
+  async snapshotConsecutiveRequestFailureCount() {
+    return {
+      consecutiveReadRequestFailureCount: this.consecutiveReadRequestFailureCount,
+      consecutiveWriteRequestFailureCount: this.consecutiveWriteRequestFailureCount
+    };
+  }
 }
-exports.PartitionKeyRangeFailoverInfo = PartitionKeyRangeFailoverInfo;
-//# sourceMappingURL=PartitionKeyRangeFailoverInfo.js.map
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  PartitionKeyRangeFailoverInfo
+});
+//# sourceMappingURL=PartitionKeyRangeFailoverInfo.js.map

package/dist/commonjs/PartitionKeyRangeFailoverInfo.js.map

@@ -1 +1,7 @@
-{"version":3,"file":"PartitionKeyRangeFailoverInfo.js","sourceRoot":"","sources":["../../src/PartitionKeyRangeFailoverInfo.ts"],"names":[],"mappings":";;;;AAAA,uCAAuC;AACvC,kCAAkC;AAClC,yCAA+D;AAC/D,kEAAkC;AAElC;;;;GAIG;AACH,MAAa,6BAA6B;IAChC,eAAe,GAAa,EAAE,CAAC;IAC/B,eAAe,CAAS;IAExB,kCAAkC,GAAW,CAAC,CAAC;IAC/C,mCAAmC,GAAW,CAAC,CAAC;IAChD,uBAAuB,GAAW,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7C,sBAAsB,GAAW,IAAI,CAAC,GAAG,EAAE,CAAC;IAE5C,qBAAqB,CAAsB;IAC3C,4BAA4B,CAAsB;IAE1D;;OAEG;IACH,YAAY,eAAuB;QACjC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,qBAAqB,GAAG,IAAA,mBAAS,EAAC,CAAC,CAAC,CAAC;QAC1C,IAAI,CAAC,4BAA4B,GAAG,IAAA,mBAAS,EAAC,CAAC,CAAC,CAAC;IACnD,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,yCAAyC,CACpD,iBAA0B;QAE1B,MAAM,EAAE,kCAAkC,EAAE,mCAAmC,EAAE,GAC/E,MAAM,IAAI,CAAC,sCAAsC,EAAE,CAAC;QAEtD,OAAO,iBAAiB;YACtB,CAAC,CAAC,kCAAkC,GAAG,oBAAS,CAAC,gCAAgC;YACjF,CAAC,CAAC,mCAAmC,GAAG,oBAAS,CAAC,iCAAiC,CAAC;IACxF,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,6BAA6B,CACxC,iBAA0B,EAC1B,yBAAiC;QAEjC,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC3C,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;gBACzC,IAAI,CAAC;oBACH,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;oBAEpF,IACE,yBAAyB,GAAG,sBAAsB;wBAClD,oBAAS,CAAC,wCAAwC,EAClD,CAAC;wBACD,IAAI,CAAC,kCAAkC,GAAG,CAAC,CAAC;wBAC5C,IAAI,CAAC,mCAAmC,GAAG,CAAC,CAAC;oBAC/C,CAAC;oBAED,IAAI,iBAAiB,EAAE,CAAC;wBACtB,IAAI,CAAC,kCAAkC,EAAE,CAAC;oBAC5C,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,mCAAmC,EAAE,CAAC;oBAC7C,CAAC;oBACD,IAAI,CAAC,sBAAsB,GAAG,yBAAyB,CAAC;oBACxD,OAAO,OAAO,EAAE,CAAC;gBACnB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChB,CAAC;wBAAS,CAAC;oBACT,6BAA6B;oBAC7B,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC;gBACrC,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,mCAAmC;QAI9C,OAAO;YACL,uBAAuB,EAAE,IAAI,CAAC,uBAAuB;YACrD,sBAAsB,EAAE,IAAI,CAAC,sBAAsB;SACpD,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,mBAAmB,CAC9B,SAA4B,EAC5B,cAAsB,EACtB,cAAsC,EACtC,mBAA2B;QAE3B,IAAI,cAAc,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;YAC5C,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC,GAAG,EAAE;gBAC1C,IAAI,CAAC;oBACH,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;wBACjC,IAAI,IAAI,CAAC,eAAe,KAAK,QAAQ,EAAE,CAAC;4BACtC,SAAS;wBACX,CAAC;wBAED,IAAI,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;4BAC5C,SAAS;wBACX,CAAC;wBAED,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;wBAC1C,IAAI,CAAC,eAAe,GAAG,QAAQ,CAAC;wBAChC,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;oBACvB,CAAC;oBACD,cAAc,CAAC,OAAO,CAAC;wBACrB,6BAA6B,EAAE,wBAAwB,mBAAmB,sBAAsB,IAAI,CAAC,eAAe,kBAAkB,IAAI,CAAC,eAAe,EAAE;qBAC7J,CAAC,CAAC;oBACH,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC;gBACxB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,GAAG,CAAC,CAAC;gBACd,CAAC;wBAAS,CAAC;oBACT,IAAI,CAAC,4BAA4B,CAAC,KAAK,EAAE,CAAC;gBAC5C,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,iFAAiF;IAC1E,kBAAkB;QACvB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,sCAAsC;QAIlD,OAAO;YACL,kCAAkC,EAAE,IAAI,CAAC,kCAAkC;YAC3E,mCAAmC,EAAE,IAAI,CAAC,mCAAmC;SAC9E,CAAC;IACJ,CAAC;CACF;AAtJD,sEAsJC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport { Constants, DiagnosticNodeInternal } from \"./index.js\";\nimport semaphore from \"semaphore\";\n\n/**\n * @hidden\n * This class manages the failover information for partition key ranges in Cosmos DB.\n * It tracks the current endpoint, failed endpoints, and the number of consecutive read/write request failures.\n */\nexport class PartitionKeyRangeFailoverInfo {\n  private failedEndPoints: string[] = [];\n  private currentEndPoint: string;\n\n  private consecutiveReadRequestFailureCount: number = 0;\n  private consecutiveWriteRequestFailureCount: number = 0;\n  private firstRequestFailureTime: number = Date.now();\n  private lastRequestFailureTime: number = Date.now();\n\n  private failureCountSemaphore: semaphore.Semaphore;\n  private tryMoveNextLocationSemaphore: semaphore.Semaphore;\n\n  /**\n   * @internal\n   */\n  constructor(currentEndpoint: string) {\n    this.currentEndPoint = currentEndpoint;\n    this.failureCountSemaphore = semaphore(1);\n    this.tryMoveNextLocationSemaphore = semaphore(1);\n  }\n\n  /**\n   * Checks if the circuit breaker can trigger a partition failover based on the failure counts.\n   * Returns true if the number of consecutive failures exceeds the defined thresholds for read or write requests.\n   */\n  public async canCircuitBreakerTriggerPartitionFailOver(\n    isReadOnlyRequest: boolean,\n  ): Promise<boolean> {\n    const { consecutiveReadRequestFailureCount, consecutiveWriteRequestFailureCount } =\n      await this.snapshotConsecutiveRequestFailureCount();\n\n    return isReadOnlyRequest\n      ? consecutiveReadRequestFailureCount > Constants.ReadRequestFailureCountThreshold\n      : consecutiveWriteRequestFailureCount > Constants.WriteRequestFailureCountThreshold;\n  }\n\n  /**\n   * Increments the failure counts for read or write requests and updates the timestamps.\n   * If the time since the last failure exceeds the reset window, it resets the failure counts.\n   */\n  public async incrementRequestFailureCounts(\n    isReadOnlyRequest: boolean,\n    currentTimeInMilliseconds: number,\n  ): Promise<void> {\n    return new Promise<void>((resolve, reject) => {\n      this.failureCountSemaphore.take(async () => {\n        try {\n          const { lastRequestFailureTime } = await this.snapshotPartitionFailoverTimestamps();\n\n          if (\n            currentTimeInMilliseconds - lastRequestFailureTime >\n            Constants.ConsecutiveFailureCountResetIntervalInMS\n          ) {\n            this.consecutiveReadRequestFailureCount = 0;\n            this.consecutiveWriteRequestFailureCount = 0;\n          }\n\n          if (isReadOnlyRequest) {\n            this.consecutiveReadRequestFailureCount++;\n          } else {\n            this.consecutiveWriteRequestFailureCount++;\n          }\n          this.lastRequestFailureTime = currentTimeInMilliseconds;\n          return resolve();\n        } catch (error) {\n          reject(error);\n        } finally {\n          // Release the semaphore lock\n          this.failureCountSemaphore.leave();\n        }\n      });\n    });\n  }\n\n  /**\n   * Returns a snapshot of the first and last request failure timestamps.\n   * This method is used to retrieve the current state of failure timestamps without modifying them.\n   */\n  public async snapshotPartitionFailoverTimestamps(): Promise<{\n    firstRequestFailureTime: number;\n    lastRequestFailureTime: number;\n  }> {\n    return {\n      firstRequestFailureTime: this.firstRequestFailureTime,\n      lastRequestFailureTime: this.lastRequestFailureTime,\n    };\n  }\n\n  /**\n   * Attempts to move to the next available location for the partition key range.\n   * If the current endpoint is the same as the failed endpoint, it tries to find a new endpoint\n   * from the provided list of endpoints. If a new endpoint is found, it updates the current endpoint\n   * and returns true. If no new endpoint is found, it returns false.\n   */\n  public async tryMoveNextLocation(\n    endPoints: readonly string[],\n    failedEndPoint: string,\n    diagnosticNode: DiagnosticNodeInternal,\n    partitionKeyRangeId: string,\n  ): Promise<boolean> {\n    if (failedEndPoint !== this.currentEndPoint) {\n      return true;\n    }\n    return new Promise((resolve, reject) => {\n      this.tryMoveNextLocationSemaphore.take(() => {\n        try {\n          for (const endpoint of endPoints) {\n            if (this.currentEndPoint === endpoint) {\n              continue;\n            }\n\n            if (this.failedEndPoints.includes(endpoint)) {\n              continue;\n            }\n\n            this.failedEndPoints.push(failedEndPoint);\n            this.currentEndPoint = endpoint;\n            return resolve(true);\n          }\n          diagnosticNode.addData({\n            partitionKeyRangeFailoverInfo: `PartitionKeyRangeId: ${partitionKeyRangeId}, failedLocations: ${this.failedEndPoints}, newLocation: ${this.currentEndPoint}`,\n          });\n          return resolve(false);\n        } catch (err) {\n          reject(err);\n        } finally {\n          this.tryMoveNextLocationSemaphore.leave();\n        }\n      });\n    });\n  }\n\n  /** Returns the current endpoint being used for partition key range operations.*/\n  public getCurrentEndPoint(): string {\n    return this.currentEndPoint;\n  }\n\n  /**\n   * Returns a snapshot of the current consecutive request failure counts for read and write requests.\n   * This method is used to retrieve the current state of failure counts without modifying them.\n   */\n  private async snapshotConsecutiveRequestFailureCount(): Promise<{\n    consecutiveReadRequestFailureCount: number;\n    consecutiveWriteRequestFailureCount: number;\n  }> {\n    return {\n      consecutiveReadRequestFailureCount: this.consecutiveReadRequestFailureCount,\n      consecutiveWriteRequestFailureCount: this.consecutiveWriteRequestFailureCount,\n    };\n  }\n}\n"]}
+{
+  "version": 3,
+  "sources": ["../../src/PartitionKeyRangeFailoverInfo.ts"],
+  "sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport type { DiagnosticNodeInternal } from \"./index.js\";\nimport { Constants } from \"./index.js\";\nimport semaphore from \"semaphore\";\n\n/**\n * @hidden\n * This class manages the failover information for partition key ranges in Cosmos DB.\n * It tracks the current endpoint, failed endpoints, and the number of consecutive read/write request failures.\n */\nexport class PartitionKeyRangeFailoverInfo {\n  private failedEndPoints: string[] = [];\n  private currentEndPoint: string;\n\n  private consecutiveReadRequestFailureCount: number = 0;\n  private consecutiveWriteRequestFailureCount: number = 0;\n  private firstRequestFailureTime: number = Date.now();\n  private lastRequestFailureTime: number = Date.now();\n\n  private failureCountSemaphore: semaphore.Semaphore;\n  private tryMoveNextLocationSemaphore: semaphore.Semaphore;\n\n  /**\n   * @internal\n   */\n  constructor(currentEndpoint: string) {\n    this.currentEndPoint = currentEndpoint;\n    this.failureCountSemaphore = semaphore(1);\n    this.tryMoveNextLocationSemaphore = semaphore(1);\n  }\n\n  /**\n   * Checks if the circuit breaker can trigger a partition failover based on the failure counts.\n   * Returns true if the number of consecutive failures exceeds the defined thresholds for read or write requests.\n   */\n  public async canCircuitBreakerTriggerPartitionFailOver(\n    isReadOnlyRequest: boolean,\n  ): Promise<boolean> {\n    const { consecutiveReadRequestFailureCount, consecutiveWriteRequestFailureCount } =\n      await this.snapshotConsecutiveRequestFailureCount();\n\n    return isReadOnlyRequest\n      ? consecutiveReadRequestFailureCount > Constants.ReadRequestFailureCountThreshold\n      : consecutiveWriteRequestFailureCount > Constants.WriteRequestFailureCountThreshold;\n  }\n\n  /**\n   * Increments the failure counts for read or write requests and updates the timestamps.\n   * If the time since the last failure exceeds the reset window, it resets the failure counts.\n   */\n  public async incrementRequestFailureCounts(\n    isReadOnlyRequest: boolean,\n    currentTimeInMilliseconds: number,\n  ): Promise<void> {\n    return new Promise<void>((resolve, reject) => {\n      this.failureCountSemaphore.take(async () => {\n        try {\n          const { lastRequestFailureTime } = await this.snapshotPartitionFailoverTimestamps();\n\n          if (\n            currentTimeInMilliseconds - lastRequestFailureTime >\n            Constants.ConsecutiveFailureCountResetIntervalInMS\n          ) {\n            this.consecutiveReadRequestFailureCount = 0;\n            this.consecutiveWriteRequestFailureCount = 0;\n          }\n\n          if (isReadOnlyRequest) {\n            this.consecutiveReadRequestFailureCount++;\n          } else {\n            this.consecutiveWriteRequestFailureCount++;\n          }\n          this.lastRequestFailureTime = currentTimeInMilliseconds;\n          return resolve();\n        } catch (error) {\n          reject(error);\n        } finally {\n          // Release the semaphore lock\n          this.failureCountSemaphore.leave();\n        }\n      });\n    });\n  }\n\n  /**\n   * Returns a snapshot of the first and last request failure timestamps.\n   * This method is used to retrieve the current state of failure timestamps without modifying them.\n   */\n  public async snapshotPartitionFailoverTimestamps(): Promise<{\n    firstRequestFailureTime: number;\n    lastRequestFailureTime: number;\n  }> {\n    return {\n      firstRequestFailureTime: this.firstRequestFailureTime,\n      lastRequestFailureTime: this.lastRequestFailureTime,\n    };\n  }\n\n  /**\n   * Attempts to move to the next available location for the partition key range.\n   * If the current endpoint is the same as the failed endpoint, it tries to find a new endpoint\n   * from the provided list of endpoints. If a new endpoint is found, it updates the current endpoint\n   * and returns true. If no new endpoint is found, it returns false.\n   */\n  public async tryMoveNextLocation(\n    endPoints: readonly string[],\n    failedEndPoint: string,\n    diagnosticNode: DiagnosticNodeInternal,\n    partitionKeyRangeId: string,\n  ): Promise<boolean> {\n    if (failedEndPoint !== this.currentEndPoint) {\n      return true;\n    }\n    return new Promise((resolve, reject) => {\n      this.tryMoveNextLocationSemaphore.take(() => {\n        try {\n          for (const endpoint of endPoints) {\n            if (this.currentEndPoint === endpoint) {\n              continue;\n            }\n\n            if (this.failedEndPoints.includes(endpoint)) {\n              continue;\n            }\n\n            this.failedEndPoints.push(failedEndPoint);\n            this.currentEndPoint = endpoint;\n            return resolve(true);\n          }\n          diagnosticNode.addData({\n            partitionKeyRangeFailoverInfo: `PartitionKeyRangeId: ${partitionKeyRangeId}, failedLocations: ${this.failedEndPoints}, newLocation: ${this.currentEndPoint}`,\n          });\n          return resolve(false);\n        } catch (err) {\n          reject(err);\n        } finally {\n          this.tryMoveNextLocationSemaphore.leave();\n        }\n      });\n    });\n  }\n\n  /** Returns the current endpoint being used for partition key range operations.*/\n  public getCurrentEndPoint(): string {\n    return this.currentEndPoint;\n  }\n\n  /**\n   * Returns a snapshot of the current consecutive request failure counts for read and write requests.\n   * This method is used to retrieve the current state of failure counts without modifying them.\n   */\n  private async snapshotConsecutiveRequestFailureCount(): Promise<{\n    consecutiveReadRequestFailureCount: number;\n    consecutiveWriteRequestFailureCount: number;\n  }> {\n    return {\n      consecutiveReadRequestFailureCount: this.consecutiveReadRequestFailureCount,\n      consecutiveWriteRequestFailureCount: this.consecutiveWriteRequestFailureCount,\n    };\n  }\n}\n"],
+  "mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAGA,mBAA0B;AAC1B,uBAAsB;AAOf,MAAM,8BAA8B;AAAA,EACjC,kBAA4B,CAAC;AAAA,EAC7B;AAAA,EAEA,qCAA6C;AAAA,EAC7C,sCAA8C;AAAA,EAC9C,0BAAkC,KAAK,IAAI;AAAA,EAC3C,yBAAiC,KAAK,IAAI;AAAA,EAE1C;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EAKR,YAAY,iBAAyB;AACnC,SAAK,kBAAkB;AACvB,SAAK,4BAAwB,iBAAAA,SAAU,CAAC;AACxC,SAAK,mCAA+B,iBAAAA,SAAU,CAAC;AAAA,EACjD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAa,0CACX,mBACkB;AAClB,UAAM,EAAE,oCAAoC,oCAAoC,IAC9E,MAAM,KAAK,uCAAuC;AAEpD,WAAO,oBACH,qCAAqC,uBAAU,mCAC/C,sCAAsC,uBAAU;AAAA,EACtD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAa,8BACX,mBACA,2BACe;AACf,WAAO,IAAI,QAAc,CAAC,SAAS,WAAW;AAC5C,WAAK,sBAAsB,KAAK,YAAY;AAC1C,YAAI;AACF,gBAAM,EAAE,uBAAuB,IAAI,MAAM,KAAK,oCAAoC;AAElF,cACE,4BAA4B,yBAC5B,uBAAU,0CACV;AACA,iBAAK,qCAAqC;AAC1C,iBAAK,sCAAsC;AAAA,UAC7C;AAEA,cAAI,mBAAmB;AACrB,iBAAK;AAAA,UACP,OAAO;AACL,iBAAK;AAAA,UACP;AACA,eAAK,yBAAyB;AAC9B,iBAAO,QAAQ;AAAA,QACjB,SAAS,OAAO;AACd,iBAAO,KAAK;AAAA,QACd,UAAE;AAEA,eAAK,sBAAsB,MAAM;AAAA,QACnC;AAAA,MACF,CAAC;AAAA,IACH,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAa,sCAGV;AACD,WAAO;AAAA,MACL,yBAAyB,KAAK;AAAA,MAC9B,wBAAwB,KAAK;AAAA,IAC/B;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAa,oBACX,WACA,gBACA,gBACA,qBACkB;AAClB,QAAI,mBAAmB,KAAK,iBAAiB;AAC3C,aAAO;AAAA,IACT;AACA,WAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,WAAK,6BAA6B,KAAK,MAAM;AAC3C,YAAI;AACF,qBAAW,YAAY,WAAW;AAChC,gBAAI,KAAK,oBAAoB,UAAU;AACrC;AAAA,YACF;AAEA,gBAAI,KAAK,gBAAgB,SAAS,QAAQ,GAAG;AAC3C;AAAA,YACF;AAEA,iBAAK,gBAAgB,KAAK,cAAc;AACxC,iBAAK,kBAAkB;AACvB,mBAAO,QAAQ,IAAI;AAAA,UACrB;AACA,yBAAe,QAAQ;AAAA,YACrB,+BAA+B,wBAAwB,mBAAmB,sBAAsB,KAAK,eAAe,kBAAkB,KAAK,eAAe;AAAA,UAC5J,CAAC;AACD,iBAAO,QAAQ,KAAK;AAAA,QACtB,SAAS,KAAK;AACZ,iBAAO,GAAG;AAAA,QACZ,UAAE;AACA,eAAK,6BAA6B,MAAM;AAAA,QAC1C;AAAA,MACF,CAAC;AAAA,IACH,CAAC;AAAA,EACH;AAAA;AAAA,EAGO,qBAA6B;AAClC,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAc,yCAGX;AACD,WAAO;AAAA,MACL,oCAAoC,KAAK;AAAA,MACzC,qCAAqC,KAAK;AAAA,IAC5C;AAAA,EACF;AACF;",
+  "names": ["semaphore"]
+}

package/dist/commonjs/auth.js

@@ -1,92 +1,100 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.setAuthorizationHeader = setAuthorizationHeader;
-exports.setAuthorizationTokenHeaderUsingMasterKey = setAuthorizationTokenHeaderUsingMasterKey;
-exports.getAuthorizationTokenUsingResourceTokens = getAuthorizationTokenUsingResourceTokens;
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT License.
-const headers_js_1 = require("./utils/headers.js");
-const index_js_1 = require("./common/index.js");
-/**
- * @hidden
- */
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var auth_exports = {};
+__export(auth_exports, {
+  getAuthorizationTokenUsingResourceTokens: () => getAuthorizationTokenUsingResourceTokens,
+  setAuthorizationHeader: () => setAuthorizationHeader,
+  setAuthorizationTokenHeaderUsingMasterKey: () => setAuthorizationTokenHeaderUsingMasterKey
+});
+module.exports = __toCommonJS(auth_exports);
+var import_headers = require("./utils/headers.js");
+var import_common = require("./common/index.js");
 async function setAuthorizationHeader(clientOptions, verb, path, resourceId, resourceType, headers) {
-    if (clientOptions.permissionFeed) {
-        clientOptions.resourceTokens = {};
-        for (const permission of clientOptions.permissionFeed) {
-            const id = (0, index_js_1.getResourceIdFromPath)(permission.resource);
-            if (!id) {
-                throw new Error(`authorization error: ${id} \
-                          is an invalid resourceId in permissionFeed`);
-            }
-            clientOptions.resourceTokens[id] = permission._token; // TODO: any
-        }
-    }
-    if (clientOptions.key) {
-        await setAuthorizationTokenHeaderUsingMasterKey(verb, resourceId, resourceType, headers, clientOptions.key);
-    }
-    else if (clientOptions.resourceTokens) {
-        headers[index_js_1.Constants.HttpHeaders.Authorization] = encodeURIComponent(getAuthorizationTokenUsingResourceTokens(clientOptions.resourceTokens, path, resourceId));
-    }
-    else if (clientOptions.tokenProvider) {
-        headers[index_js_1.Constants.HttpHeaders.Authorization] = encodeURIComponent(await clientOptions.tokenProvider({ verb, path, resourceId, resourceType, headers }));
+  if (clientOptions.permissionFeed) {
+    clientOptions.resourceTokens = {};
+    for (const permission of clientOptions.permissionFeed) {
+      const id = (0, import_common.getResourceIdFromPath)(permission.resource);
+      if (!id) {
+        throw new Error(`authorization error: ${id}                           is an invalid resourceId in permissionFeed`);
+      }
+      clientOptions.resourceTokens[id] = permission._token;
     }
+  }
+  if (clientOptions.key) {
+    await setAuthorizationTokenHeaderUsingMasterKey(
+      verb,
+      resourceId,
+      resourceType,
+      headers,
+      clientOptions.key
+    );
+  } else if (clientOptions.resourceTokens) {
+    headers[import_common.Constants.HttpHeaders.Authorization] = encodeURIComponent(
+      getAuthorizationTokenUsingResourceTokens(clientOptions.resourceTokens, path, resourceId)
+    );
+  } else if (clientOptions.tokenProvider) {
+    headers[import_common.Constants.HttpHeaders.Authorization] = encodeURIComponent(
+      await clientOptions.tokenProvider({ verb, path, resourceId, resourceType, headers })
+    );
+  }
 }
-/**
- * The default function for setting header token using the masterKey
- * @hidden
- */
 async function setAuthorizationTokenHeaderUsingMasterKey(verb, resourceId, resourceType, headers, masterKey) {
-    // TODO This should live in cosmos-sign
-    if (resourceType === index_js_1.ResourceType.offer) {
-        resourceId = resourceId && resourceId.toLowerCase();
-    }
-    headers = Object.assign(headers, await (0, headers_js_1.generateHeaders)(masterKey, verb, resourceType, resourceId));
+  if (resourceType === import_common.ResourceType.offer) {
+    resourceId = resourceId && resourceId.toLowerCase();
+  }
+  headers = Object.assign(
+    headers,
+    await (0, import_headers.generateHeaders)(masterKey, verb, resourceType, resourceId)
+  );
 }
-/**
- * @hidden
- */
-// TODO: Resource tokens
 function getAuthorizationTokenUsingResourceTokens(resourceTokens, path, resourceId) {
-    if (resourceTokens && Object.keys(resourceTokens).length > 0) {
-        // For database account access(through getDatabaseAccount API), path and resourceId are "",
-        // so in this case we return the first token to be used for creating the auth header as the
-        // service will accept any token in this case
-        if (!path && !resourceId) {
-            return resourceTokens[Object.keys(resourceTokens)[0]];
-        }
-        // If we have exact resource token for the path use it
-        if (resourceId && resourceTokens[resourceId]) {
-            return resourceTokens[resourceId];
-        }
-        // minimum valid path /dbs
-        if (!path || path.length < 4) {
-            // TODO: This should throw an error
-            return null;
-        }
-        path = (0, index_js_1.trimSlashFromLeftAndRight)(path);
-        const pathSegments = (path && path.split("/")) || [];
-        // Item path
-        if (pathSegments.length === 6) {
-            // Look for a container token matching the item path
-            const containerPath = pathSegments.slice(0, 4).map(decodeURIComponent).join("/");
-            if (resourceTokens[containerPath]) {
-                return resourceTokens[containerPath];
-            }
-        }
-        // TODO remove in v4: This is legacy behavior that lets someone use a resource token pointing ONLY at an ID
-        // It was used when _rid was exposed by the SDK, but now that we are using user provided ids it is not needed
-        // However removing it now would be a breaking change
-        // if it's an incomplete path like /dbs/db1/colls/, start from the parent resource
-        let index = pathSegments.length % 2 === 0 ? pathSegments.length - 1 : pathSegments.length - 2;
-        for (; index > 0; index -= 2) {
-            const id = decodeURI(pathSegments[index]);
-            if (resourceTokens[id]) {
-                return resourceTokens[id];
-            }
-        }
+  if (resourceTokens && Object.keys(resourceTokens).length > 0) {
+    if (!path && !resourceId) {
+      return resourceTokens[Object.keys(resourceTokens)[0]];
+    }
+    if (resourceId && resourceTokens[resourceId]) {
+      return resourceTokens[resourceId];
+    }
+    if (!path || path.length < 4) {
+      return null;
+    }
+    path = (0, import_common.trimSlashFromLeftAndRight)(path);
+    const pathSegments = path && path.split("/") || [];
+    if (pathSegments.length === 6) {
+      const containerPath = pathSegments.slice(0, 4).map(decodeURIComponent).join("/");
+      if (resourceTokens[containerPath]) {
+        return resourceTokens[containerPath];
+      }
+    }
+    let index = pathSegments.length % 2 === 0 ? pathSegments.length - 1 : pathSegments.length - 2;
+    for (; index > 0; index -= 2) {
+      const id = decodeURI(pathSegments[index]);
+      if (resourceTokens[id]) {
+        return resourceTokens[id];
+      }
     }
-    // TODO: This should throw an error
-    return null;
+  }
+  return null;
 }
-//# sourceMappingURL=auth.js.map
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  getAuthorizationTokenUsingResourceTokens,
+  setAuthorizationHeader,
+  setAuthorizationTokenHeaderUsingMasterKey
+});
+//# sourceMappingURL=auth.js.map

package/dist/commonjs/auth.js.map

@@ -1 +1,7 @@
-{"version":3,"file":"auth.js","sourceRoot":"","sources":["../../src/auth.ts"],"names":[],"mappings":";;AA2BA,wDAsCC;AAMD,8FAeC;AAMD,4FAmDC;AA/ID,uCAAuC;AACvC,kCAAkC;AAClC,mDAAqD;AAErD,gDAK2B;AAe3B;;GAEG;AACI,KAAK,UAAU,sBAAsB,CAC1C,aAAkC,EAClC,IAAgB,EAChB,IAAY,EACZ,UAAkB,EAClB,YAA0B,EAC1B,OAAsB;IAEtB,IAAI,aAAa,CAAC,cAAc,EAAE,CAAC;QACjC,aAAa,CAAC,cAAc,GAAG,EAAE,CAAC;QAClC,KAAK,MAAM,UAAU,IAAI,aAAa,CAAC,cAAc,EAAE,CAAC;YACtD,MAAM,EAAE,GAAG,IAAA,gCAAqB,EAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACtD,IAAI,CAAC,EAAE,EAAE,CAAC;gBACR,MAAM,IAAI,KAAK,CAAC,wBAAwB,EAAE;qEACmB,CAAC,CAAC;YACjE,CAAC;YAED,aAAa,CAAC,cAAc,CAAC,EAAE,CAAC,GAAI,UAAkB,CAAC,MAAM,CAAC,CAAC,YAAY;QAC7E,CAAC;IACH,CAAC;IAED,IAAI,aAAa,CAAC,GAAG,EAAE,CAAC;QACtB,MAAM,yCAAyC,CAC7C,IAAI,EACJ,UAAU,EACV,YAAY,EACZ,OAAO,EACP,aAAa,CAAC,GAAG,CAClB,CAAC;IACJ,CAAC;SAAM,IAAI,aAAa,CAAC,cAAc,EAAE,CAAC;QACxC,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,kBAAkB,CAC/D,wCAAwC,CAAC,aAAa,CAAC,cAAc,EAAE,IAAI,EAAE,UAAU,CAAC,CACzF,CAAC;IACJ,CAAC;SAAM,IAAI,aAAa,CAAC,aAAa,EAAE,CAAC;QACvC,OAAO,CAAC,oBAAS,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,kBAAkB,CAC/D,MAAM,aAAa,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CACrF,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;GAGG;AACI,KAAK,UAAU,yCAAyC,CAC7D,IAAgB,EAChB,UAAkB,EAClB,YAA0B,EAC1B,OAAsB,EACtB,SAAiB;IAEjB,uCAAuC;IACvC,IAAI,YAAY,KAAK,uBAAY,CAAC,KAAK,EAAE,CAAC;QACxC,UAAU,GAAG,UAAU,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;IACtD,CAAC;IACD,OAAO,GAAG,MAAM,CAAC,MAAM,CACrB,OAAO,EACP,MAAM,IAAA,4BAAe,EAAC,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,UAAU,CAAC,CACjE,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,wBAAwB;AACxB,SAAgB,wCAAwC,CACtD,cAAgD,EAChD,IAAY,EACZ,UAAkB;IAElB,IAAI,cAAc,IAAI,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7D,2FAA2F;QAC3F,2FAA2F;QAC3F,6CAA6C;QAC7C,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACzB,OAAO,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACxD,CAAC;QAED,sDAAsD;QACtD,IAAI,UAAU,IAAI,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;YAC7C,OAAO,cAAc,CAAC,UAAU,CAAC,CAAC;QACpC,CAAC;QAED,0BAA0B;QAC1B,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,mCAAmC;YACnC,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,GAAG,IAAA,oCAAyB,EAAC,IAAI,CAAC,CAAC;QACvC,MAAM,YAAY,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QAErD,YAAY;QACZ,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,oDAAoD;YACpD,MAAM,aAAa,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACjF,IAAI,cAAc,CAAC,aAAa,CAAC,EAAE,CAAC;gBAClC,OAAO,cAAc,CAAC,aAAa,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;QAED,2GAA2G;QAC3G,6GAA6G;QAC7G,qDAAqD;QACrD,kFAAkF;QAClF,IAAI,KAAK,GAAG,YAAY,CAAC,MAAM,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC;QAC9F,OAAO,KAAK,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,EAAE,GAAG,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;YAC1C,IAAI,cAAc,CAAC,EAAE,CAAC,EAAE,CAAC;gBACvB,OAAO,cAAc,CAAC,EAAE,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;IAED,mCAAmC;IACnC,OAAO,IAAI,CAAC;AACd,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport { generateHeaders } from \"./utils/headers.js\";\nimport type { HTTPMethod } from \"./common/index.js\";\nimport {\n  Constants,\n  getResourceIdFromPath,\n  ResourceType,\n  trimSlashFromLeftAndRight,\n} from \"./common/index.js\";\nimport type { CosmosClientOptions } from \"./CosmosClientOptions.js\";\nimport type { CosmosHeaders } from \"./queryExecutionContext/index.js\";\n\n/** @hidden */\nexport interface RequestInfo {\n  verb: HTTPMethod;\n  path: string;\n  resourceId: string;\n  resourceType: ResourceType;\n  headers: CosmosHeaders;\n}\n\nexport type TokenProvider = (requestInfo: RequestInfo) => Promise<string>;\n\n/**\n * @hidden\n */\nexport async function setAuthorizationHeader(\n  clientOptions: CosmosClientOptions,\n  verb: HTTPMethod,\n  path: string,\n  resourceId: string,\n  resourceType: ResourceType,\n  headers: CosmosHeaders,\n): Promise<void> {\n  if (clientOptions.permissionFeed) {\n    clientOptions.resourceTokens = {};\n    for (const permission of clientOptions.permissionFeed) {\n      const id = getResourceIdFromPath(permission.resource);\n      if (!id) {\n        throw new Error(`authorization error: ${id} \\\n                          is an invalid resourceId in permissionFeed`);\n      }\n\n      clientOptions.resourceTokens[id] = (permission as any)._token; // TODO: any\n    }\n  }\n\n  if (clientOptions.key) {\n    await setAuthorizationTokenHeaderUsingMasterKey(\n      verb,\n      resourceId,\n      resourceType,\n      headers,\n      clientOptions.key,\n    );\n  } else if (clientOptions.resourceTokens) {\n    headers[Constants.HttpHeaders.Authorization] = encodeURIComponent(\n      getAuthorizationTokenUsingResourceTokens(clientOptions.resourceTokens, path, resourceId),\n    );\n  } else if (clientOptions.tokenProvider) {\n    headers[Constants.HttpHeaders.Authorization] = encodeURIComponent(\n      await clientOptions.tokenProvider({ verb, path, resourceId, resourceType, headers }),\n    );\n  }\n}\n\n/**\n * The default function for setting header token using the masterKey\n * @hidden\n */\nexport async function setAuthorizationTokenHeaderUsingMasterKey(\n  verb: HTTPMethod,\n  resourceId: string,\n  resourceType: ResourceType,\n  headers: CosmosHeaders,\n  masterKey: string,\n): Promise<void> {\n  // TODO This should live in cosmos-sign\n  if (resourceType === ResourceType.offer) {\n    resourceId = resourceId && resourceId.toLowerCase();\n  }\n  headers = Object.assign(\n    headers,\n    await generateHeaders(masterKey, verb, resourceType, resourceId),\n  );\n}\n\n/**\n * @hidden\n */\n// TODO: Resource tokens\nexport function getAuthorizationTokenUsingResourceTokens(\n  resourceTokens: { [resourceId: string]: string },\n  path: string,\n  resourceId: string,\n): string {\n  if (resourceTokens && Object.keys(resourceTokens).length > 0) {\n    // For database account access(through getDatabaseAccount API), path and resourceId are \"\",\n    // so in this case we return the first token to be used for creating the auth header as the\n    // service will accept any token in this case\n    if (!path && !resourceId) {\n      return resourceTokens[Object.keys(resourceTokens)[0]];\n    }\n\n    // If we have exact resource token for the path use it\n    if (resourceId && resourceTokens[resourceId]) {\n      return resourceTokens[resourceId];\n    }\n\n    // minimum valid path /dbs\n    if (!path || path.length < 4) {\n      // TODO: This should throw an error\n      return null;\n    }\n\n    path = trimSlashFromLeftAndRight(path);\n    const pathSegments = (path && path.split(\"/\")) || [];\n\n    // Item path\n    if (pathSegments.length === 6) {\n      // Look for a container token matching the item path\n      const containerPath = pathSegments.slice(0, 4).map(decodeURIComponent).join(\"/\");\n      if (resourceTokens[containerPath]) {\n        return resourceTokens[containerPath];\n      }\n    }\n\n    // TODO remove in v4: This is legacy behavior that lets someone use a resource token pointing ONLY at an ID\n    // It was used when _rid was exposed by the SDK, but now that we are using user provided ids it is not needed\n    // However removing it now would be a breaking change\n    // if it's an incomplete path like /dbs/db1/colls/, start from the parent resource\n    let index = pathSegments.length % 2 === 0 ? pathSegments.length - 1 : pathSegments.length - 2;\n    for (; index > 0; index -= 2) {\n      const id = decodeURI(pathSegments[index]);\n      if (resourceTokens[id]) {\n        return resourceTokens[id];\n      }\n    }\n  }\n\n  // TODO: This should throw an error\n  return null;\n}\n"]}
+{
+  "version": 3,
+  "sources": ["../../src/auth.ts"],
+  "sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport { generateHeaders } from \"./utils/headers.js\";\nimport type { HTTPMethod } from \"./common/index.js\";\nimport {\n  Constants,\n  getResourceIdFromPath,\n  ResourceType,\n  trimSlashFromLeftAndRight,\n} from \"./common/index.js\";\nimport type { CosmosClientOptions } from \"./CosmosClientOptions.js\";\nimport type { CosmosHeaders } from \"./queryExecutionContext/index.js\";\n\n/** @hidden */\nexport interface RequestInfo {\n  verb: HTTPMethod;\n  path: string;\n  resourceId: string;\n  resourceType: ResourceType;\n  headers: CosmosHeaders;\n}\n\nexport type TokenProvider = (requestInfo: RequestInfo) => Promise<string>;\n\n/**\n * @hidden\n */\nexport async function setAuthorizationHeader(\n  clientOptions: CosmosClientOptions,\n  verb: HTTPMethod,\n  path: string,\n  resourceId: string,\n  resourceType: ResourceType,\n  headers: CosmosHeaders,\n): Promise<void> {\n  if (clientOptions.permissionFeed) {\n    clientOptions.resourceTokens = {};\n    for (const permission of clientOptions.permissionFeed) {\n      const id = getResourceIdFromPath(permission.resource);\n      if (!id) {\n        throw new Error(`authorization error: ${id} \\\n                          is an invalid resourceId in permissionFeed`);\n      }\n\n      clientOptions.resourceTokens[id] = (permission as any)._token; // TODO: any\n    }\n  }\n\n  if (clientOptions.key) {\n    await setAuthorizationTokenHeaderUsingMasterKey(\n      verb,\n      resourceId,\n      resourceType,\n      headers,\n      clientOptions.key,\n    );\n  } else if (clientOptions.resourceTokens) {\n    headers[Constants.HttpHeaders.Authorization] = encodeURIComponent(\n      getAuthorizationTokenUsingResourceTokens(clientOptions.resourceTokens, path, resourceId),\n    );\n  } else if (clientOptions.tokenProvider) {\n    headers[Constants.HttpHeaders.Authorization] = encodeURIComponent(\n      await clientOptions.tokenProvider({ verb, path, resourceId, resourceType, headers }),\n    );\n  }\n}\n\n/**\n * The default function for setting header token using the masterKey\n * @hidden\n */\nexport async function setAuthorizationTokenHeaderUsingMasterKey(\n  verb: HTTPMethod,\n  resourceId: string,\n  resourceType: ResourceType,\n  headers: CosmosHeaders,\n  masterKey: string,\n): Promise<void> {\n  // TODO This should live in cosmos-sign\n  if (resourceType === ResourceType.offer) {\n    resourceId = resourceId && resourceId.toLowerCase();\n  }\n  headers = Object.assign(\n    headers,\n    await generateHeaders(masterKey, verb, resourceType, resourceId),\n  );\n}\n\n/**\n * @hidden\n */\n// TODO: Resource tokens\nexport function getAuthorizationTokenUsingResourceTokens(\n  resourceTokens: { [resourceId: string]: string },\n  path: string,\n  resourceId: string,\n): string {\n  if (resourceTokens && Object.keys(resourceTokens).length > 0) {\n    // For database account access(through getDatabaseAccount API), path and resourceId are \"\",\n    // so in this case we return the first token to be used for creating the auth header as the\n    // service will accept any token in this case\n    if (!path && !resourceId) {\n      return resourceTokens[Object.keys(resourceTokens)[0]];\n    }\n\n    // If we have exact resource token for the path use it\n    if (resourceId && resourceTokens[resourceId]) {\n      return resourceTokens[resourceId];\n    }\n\n    // minimum valid path /dbs\n    if (!path || path.length < 4) {\n      // TODO: This should throw an error\n      return null;\n    }\n\n    path = trimSlashFromLeftAndRight(path);\n    const pathSegments = (path && path.split(\"/\")) || [];\n\n    // Item path\n    if (pathSegments.length === 6) {\n      // Look for a container token matching the item path\n      const containerPath = pathSegments.slice(0, 4).map(decodeURIComponent).join(\"/\");\n      if (resourceTokens[containerPath]) {\n        return resourceTokens[containerPath];\n      }\n    }\n\n    // TODO remove in v4: This is legacy behavior that lets someone use a resource token pointing ONLY at an ID\n    // It was used when _rid was exposed by the SDK, but now that we are using user provided ids it is not needed\n    // However removing it now would be a breaking change\n    // if it's an incomplete path like /dbs/db1/colls/, start from the parent resource\n    let index = pathSegments.length % 2 === 0 ? pathSegments.length - 1 : pathSegments.length - 2;\n    for (; index > 0; index -= 2) {\n      const id = decodeURI(pathSegments[index]);\n      if (resourceTokens[id]) {\n        return resourceTokens[id];\n      }\n    }\n  }\n\n  // TODO: This should throw an error\n  return null;\n}\n"],
+  "mappings": ";;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAEA,qBAAgC;AAEhC,oBAKO;AAkBP,eAAsB,uBACpB,eACA,MACA,MACA,YACA,cACA,SACe;AACf,MAAI,cAAc,gBAAgB;AAChC,kBAAc,iBAAiB,CAAC;AAChC,eAAW,cAAc,cAAc,gBAAgB;AACrD,YAAM,SAAK,qCAAsB,WAAW,QAAQ;AACpD,UAAI,CAAC,IAAI;AACP,cAAM,IAAI,MAAM,wBAAwB,EAAE,uEACmB;AAAA,MAC/D;AAEA,oBAAc,eAAe,EAAE,IAAK,WAAmB;AAAA,IACzD;AAAA,EACF;AAEA,MAAI,cAAc,KAAK;AACrB,UAAM;AAAA,MACJ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,cAAc;AAAA,IAChB;AAAA,EACF,WAAW,cAAc,gBAAgB;AACvC,YAAQ,wBAAU,YAAY,aAAa,IAAI;AAAA,MAC7C,yCAAyC,cAAc,gBAAgB,MAAM,UAAU;AAAA,IACzF;AAAA,EACF,WAAW,cAAc,eAAe;AACtC,YAAQ,wBAAU,YAAY,aAAa,IAAI;AAAA,MAC7C,MAAM,cAAc,cAAc,EAAE,MAAM,MAAM,YAAY,cAAc,QAAQ,CAAC;AAAA,IACrF;AAAA,EACF;AACF;AAMA,eAAsB,0CACpB,MACA,YACA,cACA,SACA,WACe;AAEf,MAAI,iBAAiB,2BAAa,OAAO;AACvC,iBAAa,cAAc,WAAW,YAAY;AAAA,EACpD;AACA,YAAU,OAAO;AAAA,IACf;AAAA,IACA,UAAM,gCAAgB,WAAW,MAAM,cAAc,UAAU;AAAA,EACjE;AACF;AAMO,SAAS,yCACd,gBACA,MACA,YACQ;AACR,MAAI,kBAAkB,OAAO,KAAK,cAAc,EAAE,SAAS,GAAG;AAI5D,QAAI,CAAC,QAAQ,CAAC,YAAY;AACxB,aAAO,eAAe,OAAO,KAAK,cAAc,EAAE,CAAC,CAAC;AAAA,IACtD;AAGA,QAAI,cAAc,eAAe,UAAU,GAAG;AAC5C,aAAO,eAAe,UAAU;AAAA,IAClC;AAGA,QAAI,CAAC,QAAQ,KAAK,SAAS,GAAG;AAE5B,aAAO;AAAA,IACT;AAEA,eAAO,yCAA0B,IAAI;AACrC,UAAM,eAAgB,QAAQ,KAAK,MAAM,GAAG,KAAM,CAAC;AAGnD,QAAI,aAAa,WAAW,GAAG;AAE7B,YAAM,gBAAgB,aAAa,MAAM,GAAG,CAAC,EAAE,IAAI,kBAAkB,EAAE,KAAK,GAAG;AAC/E,UAAI,eAAe,aAAa,GAAG;AACjC,eAAO,eAAe,aAAa;AAAA,MACrC;AAAA,IACF;AAMA,QAAI,QAAQ,aAAa,SAAS,MAAM,IAAI,aAAa,SAAS,IAAI,aAAa,SAAS;AAC5F,WAAO,QAAQ,GAAG,SAAS,GAAG;AAC5B,YAAM,KAAK,UAAU,aAAa,KAAK,CAAC;AACxC,UAAI,eAAe,EAAE,GAAG;AACtB,eAAO,eAAe,EAAE;AAAA,MAC1B;AAAA,IACF;AAAA,EACF;AAGA,SAAO;AACT;",
+  "names": []
+}