@azure/cosmos 3.14.2-alpha.20211111.1 → 3.15.0

package/dist-esm/src/ClientContext.js

@@ -0,0 +1,549 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+import { v4 } from "uuid";
+const uuid = v4;
+import { bearerTokenAuthenticationPolicy, createEmptyPipeline } from "@azure/core-rest-pipeline";
+import { Constants, HTTPMethod, OperationType, ResourceType } from "./common/constants";
+import { getIdFromLink, getPathFromLink, parseLink } from "./common/helper";
+import { StatusCodes, SubStatusCodes } from "./common/statusCodes";
+import { ConsistencyLevel, DatabaseAccount } from "./documents";
+import { executePlugins, PluginOn } from "./plugins/Plugin";
+import { QueryIterator } from "./queryIterator";
+import { getHeaders } from "./request/request";
+import { request as executeRequest } from "./request/RequestHandler";
+import { SessionContainer } from "./session/sessionContainer";
+import { sanitizeEndpoint } from "./utils/checkURL";
+import { createClientLogger } from "@azure/logger";
+const logger = createClientLogger("ClientContext");
+const QueryJsonContentType = "application/query+json";
+/**
+ * @hidden
+ * @hidden
+ */
+export class ClientContext {
+    constructor(cosmosClientOptions, globalEndpointManager) {
+        this.cosmosClientOptions = cosmosClientOptions;
+        this.globalEndpointManager = globalEndpointManager;
+        this.connectionPolicy = cosmosClientOptions.connectionPolicy;
+        this.sessionContainer = new SessionContainer();
+        this.partitionKeyDefinitionCache = {};
+        this.pipeline = null;
+        if (cosmosClientOptions.aadCredentials) {
+            this.pipeline = createEmptyPipeline();
+            const hrefEndpoint = sanitizeEndpoint(cosmosClientOptions.endpoint);
+            const scope = `${hrefEndpoint}/.default`;
+            this.pipeline.addPolicy(bearerTokenAuthenticationPolicy({
+                credential: cosmosClientOptions.aadCredentials,
+                scopes: scope,
+                challengeCallbacks: {
+                    async authorizeRequest({ request, getAccessToken }) {
+                        const tokenResponse = await getAccessToken([scope], {});
+                        const AUTH_PREFIX = `type=aad&ver=1.0&sig=`;
+                        const authorizationToken = `${AUTH_PREFIX}${tokenResponse.token}`;
+                        request.headers.set("Authorization", authorizationToken);
+                    }
+                }
+            }));
+        }
+    }
+    /** @hidden */
+    async read({ path, resourceType, resourceId, options = {}, partitionKey }) {
+        try {
+            const request = {
+                globalEndpointManager: this.globalEndpointManager,
+                requestAgent: this.cosmosClientOptions.agent,
+                connectionPolicy: this.connectionPolicy,
+                method: HTTPMethod.get,
+                path,
+                operationType: OperationType.Read,
+                client: this,
+                resourceId,
+                options,
+                resourceType,
+                plugins: this.cosmosClientOptions.plugins,
+                partitionKey,
+                pipeline: this.pipeline
+            };
+            request.headers = await this.buildHeaders(request);
+            this.applySessionToken(request);
+            // read will use ReadEndpoint since it uses GET operation
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+            const response = await executePlugins(request, executeRequest, PluginOn.operation);
+            this.captureSessionToken(undefined, path, OperationType.Read, response.headers);
+            return response;
+        }
+        catch (err) {
+            this.captureSessionToken(err, path, OperationType.Upsert, err.headers);
+            throw err;
+        }
+    }
+    async queryFeed({ path, resourceType, resourceId, resultFn, query, options, partitionKeyRangeId, partitionKey }) {
+        // Query operations will use ReadEndpoint even though it uses
+        // GET(for queryFeed) and POST(for regular query operations)
+        const request = {
+            globalEndpointManager: this.globalEndpointManager,
+            requestAgent: this.cosmosClientOptions.agent,
+            connectionPolicy: this.connectionPolicy,
+            method: HTTPMethod.get,
+            path,
+            operationType: OperationType.Query,
+            client: this,
+            partitionKeyRangeId,
+            resourceId,
+            resourceType,
+            options,
+            body: query,
+            plugins: this.cosmosClientOptions.plugins,
+            partitionKey,
+            pipeline: this.pipeline
+        };
+        const requestId = uuid();
+        if (query !== undefined) {
+            request.method = HTTPMethod.post;
+        }
+        request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+        request.headers = await this.buildHeaders(request);
+        if (query !== undefined) {
+            request.headers[Constants.HttpHeaders.IsQuery] = "true";
+            request.headers[Constants.HttpHeaders.ContentType] = QueryJsonContentType;
+            if (typeof query === "string") {
+                request.body = { query }; // Converts query text to query object.
+            }
+        }
+        this.applySessionToken(request);
+        logger.info("query " +
+            requestId +
+            " started" +
+            (request.partitionKeyRangeId ? " pkrid: " + request.partitionKeyRangeId : ""));
+        logger.verbose(request);
+        const start = Date.now();
+        const response = await executeRequest(request);
+        logger.info("query " + requestId + " finished - " + (Date.now() - start) + "ms");
+        this.captureSessionToken(undefined, path, OperationType.Query, response.headers);
+        return this.processQueryFeedResponse(response, !!query, resultFn);
+    }
+    async getQueryPlan(path, resourceType, resourceId, query, options = {}) {
+        const request = {
+            globalEndpointManager: this.globalEndpointManager,
+            requestAgent: this.cosmosClientOptions.agent,
+            connectionPolicy: this.connectionPolicy,
+            method: HTTPMethod.post,
+            path,
+            operationType: OperationType.Read,
+            client: this,
+            resourceId,
+            resourceType,
+            options,
+            body: query,
+            plugins: this.cosmosClientOptions.plugins,
+            pipeline: this.pipeline
+        };
+        request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+        request.headers = await this.buildHeaders(request);
+        request.headers[Constants.HttpHeaders.IsQueryPlan] = "True";
+        request.headers[Constants.HttpHeaders.QueryVersion] = "1.4";
+        request.headers[Constants.HttpHeaders.SupportedQueryFeatures] =
+            "NonValueAggregate, Aggregate, Distinct, MultipleOrderBy, OffsetAndLimit, OrderBy, Top, CompositeAggregate, GroupBy, MultipleAggregates";
+        request.headers[Constants.HttpHeaders.ContentType] = QueryJsonContentType;
+        if (typeof query === "string") {
+            request.body = { query }; // Converts query text to query object.
+        }
+        this.applySessionToken(request);
+        const response = await executeRequest(request);
+        this.captureSessionToken(undefined, path, OperationType.Query, response.headers);
+        return response;
+    }
+    queryPartitionKeyRanges(collectionLink, query, options) {
+        const path = getPathFromLink(collectionLink, ResourceType.pkranges);
+        const id = getIdFromLink(collectionLink);
+        const cb = (innerOptions) => {
+            return this.queryFeed({
+                path,
+                resourceType: ResourceType.pkranges,
+                resourceId: id,
+                resultFn: (result) => result.PartitionKeyRanges,
+                query,
+                options: innerOptions
+            });
+        };
+        return new QueryIterator(this, query, options, cb);
+    }
+    async delete({ path, resourceType, resourceId, options = {}, partitionKey }) {
+        try {
+            const request = {
+                globalEndpointManager: this.globalEndpointManager,
+                requestAgent: this.cosmosClientOptions.agent,
+                connectionPolicy: this.connectionPolicy,
+                method: HTTPMethod.delete,
+                client: this,
+                operationType: OperationType.Delete,
+                path,
+                resourceType,
+                options,
+                resourceId,
+                plugins: this.cosmosClientOptions.plugins,
+                partitionKey,
+                pipeline: this.pipeline
+            };
+            request.headers = await this.buildHeaders(request);
+            this.applySessionToken(request);
+            // deleteResource will use WriteEndpoint since it uses DELETE operation
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+            const response = await executePlugins(request, executeRequest, PluginOn.operation);
+            if (parseLink(path).type !== "colls") {
+                this.captureSessionToken(undefined, path, OperationType.Delete, response.headers);
+            }
+            else {
+                this.clearSessionToken(path);
+            }
+            return response;
+        }
+        catch (err) {
+            this.captureSessionToken(err, path, OperationType.Upsert, err.headers);
+            throw err;
+        }
+    }
+    async patch({ body, path, resourceType, resourceId, options = {}, partitionKey }) {
+        try {
+            const request = {
+                globalEndpointManager: this.globalEndpointManager,
+                requestAgent: this.cosmosClientOptions.agent,
+                connectionPolicy: this.connectionPolicy,
+                method: HTTPMethod.patch,
+                client: this,
+                operationType: OperationType.Patch,
+                path,
+                resourceType,
+                body,
+                resourceId,
+                options,
+                plugins: this.cosmosClientOptions.plugins,
+                partitionKey
+            };
+            request.headers = await this.buildHeaders(request);
+            this.applySessionToken(request);
+            // patch will use WriteEndpoint
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+            const response = await executePlugins(request, executeRequest, PluginOn.operation);
+            this.captureSessionToken(undefined, path, OperationType.Patch, response.headers);
+            return response;
+        }
+        catch (err) {
+            this.captureSessionToken(err, path, OperationType.Upsert, err.headers);
+            throw err;
+        }
+    }
+    async create({ body, path, resourceType, resourceId, options = {}, partitionKey }) {
+        try {
+            const request = {
+                globalEndpointManager: this.globalEndpointManager,
+                requestAgent: this.cosmosClientOptions.agent,
+                connectionPolicy: this.connectionPolicy,
+                method: HTTPMethod.post,
+                client: this,
+                operationType: OperationType.Create,
+                path,
+                resourceType,
+                resourceId,
+                body,
+                options,
+                plugins: this.cosmosClientOptions.plugins,
+                partitionKey,
+                pipeline: this.pipeline
+            };
+            request.headers = await this.buildHeaders(request);
+            // create will use WriteEndpoint since it uses POST operation
+            this.applySessionToken(request);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+            const response = await executePlugins(request, executeRequest, PluginOn.operation);
+            this.captureSessionToken(undefined, path, OperationType.Create, response.headers);
+            return response;
+        }
+        catch (err) {
+            this.captureSessionToken(err, path, OperationType.Upsert, err.headers);
+            throw err;
+        }
+    }
+    processQueryFeedResponse(res, isQuery, resultFn) {
+        if (isQuery) {
+            return { result: resultFn(res.result), headers: res.headers, code: res.code };
+        }
+        else {
+            const newResult = resultFn(res.result).map((body) => body);
+            return { result: newResult, headers: res.headers, code: res.code };
+        }
+    }
+    applySessionToken(requestContext) {
+        const request = this.getSessionParams(requestContext.path);
+        if (requestContext.headers && requestContext.headers[Constants.HttpHeaders.SessionToken]) {
+            return;
+        }
+        const sessionConsistency = requestContext.headers[Constants.HttpHeaders.ConsistencyLevel];
+        if (!sessionConsistency) {
+            return;
+        }
+        if (sessionConsistency !== ConsistencyLevel.Session) {
+            return;
+        }
+        if (request.resourceAddress) {
+            const sessionToken = this.sessionContainer.get(request);
+            if (sessionToken) {
+                requestContext.headers[Constants.HttpHeaders.SessionToken] = sessionToken;
+            }
+        }
+    }
+    async replace({ body, path, resourceType, resourceId, options = {}, partitionKey }) {
+        try {
+            const request = {
+                globalEndpointManager: this.globalEndpointManager,
+                requestAgent: this.cosmosClientOptions.agent,
+                connectionPolicy: this.connectionPolicy,
+                method: HTTPMethod.put,
+                client: this,
+                operationType: OperationType.Replace,
+                path,
+                resourceType,
+                body,
+                resourceId,
+                options,
+                plugins: this.cosmosClientOptions.plugins,
+                partitionKey,
+                pipeline: this.pipeline
+            };
+            request.headers = await this.buildHeaders(request);
+            this.applySessionToken(request);
+            // replace will use WriteEndpoint since it uses PUT operation
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+            const response = await executePlugins(request, executeRequest, PluginOn.operation);
+            this.captureSessionToken(undefined, path, OperationType.Replace, response.headers);
+            return response;
+        }
+        catch (err) {
+            this.captureSessionToken(err, path, OperationType.Upsert, err.headers);
+            throw err;
+        }
+    }
+    async upsert({ body, path, resourceType, resourceId, options = {}, partitionKey }) {
+        try {
+            const request = {
+                globalEndpointManager: this.globalEndpointManager,
+                requestAgent: this.cosmosClientOptions.agent,
+                connectionPolicy: this.connectionPolicy,
+                method: HTTPMethod.post,
+                client: this,
+                operationType: OperationType.Upsert,
+                path,
+                resourceType,
+                body,
+                resourceId,
+                options,
+                plugins: this.cosmosClientOptions.plugins,
+                partitionKey,
+                pipeline: this.pipeline
+            };
+            request.headers = await this.buildHeaders(request);
+            request.headers[Constants.HttpHeaders.IsUpsert] = true;
+            this.applySessionToken(request);
+            // upsert will use WriteEndpoint since it uses POST operation
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+            const response = await executePlugins(request, executeRequest, PluginOn.operation);
+            this.captureSessionToken(undefined, path, OperationType.Upsert, response.headers);
+            return response;
+        }
+        catch (err) {
+            this.captureSessionToken(err, path, OperationType.Upsert, err.headers);
+            throw err;
+        }
+    }
+    async execute({ sprocLink, params, options = {}, partitionKey }) {
+        // Accept a single parameter or an array of parameters.
+        // Didn't add type annotation for this because we should legacy this behavior
+        if (params !== null && params !== undefined && !Array.isArray(params)) {
+            params = [params];
+        }
+        const path = getPathFromLink(sprocLink);
+        const id = getIdFromLink(sprocLink);
+        const request = {
+            globalEndpointManager: this.globalEndpointManager,
+            requestAgent: this.cosmosClientOptions.agent,
+            connectionPolicy: this.connectionPolicy,
+            method: HTTPMethod.post,
+            client: this,
+            operationType: OperationType.Execute,
+            path,
+            resourceType: ResourceType.sproc,
+            options,
+            resourceId: id,
+            body: params,
+            plugins: this.cosmosClientOptions.plugins,
+            partitionKey,
+            pipeline: this.pipeline
+        };
+        request.headers = await this.buildHeaders(request);
+        // executeStoredProcedure will use WriteEndpoint since it uses POST operation
+        request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+        return executePlugins(request, executeRequest, PluginOn.operation);
+    }
+    /**
+     * Gets the Database account information.
+     * @param options - `urlConnection` in the options is the endpoint url whose database account needs to be retrieved.
+     * If not present, current client's url will be used.
+     */
+    async getDatabaseAccount(options = {}) {
+        const endpoint = options.urlConnection || this.cosmosClientOptions.endpoint;
+        const request = {
+            endpoint,
+            globalEndpointManager: this.globalEndpointManager,
+            requestAgent: this.cosmosClientOptions.agent,
+            connectionPolicy: this.connectionPolicy,
+            method: HTTPMethod.get,
+            client: this,
+            operationType: OperationType.Read,
+            path: "",
+            resourceType: ResourceType.none,
+            options,
+            plugins: this.cosmosClientOptions.plugins,
+            pipeline: this.pipeline
+        };
+        request.headers = await this.buildHeaders(request);
+        // await options.beforeOperation({ endpoint, request, headers: requestHeaders });
+        const { result, headers } = await executePlugins(request, executeRequest, PluginOn.operation);
+        const databaseAccount = new DatabaseAccount(result, headers);
+        return { result: databaseAccount, headers };
+    }
+    getWriteEndpoint() {
+        return this.globalEndpointManager.getWriteEndpoint();
+    }
+    getReadEndpoint() {
+        return this.globalEndpointManager.getReadEndpoint();
+    }
+    getWriteEndpoints() {
+        return this.globalEndpointManager.getWriteEndpoints();
+    }
+    getReadEndpoints() {
+        return this.globalEndpointManager.getReadEndpoints();
+    }
+    async batch({ body, path, partitionKey, resourceId, options = {} }) {
+        try {
+            const request = {
+                globalEndpointManager: this.globalEndpointManager,
+                requestAgent: this.cosmosClientOptions.agent,
+                connectionPolicy: this.connectionPolicy,
+                method: HTTPMethod.post,
+                client: this,
+                operationType: OperationType.Batch,
+                path,
+                body,
+                resourceType: ResourceType.item,
+                resourceId,
+                plugins: this.cosmosClientOptions.plugins,
+                options,
+                pipeline: this.pipeline,
+                partitionKey
+            };
+            request.headers = await this.buildHeaders(request);
+            request.headers[Constants.HttpHeaders.IsBatchRequest] = true;
+            request.headers[Constants.HttpHeaders.IsBatchAtomic] = true;
+            this.applySessionToken(request);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+            const response = await executePlugins(request, executeRequest, PluginOn.operation);
+            this.captureSessionToken(undefined, path, OperationType.Batch, response.headers);
+            return response;
+        }
+        catch (err) {
+            this.captureSessionToken(err, path, OperationType.Upsert, err.headers);
+            throw err;
+        }
+    }
+    async bulk({ body, path, partitionKeyRangeId, resourceId, bulkOptions = {}, options = {} }) {
+        try {
+            const request = {
+                globalEndpointManager: this.globalEndpointManager,
+                requestAgent: this.cosmosClientOptions.agent,
+                connectionPolicy: this.connectionPolicy,
+                method: HTTPMethod.post,
+                client: this,
+                operationType: OperationType.Batch,
+                path,
+                body,
+                resourceType: ResourceType.item,
+                resourceId,
+                plugins: this.cosmosClientOptions.plugins,
+                options,
+                pipeline: this.pipeline
+            };
+            request.headers = await this.buildHeaders(request);
+            request.headers[Constants.HttpHeaders.IsBatchRequest] = true;
+            request.headers[Constants.HttpHeaders.PartitionKeyRangeID] = partitionKeyRangeId;
+            request.headers[Constants.HttpHeaders.IsBatchAtomic] = false;
+            request.headers[Constants.HttpHeaders.BatchContinueOnError] =
+                bulkOptions.continueOnError || false;
+            this.applySessionToken(request);
+            request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(request.resourceType, request.operationType);
+            const response = await executePlugins(request, executeRequest, PluginOn.operation);
+            this.captureSessionToken(undefined, path, OperationType.Batch, response.headers);
+            return response;
+        }
+        catch (err) {
+            this.captureSessionToken(err, path, OperationType.Upsert, err.headers);
+            throw err;
+        }
+    }
+    captureSessionToken(err, path, operationType, resHeaders) {
+        const request = this.getSessionParams(path);
+        request.operationType = operationType;
+        if (!err ||
+            (!this.isMasterResource(request.resourceType) &&
+                (err.code === StatusCodes.PreconditionFailed ||
+                    err.code === StatusCodes.Conflict ||
+                    (err.code === StatusCodes.NotFound &&
+                        err.substatus !== SubStatusCodes.ReadSessionNotAvailable)))) {
+            this.sessionContainer.set(request, resHeaders);
+        }
+    }
+    clearSessionToken(path) {
+        const request = this.getSessionParams(path);
+        this.sessionContainer.remove(request);
+    }
+    getSessionParams(resourceLink) {
+        const resourceId = null;
+        let resourceAddress = null;
+        const parserOutput = parseLink(resourceLink);
+        resourceAddress = parserOutput.objectBody.self;
+        const resourceType = parserOutput.type;
+        return {
+            resourceId,
+            resourceAddress,
+            resourceType,
+            isNameBased: true
+        };
+    }
+    isMasterResource(resourceType) {
+        if (resourceType === Constants.Path.OffersPathSegment ||
+            resourceType === Constants.Path.DatabasesPathSegment ||
+            resourceType === Constants.Path.UsersPathSegment ||
+            resourceType === Constants.Path.PermissionsPathSegment ||
+            resourceType === Constants.Path.TopologyPathSegment ||
+            resourceType === Constants.Path.DatabaseAccountPathSegment ||
+            resourceType === Constants.Path.PartitionKeyRangesPathSegment ||
+            resourceType === Constants.Path.CollectionsPathSegment) {
+            return true;
+        }
+        return false;
+    }
+    buildHeaders(requestContext) {
+        return getHeaders({
+            clientOptions: this.cosmosClientOptions,
+            defaultHeaders: Object.assign(Object.assign({}, this.cosmosClientOptions.defaultHeaders), requestContext.options.initialHeaders),
+            verb: requestContext.method,
+            path: requestContext.path,
+            resourceId: requestContext.resourceId,
+            resourceType: requestContext.resourceType,
+            options: requestContext.options,
+            partitionKeyRangeId: requestContext.partitionKeyRangeId,
+            useMultipleWriteLocations: this.connectionPolicy.useMultipleWriteLocations,
+            partitionKey: requestContext.partitionKey
+        });
+    }
+}
+//# sourceMappingURL=ClientContext.js.map

