@azure/core-lro 3.3.2-alpha.20260311.2 → 3.4.0

package/dist/esm/http/operation.js

@@ -1,259 +1,257 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
 import { pollOperation } from "../poller/operation.js";
 import { logger } from "../logger.js";
 function getOperationLocationPollingUrl(inputs) {
-  const { azureAsyncOperation, operationLocation } = inputs;
-  return operationLocation ?? azureAsyncOperation;
+    const { azureAsyncOperation, operationLocation } = inputs;
+    return operationLocation ?? azureAsyncOperation;
 }
 function getLocationHeader(rawResponse) {
-  return rawResponse.headers["location"];
+    return rawResponse.headers["location"];
 }
 function getOperationLocationHeader(rawResponse) {
-  return rawResponse.headers["operation-location"];
+    return rawResponse.headers["operation-location"];
 }
 function getAzureAsyncOperationHeader(rawResponse) {
-  return rawResponse.headers["azure-asyncoperation"];
+    return rawResponse.headers["azure-asyncoperation"];
 }
 function findResourceLocation(inputs) {
-  const { location, requestMethod, requestPath, resourceLocationConfig, skipFinalGet } = inputs;
-  if (skipFinalGet) {
-    return void 0;
-  }
-  switch (requestMethod) {
-    case "PUT": {
-      return requestPath;
+    const { location, requestMethod, requestPath, resourceLocationConfig, skipFinalGet } = inputs;
+    // If skipFinalGet is true, return undefined to skip the final GET request
+    if (skipFinalGet) {
+        return undefined;
     }
-    case "DELETE": {
-      return void 0;
+    switch (requestMethod) {
+        case "PUT": {
+            return requestPath;
+        }
+        case "DELETE": {
+            return undefined;
+        }
+        case "PATCH": {
+            return getDefault() ?? requestPath;
+        }
+        default: {
+            return getDefault();
+        }
     }
-    case "PATCH": {
-      return getDefault() ?? requestPath;
+    function getDefault() {
+        switch (resourceLocationConfig) {
+            case "operation-location":
+            case "azure-async-operation": {
+                return undefined;
+            }
+            case "original-uri": {
+                return requestPath;
+            }
+            case "location":
+            default: {
+                return location;
+            }
+        }
     }
-    default: {
-      return getDefault();
+}
+export function inferLroMode(rawResponse, resourceLocationConfig, skipFinalGet) {
+    const requestPath = rawResponse.request.url;
+    const requestMethod = rawResponse.request.method;
+    const operationLocation = getOperationLocationHeader(rawResponse);
+    const azureAsyncOperation = getAzureAsyncOperationHeader(rawResponse);
+    const pollingUrl = getOperationLocationPollingUrl({ operationLocation, azureAsyncOperation });
+    const location = getLocationHeader(rawResponse);
+    const normalizedRequestMethod = requestMethod?.toLocaleUpperCase();
+    if (pollingUrl !== undefined) {
+        return {
+            mode: "OperationLocation",
+            operationLocation: pollingUrl,
+            resourceLocation: findResourceLocation({
+                requestMethod: normalizedRequestMethod,
+                location,
+                requestPath,
+                resourceLocationConfig,
+                skipFinalGet,
+            }),
+            initialRequestUrl: requestPath,
+            requestMethod,
+        };
     }
-  }
-  function getDefault() {
-    switch (resourceLocationConfig) {
-      case "operation-location":
-      case "azure-async-operation": {
-        return void 0;
-      }
-      case "original-uri": {
-        return requestPath;
-      }
-      case "location":
-      default: {
-        return location;
-      }
+    else if (location !== undefined) {
+        return {
+            mode: "ResourceLocation",
+            operationLocation: location,
+            initialRequestUrl: requestPath,
+            requestMethod,
+        };
+    }
+    else if (normalizedRequestMethod === "PUT" && requestPath) {
+        return {
+            mode: "Body",
+            operationLocation: requestPath,
+            initialRequestUrl: requestPath,
+            requestMethod,
+        };
+    }
+    else {
+        return undefined;
     }
-  }
-}
-function inferLroMode(rawResponse, resourceLocationConfig, skipFinalGet) {
-  const requestPath = rawResponse.request.url;
-  const requestMethod = rawResponse.request.method;
-  const operationLocation = getOperationLocationHeader(rawResponse);
-  const azureAsyncOperation = getAzureAsyncOperationHeader(rawResponse);
-  const pollingUrl = getOperationLocationPollingUrl({ operationLocation, azureAsyncOperation });
-  const location = getLocationHeader(rawResponse);
-  const normalizedRequestMethod = requestMethod?.toLocaleUpperCase();
-  if (pollingUrl !== void 0) {
-    return {
-      mode: "OperationLocation",
-      operationLocation: pollingUrl,
-      resourceLocation: findResourceLocation({
-        requestMethod: normalizedRequestMethod,
-        location,
-        requestPath,
-        resourceLocationConfig,
-        skipFinalGet
-      }),
-      initialRequestUrl: requestPath,
-      requestMethod
-    };
-  } else if (location !== void 0) {
-    return {
-      mode: "ResourceLocation",
-      operationLocation: location,
-      initialRequestUrl: requestPath,
-      requestMethod
-    };
-  } else if (normalizedRequestMethod === "PUT" && requestPath) {
-    return {
-      mode: "Body",
-      operationLocation: requestPath,
-      initialRequestUrl: requestPath,
-      requestMethod
-    };
-  } else {
-    return void 0;
-  }
 }
 function transformStatus(inputs) {
-  const { status, statusCode } = inputs;
-  if (typeof status !== "string" && status !== void 0) {
-    throw new Error(
-      `Polling was unsuccessful. Expected status to have a string value or no value but it has instead: ${status}. This doesn't necessarily indicate the operation has failed. Check your Azure subscription or resource status for more information.`
-    );
-  }
-  logger.verbose(`LRO: Transforming status: ${status} with status code: ${statusCode}.`);
-  const lowerCaseStatus = status?.toLocaleLowerCase();
-  if (!lowerCaseStatus) {
-    return toOperationStatus(statusCode);
-  }
-  if (lowerCaseStatus.includes("succeeded")) {
-    return "succeeded";
-  }
-  if (lowerCaseStatus.includes("fail")) {
-    return "failed";
-  }
-  if (["canceled", "cancelled"].includes(lowerCaseStatus)) {
-    return "canceled";
-  }
-  return "running";
+    const { status, statusCode } = inputs;
+    if (typeof status !== "string" && status !== undefined) {
+        throw new Error(`Polling was unsuccessful. Expected status to have a string value or no value but it has instead: ${status}. This doesn't necessarily indicate the operation has failed. Check your Azure subscription or resource status for more information.`);
+    }
+    logger.verbose(`LRO: Transforming status: ${status} with status code: ${statusCode}.`);
+    const lowerCaseStatus = status?.toLocaleLowerCase();
+    if (!lowerCaseStatus) {
+        return toOperationStatus(statusCode);
+    }
+    if (lowerCaseStatus.includes("succeeded")) {
+        return "succeeded";
+    }
+    if (lowerCaseStatus.includes("fail")) {
+        return "failed";
+    }
+    if (["canceled", "cancelled"].includes(lowerCaseStatus)) {
+        return "canceled";
+    }
+    return "running";
 }
 function getStatus(rawResponse) {
-  const { status } = rawResponse.body ?? {};
-  return transformStatus({ status, statusCode: rawResponse.statusCode });
+    const { status } = rawResponse.body ?? {};
+    return transformStatus({ status, statusCode: rawResponse.statusCode });
 }
 function getProvisioningState(rawResponse) {
-  const { properties, provisioningState } = rawResponse.body ?? {};
-  const status = properties?.provisioningState ?? provisioningState;
-  return transformStatus({ status, statusCode: rawResponse.statusCode });
+    const { properties, provisioningState } = rawResponse.body ?? {};
+    const status = properties?.provisioningState ?? provisioningState;
+    return transformStatus({ status, statusCode: rawResponse.statusCode });
 }
 function toOperationStatus(statusCode) {
-  if (statusCode === 202) {
-    return "running";
-  } else if (statusCode < 300) {
-    return "succeeded";
-  } else {
-    return "failed";
-  }
-}
-function parseRetryAfter({ rawResponse }) {
-  const retryAfter = rawResponse.headers["retry-after"];
-  if (retryAfter !== void 0) {
-    const retryAfterInSeconds = parseInt(retryAfter);
-    return isNaN(retryAfterInSeconds) ? calculatePollingIntervalFromDate(new Date(retryAfter)) : retryAfterInSeconds * 1e3;
-  }
-  return void 0;
-}
-function getErrorFromResponse(response) {
-  const error = accessBodyProperty(response, "error");
-  if (!error) {
-    logger.warning(
-      `The long-running operation failed but there is no error property in the response's body`
-    );
-    return;
-  }
-  if (!error.code || !error.message) {
-    logger.warning(
-      `The long-running operation failed but the error property in the response's body doesn't contain code or message`
-    );
-    return;
-  }
-  return error;
-}
-function calculatePollingIntervalFromDate(retryAfterDate) {
-  const timeNow = Math.floor((/* @__PURE__ */ new Date()).getTime());
-  const retryAfterTime = retryAfterDate.getTime();
-  if (timeNow < retryAfterTime) {
-    return retryAfterTime - timeNow;
-  }
-  return void 0;
-}
-function getStatusFromInitialResponse(inputs) {
-  const { response, state, operationLocation } = inputs;
-  function helper() {
-    const mode = state.config.metadata?.["mode"];
-    switch (mode) {
-      case void 0:
-        return toOperationStatus(response.rawResponse.statusCode);
-      case "Body":
-        return getOperationStatus(response, state);
-      default:
+    if (statusCode === 202) {
         return "running";
     }
-  }
-  const status = helper();
-  return status === "running" && operationLocation === void 0 ? "succeeded" : status;
+    else if (statusCode < 300) {
+        return "succeeded";
+    }
+    else {
+        return "failed";
+    }
 }
-function getOperationLocation({ rawResponse }, state) {
-  const mode = state.config.metadata?.["mode"];
-  switch (mode) {
-    case "OperationLocation": {
-      return getOperationLocationPollingUrl({
-        operationLocation: getOperationLocationHeader(rawResponse),
-        azureAsyncOperation: getAzureAsyncOperationHeader(rawResponse)
-      });
+export function parseRetryAfter({ rawResponse }) {
+    const retryAfter = rawResponse.headers["retry-after"];
+    if (retryAfter !== undefined) {
+        // Retry-After header value is either in HTTP date format, or in seconds
+        const retryAfterInSeconds = parseInt(retryAfter);
+        return isNaN(retryAfterInSeconds)
+            ? calculatePollingIntervalFromDate(new Date(retryAfter))
+            : retryAfterInSeconds * 1000;
     }
-    case "ResourceLocation": {
-      return getLocationHeader(rawResponse);
+    return undefined;
+}
+export function getErrorFromResponse(response) {
+    const error = accessBodyProperty(response, "error");
+    if (!error) {
+        logger.warning(`The long-running operation failed but there is no error property in the response's body`);
+        return;
     }
-    case "Body":
-    default: {
-      return void 0;
+    if (!error.code || !error.message) {
+        logger.warning(`The long-running operation failed but the error property in the response's body doesn't contain code or message`);
+        return;
     }
-  }
+    return error;
 }
-function getOperationStatus({ rawResponse }, state) {
-  const mode = state.config.metadata?.["mode"];
-  switch (mode) {
-    case "OperationLocation": {
-      return getStatus(rawResponse);
+function calculatePollingIntervalFromDate(retryAfterDate) {
+    const timeNow = Math.floor(new Date().getTime());
+    const retryAfterTime = retryAfterDate.getTime();
+    if (timeNow < retryAfterTime) {
+        return retryAfterTime - timeNow;
     }
-    case "ResourceLocation": {
-      return toOperationStatus(rawResponse.statusCode);
+    return undefined;
+}
+export function getStatusFromInitialResponse(inputs) {
+    const { response, state, operationLocation } = inputs;
+    function helper() {
+        const mode = state.config.metadata?.["mode"];
+        switch (mode) {
+            case undefined:
+                return toOperationStatus(response.rawResponse.statusCode);
+            case "Body":
+                return getOperationStatus(response, state);
+            default:
+                return "running";
+        }
     }
-    case "Body": {
-      return getProvisioningState(rawResponse);
+    const status = helper();
+    return status === "running" && operationLocation === undefined ? "succeeded" : status;
+}
+export function getOperationLocation({ rawResponse }, state) {
+    const mode = state.config.metadata?.["mode"];
+    switch (mode) {
+        case "OperationLocation": {
+            return getOperationLocationPollingUrl({
+                operationLocation: getOperationLocationHeader(rawResponse),
+                azureAsyncOperation: getAzureAsyncOperationHeader(rawResponse),
+            });
+        }
+        case "ResourceLocation": {
+            return getLocationHeader(rawResponse);
+        }
+        case "Body":
+        default: {
+            return undefined;
+        }
+    }
+}
+export function getOperationStatus({ rawResponse }, state) {
+    const mode = state.config.metadata?.["mode"];
+    switch (mode) {
+        case "OperationLocation": {
+            return getStatus(rawResponse);
+        }
+        case "ResourceLocation": {
+            return toOperationStatus(rawResponse.statusCode);
+        }
+        case "Body": {
+            return getProvisioningState(rawResponse);
+        }
+        default:
+            throw new Error(`Internal error: Unexpected operation mode: ${mode}`);
     }
-    default:
-      throw new Error(`Internal error: Unexpected operation mode: ${mode}`);
-  }
 }
 function accessBodyProperty({ flatResponse, rawResponse }, prop) {
-  return flatResponse?.[prop] ?? rawResponse.body?.[prop];
+    return flatResponse?.[prop] ?? rawResponse.body?.[prop];
 }
-function getResourceLocation(res, state) {
-  const loc = accessBodyProperty(res, "resourceLocation");
-  if (loc && typeof loc === "string") {
-    state.config.resourceLocation = loc;
-  }
-  return state.config.resourceLocation;
+export function getResourceLocation(res, state) {
+    const loc = accessBodyProperty(res, "resourceLocation");
+    if (loc && typeof loc === "string") {
+        state.config.resourceLocation = loc;
+    }
+    return state.config.resourceLocation;
 }
-function isOperationError(e) {
-  return e.name === "RestError";
+export function isOperationError(e) {
+    return e.name === "RestError";
 }
-async function pollHttpOperation(inputs) {
-  const { lro, options, processResult, updateState, setDelay, state, setErrorAsResult } = inputs;
-  return pollOperation({
-    state,
-    setDelay,
-    processResult: processResult ? ({ flatResponse }, inputState) => processResult(flatResponse, inputState) : ({ flatResponse }) => flatResponse,
-    getError: getErrorFromResponse,
-    updateState,
-    getPollingInterval: parseRetryAfter,
-    getOperationLocation,
-    getOperationStatus,
-    isOperationError,
-    getResourceLocation,
-    options,
-    /**
-     * The expansion here is intentional because `lro` could be an object that
-     * references an inner this, so we need to preserve a reference to it.
-     */
-    poll: async (location, inputOptions) => lro.sendPollRequest(location, inputOptions),
-    setErrorAsResult
-  });
+/** Polls the long-running operation. */
+export async function pollHttpOperation(inputs) {
+    const { lro, options, processResult, updateState, setDelay, state, setErrorAsResult } = inputs;
+    return pollOperation({
+        state,
+        setDelay,
+        processResult: processResult
+            ? ({ flatResponse }, inputState) => processResult(flatResponse, inputState)
+            : ({ flatResponse }) => flatResponse,
+        getError: getErrorFromResponse,
+        updateState,
+        getPollingInterval: parseRetryAfter,
+        getOperationLocation,
+        getOperationStatus,
+        isOperationError,
+        getResourceLocation,
+        options,
+        /**
+         * The expansion here is intentional because `lro` could be an object that
+         * references an inner this, so we need to preserve a reference to it.
+         */
+        poll: (location, inputOptions) => lro.sendPollRequest(location, inputOptions),
+        setErrorAsResult,
+    });
 }
-export {
-  getErrorFromResponse,
-  getOperationLocation,
-  getOperationStatus,
-  getResourceLocation,
-  getStatusFromInitialResponse,
-  inferLroMode,
-  isOperationError,
-  parseRetryAfter,
-  pollHttpOperation
-};
+//# sourceMappingURL=operation.js.map

package/dist/esm/http/operation.js.map

@@ -1,7 +1 @@
-{
-  "version": 3,
-  "sources": ["/mnt/vss/_work/1/s/sdk/core/core-lro/src/http/operation.ts"],
-  "sourcesContent": ["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  HttpOperationMode,\n  RunningOperation,\n  ResourceLocationConfig,\n  OperationResponse,\n  RawResponse,\n  ResponseBody,\n} from \"./models.js\";\nimport type {\n  LroError,\n  OperationConfig,\n  OperationState,\n  OperationStatus,\n  RestorableOperationState,\n} from \"../poller/models.js\";\nimport { pollOperation } from \"../poller/operation.js\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport { logger } from \"../logger.js\";\n\nfunction getOperationLocationPollingUrl(inputs: {\n  operationLocation?: string;\n  azureAsyncOperation?: string;\n}): string | undefined {\n  const { azureAsyncOperation, operationLocation } = inputs;\n  return operationLocation ?? azureAsyncOperation;\n}\n\nfunction getLocationHeader(rawResponse: RawResponse): string | undefined {\n  return rawResponse.headers[\"location\"];\n}\n\nfunction getOperationLocationHeader(rawResponse: RawResponse): string | undefined {\n  return rawResponse.headers[\"operation-location\"];\n}\n\nfunction getAzureAsyncOperationHeader(rawResponse: RawResponse): string | undefined {\n  return rawResponse.headers[\"azure-asyncoperation\"];\n}\n\nfunction findResourceLocation(inputs: {\n  requestMethod?: string;\n  location?: string;\n  requestPath?: string;\n  resourceLocationConfig?: ResourceLocationConfig;\n  skipFinalGet?: boolean;\n}): string | undefined {\n  const { location, requestMethod, requestPath, resourceLocationConfig, skipFinalGet } = inputs;\n\n  // If skipFinalGet is true, return undefined to skip the final GET request\n  if (skipFinalGet) {\n    return undefined;\n  }\n\n  switch (requestMethod) {\n    case \"PUT\": {\n      return requestPath;\n    }\n    case \"DELETE\": {\n      return undefined;\n    }\n    case \"PATCH\": {\n      return getDefault() ?? requestPath;\n    }\n    default: {\n      return getDefault();\n    }\n  }\n\n  function getDefault(): string | undefined {\n    switch (resourceLocationConfig) {\n      case \"operation-location\":\n      case \"azure-async-operation\": {\n        return undefined;\n      }\n      case \"original-uri\": {\n        return requestPath;\n      }\n      case \"location\":\n      default: {\n        return location;\n      }\n    }\n  }\n}\n\nexport function inferLroMode(\n  rawResponse: RawResponse,\n  resourceLocationConfig?: ResourceLocationConfig,\n  skipFinalGet?: boolean,\n): (OperationConfig & { mode: HttpOperationMode }) | undefined {\n  const requestPath = rawResponse.request.url;\n  const requestMethod = rawResponse.request.method;\n  const operationLocation = getOperationLocationHeader(rawResponse);\n  const azureAsyncOperation = getAzureAsyncOperationHeader(rawResponse);\n  const pollingUrl = getOperationLocationPollingUrl({ operationLocation, azureAsyncOperation });\n  const location = getLocationHeader(rawResponse);\n  const normalizedRequestMethod = requestMethod?.toLocaleUpperCase();\n  if (pollingUrl !== undefined) {\n    return {\n      mode: \"OperationLocation\",\n      operationLocation: pollingUrl,\n      resourceLocation: findResourceLocation({\n        requestMethod: normalizedRequestMethod,\n        location,\n        requestPath,\n        resourceLocationConfig,\n        skipFinalGet,\n      }),\n      initialRequestUrl: requestPath,\n      requestMethod,\n    };\n  } else if (location !== undefined) {\n    return {\n      mode: \"ResourceLocation\",\n      operationLocation: location,\n      initialRequestUrl: requestPath,\n      requestMethod,\n    };\n  } else if (normalizedRequestMethod === \"PUT\" && requestPath) {\n    return {\n      mode: \"Body\",\n      operationLocation: requestPath,\n      initialRequestUrl: requestPath,\n      requestMethod,\n    };\n  } else {\n    return undefined;\n  }\n}\n\nfunction transformStatus(inputs: { status: unknown; statusCode: number }): OperationStatus {\n  const { status, statusCode } = inputs;\n  if (typeof status !== \"string\" && status !== undefined) {\n    throw new Error(\n      `Polling was unsuccessful. Expected status to have a string value or no value but it has instead: ${status}. This doesn't necessarily indicate the operation has failed. Check your Azure subscription or resource status for more information.`,\n    );\n  }\n  logger.verbose(`LRO: Transforming status: ${status} with status code: ${statusCode}.`);\n  const lowerCaseStatus = status?.toLocaleLowerCase();\n  if (!lowerCaseStatus) {\n    return toOperationStatus(statusCode);\n  }\n  if (lowerCaseStatus.includes(\"succeeded\")) {\n    return \"succeeded\";\n  }\n  if (lowerCaseStatus.includes(\"fail\")) {\n    return \"failed\";\n  }\n  if ([\"canceled\", \"cancelled\"].includes(lowerCaseStatus)) {\n    return \"canceled\";\n  }\n  return \"running\";\n}\n\nfunction getStatus(rawResponse: RawResponse): OperationStatus {\n  const { status } = (rawResponse.body as ResponseBody) ?? {};\n  return transformStatus({ status, statusCode: rawResponse.statusCode });\n}\n\nfunction getProvisioningState(rawResponse: RawResponse): OperationStatus {\n  const { properties, provisioningState } = (rawResponse.body as ResponseBody) ?? {};\n  const status = properties?.provisioningState ?? provisioningState;\n  return transformStatus({ status, statusCode: rawResponse.statusCode });\n}\n\nfunction toOperationStatus(statusCode: number): OperationStatus {\n  if (statusCode === 202) {\n    return \"running\";\n  } else if (statusCode < 300) {\n    return \"succeeded\";\n  } else {\n    return \"failed\";\n  }\n}\n\nexport function parseRetryAfter<T>({ rawResponse }: OperationResponse<T>): number | undefined {\n  const retryAfter: string | undefined = rawResponse.headers[\"retry-after\"];\n  if (retryAfter !== undefined) {\n    // Retry-After header value is either in HTTP date format, or in seconds\n    const retryAfterInSeconds = parseInt(retryAfter);\n    return isNaN(retryAfterInSeconds)\n      ? calculatePollingIntervalFromDate(new Date(retryAfter))\n      : retryAfterInSeconds * 1000;\n  }\n  return undefined;\n}\n\nexport function getErrorFromResponse<T>(response: OperationResponse<T>): LroError | undefined {\n  const error = accessBodyProperty(response, \"error\");\n  if (!error) {\n    logger.warning(\n      `The long-running operation failed but there is no error property in the response's body`,\n    );\n    return;\n  }\n  if (!error.code || !error.message) {\n    logger.warning(\n      `The long-running operation failed but the error property in the response's body doesn't contain code or message`,\n    );\n    return;\n  }\n  return error as LroError;\n}\n\nfunction calculatePollingIntervalFromDate(retryAfterDate: Date): number | undefined {\n  const timeNow = Math.floor(new Date().getTime());\n  const retryAfterTime = retryAfterDate.getTime();\n  if (timeNow < retryAfterTime) {\n    return retryAfterTime - timeNow;\n  }\n  return undefined;\n}\n\nexport function getStatusFromInitialResponse<\n  TResult,\n  TState extends OperationState<TResult>,\n>(inputs: {\n  response: OperationResponse<unknown>;\n  state: RestorableOperationState<TResult, TState>;\n  operationLocation?: string;\n}): OperationStatus {\n  const { response, state, operationLocation } = inputs;\n  function helper(): OperationStatus {\n    const mode = state.config.metadata?.[\"mode\"];\n    switch (mode) {\n      case undefined:\n        return toOperationStatus(response.rawResponse.statusCode);\n      case \"Body\":\n        return getOperationStatus(response, state);\n      default:\n        return \"running\";\n    }\n  }\n  const status = helper();\n  return status === \"running\" && operationLocation === undefined ? \"succeeded\" : status;\n}\n\nexport function getOperationLocation<TResult, TState extends OperationState<TResult>>(\n  { rawResponse }: OperationResponse,\n  state: RestorableOperationState<TResult, TState>,\n): string | undefined {\n  const mode = state.config.metadata?.[\"mode\"];\n  switch (mode) {\n    case \"OperationLocation\": {\n      return getOperationLocationPollingUrl({\n        operationLocation: getOperationLocationHeader(rawResponse),\n        azureAsyncOperation: getAzureAsyncOperationHeader(rawResponse),\n      });\n    }\n    case \"ResourceLocation\": {\n      return getLocationHeader(rawResponse);\n    }\n    case \"Body\":\n    default: {\n      return undefined;\n    }\n  }\n}\n\nexport function getOperationStatus<TResult, TState extends OperationState<TResult>>(\n  { rawResponse }: OperationResponse,\n  state: RestorableOperationState<TResult, TState>,\n): OperationStatus {\n  const mode = state.config.metadata?.[\"mode\"];\n  switch (mode) {\n    case \"OperationLocation\": {\n      return getStatus(rawResponse);\n    }\n    case \"ResourceLocation\": {\n      return toOperationStatus(rawResponse.statusCode);\n    }\n    case \"Body\": {\n      return getProvisioningState(rawResponse);\n    }\n    default:\n      throw new Error(`Internal error: Unexpected operation mode: ${mode}`);\n  }\n}\n\nfunction accessBodyProperty<P extends string>(\n  { flatResponse, rawResponse }: OperationResponse,\n  prop: P,\n): ResponseBody[P] {\n  return (flatResponse as ResponseBody)?.[prop] ?? (rawResponse.body as ResponseBody)?.[prop];\n}\n\nexport function getResourceLocation<TResult, TState extends OperationState<TResult>>(\n  res: OperationResponse,\n  state: RestorableOperationState<TResult, TState>,\n): string | undefined {\n  const loc = accessBodyProperty(res, \"resourceLocation\");\n  if (loc && typeof loc === \"string\") {\n    state.config.resourceLocation = loc;\n  }\n  return state.config.resourceLocation;\n}\n\nexport function isOperationError(e: Error): boolean {\n  return e.name === \"RestError\";\n}\n\n/** Polls the long-running operation. */\nexport async function pollHttpOperation<TState extends OperationState<TResult>, TResult>(inputs: {\n  lro: RunningOperation;\n  processResult?: (result: unknown, state: TState) => Promise<TResult>;\n  updateState?: (state: TState, lastResponse: OperationResponse) => void;\n  isDone?: (lastResponse: OperationResponse, state: TState) => boolean;\n  setDelay: (intervalInMs: number) => void;\n  options?: { abortSignal?: AbortSignalLike };\n  state: RestorableOperationState<TResult, TState>;\n  setErrorAsResult: boolean;\n}): Promise<void> {\n  const { lro, options, processResult, updateState, setDelay, state, setErrorAsResult } = inputs;\n  return pollOperation({\n    state,\n    setDelay,\n    processResult: processResult\n      ? ({ flatResponse }, inputState) => processResult(flatResponse, inputState)\n      : ({ flatResponse }) => flatResponse as Promise<TResult>,\n    getError: getErrorFromResponse,\n    updateState,\n    getPollingInterval: parseRetryAfter,\n    getOperationLocation,\n    getOperationStatus,\n    isOperationError,\n    getResourceLocation,\n    options,\n    /**\n     * The expansion here is intentional because `lro` could be an object that\n     * references an inner this, so we need to preserve a reference to it.\n     */\n    poll: async (location: string, inputOptions?: { abortSignal?: AbortSignalLike }) =>\n      lro.sendPollRequest(location, inputOptions),\n    setErrorAsResult,\n  });\n}\n"],
-  "mappings": "AAkBA,SAAS,qBAAqB;AAE9B,SAAS,cAAc;AAEvB,SAAS,+BAA+B,QAGjB;AACrB,QAAM,EAAE,qBAAqB,kBAAkB,IAAI;AACnD,SAAO,qBAAqB;AAC9B;AAEA,SAAS,kBAAkB,aAA8C;AACvE,SAAO,YAAY,QAAQ,UAAU;AACvC;AAEA,SAAS,2BAA2B,aAA8C;AAChF,SAAO,YAAY,QAAQ,oBAAoB;AACjD;AAEA,SAAS,6BAA6B,aAA8C;AAClF,SAAO,YAAY,QAAQ,sBAAsB;AACnD;AAEA,SAAS,qBAAqB,QAMP;AACrB,QAAM,EAAE,UAAU,eAAe,aAAa,wBAAwB,aAAa,IAAI;AAGvF,MAAI,cAAc;AAChB,WAAO;AAAA,EACT;AAEA,UAAQ,eAAe;AAAA,IACrB,KAAK,OAAO;AACV,aAAO;AAAA,IACT;AAAA,IACA,KAAK,UAAU;AACb,aAAO;AAAA,IACT;AAAA,IACA,KAAK,SAAS;AACZ,aAAO,WAAW,KAAK;AAAA,IACzB;AAAA,IACA,SAAS;AACP,aAAO,WAAW;AAAA,IACpB;AAAA,EACF;AAEA,WAAS,aAAiC;AACxC,YAAQ,wBAAwB;AAAA,MAC9B,KAAK;AAAA,MACL,KAAK,yBAAyB;AAC5B,eAAO;AAAA,MACT;AAAA,MACA,KAAK,gBAAgB;AACnB,eAAO;AAAA,MACT;AAAA,MACA,KAAK;AAAA,MACL,SAAS;AACP,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,aACd,aACA,wBACA,cAC6D;AAC7D,QAAM,cAAc,YAAY,QAAQ;AACxC,QAAM,gBAAgB,YAAY,QAAQ;AAC1C,QAAM,oBAAoB,2BAA2B,WAAW;AAChE,QAAM,sBAAsB,6BAA6B,WAAW;AACpE,QAAM,aAAa,+BAA+B,EAAE,mBAAmB,oBAAoB,CAAC;AAC5F,QAAM,WAAW,kBAAkB,WAAW;AAC9C,QAAM,0BAA0B,eAAe,kBAAkB;AACjE,MAAI,eAAe,QAAW;AAC5B,WAAO;AAAA,MACL,MAAM;AAAA,MACN,mBAAmB;AAAA,MACnB,kBAAkB,qBAAqB;AAAA,QACrC,eAAe;AAAA,QACf;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,MACD,mBAAmB;AAAA,MACnB;AAAA,IACF;AAAA,EACF,WAAW,aAAa,QAAW;AACjC,WAAO;AAAA,MACL,MAAM;AAAA,MACN,mBAAmB;AAAA,MACnB,mBAAmB;AAAA,MACnB;AAAA,IACF;AAAA,EACF,WAAW,4BAA4B,SAAS,aAAa;AAC3D,WAAO;AAAA,MACL,MAAM;AAAA,MACN,mBAAmB;AAAA,MACnB,mBAAmB;AAAA,MACnB;AAAA,IACF;AAAA,EACF,OAAO;AACL,WAAO;AAAA,EACT;AACF;AAEA,SAAS,gBAAgB,QAAkE;AACzF,QAAM,EAAE,QAAQ,WAAW,IAAI;AAC/B,MAAI,OAAO,WAAW,YAAY,WAAW,QAAW;AACtD,UAAM,IAAI;AAAA,MACR,oGAAoG,MAAM;AAAA,IAC5G;AAAA,EACF;AACA,SAAO,QAAQ,6BAA6B,MAAM,sBAAsB,UAAU,GAAG;AACrF,QAAM,kBAAkB,QAAQ,kBAAkB;AAClD,MAAI,CAAC,iBAAiB;AACpB,WAAO,kBAAkB,UAAU;AAAA,EACrC;AACA,MAAI,gBAAgB,SAAS,WAAW,GAAG;AACzC,WAAO;AAAA,EACT;AACA,MAAI,gBAAgB,SAAS,MAAM,GAAG;AACpC,WAAO;AAAA,EACT;AACA,MAAI,CAAC,YAAY,WAAW,EAAE,SAAS,eAAe,GAAG;AACvD,WAAO;AAAA,EACT;AACA,SAAO;AACT;AAEA,SAAS,UAAU,aAA2C;AAC5D,QAAM,EAAE,OAAO,IAAK,YAAY,QAAyB,CAAC;AAC1D,SAAO,gBAAgB,EAAE,QAAQ,YAAY,YAAY,WAAW,CAAC;AACvE;AAEA,SAAS,qBAAqB,aAA2C;AACvE,QAAM,EAAE,YAAY,kBAAkB,IAAK,YAAY,QAAyB,CAAC;AACjF,QAAM,SAAS,YAAY,qBAAqB;AAChD,SAAO,gBAAgB,EAAE,QAAQ,YAAY,YAAY,WAAW,CAAC;AACvE;AAEA,SAAS,kBAAkB,YAAqC;AAC9D,MAAI,eAAe,KAAK;AACtB,WAAO;AAAA,EACT,WAAW,aAAa,KAAK;AAC3B,WAAO;AAAA,EACT,OAAO;AACL,WAAO;AAAA,EACT;AACF;AAEO,SAAS,gBAAmB,EAAE,YAAY,GAA6C;AAC5F,QAAM,aAAiC,YAAY,QAAQ,aAAa;AACxE,MAAI,eAAe,QAAW;AAE5B,UAAM,sBAAsB,SAAS,UAAU;AAC/C,WAAO,MAAM,mBAAmB,IAC5B,iCAAiC,IAAI,KAAK,UAAU,CAAC,IACrD,sBAAsB;AAAA,EAC5B;AACA,SAAO;AACT;AAEO,SAAS,qBAAwB,UAAsD;AAC5F,QAAM,QAAQ,mBAAmB,UAAU,OAAO;AAClD,MAAI,CAAC,OAAO;AACV,WAAO;AAAA,MACL;AAAA,IACF;AACA;AAAA,EACF;AACA,MAAI,CAAC,MAAM,QAAQ,CAAC,MAAM,SAAS;AACjC,WAAO;AAAA,MACL;AAAA,IACF;AACA;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,iCAAiC,gBAA0C;AAClF,QAAM,UAAU,KAAK,OAAM,oBAAI,KAAK,GAAE,QAAQ,CAAC;AAC/C,QAAM,iBAAiB,eAAe,QAAQ;AAC9C,MAAI,UAAU,gBAAgB;AAC5B,WAAO,iBAAiB;AAAA,EAC1B;AACA,SAAO;AACT;AAEO,SAAS,6BAGd,QAIkB;AAClB,QAAM,EAAE,UAAU,OAAO,kBAAkB,IAAI;AAC/C,WAAS,SAA0B;AACjC,UAAM,OAAO,MAAM,OAAO,WAAW,MAAM;AAC3C,YAAQ,MAAM;AAAA,MACZ,KAAK;AACH,eAAO,kBAAkB,SAAS,YAAY,UAAU;AAAA,MAC1D,KAAK;AACH,eAAO,mBAAmB,UAAU,KAAK;AAAA,MAC3C;AACE,eAAO;AAAA,IACX;AAAA,EACF;AACA,QAAM,SAAS,OAAO;AACtB,SAAO,WAAW,aAAa,sBAAsB,SAAY,cAAc;AACjF;AAEO,SAAS,qBACd,EAAE,YAAY,GACd,OACoB;AACpB,QAAM,OAAO,MAAM,OAAO,WAAW,MAAM;AAC3C,UAAQ,MAAM;AAAA,IACZ,KAAK,qBAAqB;AACxB,aAAO,+BAA+B;AAAA,QACpC,mBAAmB,2BAA2B,WAAW;AAAA,QACzD,qBAAqB,6BAA6B,WAAW;AAAA,MAC/D,CAAC;AAAA,IACH;AAAA,IACA,KAAK,oBAAoB;AACvB,aAAO,kBAAkB,WAAW;AAAA,IACtC;AAAA,IACA,KAAK;AAAA,IACL,SAAS;AACP,aAAO;AAAA,IACT;AAAA,EACF;AACF;AAEO,SAAS,mBACd,EAAE,YAAY,GACd,OACiB;AACjB,QAAM,OAAO,MAAM,OAAO,WAAW,MAAM;AAC3C,UAAQ,MAAM;AAAA,IACZ,KAAK,qBAAqB;AACxB,aAAO,UAAU,WAAW;AAAA,IAC9B;AAAA,IACA,KAAK,oBAAoB;AACvB,aAAO,kBAAkB,YAAY,UAAU;AAAA,IACjD;AAAA,IACA,KAAK,QAAQ;AACX,aAAO,qBAAqB,WAAW;AAAA,IACzC;AAAA,IACA;AACE,YAAM,IAAI,MAAM,8CAA8C,IAAI,EAAE;AAAA,EACxE;AACF;AAEA,SAAS,mBACP,EAAE,cAAc,YAAY,GAC5B,MACiB;AACjB,SAAQ,eAAgC,IAAI,KAAM,YAAY,OAAwB,IAAI;AAC5F;AAEO,SAAS,oBACd,KACA,OACoB;AACpB,QAAM,MAAM,mBAAmB,KAAK,kBAAkB;AACtD,MAAI,OAAO,OAAO,QAAQ,UAAU;AAClC,UAAM,OAAO,mBAAmB;AAAA,EAClC;AACA,SAAO,MAAM,OAAO;AACtB;AAEO,SAAS,iBAAiB,GAAmB;AAClD,SAAO,EAAE,SAAS;AACpB;AAGA,eAAsB,kBAAmE,QASvE;AAChB,QAAM,EAAE,KAAK,SAAS,eAAe,aAAa,UAAU,OAAO,iBAAiB,IAAI;AACxF,SAAO,cAAc;AAAA,IACnB;AAAA,IACA;AAAA,IACA,eAAe,gBACX,CAAC,EAAE,aAAa,GAAG,eAAe,cAAc,cAAc,UAAU,IACxE,CAAC,EAAE,aAAa,MAAM;AAAA,IAC1B,UAAU;AAAA,IACV;AAAA,IACA,oBAAoB;AAAA,IACpB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA;AAAA;AAAA;AAAA;AAAA,IAKA,MAAM,OAAO,UAAkB,iBAC7B,IAAI,gBAAgB,UAAU,YAAY;AAAA,IAC5C;AAAA,EACF,CAAC;AACH;",
-  "names": []
-}
+{"version":3,"file":"operation.js","sourceRoot":"","sources":["../../../src/http/operation.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAiBlC,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAEvD,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC,SAAS,8BAA8B,CAAC,MAGvC;IACC,MAAM,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,GAAG,MAAM,CAAC;IAC1D,OAAO,iBAAiB,IAAI,mBAAmB,CAAC;AAClD,CAAC;AAED,SAAS,iBAAiB,CAAC,WAAwB;IACjD,OAAO,WAAW,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;AACzC,CAAC;AAED,SAAS,0BAA0B,CAAC,WAAwB;IAC1D,OAAO,WAAW,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,4BAA4B,CAAC,WAAwB;IAC5D,OAAO,WAAW,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;AACrD,CAAC;AAED,SAAS,oBAAoB,CAAC,MAM7B;IACC,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,WAAW,EAAE,sBAAsB,EAAE,YAAY,EAAE,GAAG,MAAM,CAAC;IAE9F,0EAA0E;IAC1E,IAAI,YAAY,EAAE,CAAC;QACjB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,QAAQ,aAAa,EAAE,CAAC;QACtB,KAAK,KAAK,CAAC,CAAC,CAAC;YACX,OAAO,WAAW,CAAC;QACrB,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,OAAO,UAAU,EAAE,IAAI,WAAW,CAAC;QACrC,CAAC;QACD,OAAO,CAAC,CAAC,CAAC;YACR,OAAO,UAAU,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;IAED,SAAS,UAAU;QACjB,QAAQ,sBAAsB,EAAE,CAAC;YAC/B,KAAK,oBAAoB,CAAC;YAC1B,KAAK,uBAAuB,CAAC,CAAC,CAAC;gBAC7B,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,OAAO,WAAW,CAAC;YACrB,CAAC;YACD,KAAK,UAAU,CAAC;YAChB,OAAO,CAAC,CAAC,CAAC;gBACR,OAAO,QAAQ,CAAC;YAClB,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,WAAwB,EACxB,sBAA+C,EAC/C,YAAsB;IAEtB,MAAM,WAAW,GAAG,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC;IAC5C,MAAM,aAAa,GAAG,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;IACjD,MAAM,iBAAiB,GAAG,0BAA0B,CAAC,WAAW,CAAC,CAAC;IAClE,MAAM,mBAAmB,GAAG,4BAA4B,CAAC,WAAW,CAAC,CAAC;IACtE,MAAM,UAAU,GAAG,8BAA8B,CAAC,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC9F,MAAM,QAAQ,GAAG,iBAAiB,CAAC,WAAW,CAAC,CAAC;IAChD,MAAM,uBAAuB,GAAG,aAAa,EAAE,iBAAiB,EAAE,CAAC;IACnE,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC7B,OAAO;YACL,IAAI,EAAE,mBAAmB;YACzB,iBAAiB,EAAE,UAAU;YAC7B,gBAAgB,EAAE,oBAAoB,CAAC;gBACrC,aAAa,EAAE,uBAAuB;gBACtC,QAAQ;gBACR,WAAW;gBACX,sBAAsB;gBACtB,YAAY;aACb,CAAC;YACF,iBAAiB,EAAE,WAAW;YAC9B,aAAa;SACd,CAAC;IACJ,CAAC;SAAM,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAClC,OAAO;YACL,IAAI,EAAE,kBAAkB;YACxB,iBAAiB,EAAE,QAAQ;YAC3B,iBAAiB,EAAE,WAAW;YAC9B,aAAa;SACd,CAAC;IACJ,CAAC;SAAM,IAAI,uBAAuB,KAAK,KAAK,IAAI,WAAW,EAAE,CAAC;QAC5D,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE,WAAW;YAC9B,iBAAiB,EAAE,WAAW;YAC9B,aAAa;SACd,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,MAA+C;IACtE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC;IACtC,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACvD,MAAM,IAAI,KAAK,CACb,oGAAoG,MAAM,sIAAsI,CACjP,CAAC;IACJ,CAAC;IACD,MAAM,CAAC,OAAO,CAAC,6BAA6B,MAAM,sBAAsB,UAAU,GAAG,CAAC,CAAC;IACvF,MAAM,eAAe,GAAG,MAAM,EAAE,iBAAiB,EAAE,CAAC;IACpD,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,iBAAiB,CAAC,UAAU,CAAC,CAAC;IACvC,CAAC;IACD,IAAI,eAAe,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;QAC1C,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,IAAI,eAAe,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QACrC,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;QACxD,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,SAAS,CAAC,WAAwB;IACzC,MAAM,EAAE,MAAM,EAAE,GAAI,WAAW,CAAC,IAAqB,IAAI,EAAE,CAAC;IAC5D,OAAO,eAAe,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,oBAAoB,CAAC,WAAwB;IACpD,MAAM,EAAE,UAAU,EAAE,iBAAiB,EAAE,GAAI,WAAW,CAAC,IAAqB,IAAI,EAAE,CAAC;IACnF,MAAM,MAAM,GAAG,UAAU,EAAE,iBAAiB,IAAI,iBAAiB,CAAC;IAClE,OAAO,eAAe,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,iBAAiB,CAAC,UAAkB;IAC3C,IAAI,UAAU,KAAK,GAAG,EAAE,CAAC;QACvB,OAAO,SAAS,CAAC;IACnB,CAAC;SAAM,IAAI,UAAU,GAAG,GAAG,EAAE,CAAC;QAC5B,OAAO,WAAW,CAAC;IACrB,CAAC;SAAM,CAAC;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,eAAe,CAAI,EAAE,WAAW,EAAwB;IACtE,MAAM,UAAU,GAAuB,WAAW,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAC1E,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC7B,wEAAwE;QACxE,MAAM,mBAAmB,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;QACjD,OAAO,KAAK,CAAC,mBAAmB,CAAC;YAC/B,CAAC,CAAC,gCAAgC,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC;YACxD,CAAC,CAAC,mBAAmB,GAAG,IAAI,CAAC;IACjC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAI,QAA8B;IACpE,MAAM,KAAK,GAAG,kBAAkB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACpD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,CAAC,OAAO,CACZ,yFAAyF,CAC1F,CAAC;QACF,OAAO;IACT,CAAC;IACD,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QAClC,MAAM,CAAC,OAAO,CACZ,iHAAiH,CAClH,CAAC;QACF,OAAO;IACT,CAAC;IACD,OAAO,KAAiB,CAAC;AAC3B,CAAC;AAED,SAAS,gCAAgC,CAAC,cAAoB;IAC5D,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;IACjD,MAAM,cAAc,GAAG,cAAc,CAAC,OAAO,EAAE,CAAC;IAChD,IAAI,OAAO,GAAG,cAAc,EAAE,CAAC;QAC7B,OAAO,cAAc,GAAG,OAAO,CAAC;IAClC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,4BAA4B,CAG1C,MAID;IACC,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,iBAAiB,EAAE,GAAG,MAAM,CAAC;IACtD,SAAS,MAAM;QACb,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC;QAC7C,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,SAAS;gBACZ,OAAO,iBAAiB,CAAC,QAAQ,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YAC5D,KAAK,MAAM;gBACT,OAAO,kBAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAC7C;gBACE,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;IACxB,OAAO,MAAM,KAAK,SAAS,IAAI,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC;AACxF,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,EAAE,WAAW,EAAqB,EAClC,KAAgD;IAEhD,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC;IAC7C,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,mBAAmB,CAAC,CAAC,CAAC;YACzB,OAAO,8BAA8B,CAAC;gBACpC,iBAAiB,EAAE,0BAA0B,CAAC,WAAW,CAAC;gBAC1D,mBAAmB,EAAE,4BAA4B,CAAC,WAAW,CAAC;aAC/D,CAAC,CAAC;QACL,CAAC;QACD,KAAK,kBAAkB,CAAC,CAAC,CAAC;YACxB,OAAO,iBAAiB,CAAC,WAAW,CAAC,CAAC;QACxC,CAAC;QACD,KAAK,MAAM,CAAC;QACZ,OAAO,CAAC,CAAC,CAAC;YACR,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,EAAE,WAAW,EAAqB,EAClC,KAAgD;IAEhD,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,CAAC;IAC7C,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,mBAAmB,CAAC,CAAC,CAAC;YACzB,OAAO,SAAS,CAAC,WAAW,CAAC,CAAC;QAChC,CAAC;QACD,KAAK,kBAAkB,CAAC,CAAC,CAAC;YACxB,OAAO,iBAAiB,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QACnD,CAAC;QACD,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,OAAO,oBAAoB,CAAC,WAAW,CAAC,CAAC;QAC3C,CAAC;QACD;YACE,MAAM,IAAI,KAAK,CAAC,8CAA8C,IAAI,EAAE,CAAC,CAAC;IAC1E,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CACzB,EAAE,YAAY,EAAE,WAAW,EAAqB,EAChD,IAAO;IAEP,OAAQ,YAA6B,EAAE,CAAC,IAAI,CAAC,IAAK,WAAW,CAAC,IAAqB,EAAE,CAAC,IAAI,CAAC,CAAC;AAC9F,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,GAAsB,EACtB,KAAgD;IAEhD,MAAM,GAAG,GAAG,kBAAkB,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;IACxD,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;QACnC,KAAK,CAAC,MAAM,CAAC,gBAAgB,GAAG,GAAG,CAAC;IACtC,CAAC;IACD,OAAO,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,CAAQ;IACvC,OAAO,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC;AAChC,CAAC;AAED,wCAAwC;AACxC,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAkD,MASxF;IACC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,QAAQ,EAAE,KAAK,EAAE,gBAAgB,EAAE,GAAG,MAAM,CAAC;IAC/F,OAAO,aAAa,CAAC;QACnB,KAAK;QACL,QAAQ;QACR,aAAa,EAAE,aAAa;YAC1B,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,EAAE,UAAU,EAAE,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,UAAU,CAAC;YAC3E,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,YAAgC;QAC1D,QAAQ,EAAE,oBAAoB;QAC9B,WAAW;QACX,kBAAkB,EAAE,eAAe;QACnC,oBAAoB;QACpB,kBAAkB;QAClB,gBAAgB;QAChB,mBAAmB;QACnB,OAAO;QACP;;;WAGG;QACH,IAAI,EAAE,CAAC,QAAgB,EAAE,YAAgD,EAAE,EAAE,CAC3E,GAAG,CAAC,eAAe,CAAC,QAAQ,EAAE,YAAY,CAAC;QAC7C,gBAAgB;KACjB,CAAC,CAAC;AACL,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  HttpOperationMode,\n  RunningOperation,\n  ResourceLocationConfig,\n  OperationResponse,\n  RawResponse,\n  ResponseBody,\n} from \"./models.js\";\nimport type {\n  LroError,\n  OperationConfig,\n  OperationState,\n  OperationStatus,\n  RestorableOperationState,\n} from \"../poller/models.js\";\nimport { pollOperation } from \"../poller/operation.js\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport { logger } from \"../logger.js\";\n\nfunction getOperationLocationPollingUrl(inputs: {\n  operationLocation?: string;\n  azureAsyncOperation?: string;\n}): string | undefined {\n  const { azureAsyncOperation, operationLocation } = inputs;\n  return operationLocation ?? azureAsyncOperation;\n}\n\nfunction getLocationHeader(rawResponse: RawResponse): string | undefined {\n  return rawResponse.headers[\"location\"];\n}\n\nfunction getOperationLocationHeader(rawResponse: RawResponse): string | undefined {\n  return rawResponse.headers[\"operation-location\"];\n}\n\nfunction getAzureAsyncOperationHeader(rawResponse: RawResponse): string | undefined {\n  return rawResponse.headers[\"azure-asyncoperation\"];\n}\n\nfunction findResourceLocation(inputs: {\n  requestMethod?: string;\n  location?: string;\n  requestPath?: string;\n  resourceLocationConfig?: ResourceLocationConfig;\n  skipFinalGet?: boolean;\n}): string | undefined {\n  const { location, requestMethod, requestPath, resourceLocationConfig, skipFinalGet } = inputs;\n\n  // If skipFinalGet is true, return undefined to skip the final GET request\n  if (skipFinalGet) {\n    return undefined;\n  }\n\n  switch (requestMethod) {\n    case \"PUT\": {\n      return requestPath;\n    }\n    case \"DELETE\": {\n      return undefined;\n    }\n    case \"PATCH\": {\n      return getDefault() ?? requestPath;\n    }\n    default: {\n      return getDefault();\n    }\n  }\n\n  function getDefault(): string | undefined {\n    switch (resourceLocationConfig) {\n      case \"operation-location\":\n      case \"azure-async-operation\": {\n        return undefined;\n      }\n      case \"original-uri\": {\n        return requestPath;\n      }\n      case \"location\":\n      default: {\n        return location;\n      }\n    }\n  }\n}\n\nexport function inferLroMode(\n  rawResponse: RawResponse,\n  resourceLocationConfig?: ResourceLocationConfig,\n  skipFinalGet?: boolean,\n): (OperationConfig & { mode: HttpOperationMode }) | undefined {\n  const requestPath = rawResponse.request.url;\n  const requestMethod = rawResponse.request.method;\n  const operationLocation = getOperationLocationHeader(rawResponse);\n  const azureAsyncOperation = getAzureAsyncOperationHeader(rawResponse);\n  const pollingUrl = getOperationLocationPollingUrl({ operationLocation, azureAsyncOperation });\n  const location = getLocationHeader(rawResponse);\n  const normalizedRequestMethod = requestMethod?.toLocaleUpperCase();\n  if (pollingUrl !== undefined) {\n    return {\n      mode: \"OperationLocation\",\n      operationLocation: pollingUrl,\n      resourceLocation: findResourceLocation({\n        requestMethod: normalizedRequestMethod,\n        location,\n        requestPath,\n        resourceLocationConfig,\n        skipFinalGet,\n      }),\n      initialRequestUrl: requestPath,\n      requestMethod,\n    };\n  } else if (location !== undefined) {\n    return {\n      mode: \"ResourceLocation\",\n      operationLocation: location,\n      initialRequestUrl: requestPath,\n      requestMethod,\n    };\n  } else if (normalizedRequestMethod === \"PUT\" && requestPath) {\n    return {\n      mode: \"Body\",\n      operationLocation: requestPath,\n      initialRequestUrl: requestPath,\n      requestMethod,\n    };\n  } else {\n    return undefined;\n  }\n}\n\nfunction transformStatus(inputs: { status: unknown; statusCode: number }): OperationStatus {\n  const { status, statusCode } = inputs;\n  if (typeof status !== \"string\" && status !== undefined) {\n    throw new Error(\n      `Polling was unsuccessful. Expected status to have a string value or no value but it has instead: ${status}. This doesn't necessarily indicate the operation has failed. Check your Azure subscription or resource status for more information.`,\n    );\n  }\n  logger.verbose(`LRO: Transforming status: ${status} with status code: ${statusCode}.`);\n  const lowerCaseStatus = status?.toLocaleLowerCase();\n  if (!lowerCaseStatus) {\n    return toOperationStatus(statusCode);\n  }\n  if (lowerCaseStatus.includes(\"succeeded\")) {\n    return \"succeeded\";\n  }\n  if (lowerCaseStatus.includes(\"fail\")) {\n    return \"failed\";\n  }\n  if ([\"canceled\", \"cancelled\"].includes(lowerCaseStatus)) {\n    return \"canceled\";\n  }\n  return \"running\";\n}\n\nfunction getStatus(rawResponse: RawResponse): OperationStatus {\n  const { status } = (rawResponse.body as ResponseBody) ?? {};\n  return transformStatus({ status, statusCode: rawResponse.statusCode });\n}\n\nfunction getProvisioningState(rawResponse: RawResponse): OperationStatus {\n  const { properties, provisioningState } = (rawResponse.body as ResponseBody) ?? {};\n  const status = properties?.provisioningState ?? provisioningState;\n  return transformStatus({ status, statusCode: rawResponse.statusCode });\n}\n\nfunction toOperationStatus(statusCode: number): OperationStatus {\n  if (statusCode === 202) {\n    return \"running\";\n  } else if (statusCode < 300) {\n    return \"succeeded\";\n  } else {\n    return \"failed\";\n  }\n}\n\nexport function parseRetryAfter<T>({ rawResponse }: OperationResponse<T>): number | undefined {\n  const retryAfter: string | undefined = rawResponse.headers[\"retry-after\"];\n  if (retryAfter !== undefined) {\n    // Retry-After header value is either in HTTP date format, or in seconds\n    const retryAfterInSeconds = parseInt(retryAfter);\n    return isNaN(retryAfterInSeconds)\n      ? calculatePollingIntervalFromDate(new Date(retryAfter))\n      : retryAfterInSeconds * 1000;\n  }\n  return undefined;\n}\n\nexport function getErrorFromResponse<T>(response: OperationResponse<T>): LroError | undefined {\n  const error = accessBodyProperty(response, \"error\");\n  if (!error) {\n    logger.warning(\n      `The long-running operation failed but there is no error property in the response's body`,\n    );\n    return;\n  }\n  if (!error.code || !error.message) {\n    logger.warning(\n      `The long-running operation failed but the error property in the response's body doesn't contain code or message`,\n    );\n    return;\n  }\n  return error as LroError;\n}\n\nfunction calculatePollingIntervalFromDate(retryAfterDate: Date): number | undefined {\n  const timeNow = Math.floor(new Date().getTime());\n  const retryAfterTime = retryAfterDate.getTime();\n  if (timeNow < retryAfterTime) {\n    return retryAfterTime - timeNow;\n  }\n  return undefined;\n}\n\nexport function getStatusFromInitialResponse<\n  TResult,\n  TState extends OperationState<TResult>,\n>(inputs: {\n  response: OperationResponse<unknown>;\n  state: RestorableOperationState<TResult, TState>;\n  operationLocation?: string;\n}): OperationStatus {\n  const { response, state, operationLocation } = inputs;\n  function helper(): OperationStatus {\n    const mode = state.config.metadata?.[\"mode\"];\n    switch (mode) {\n      case undefined:\n        return toOperationStatus(response.rawResponse.statusCode);\n      case \"Body\":\n        return getOperationStatus(response, state);\n      default:\n        return \"running\";\n    }\n  }\n  const status = helper();\n  return status === \"running\" && operationLocation === undefined ? \"succeeded\" : status;\n}\n\nexport function getOperationLocation<TResult, TState extends OperationState<TResult>>(\n  { rawResponse }: OperationResponse,\n  state: RestorableOperationState<TResult, TState>,\n): string | undefined {\n  const mode = state.config.metadata?.[\"mode\"];\n  switch (mode) {\n    case \"OperationLocation\": {\n      return getOperationLocationPollingUrl({\n        operationLocation: getOperationLocationHeader(rawResponse),\n        azureAsyncOperation: getAzureAsyncOperationHeader(rawResponse),\n      });\n    }\n    case \"ResourceLocation\": {\n      return getLocationHeader(rawResponse);\n    }\n    case \"Body\":\n    default: {\n      return undefined;\n    }\n  }\n}\n\nexport function getOperationStatus<TResult, TState extends OperationState<TResult>>(\n  { rawResponse }: OperationResponse,\n  state: RestorableOperationState<TResult, TState>,\n): OperationStatus {\n  const mode = state.config.metadata?.[\"mode\"];\n  switch (mode) {\n    case \"OperationLocation\": {\n      return getStatus(rawResponse);\n    }\n    case \"ResourceLocation\": {\n      return toOperationStatus(rawResponse.statusCode);\n    }\n    case \"Body\": {\n      return getProvisioningState(rawResponse);\n    }\n    default:\n      throw new Error(`Internal error: Unexpected operation mode: ${mode}`);\n  }\n}\n\nfunction accessBodyProperty<P extends string>(\n  { flatResponse, rawResponse }: OperationResponse,\n  prop: P,\n): ResponseBody[P] {\n  return (flatResponse as ResponseBody)?.[prop] ?? (rawResponse.body as ResponseBody)?.[prop];\n}\n\nexport function getResourceLocation<TResult, TState extends OperationState<TResult>>(\n  res: OperationResponse,\n  state: RestorableOperationState<TResult, TState>,\n): string | undefined {\n  const loc = accessBodyProperty(res, \"resourceLocation\");\n  if (loc && typeof loc === \"string\") {\n    state.config.resourceLocation = loc;\n  }\n  return state.config.resourceLocation;\n}\n\nexport function isOperationError(e: Error): boolean {\n  return e.name === \"RestError\";\n}\n\n/** Polls the long-running operation. */\nexport async function pollHttpOperation<TState extends OperationState<TResult>, TResult>(inputs: {\n  lro: RunningOperation;\n  processResult?: (result: unknown, state: TState) => Promise<TResult>;\n  updateState?: (state: TState, lastResponse: OperationResponse) => void;\n  isDone?: (lastResponse: OperationResponse, state: TState) => boolean;\n  setDelay: (intervalInMs: number) => void;\n  options?: { abortSignal?: AbortSignalLike };\n  state: RestorableOperationState<TResult, TState>;\n  setErrorAsResult: boolean;\n}): Promise<void> {\n  const { lro, options, processResult, updateState, setDelay, state, setErrorAsResult } = inputs;\n  return pollOperation({\n    state,\n    setDelay,\n    processResult: processResult\n      ? ({ flatResponse }, inputState) => processResult(flatResponse, inputState)\n      : ({ flatResponse }) => flatResponse as Promise<TResult>,\n    getError: getErrorFromResponse,\n    updateState,\n    getPollingInterval: parseRetryAfter,\n    getOperationLocation,\n    getOperationStatus,\n    isOperationError,\n    getResourceLocation,\n    options,\n    /**\n     * The expansion here is intentional because `lro` could be an object that\n     * references an inner this, so we need to preserve a reference to it.\n     */\n    poll: (location: string, inputOptions?: { abortSignal?: AbortSignalLike }) =>\n      lro.sendPollRequest(location, inputOptions),\n    setErrorAsResult,\n  });\n}\n"]}