@autorest/openapi-to-typespec/src/utils/lro.ts

Autorest plugin to scaffold a Typespec definition from an OpenAPI document

import { CodeModel, isObjectSchema, Operation } from "@autorest/codemodel";
import { TspLroHeaders } from "../interfaces";
import { isResponseSchema } from "./schemas";

export function markLRO(codeModel: CodeModel) {
  for (const operationGroup of codeModel.operationGroups) {
    for (const operation of operationGroup.operations) {
      operation.extensions ?? {};
      if (!hasLROExtension(operation)) {
        continue;
      }

      for (const response of operation.responses ?? []) {
        if (!isResponseSchema(response)) {
          continue;
        }

        if (!isObjectSchema(response.schema)) {
          continue;
        }

        response.schema.language.default.lro = {
          ...response.schema.language.default.lro,
          isLRO: true,
        };

        // TODO: mark the response as an LRO
      }
    }
  }
}

export function hasLROExtension(operation: Operation) {
  return (
    operation.extensions?.["x-ms-long-running-operation"] ||
    operation.extensions?.["x-ms-long-running-operation-options"]
  );
}

export function generateLroHeaders(lroHeaders: TspLroHeaders): string {
  if (lroHeaders === "Azure-AsyncOperation") {
    return "ArmAsyncOperationHeader & Azure.Core.Foundations.RetryAfterHeader";
  } else if (lroHeaders === "Location") {
    return "ArmLroLocationHeader & Azure.Core.Foundations.RetryAfterHeader";
  }
  throw new Error(`Unknown LRO header: ${lroHeaders}`);
}