package/dist-esm/src/ClientContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ClientContext.js","sourceRoot":"","sources":["../../src/ClientContext.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,OAAO,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAC1B,MAAM,IAAI,GAAG,EAAE,CAAC;AAChB,OAAO,EACL,+BAA+B,EAC/B,mBAAmB,EAEpB,MAAM,2BAA2B,CAAC;AAGnC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACxF,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEnE,OAAO,EAAoB,gBAAgB,EAAE,eAAe,EAAgB,MAAM,aAAa,CAAC;AAEhG,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAG5D,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAIhD,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/C,OAAO,EAAE,OAAO,IAAI,cAAc,EAAE,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAG9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAe,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAEhE,MAAM,MAAM,GAAgB,kBAAkB,CAAC,eAAe,CAAC,CAAC;AAEhE,MAAM,oBAAoB,GAAG,wBAAwB,CAAC;AAEtD;;;GAGG;AACH,MAAM,OAAO,aAAa;IAKxB,YACU,mBAAwC,EACxC,qBAA4C;QAD5C,wBAAmB,GAAnB,mBAAmB,CAAqB;QACxC,0BAAqB,GAArB,qBAAqB,CAAuB;QAEpD,IAAI,CAAC,gBAAgB,GAAG,mBAAmB,CAAC,gBAAgB,CAAC;QAC7D,IAAI,CAAC,gBAAgB,GAAG,IAAI,gBAAgB,EAAE,CAAC;QAC/C,IAAI,CAAC,2BAA2B,GAAG,EAAE,CAAC;QACtC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,mBAAmB,CAAC,cAAc,EAAE;YACtC,IAAI,CAAC,QAAQ,GAAG,mBAAmB,EAAE,CAAC;YACtC,MAAM,YAAY,GAAG,gBAAgB,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YACpE,MAAM,KAAK,GAAG,GAAG,YAAY,WAAW,CAAC;YACzC,IAAI,CAAC,QAAQ,CAAC,SAAS,CACrB,+BAA+B,CAAC;gBAC9B,UAAU,EAAE,mBAAmB,CAAC,cAAc;gBAC9C,MAAM,EAAE,KAAK;gBACb,kBAAkB,EAAE;oBAClB,KAAK,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE;wBAChD,MAAM,aAAa,GAAG,MAAM,cAAc,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;wBACxD,MAAM,WAAW,GAAG,uBAAuB,CAAC;wBAC5C,MAAM,kBAAkB,GAAG,GAAG,WAAW,GAAG,aAAa,CAAC,KAAK,EAAE,CAAC;wBAClE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,kBAAkB,CAAC,CAAC;oBAC3D,CAAC;iBACF;aACF,CAAC,CACH,CAAC;SACH;IACH,CAAC;IACD,cAAc;IACP,KAAK,CAAC,IAAI,CAAI,EACnB,IAAI,EACJ,YAAY,EACZ,UAAU,EACV,OAAO,GAAG,EAAE,EACZ,YAAY,EAOb;QACC,IAAI;YACF,MAAM,OAAO,GAAmB;gBAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;gBACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;gBAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,MAAM,EAAE,UAAU,CAAC,GAAG;gBACtB,IAAI;gBACJ,aAAa,EAAE,aAAa,CAAC,IAAI;gBACjC,MAAM,EAAE,IAAI;gBACZ,UAAU;gBACV,OAAO;gBACP,YAAY;gBACZ,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;gBACzC,YAAY;gBACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC;YAEF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACnD,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAEhC,yDAAyD;YACzD,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YACnF,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YAChF,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAG,GAAqB,CAAC,OAAO,CAAC,CAAC;YAC1F,MAAM,GAAG,CAAC;SACX;IACH,CAAC;IAEM,KAAK,CAAC,SAAS,CAAI,EACxB,IAAI,EACJ,YAAY,EACZ,UAAU,EACV,QAAQ,EACR,KAAK,EACL,OAAO,EACP,mBAAmB,EACnB,YAAY,EAUb;QACC,6DAA6D;QAC7D,4DAA4D;QAE5D,MAAM,OAAO,GAAmB;YAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;YACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;YAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,MAAM,EAAE,UAAU,CAAC,GAAG;YACtB,IAAI;YACJ,aAAa,EAAE,aAAa,CAAC,KAAK;YAClC,MAAM,EAAE,IAAI;YACZ,mBAAmB;YACnB,UAAU;YACV,YAAY;YACZ,OAAO;YACP,IAAI,EAAE,KAAK;YACX,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;YACzC,YAAY;YACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,EAAE,CAAC;QACzB,IAAI,KAAK,KAAK,SAAS,EAAE;YACvB,OAAO,CAAC,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC;SAClC;QACD,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;QACF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACnD,IAAI,KAAK,KAAK,SAAS,EAAE;YACvB,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG,MAAM,CAAC;YACxD,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,WAAW,CAAC,GAAG,oBAAoB,CAAC;YAC1E,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;gBAC7B,OAAO,CAAC,IAAI,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,uCAAuC;aAClE;SACF;QACD,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,CAAC,IAAI,CACT,QAAQ;YACN,SAAS;YACT,UAAU;YACV,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAC,UAAU,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC,CAChF,CAAC;QACF,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,CAAC;QAC/C,MAAM,CAAC,IAAI,CAAC,QAAQ,GAAG,SAAS,GAAG,cAAc,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,GAAG,IAAI,CAAC,CAAC;QACjF,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;QACjF,OAAO,IAAI,CAAC,wBAAwB,CAAC,QAAQ,EAAE,CAAC,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IACpE,CAAC;IAEM,KAAK,CAAC,YAAY,CACvB,IAAY,EACZ,YAA0B,EAC1B,UAAkB,EAClB,KAA4B,EAC5B,UAAuB,EAAE;QAEzB,MAAM,OAAO,GAAmB;YAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;YACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;YAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,MAAM,EAAE,UAAU,CAAC,IAAI;YACvB,IAAI;YACJ,aAAa,EAAE,aAAa,CAAC,IAAI;YACjC,MAAM,EAAE,IAAI;YACZ,UAAU;YACV,YAAY;YACZ,OAAO;YACP,IAAI,EAAE,KAAK;YACX,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;YACzC,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC;QAEF,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;QACF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACnD,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,WAAW,CAAC,GAAG,MAAM,CAAC;QAC5D,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,GAAG,KAAK,CAAC;QAC5D,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,sBAAsB,CAAC;YAC3D,wIAAwI,CAAC;QAC3I,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,WAAW,CAAC,GAAG,oBAAoB,CAAC;QAC1E,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;YAC7B,OAAO,CAAC,IAAI,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,uCAAuC;SAClE;QAED,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;QACjF,OAAO,QAAe,CAAC;IACzB,CAAC;IAEM,uBAAuB,CAC5B,cAAsB,EACtB,KAA6B,EAC7B,OAAqB;QAErB,MAAM,IAAI,GAAG,eAAe,CAAC,cAAc,EAAE,YAAY,CAAC,QAAQ,CAAC,CAAC;QACpE,MAAM,EAAE,GAAG,aAAa,CAAC,cAAc,CAAC,CAAC;QACzC,MAAM,EAAE,GAA0B,CAAC,YAAY,EAAE,EAAE;YACjD,OAAO,IAAI,CAAC,SAAS,CAAC;gBACpB,IAAI;gBACJ,YAAY,EAAE,YAAY,CAAC,QAAQ;gBACnC,UAAU,EAAE,EAAE;gBACd,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,kBAAkB;gBAC/C,KAAK;gBACL,OAAO,EAAE,YAAY;aACtB,CAAC,CAAC;QACL,CAAC,CAAC;QACF,OAAO,IAAI,aAAa,CAAoB,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IACxE,CAAC;IAEM,KAAK,CAAC,MAAM,CAAI,EACrB,IAAI,EACJ,YAAY,EACZ,UAAU,EACV,OAAO,GAAG,EAAE,EACZ,YAAY,EAOb;QACC,IAAI;YACF,MAAM,OAAO,GAAmB;gBAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;gBACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;gBAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,MAAM,EAAE,UAAU,CAAC,MAAM;gBACzB,MAAM,EAAE,IAAI;gBACZ,aAAa,EAAE,aAAa,CAAC,MAAM;gBACnC,IAAI;gBACJ,YAAY;gBACZ,OAAO;gBACP,UAAU;gBACV,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;gBACzC,YAAY;gBACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC;YAEF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACnD,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAChC,uEAAuE;YACvE,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YACnF,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE;gBACpC,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;aACnF;iBAAM;gBACL,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;aAC9B;YACD,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAG,GAAqB,CAAC,OAAO,CAAC,CAAC;YAC1F,MAAM,GAAG,CAAC;SACX;IACH,CAAC;IAEM,KAAK,CAAC,KAAK,CAAI,EACpB,IAAI,EACJ,IAAI,EACJ,YAAY,EACZ,UAAU,EACV,OAAO,GAAG,EAAE,EACZ,YAAY,EAQb;QACC,IAAI;YACF,MAAM,OAAO,GAAmB;gBAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;gBACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;gBAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,MAAM,EAAE,UAAU,CAAC,KAAK;gBACxB,MAAM,EAAE,IAAI;gBACZ,aAAa,EAAE,aAAa,CAAC,KAAK;gBAClC,IAAI;gBACJ,YAAY;gBACZ,IAAI;gBACJ,UAAU;gBACV,OAAO;gBACP,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;gBACzC,YAAY;aACb,CAAC;YAEF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACnD,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAEhC,+BAA+B;YAC/B,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YACnF,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YACjF,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAG,GAAqB,CAAC,OAAO,CAAC,CAAC;YAC1F,MAAM,GAAG,CAAC;SACX;IACH,CAAC;IAEM,KAAK,CAAC,MAAM,CAAW,EAC5B,IAAI,EACJ,IAAI,EACJ,YAAY,EACZ,UAAU,EACV,OAAO,GAAG,EAAE,EACZ,YAAY,EAQb;QACC,IAAI;YACF,MAAM,OAAO,GAAmB;gBAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;gBACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;gBAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,MAAM,EAAE,UAAU,CAAC,IAAI;gBACvB,MAAM,EAAE,IAAI;gBACZ,aAAa,EAAE,aAAa,CAAC,MAAM;gBACnC,IAAI;gBACJ,YAAY;gBACZ,UAAU;gBACV,IAAI;gBACJ,OAAO;gBACP,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;gBACzC,YAAY;gBACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC;YAEF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACnD,6DAA6D;YAC7D,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAEhC,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YACnF,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YAClF,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAG,GAAqB,CAAC,OAAO,CAAC,CAAC;YAC1F,MAAM,GAAG,CAAC;SACX;IACH,CAAC;IAEO,wBAAwB,CAC9B,GAAkB,EAClB,OAAgB,EAChB,QAAmD;QAEnD,IAAI,OAAO,EAAE;YACX,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC;SAC/E;aAAM;YACL,MAAM,SAAS,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;YAChE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC;SACpE;IACH,CAAC;IAEO,iBAAiB,CAAC,cAA8B;QACtD,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAE3D,IAAI,cAAc,CAAC,OAAO,IAAI,cAAc,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE;YACxF,OAAO;SACR;QAED,MAAM,kBAAkB,GAAqB,cAAc,CAAC,OAAO,CACjE,SAAS,CAAC,WAAW,CAAC,gBAAgB,CACnB,CAAC;QACtB,IAAI,CAAC,kBAAkB,EAAE;YACvB,OAAO;SACR;QAED,IAAI,kBAAkB,KAAK,gBAAgB,CAAC,OAAO,EAAE;YACnD,OAAO;SACR;QAED,IAAI,OAAO,CAAC,eAAe,EAAE;YAC3B,MAAM,YAAY,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACxD,IAAI,YAAY,EAAE;gBAChB,cAAc,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,YAAY,CAAC,GAAG,YAAY,CAAC;aAC3E;SACF;IACH,CAAC;IAEM,KAAK,CAAC,OAAO,CAAI,EACtB,IAAI,EACJ,IAAI,EACJ,YAAY,EACZ,UAAU,EACV,OAAO,GAAG,EAAE,EACZ,YAAY,EAQb;QACC,IAAI;YACF,MAAM,OAAO,GAAmB;gBAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;gBACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;gBAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,MAAM,EAAE,UAAU,CAAC,GAAG;gBACtB,MAAM,EAAE,IAAI;gBACZ,aAAa,EAAE,aAAa,CAAC,OAAO;gBACpC,IAAI;gBACJ,YAAY;gBACZ,IAAI;gBACJ,UAAU;gBACV,OAAO;gBACP,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;gBACzC,YAAY;gBACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC;YAEF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACnD,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAEhC,6DAA6D;YAC7D,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YACnF,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YACnF,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAG,GAAqB,CAAC,OAAO,CAAC,CAAC;YAC1F,MAAM,GAAG,CAAC;SACX;IACH,CAAC;IAEM,KAAK,CAAC,MAAM,CAAW,EAC5B,IAAI,EACJ,IAAI,EACJ,YAAY,EACZ,UAAU,EACV,OAAO,GAAG,EAAE,EACZ,YAAY,EAQb;QACC,IAAI;YACF,MAAM,OAAO,GAAmB;gBAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;gBACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;gBAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,MAAM,EAAE,UAAU,CAAC,IAAI;gBACvB,MAAM,EAAE,IAAI;gBACZ,aAAa,EAAE,aAAa,CAAC,MAAM;gBACnC,IAAI;gBACJ,YAAY;gBACZ,IAAI;gBACJ,UAAU;gBACV,OAAO;gBACP,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;gBACzC,YAAY;gBACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC;YAEF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACnD,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC;YACvD,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAEhC,6DAA6D;YAC7D,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YACnF,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YAClF,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAG,GAAqB,CAAC,OAAO,CAAC,CAAC;YAC1F,MAAM,GAAG,CAAC;SACX;IACH,CAAC;IAEM,KAAK,CAAC,OAAO,CAAI,EACtB,SAAS,EACT,MAAM,EACN,OAAO,GAAG,EAAE,EACZ,YAAY,EAMb;QACC,uDAAuD;QACvD,6EAA6E;QAC7E,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;YACrE,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC;SACnB;QACD,MAAM,IAAI,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;QACxC,MAAM,EAAE,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;QAEpC,MAAM,OAAO,GAAmB;YAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;YACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;YAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,MAAM,EAAE,UAAU,CAAC,IAAI;YACvB,MAAM,EAAE,IAAI;YACZ,aAAa,EAAE,aAAa,CAAC,OAAO;YACpC,IAAI;YACJ,YAAY,EAAE,YAAY,CAAC,KAAK;YAChC,OAAO;YACP,UAAU,EAAE,EAAE;YACd,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;YACzC,YAAY;YACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC;QAEF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACnD,6EAA6E;QAC7E,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;QACF,OAAO,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;IACrE,CAAC;IAED;;;;OAIG;IACI,KAAK,CAAC,kBAAkB,CAC7B,UAA0B,EAAE;QAE5B,MAAM,QAAQ,GAAG,OAAO,CAAC,aAAa,IAAI,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC;QAC5E,MAAM,OAAO,GAAmB;YAC9B,QAAQ;YACR,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;YACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;YAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,MAAM,EAAE,UAAU,CAAC,GAAG;YACtB,MAAM,EAAE,IAAI;YACZ,aAAa,EAAE,aAAa,CAAC,IAAI;YACjC,IAAI,EAAE,EAAE;YACR,YAAY,EAAE,YAAY,CAAC,IAAI;YAC/B,OAAO;YACP,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;YACzC,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC;QAEF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACnD,iFAAiF;QACjF,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;QAE9F,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAE7D,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,OAAO,EAAE,CAAC;IAC9C,CAAC;IAEM,gBAAgB;QACrB,OAAO,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,EAAE,CAAC;IACvD,CAAC;IAEM,eAAe;QACpB,OAAO,IAAI,CAAC,qBAAqB,CAAC,eAAe,EAAE,CAAC;IACtD,CAAC;IAEM,iBAAiB;QACtB,OAAO,IAAI,CAAC,qBAAqB,CAAC,iBAAiB,EAAE,CAAC;IACxD,CAAC;IAEM,gBAAgB;QACrB,OAAO,IAAI,CAAC,qBAAqB,CAAC,gBAAgB,EAAE,CAAC;IACvD,CAAC;IAEM,KAAK,CAAC,KAAK,CAAI,EACpB,IAAI,EACJ,IAAI,EACJ,YAAY,EACZ,UAAU,EACV,OAAO,GAAG,EAAE,EAOb;QACC,IAAI;YACF,MAAM,OAAO,GAAmB;gBAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;gBACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;gBAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,MAAM,EAAE,UAAU,CAAC,IAAI;gBACvB,MAAM,EAAE,IAAI;gBACZ,aAAa,EAAE,aAAa,CAAC,KAAK;gBAClC,IAAI;gBACJ,IAAI;gBACJ,YAAY,EAAE,YAAY,CAAC,IAAI;gBAC/B,UAAU;gBACV,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;gBACzC,OAAO;gBACP,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,YAAY;aACb,CAAC;YAEF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACnD,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC;YAC7D,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC;YAE5D,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAEhC,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YACnF,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YACjF,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAG,GAAqB,CAAC,OAAO,CAAC,CAAC;YAC1F,MAAM,GAAG,CAAC;SACX;IACH,CAAC;IAEM,KAAK,CAAC,IAAI,CAAI,EACnB,IAAI,EACJ,IAAI,EACJ,mBAAmB,EACnB,UAAU,EACV,WAAW,GAAG,EAAE,EAChB,OAAO,GAAG,EAAE,EAQb;QACC,IAAI;YACF,MAAM,OAAO,GAAmB;gBAC9B,qBAAqB,EAAE,IAAI,CAAC,qBAAqB;gBACjD,YAAY,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK;gBAC5C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;gBACvC,MAAM,EAAE,UAAU,CAAC,IAAI;gBACvB,MAAM,EAAE,IAAI;gBACZ,aAAa,EAAE,aAAa,CAAC,KAAK;gBAClC,IAAI;gBACJ,IAAI;gBACJ,YAAY,EAAE,YAAY,CAAC,IAAI;gBAC/B,UAAU;gBACV,OAAO,EAAE,IAAI,CAAC,mBAAmB,CAAC,OAAO;gBACzC,OAAO;gBACP,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC;YAEF,OAAO,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACnD,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC;YAC7D,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,mBAAmB,CAAC,GAAG,mBAAmB,CAAC;YACjF,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,KAAK,CAAC;YAC7D,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,WAAW,CAAC,oBAAoB,CAAC;gBACzD,WAAW,CAAC,eAAe,IAAI,KAAK,CAAC;YAEvC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAEhC,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,sBAAsB,CACxE,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,aAAa,CACtB,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,cAAc,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;YACnF,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YACjF,OAAO,QAAQ,CAAC;SACjB;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,IAAI,EAAE,aAAa,CAAC,MAAM,EAAG,GAAqB,CAAC,OAAO,CAAC,CAAC;YAC1F,MAAM,GAAG,CAAC;SACX;IACH,CAAC;IAEO,mBAAmB,CACzB,GAAkB,EAClB,IAAY,EACZ,aAA4B,EAC5B,UAAyB;QAEzB,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC5C,OAAO,CAAC,aAAa,GAAG,aAAa,CAAC;QACtC,IACE,CAAC,GAAG;YACJ,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,YAAY,CAAC;gBAC3C,CAAC,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC,kBAAkB;oBAC1C,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC,QAAQ;oBACjC,CAAC,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC,QAAQ;wBAChC,GAAG,CAAC,SAAS,KAAK,cAAc,CAAC,uBAAuB,CAAC,CAAC,CAAC,EACjE;YACA,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;SAChD;IACH,CAAC;IAEM,iBAAiB,CAAC,IAAY;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAEO,gBAAgB,CAAC,YAAoB;QAC3C,MAAM,UAAU,GAAW,IAAI,CAAC;QAChC,IAAI,eAAe,GAAW,IAAI,CAAC;QACnC,MAAM,YAAY,GAAG,SAAS,CAAC,YAAY,CAAC,CAAC;QAE7C,eAAe,GAAG,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC;QAE/C,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,CAAC;QACvC,OAAO;YACL,UAAU;YACV,eAAe;YACf,YAAY;YACZ,WAAW,EAAE,IAAI;SAClB,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,YAAoB;QAC3C,IACE,YAAY,KAAK,SAAS,CAAC,IAAI,CAAC,iBAAiB;YACjD,YAAY,KAAK,SAAS,CAAC,IAAI,CAAC,oBAAoB;YACpD,YAAY,KAAK,SAAS,CAAC,IAAI,CAAC,gBAAgB;YAChD,YAAY,KAAK,SAAS,CAAC,IAAI,CAAC,sBAAsB;YACtD,YAAY,KAAK,SAAS,CAAC,IAAI,CAAC,mBAAmB;YACnD,YAAY,KAAK,SAAS,CAAC,IAAI,CAAC,0BAA0B;YAC1D,YAAY,KAAK,SAAS,CAAC,IAAI,CAAC,6BAA6B;YAC7D,YAAY,KAAK,SAAS,CAAC,IAAI,CAAC,sBAAsB,EACtD;YACA,OAAO,IAAI,CAAC;SACb;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,YAAY,CAAC,cAA8B;QACjD,OAAO,UAAU,CAAC;YAChB,aAAa,EAAE,IAAI,CAAC,mBAAmB;YACvC,cAAc,kCACT,IAAI,CAAC,mBAAmB,CAAC,cAAc,GACvC,cAAc,CAAC,OAAO,CAAC,cAAc,CACzC;YACD,IAAI,EAAE,cAAc,CAAC,MAAM;YAC3B,IAAI,EAAE,cAAc,CAAC,IAAI;YACzB,UAAU,EAAE,cAAc,CAAC,UAAU;YACrC,YAAY,EAAE,cAAc,CAAC,YAAY;YACzC,OAAO,EAAE,cAAc,CAAC,OAAO;YAC/B,mBAAmB,EAAE,cAAc,CAAC,mBAAmB;YACvD,yBAAyB,EAAE,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;YAC1E,YAAY,EAAE,cAAc,CAAC,YAAY;SAC1C,CAAC,CAAC;IACL,CAAC;CACF","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT license.\nimport { v4 } from \"uuid\";\nconst uuid = v4;\nimport {\n  bearerTokenAuthenticationPolicy,\n  createEmptyPipeline,\n  Pipeline\n} from \"@azure/core-rest-pipeline\";\nimport { PartitionKeyRange } from \"./client/Container/PartitionKeyRange\";\nimport { Resource } from \"./client/Resource\";\nimport { Constants, HTTPMethod, OperationType, ResourceType } from \"./common/constants\";\nimport { getIdFromLink, getPathFromLink, parseLink } from \"./common/helper\";\nimport { StatusCodes, SubStatusCodes } from \"./common/statusCodes\";\nimport { CosmosClientOptions } from \"./CosmosClientOptions\";\nimport { ConnectionPolicy, ConsistencyLevel, DatabaseAccount, PartitionKey } from \"./documents\";\nimport { GlobalEndpointManager } from \"./globalEndpointManager\";\nimport { executePlugins, PluginOn } from \"./plugins/Plugin\";\nimport { FetchFunctionCallback, SqlQuerySpec } from \"./queryExecutionContext\";\nimport { CosmosHeaders } from \"./queryExecutionContext/CosmosHeaders\";\nimport { QueryIterator } from \"./queryIterator\";\nimport { ErrorResponse } from \"./request\";\nimport { FeedOptions, RequestOptions, Response } from \"./request\";\nimport { PartitionedQueryExecutionInfo } from \"./request/ErrorResponse\";\nimport { getHeaders } from \"./request/request\";\nimport { RequestContext } from \"./request/RequestContext\";\nimport { request as executeRequest } from \"./request/RequestHandler\";\nimport { SessionContainer } from \"./session/sessionContainer\";\nimport { SessionContext } from \"./session/SessionContext\";\nimport { BulkOptions } from \"./utils/batch\";\nimport { sanitizeEndpoint } from \"./utils/checkURL\";\nimport { AzureLogger, createClientLogger } from \"@azure/logger\";\n\nconst logger: AzureLogger = createClientLogger(\"ClientContext\");\n\nconst QueryJsonContentType = \"application/query+json\";\n\n/**\n * @hidden\n * @hidden\n */\nexport class ClientContext {\n  private readonly sessionContainer: SessionContainer;\n  private connectionPolicy: ConnectionPolicy;\n  private pipeline: Pipeline;\n  public partitionKeyDefinitionCache: { [containerUrl: string]: any }; // TODO: PartitionKeyDefinitionCache\n  public constructor(\n    private cosmosClientOptions: CosmosClientOptions,\n    private globalEndpointManager: GlobalEndpointManager\n  ) {\n    this.connectionPolicy = cosmosClientOptions.connectionPolicy;\n    this.sessionContainer = new SessionContainer();\n    this.partitionKeyDefinitionCache = {};\n    this.pipeline = null;\n    if (cosmosClientOptions.aadCredentials) {\n      this.pipeline = createEmptyPipeline();\n      const hrefEndpoint = sanitizeEndpoint(cosmosClientOptions.endpoint);\n      const scope = `${hrefEndpoint}/.default`;\n      this.pipeline.addPolicy(\n        bearerTokenAuthenticationPolicy({\n          credential: cosmosClientOptions.aadCredentials,\n          scopes: scope,\n          challengeCallbacks: {\n            async authorizeRequest({ request, getAccessToken }) {\n              const tokenResponse = await getAccessToken([scope], {});\n              const AUTH_PREFIX = `type=aad&ver=1.0&sig=`;\n              const authorizationToken = `${AUTH_PREFIX}${tokenResponse.token}`;\n              request.headers.set(\"Authorization\", authorizationToken);\n            }\n          }\n        })\n      );\n    }\n  }\n  /** @hidden */\n  public async read<T>({\n    path,\n    resourceType,\n    resourceId,\n    options = {},\n    partitionKey\n  }: {\n    path: string;\n    resourceType: ResourceType;\n    resourceId: string;\n    options?: RequestOptions;\n    partitionKey?: PartitionKey;\n  }): Promise<Response<T & Resource>> {\n    try {\n      const request: RequestContext = {\n        globalEndpointManager: this.globalEndpointManager,\n        requestAgent: this.cosmosClientOptions.agent,\n        connectionPolicy: this.connectionPolicy,\n        method: HTTPMethod.get,\n        path,\n        operationType: OperationType.Read,\n        client: this,\n        resourceId,\n        options,\n        resourceType,\n        plugins: this.cosmosClientOptions.plugins,\n        partitionKey,\n        pipeline: this.pipeline\n      };\n\n      request.headers = await this.buildHeaders(request);\n      this.applySessionToken(request);\n\n      // read will use ReadEndpoint since it uses GET operation\n      request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n        request.resourceType,\n        request.operationType\n      );\n      const response = await executePlugins(request, executeRequest, PluginOn.operation);\n      this.captureSessionToken(undefined, path, OperationType.Read, response.headers);\n      return response;\n    } catch (err) {\n      this.captureSessionToken(err, path, OperationType.Upsert, (err as ErrorResponse).headers);\n      throw err;\n    }\n  }\n\n  public async queryFeed<T>({\n    path,\n    resourceType,\n    resourceId,\n    resultFn,\n    query,\n    options,\n    partitionKeyRangeId,\n    partitionKey\n  }: {\n    path: string;\n    resourceType: ResourceType;\n    resourceId: string;\n    resultFn: (result: { [key: string]: any }) => any[];\n    query: SqlQuerySpec | string;\n    options: FeedOptions;\n    partitionKeyRangeId?: string;\n    partitionKey?: PartitionKey;\n  }): Promise<Response<T & Resource>> {\n    // Query operations will use ReadEndpoint even though it uses\n    // GET(for queryFeed) and POST(for regular query operations)\n\n    const request: RequestContext = {\n      globalEndpointManager: this.globalEndpointManager,\n      requestAgent: this.cosmosClientOptions.agent,\n      connectionPolicy: this.connectionPolicy,\n      method: HTTPMethod.get,\n      path,\n      operationType: OperationType.Query,\n      client: this,\n      partitionKeyRangeId,\n      resourceId,\n      resourceType,\n      options,\n      body: query,\n      plugins: this.cosmosClientOptions.plugins,\n      partitionKey,\n      pipeline: this.pipeline\n    };\n    const requestId = uuid();\n    if (query !== undefined) {\n      request.method = HTTPMethod.post;\n    }\n    request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n      request.resourceType,\n      request.operationType\n    );\n    request.headers = await this.buildHeaders(request);\n    if (query !== undefined) {\n      request.headers[Constants.HttpHeaders.IsQuery] = \"true\";\n      request.headers[Constants.HttpHeaders.ContentType] = QueryJsonContentType;\n      if (typeof query === \"string\") {\n        request.body = { query }; // Converts query text to query object.\n      }\n    }\n    this.applySessionToken(request);\n    logger.info(\n      \"query \" +\n        requestId +\n        \" started\" +\n        (request.partitionKeyRangeId ? \" pkrid: \" + request.partitionKeyRangeId : \"\")\n    );\n    logger.verbose(request);\n    const start = Date.now();\n    const response = await executeRequest(request);\n    logger.info(\"query \" + requestId + \" finished - \" + (Date.now() - start) + \"ms\");\n    this.captureSessionToken(undefined, path, OperationType.Query, response.headers);\n    return this.processQueryFeedResponse(response, !!query, resultFn);\n  }\n\n  public async getQueryPlan(\n    path: string,\n    resourceType: ResourceType,\n    resourceId: string,\n    query: SqlQuerySpec | string,\n    options: FeedOptions = {}\n  ): Promise<Response<PartitionedQueryExecutionInfo>> {\n    const request: RequestContext = {\n      globalEndpointManager: this.globalEndpointManager,\n      requestAgent: this.cosmosClientOptions.agent,\n      connectionPolicy: this.connectionPolicy,\n      method: HTTPMethod.post,\n      path,\n      operationType: OperationType.Read,\n      client: this,\n      resourceId,\n      resourceType,\n      options,\n      body: query,\n      plugins: this.cosmosClientOptions.plugins,\n      pipeline: this.pipeline\n    };\n\n    request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n      request.resourceType,\n      request.operationType\n    );\n    request.headers = await this.buildHeaders(request);\n    request.headers[Constants.HttpHeaders.IsQueryPlan] = \"True\";\n    request.headers[Constants.HttpHeaders.QueryVersion] = \"1.4\";\n    request.headers[Constants.HttpHeaders.SupportedQueryFeatures] =\n      \"NonValueAggregate, Aggregate, Distinct, MultipleOrderBy, OffsetAndLimit, OrderBy, Top, CompositeAggregate, GroupBy, MultipleAggregates\";\n    request.headers[Constants.HttpHeaders.ContentType] = QueryJsonContentType;\n    if (typeof query === \"string\") {\n      request.body = { query }; // Converts query text to query object.\n    }\n\n    this.applySessionToken(request);\n    const response = await executeRequest(request);\n    this.captureSessionToken(undefined, path, OperationType.Query, response.headers);\n    return response as any;\n  }\n\n  public queryPartitionKeyRanges(\n    collectionLink: string,\n    query?: string | SqlQuerySpec,\n    options?: FeedOptions\n  ): QueryIterator<PartitionKeyRange> {\n    const path = getPathFromLink(collectionLink, ResourceType.pkranges);\n    const id = getIdFromLink(collectionLink);\n    const cb: FetchFunctionCallback = (innerOptions) => {\n      return this.queryFeed({\n        path,\n        resourceType: ResourceType.pkranges,\n        resourceId: id,\n        resultFn: (result) => result.PartitionKeyRanges,\n        query,\n        options: innerOptions\n      });\n    };\n    return new QueryIterator<PartitionKeyRange>(this, query, options, cb);\n  }\n\n  public async delete<T>({\n    path,\n    resourceType,\n    resourceId,\n    options = {},\n    partitionKey\n  }: {\n    path: string;\n    resourceType: ResourceType;\n    resourceId: string;\n    options?: RequestOptions;\n    partitionKey?: PartitionKey;\n  }): Promise<Response<T & Resource>> {\n    try {\n      const request: RequestContext = {\n        globalEndpointManager: this.globalEndpointManager,\n        requestAgent: this.cosmosClientOptions.agent,\n        connectionPolicy: this.connectionPolicy,\n        method: HTTPMethod.delete,\n        client: this,\n        operationType: OperationType.Delete,\n        path,\n        resourceType,\n        options,\n        resourceId,\n        plugins: this.cosmosClientOptions.plugins,\n        partitionKey,\n        pipeline: this.pipeline\n      };\n\n      request.headers = await this.buildHeaders(request);\n      this.applySessionToken(request);\n      // deleteResource will use WriteEndpoint since it uses DELETE operation\n      request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n        request.resourceType,\n        request.operationType\n      );\n      const response = await executePlugins(request, executeRequest, PluginOn.operation);\n      if (parseLink(path).type !== \"colls\") {\n        this.captureSessionToken(undefined, path, OperationType.Delete, response.headers);\n      } else {\n        this.clearSessionToken(path);\n      }\n      return response;\n    } catch (err) {\n      this.captureSessionToken(err, path, OperationType.Upsert, (err as ErrorResponse).headers);\n      throw err;\n    }\n  }\n\n  public async patch<T>({\n    body,\n    path,\n    resourceType,\n    resourceId,\n    options = {},\n    partitionKey\n  }: {\n    body: any;\n    path: string;\n    resourceType: ResourceType;\n    resourceId: string;\n    options?: RequestOptions;\n    partitionKey?: PartitionKey;\n  }): Promise<Response<T & Resource>> {\n    try {\n      const request: RequestContext = {\n        globalEndpointManager: this.globalEndpointManager,\n        requestAgent: this.cosmosClientOptions.agent,\n        connectionPolicy: this.connectionPolicy,\n        method: HTTPMethod.patch,\n        client: this,\n        operationType: OperationType.Patch,\n        path,\n        resourceType,\n        body,\n        resourceId,\n        options,\n        plugins: this.cosmosClientOptions.plugins,\n        partitionKey\n      };\n\n      request.headers = await this.buildHeaders(request);\n      this.applySessionToken(request);\n\n      // patch will use WriteEndpoint\n      request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n        request.resourceType,\n        request.operationType\n      );\n      const response = await executePlugins(request, executeRequest, PluginOn.operation);\n      this.captureSessionToken(undefined, path, OperationType.Patch, response.headers);\n      return response;\n    } catch (err) {\n      this.captureSessionToken(err, path, OperationType.Upsert, (err as ErrorResponse).headers);\n      throw err;\n    }\n  }\n\n  public async create<T, U = T>({\n    body,\n    path,\n    resourceType,\n    resourceId,\n    options = {},\n    partitionKey\n  }: {\n    body: T;\n    path: string;\n    resourceType: ResourceType;\n    resourceId: string;\n    options?: RequestOptions;\n    partitionKey?: PartitionKey;\n  }): Promise<Response<T & U & Resource>> {\n    try {\n      const request: RequestContext = {\n        globalEndpointManager: this.globalEndpointManager,\n        requestAgent: this.cosmosClientOptions.agent,\n        connectionPolicy: this.connectionPolicy,\n        method: HTTPMethod.post,\n        client: this,\n        operationType: OperationType.Create,\n        path,\n        resourceType,\n        resourceId,\n        body,\n        options,\n        plugins: this.cosmosClientOptions.plugins,\n        partitionKey,\n        pipeline: this.pipeline\n      };\n\n      request.headers = await this.buildHeaders(request);\n      // create will use WriteEndpoint since it uses POST operation\n      this.applySessionToken(request);\n\n      request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n        request.resourceType,\n        request.operationType\n      );\n      const response = await executePlugins(request, executeRequest, PluginOn.operation);\n      this.captureSessionToken(undefined, path, OperationType.Create, response.headers);\n      return response;\n    } catch (err) {\n      this.captureSessionToken(err, path, OperationType.Upsert, (err as ErrorResponse).headers);\n      throw err;\n    }\n  }\n\n  private processQueryFeedResponse(\n    res: Response<any>,\n    isQuery: boolean,\n    resultFn: (result: { [key: string]: any }) => any[]\n  ): Response<any> {\n    if (isQuery) {\n      return { result: resultFn(res.result), headers: res.headers, code: res.code };\n    } else {\n      const newResult = resultFn(res.result).map((body: any) => body);\n      return { result: newResult, headers: res.headers, code: res.code };\n    }\n  }\n\n  private applySessionToken(requestContext: RequestContext): void {\n    const request = this.getSessionParams(requestContext.path);\n\n    if (requestContext.headers && requestContext.headers[Constants.HttpHeaders.SessionToken]) {\n      return;\n    }\n\n    const sessionConsistency: ConsistencyLevel = requestContext.headers[\n      Constants.HttpHeaders.ConsistencyLevel\n    ] as ConsistencyLevel;\n    if (!sessionConsistency) {\n      return;\n    }\n\n    if (sessionConsistency !== ConsistencyLevel.Session) {\n      return;\n    }\n\n    if (request.resourceAddress) {\n      const sessionToken = this.sessionContainer.get(request);\n      if (sessionToken) {\n        requestContext.headers[Constants.HttpHeaders.SessionToken] = sessionToken;\n      }\n    }\n  }\n\n  public async replace<T>({\n    body,\n    path,\n    resourceType,\n    resourceId,\n    options = {},\n    partitionKey\n  }: {\n    body: any;\n    path: string;\n    resourceType: ResourceType;\n    resourceId: string;\n    options?: RequestOptions;\n    partitionKey?: PartitionKey;\n  }): Promise<Response<T & Resource>> {\n    try {\n      const request: RequestContext = {\n        globalEndpointManager: this.globalEndpointManager,\n        requestAgent: this.cosmosClientOptions.agent,\n        connectionPolicy: this.connectionPolicy,\n        method: HTTPMethod.put,\n        client: this,\n        operationType: OperationType.Replace,\n        path,\n        resourceType,\n        body,\n        resourceId,\n        options,\n        plugins: this.cosmosClientOptions.plugins,\n        partitionKey,\n        pipeline: this.pipeline\n      };\n\n      request.headers = await this.buildHeaders(request);\n      this.applySessionToken(request);\n\n      // replace will use WriteEndpoint since it uses PUT operation\n      request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n        request.resourceType,\n        request.operationType\n      );\n      const response = await executePlugins(request, executeRequest, PluginOn.operation);\n      this.captureSessionToken(undefined, path, OperationType.Replace, response.headers);\n      return response;\n    } catch (err) {\n      this.captureSessionToken(err, path, OperationType.Upsert, (err as ErrorResponse).headers);\n      throw err;\n    }\n  }\n\n  public async upsert<T, U = T>({\n    body,\n    path,\n    resourceType,\n    resourceId,\n    options = {},\n    partitionKey\n  }: {\n    body: T;\n    path: string;\n    resourceType: ResourceType;\n    resourceId: string;\n    options?: RequestOptions;\n    partitionKey?: PartitionKey;\n  }): Promise<Response<T & U & Resource>> {\n    try {\n      const request: RequestContext = {\n        globalEndpointManager: this.globalEndpointManager,\n        requestAgent: this.cosmosClientOptions.agent,\n        connectionPolicy: this.connectionPolicy,\n        method: HTTPMethod.post,\n        client: this,\n        operationType: OperationType.Upsert,\n        path,\n        resourceType,\n        body,\n        resourceId,\n        options,\n        plugins: this.cosmosClientOptions.plugins,\n        partitionKey,\n        pipeline: this.pipeline\n      };\n\n      request.headers = await this.buildHeaders(request);\n      request.headers[Constants.HttpHeaders.IsUpsert] = true;\n      this.applySessionToken(request);\n\n      // upsert will use WriteEndpoint since it uses POST operation\n      request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n        request.resourceType,\n        request.operationType\n      );\n      const response = await executePlugins(request, executeRequest, PluginOn.operation);\n      this.captureSessionToken(undefined, path, OperationType.Upsert, response.headers);\n      return response;\n    } catch (err) {\n      this.captureSessionToken(err, path, OperationType.Upsert, (err as ErrorResponse).headers);\n      throw err;\n    }\n  }\n\n  public async execute<T>({\n    sprocLink,\n    params,\n    options = {},\n    partitionKey\n  }: {\n    sprocLink: string;\n    params?: any[];\n    options?: RequestOptions;\n    partitionKey?: PartitionKey;\n  }): Promise<Response<T>> {\n    // Accept a single parameter or an array of parameters.\n    // Didn't add type annotation for this because we should legacy this behavior\n    if (params !== null && params !== undefined && !Array.isArray(params)) {\n      params = [params];\n    }\n    const path = getPathFromLink(sprocLink);\n    const id = getIdFromLink(sprocLink);\n\n    const request: RequestContext = {\n      globalEndpointManager: this.globalEndpointManager,\n      requestAgent: this.cosmosClientOptions.agent,\n      connectionPolicy: this.connectionPolicy,\n      method: HTTPMethod.post,\n      client: this,\n      operationType: OperationType.Execute,\n      path,\n      resourceType: ResourceType.sproc,\n      options,\n      resourceId: id,\n      body: params,\n      plugins: this.cosmosClientOptions.plugins,\n      partitionKey,\n      pipeline: this.pipeline\n    };\n\n    request.headers = await this.buildHeaders(request);\n    // executeStoredProcedure will use WriteEndpoint since it uses POST operation\n    request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n      request.resourceType,\n      request.operationType\n    );\n    return executePlugins(request, executeRequest, PluginOn.operation);\n  }\n\n  /**\n   * Gets the Database account information.\n   * @param options - `urlConnection` in the options is the endpoint url whose database account needs to be retrieved.\n   * If not present, current client's url will be used.\n   */\n  public async getDatabaseAccount(\n    options: RequestOptions = {}\n  ): Promise<Response<DatabaseAccount>> {\n    const endpoint = options.urlConnection || this.cosmosClientOptions.endpoint;\n    const request: RequestContext = {\n      endpoint,\n      globalEndpointManager: this.globalEndpointManager,\n      requestAgent: this.cosmosClientOptions.agent,\n      connectionPolicy: this.connectionPolicy,\n      method: HTTPMethod.get,\n      client: this,\n      operationType: OperationType.Read,\n      path: \"\",\n      resourceType: ResourceType.none,\n      options,\n      plugins: this.cosmosClientOptions.plugins,\n      pipeline: this.pipeline\n    };\n\n    request.headers = await this.buildHeaders(request);\n    // await options.beforeOperation({ endpoint, request, headers: requestHeaders });\n    const { result, headers } = await executePlugins(request, executeRequest, PluginOn.operation);\n\n    const databaseAccount = new DatabaseAccount(result, headers);\n\n    return { result: databaseAccount, headers };\n  }\n\n  public getWriteEndpoint(): Promise<string> {\n    return this.globalEndpointManager.getWriteEndpoint();\n  }\n\n  public getReadEndpoint(): Promise<string> {\n    return this.globalEndpointManager.getReadEndpoint();\n  }\n\n  public getWriteEndpoints(): Promise<readonly string[]> {\n    return this.globalEndpointManager.getWriteEndpoints();\n  }\n\n  public getReadEndpoints(): Promise<readonly string[]> {\n    return this.globalEndpointManager.getReadEndpoints();\n  }\n\n  public async batch<T>({\n    body,\n    path,\n    partitionKey,\n    resourceId,\n    options = {}\n  }: {\n    body: T;\n    path: string;\n    partitionKey: string;\n    resourceId: string;\n    options?: RequestOptions;\n  }): Promise<Response<any>> {\n    try {\n      const request: RequestContext = {\n        globalEndpointManager: this.globalEndpointManager,\n        requestAgent: this.cosmosClientOptions.agent,\n        connectionPolicy: this.connectionPolicy,\n        method: HTTPMethod.post,\n        client: this,\n        operationType: OperationType.Batch,\n        path,\n        body,\n        resourceType: ResourceType.item,\n        resourceId,\n        plugins: this.cosmosClientOptions.plugins,\n        options,\n        pipeline: this.pipeline,\n        partitionKey\n      };\n\n      request.headers = await this.buildHeaders(request);\n      request.headers[Constants.HttpHeaders.IsBatchRequest] = true;\n      request.headers[Constants.HttpHeaders.IsBatchAtomic] = true;\n\n      this.applySessionToken(request);\n\n      request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n        request.resourceType,\n        request.operationType\n      );\n      const response = await executePlugins(request, executeRequest, PluginOn.operation);\n      this.captureSessionToken(undefined, path, OperationType.Batch, response.headers);\n      return response;\n    } catch (err) {\n      this.captureSessionToken(err, path, OperationType.Upsert, (err as ErrorResponse).headers);\n      throw err;\n    }\n  }\n\n  public async bulk<T>({\n    body,\n    path,\n    partitionKeyRangeId,\n    resourceId,\n    bulkOptions = {},\n    options = {}\n  }: {\n    body: T;\n    path: string;\n    partitionKeyRangeId: string;\n    resourceId: string;\n    bulkOptions?: BulkOptions;\n    options?: RequestOptions;\n  }): Promise<Response<any>> {\n    try {\n      const request: RequestContext = {\n        globalEndpointManager: this.globalEndpointManager,\n        requestAgent: this.cosmosClientOptions.agent,\n        connectionPolicy: this.connectionPolicy,\n        method: HTTPMethod.post,\n        client: this,\n        operationType: OperationType.Batch,\n        path,\n        body,\n        resourceType: ResourceType.item,\n        resourceId,\n        plugins: this.cosmosClientOptions.plugins,\n        options,\n        pipeline: this.pipeline\n      };\n\n      request.headers = await this.buildHeaders(request);\n      request.headers[Constants.HttpHeaders.IsBatchRequest] = true;\n      request.headers[Constants.HttpHeaders.PartitionKeyRangeID] = partitionKeyRangeId;\n      request.headers[Constants.HttpHeaders.IsBatchAtomic] = false;\n      request.headers[Constants.HttpHeaders.BatchContinueOnError] =\n        bulkOptions.continueOnError || false;\n\n      this.applySessionToken(request);\n\n      request.endpoint = await this.globalEndpointManager.resolveServiceEndpoint(\n        request.resourceType,\n        request.operationType\n      );\n      const response = await executePlugins(request, executeRequest, PluginOn.operation);\n      this.captureSessionToken(undefined, path, OperationType.Batch, response.headers);\n      return response;\n    } catch (err) {\n      this.captureSessionToken(err, path, OperationType.Upsert, (err as ErrorResponse).headers);\n      throw err;\n    }\n  }\n\n  private captureSessionToken(\n    err: ErrorResponse,\n    path: string,\n    operationType: OperationType,\n    resHeaders: CosmosHeaders\n  ): void {\n    const request = this.getSessionParams(path);\n    request.operationType = operationType;\n    if (\n      !err ||\n      (!this.isMasterResource(request.resourceType) &&\n        (err.code === StatusCodes.PreconditionFailed ||\n          err.code === StatusCodes.Conflict ||\n          (err.code === StatusCodes.NotFound &&\n            err.substatus !== SubStatusCodes.ReadSessionNotAvailable)))\n    ) {\n      this.sessionContainer.set(request, resHeaders);\n    }\n  }\n\n  public clearSessionToken(path: string): void {\n    const request = this.getSessionParams(path);\n    this.sessionContainer.remove(request);\n  }\n\n  private getSessionParams(resourceLink: string): SessionContext {\n    const resourceId: string = null;\n    let resourceAddress: string = null;\n    const parserOutput = parseLink(resourceLink);\n\n    resourceAddress = parserOutput.objectBody.self;\n\n    const resourceType = parserOutput.type;\n    return {\n      resourceId,\n      resourceAddress,\n      resourceType,\n      isNameBased: true\n    };\n  }\n\n  private isMasterResource(resourceType: string): boolean {\n    if (\n      resourceType === Constants.Path.OffersPathSegment ||\n      resourceType === Constants.Path.DatabasesPathSegment ||\n      resourceType === Constants.Path.UsersPathSegment ||\n      resourceType === Constants.Path.PermissionsPathSegment ||\n      resourceType === Constants.Path.TopologyPathSegment ||\n      resourceType === Constants.Path.DatabaseAccountPathSegment ||\n      resourceType === Constants.Path.PartitionKeyRangesPathSegment ||\n      resourceType === Constants.Path.CollectionsPathSegment\n    ) {\n      return true;\n    }\n\n    return false;\n  }\n\n  private buildHeaders(requestContext: RequestContext): Promise<CosmosHeaders> {\n    return getHeaders({\n      clientOptions: this.cosmosClientOptions,\n      defaultHeaders: {\n        ...this.cosmosClientOptions.defaultHeaders,\n        ...requestContext.options.initialHeaders\n      },\n      verb: requestContext.method,\n      path: requestContext.path,\n      resourceId: requestContext.resourceId,\n      resourceType: requestContext.resourceType,\n      options: requestContext.options,\n      partitionKeyRangeId: requestContext.partitionKeyRangeId,\n      useMultipleWriteLocations: this.connectionPolicy.useMultipleWriteLocations,\n      partitionKey: requestContext.partitionKey\n    });\n  }\n}\n"]}

