Introduce more usages to describe things related with the states of LRO

.chronus/changes/remove-lro-workaround-2024-10-19-9-4-1.md

@@ -0,0 +1,9 @@
+---
+# Change versionKind to one of: internal, fix, dependencies, feature, deprecation, breaking
+changeKind: feature
+packages:
+  - "@azure-tools/typespec-client-generator-core"
+---
+
+1. Introduce new usage: `LroInitial`, `LroPolling`, `LroFinalEnvelope`.
+2. usage and access now properly propagate on polling model, final result and final envelop result of `lroMetadata`.

packages/typespec-client-generator-core/src/interfaces.ts

@@ -751,6 +751,12 @@ export enum UsageFlags {
   Xml = 1 << 9,
   // Set when type is used for exception output.
   Exception = 1 << 10,
+  // Set when type is used as LRO initial response.
+  LroInitial = 1 << 11,
+  // Set when type is used as LRO polling response.
+  LroPolling = 1 << 12,
+  // Set when type is used as LRO final envelop response.
+  LroFinalEnvelope = 1 << 13,
 }
 
 interface SdkExampleBase {

packages/typespec-client-generator-core/src/types.ts

@@ -1596,6 +1596,7 @@ function updateTypesFromOperation(
     diagnostics.pipe(updateUsageOrAccess(context, access, sdkType));
   }
 
+  const lroMetaData = getLroMetadata(program, operation);
   for (const response of httpOperation.responses) {
     for (const innerResponse of response.responses) {
       if (innerResponse.body?.type && !isNeverOrVoidType(innerResponse.body.type)) {
@@ -1607,6 +1608,9 @@ function updateTypesFromOperation(
         if (generateConvenient) {
           if (response.statusCodes === "*" || isErrorModel(context.program, body)) {
             diagnostics.pipe(updateUsageOrAccess(context, UsageFlags.Exception, sdkType));
+          } else if (lroMetaData !== undefined) {
+            // when the operation is an lro, the response should be its initial response.
+            diagnostics.pipe(updateUsageOrAccess(context, UsageFlags.LroInitial, sdkType));
           } else {
             diagnostics.pipe(updateUsageOrAccess(context, UsageFlags.Output, sdkType));
           }
@@ -1634,29 +1638,35 @@ function updateTypesFromOperation(
       }
     }
   }
-  const lroMetaData = getLroMetadata(program, operation);
+
   if (lroMetaData && generateConvenient) {
-    if (lroMetaData.finalResult !== undefined && lroMetaData.finalResult !== "void") {
-      const sdkType = diagnostics.pipe(
-        getClientTypeWithDiagnostics(context, lroMetaData.finalResult, operation),
-      );
-      diagnostics.pipe(updateUsageOrAccess(context, UsageFlags.Output, sdkType));
-      const access = getAccessOverride(context, operation) ?? "public";
-      diagnostics.pipe(updateUsageOrAccess(context, access, sdkType));
-
-      if (!context.arm) {
-        // TODO: currently skipping adding of envelopeResult due to arm error
-        // https://github.com/Azure/typespec-azure/issues/311
-        const sdkType = diagnostics.pipe(
-          getClientTypeWithDiagnostics(context, lroMetaData.envelopeResult, operation),
-        );
-        diagnostics.pipe(updateUsageOrAccess(context, UsageFlags.Output, sdkType));
-        const access = getAccessOverride(context, operation) ?? "public";
-        diagnostics.pipe(updateUsageOrAccess(context, access, sdkType));
-      }
-    }
+    // the final result will be normal output usage.
+    updateUsageOrAccessForLroComponent(lroMetaData.finalResult, UsageFlags.Output);
+
+    // the final envelope result will have LroFinalEnvelope.
+    updateUsageOrAccessForLroComponent(
+      lroMetaData.finalEnvelopeResult,
+      UsageFlags.LroFinalEnvelope,
+    );
+
+    // the polling model will have LroPolling.
+    updateUsageOrAccessForLroComponent(
+      lroMetaData.pollingInfo.responseModel,
+      UsageFlags.LroPolling,
+    );
   }
   return diagnostics.wrap(undefined);
+
+  function updateUsageOrAccessForLroComponent(
+    model: Model | "void" | undefined,
+    usage: UsageFlags,
+  ) {
+    if (model === undefined || model === "void") return;
+    const sdkType = diagnostics.pipe(getClientTypeWithDiagnostics(context, model, operation));
+    diagnostics.pipe(updateUsageOrAccess(context, usage, sdkType));
+    const access = getAccessOverride(context, operation) ?? "public";
+    diagnostics.pipe(updateUsageOrAccess(context, access, sdkType));
+  }
 }
 
 function updateAccessOverride(context: TCGCContext): [void, readonly Diagnostic[]] {