package/dist-esm/src/CosmosClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CosmosClient.d.ts","sourceRoot":"","sources":["../../src/CosmosClient.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAK/C,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,eAAe,EAA2B,MAAM,aAAa,CAAC;AAEvE,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAG7D;;;;;;;;;;;;;;;;;;GAkBG;AACH,qBAAa,YAAY;IACvB;;;;;;;;;OASG;IACH,SAAgB,SAAS,EAAE,SAAS,CAAC;IACrC;;;;OAIG;IACH,SAAgB,MAAM,EAAE,MAAM,CAAC;IAC/B,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,iBAAiB,CAAe;IACxC;;OAEG;gBACS,gBAAgB,EAAE,MAAM;IACpC;;;OAGG;gBACS,OAAO,EAAE,mBAAmB;IAkDxC;;OAEG;IACU,kBAAkB,CAC7B,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;IAK7C;;;;OAIG;IACI,gBAAgB,IAAI,OAAO,CAAC,MAAM,CAAC;IAI1C;;;;OAIG;IACI,eAAe,IAAI,OAAO,CAAC,MAAM,CAAC;IAIzC;;;;OAIG;IACI,iBAAiB,IAAI,OAAO,CAAC,SAAS,MAAM,EAAE,CAAC;IAItD;;;;OAIG;IACI,gBAAgB,IAAI,OAAO,CAAC,SAAS,MAAM,EAAE,CAAC;IAIrD;;;;;;;;;;;;;;;OAeG;IACI,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,QAAQ;IAIrC;;;OAGG;IACI,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,KAAK;IAI/B;;OAEG;IACI,OAAO,IAAI,IAAI;YAIR,6BAA6B;CAe5C"}