@azure-tools/typespec-azure-core 0.34.0-dev.1 → 0.34.0-dev.10

package/README.md

@@ -17,7 +17,9 @@ npm install @azure-tools/typespec-azure-core
 - [`@fixed`](#@fixed)
 - [`@items`](#@items)
 - [`@lroCanceled`](#@lrocanceled)
+- [`@lroErrorResult`](#@lroerrorresult)
 - [`@lroFailed`](#@lrofailed)
+- [`@lroResult`](#@lroresult)
 - [`@lroStatus`](#@lrostatus)
 - [`@lroSucceeded`](#@lrosucceeded)
 - [`@nextLink`](#@nextlink)
@@ -26,6 +28,7 @@ npm install @azure-tools/typespec-azure-core
 - [`@pagedResult`](#@pagedresult)
 - [`@pollingLocation`](#@pollinglocation)
 - [`@pollingOperation`](#@pollingoperation)
+- [`@pollingOperationParameter`](#@pollingoperationparameter)
 
 #### `@finalLocation`
 
@@ -97,6 +100,7 @@ None
 
 #### `@lroCanceled`
 
+Used for custom StatusMonitor implementation.
 Identifies an EnumMember as a long-running "Canceled" terminal state.
 
 ```typespec
@@ -111,8 +115,27 @@ Identifies an EnumMember as a long-running "Canceled" terminal state.
 
 None
 
+#### `@lroErrorResult`
+
+Used for custom StatusMonitor implementation.
+Identifies a model property of a StatusMonitor as containing the result
+of a long-running operation that terminates unsuccessfully (Failed).
+
+```typespec
+@Azure.Core.lroErrorResult
+```
+
+##### Target
+
+`ModelProperty`
+
+##### Parameters
+
+None
+
 #### `@lroFailed`
 
+Used for custom StatusMonitor implementation.
 Identifies an enum member as a long-running "Failed" terminal state.
 
 ```typespec
@@ -127,8 +150,27 @@ Identifies an enum member as a long-running "Failed" terminal state.
 
 None
 
+#### `@lroResult`
+
+Used for custom StatusMonitor implementation.
+Identifies a model property of a StatusMonitor as containing the result
+of a long-running operation that terminates successfully (Succeeded).
+
+```typespec
+@Azure.Core.lroResult
+```
+
+##### Target
+
+`ModelProperty`
+
+##### Parameters
+
+None
+
 #### `@lroStatus`
 
+Used for custom StatusMonitor implementation.
 Identifies an Enum or ModelProperty as containing long-running operation
 status.
 
@@ -146,6 +188,7 @@ None
 
 #### `@lroSucceeded`
 
+Used for custom StatusMonitor implementation.
 Identifies an EnumMember as a long-running "Succeeded" terminal state.
 
 ```typespec
@@ -266,6 +309,24 @@ Identifies that an operation is a polling operation for an LRO.
 | linkedOperation | `Operation` | The linked Operation                                                                                                |
 | parameters      | `model {}`  | Map of `RequestParameter<T>` and/or `ResponseProperty<T>` that will<br />be passed to the linked operation request. |
 
+#### `@pollingOperationParameter`
+
+Used to define how to call custom polling operations for long-running operations.
+
+```typespec
+@Azure.Core.pollingOperationParameter(targetParameter?: ModelProperty | string)
+```
+
+##### Target
+
+`ModelProperty`
+
+##### Parameters
+
+| Name            | Type                            | Description                                                                                                                                                                                        |
+| --------------- | ------------------------------- | -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| targetParameter | `union ModelProperty \| string` | A reference to the polling operation parameter this parameter<br />provides a value for, or the name of that parameter. The default value is the name of<br />the decorated parameter or property. |
+
 ### Azure.Core.Foundations
 
 - [`@omitKeyProperties`](#@omitkeyproperties)

package/dist/src/decorators.d.ts

@@ -1,4 +1,6 @@
-import { DecoratorContext, Diagnostic, Enum, EnumMember, Model, ModelProperty, Operation, Program, Scalar, Type } from "@typespec/compiler";
+import { DecoratorContext, Diagnostic, Enum, EnumMember, Model, ModelProperty, Operation, Program, Scalar, StringLiteral, Type } from "@typespec/compiler";
+import { OperationLink } from "./lro-helpers.js";
+import { PropertyMap, ResultInfo } from "./lro-info.js";
 export declare const namespace = "Azure.Core";
 export declare const PollingOperationKey: string;
 export declare const FinalOperationKey = "final";
@@ -8,9 +10,15 @@ export declare function $pagedResult(context: DecoratorContext, entity: Model):
 export interface PagedResultMetadata {
     modelType: Model;
     itemsProperty?: ModelProperty;
+    /** @deprecated use itemsSegments  */
     itemsPath?: string;
+    /** Path to the items property. */
+    itemsSegments?: string[];
     nextLinkProperty?: ModelProperty;
+    /** @deprecated use nextLinkSegments */
     nextLinkPath?: string;
+    /** Path to the next link property. */
+    nextLinkSegments?: string[];
     nextLinkOperation?: Operation;
 }
 /**
@@ -34,9 +42,9 @@ export declare function getNextLink(program: Program, entity: Type): boolean | u
  *  additional states defined for the operation.
  **/
 export interface LongRunningStates {
-    succeededState: string;
-    failedState: string;
-    canceledState: string;
+    succeededState: string[];
+    failedState: string[];
+    canceledState: string[];
     states: string[];
 }
 export declare function $lroStatus(context: DecoratorContext, entity: Enum | ModelProperty): void;
@@ -45,6 +53,18 @@ export declare function extractLroStates(program: Program, entity: Type): [LongR
  *  Returns the `LongRunningStates` associated with `entity`.
  */
 export declare function getLongRunningStates(program: Program, entity: Enum | Model | Scalar | ModelProperty): LongRunningStates | undefined;
+/**
+ * Return the property that contains the lro status
+ * @param program The program to process
+ * @param target The model to check for lro status
+ */
+export declare function getLroStatusProperty(program: Program, target: Model): ModelProperty | undefined;
+export declare function $lroResult(context: DecoratorContext, entity: ModelProperty): void;
+export declare function getLroResult(program: Program, entity: Model | ModelProperty, useDefault?: boolean): ModelProperty | undefined;
+export declare function $lroErrorResult(context: DecoratorContext, entity: ModelProperty): void;
+export declare function getLroErrorResult(program: Program, entity: Model | ModelProperty, useDefault?: boolean): ModelProperty | undefined;
+export declare function $pollingOperationParameter(context: DecoratorContext, entity: ModelProperty, target?: ModelProperty | StringLiteral): void;
+export declare function getPollingOperationParameter(program: Program, entity: ModelProperty): string | ModelProperty | undefined;
 export declare function $lroSucceeded(context: DecoratorContext, entity: EnumMember): void;
 /**
  *  Returns `true` if the enum member represents a "succeeded" state.
@@ -75,6 +95,9 @@ export interface OperationLinkMetadata {
     parameters?: Type;
     linkedOperation: Operation;
     linkType: string;
+    link?: OperationLink;
+    parameterMap?: Map<string, PropertyMap>;
+    result?: ResultInfo;
 }
 export declare function $operationLink(context: DecoratorContext, entity: Operation, linkedOperation: Operation, linkType: string, parameters?: Model): void;
 /**

package/dist/src/decorators.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"decorators.d.ts","sourceRoot":"","sources":["../../src/decorators.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,gBAAgB,EAChB,UAAU,EACV,IAAI,EACJ,UAAU,EAQV,KAAK,EACL,aAAa,EACb,SAAS,EACT,OAAO,EACP,MAAM,EAEN,IAAI,EACL,MAAM,oBAAoB,CAAC;AAS5B,eAAO,MAAM,SAAS,eAAe,CAAC;AAKtC,eAAO,MAAM,mBAAmB,EAAE,MAAkB,CAAC;AACrD,eAAO,MAAM,iBAAiB,UAAU,CAAC;AAMzC,wBAAgB,MAAM,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,IAAI,QAE7D;AAED,wBAAgB,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,GAAG,OAAO,CAE/D;AAMD,wBAAgB,YAAY,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,QAEpE;AAED,MAAM,WAAW,mBAAmB;IAClC,SAAS,EAAE,KAAK,CAAC;IACjB,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,aAAa,CAAC;IACjC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,iBAAiB,CAAC,EAAE,SAAS,CAAC;CAC/B;AAkED;;;;GAIG;AACH,wBAAgB,cAAc,CAC5B,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,KAAK,GAAG,SAAS,GACxB,mBAAmB,GAAG,SAAS,CAsEjC;AAID,wBAAgB,MAAM,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,aAAa,QAEtE;AAED;;GAEG;AACH,wBAAgB,QAAQ,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,GAAG,OAAO,GAAG,SAAS,CAE5E;AAID,wBAAgB,SAAS,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,aAAa,QAEzE;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,GAAG,OAAO,GAAG,SAAS,CAE/E;AAED;;;IAGI;AACJ,MAAM,WAAW,iBAAiB;IAEhC,cAAc,EAAE,MAAM,CAAC;IAGvB,WAAW,EAAE,MAAM,CAAC;IAGpB,aAAa,EAAE,MAAM,CAAC;IAGtB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAID,wBAAgB,UAAU,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,IAAI,GAAG,aAAa,QAIjF;AA2BD,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,IAAI,GACX,CAAC,iBAAiB,GAAG,SAAS,EAAE,SAAS,UAAU,EAAE,CAAC,CAkExD;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,IAAI,GAAG,KAAK,GAAG,MAAM,GAAG,aAAa,GAC5C,iBAAiB,GAAG,SAAS,CAU/B;AAMD,wBAAgB,aAAa,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,UAAU,QAE1E;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,WAEvE;AAMD,wBAAgB,YAAY,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,UAAU,QAEzE;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,WAEtE;AAMD,wBAAgB,UAAU,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,UAAU,QAEvE;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,GAAG,OAAO,CAE9E;AAMD,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,aAAa,QAEhF;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,GAAG,OAAO,CAElF;AAMD,wBAAgB,cAAc,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,aAAa,QAE9E;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,GAAG,OAAO,CAEhF;AAED,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,QAO1E;AAED,MAAM,WAAW,qBAAqB;IACpC,UAAU,CAAC,EAAE,IAAI,CAAC;IAClB,eAAe,EAAE,SAAS,CAAC;IAC3B,QAAQ,EAAE,MAAM,CAAC;CAClB;AAID,wBAAgB,cAAc,CAC5B,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,eAAe,EAAE,SAAS,EAC1B,QAAQ,EAAE,MAAM,EAChB,UAAU,CAAC,EAAE,KAAK,QAwFnB;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,SAAS,EACjB,QAAQ,EAAE,MAAM,GACf,qBAAqB,GAAG,SAAS,CASnC;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,SAAS,GAChB,GAAG,CAAC,MAAM,EAAE,qBAAqB,CAAC,GAAG,SAAS,CAEhD;AAED,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,eAAe,EAAE,SAAS,EAC1B,UAAU,CAAC,EAAE,KAAK,QAcnB;AAED,wBAAgB,eAAe,CAC7B,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,eAAe,EAAE,SAAS,EAC1B,UAAU,CAAC,EAAE,KAAK,QAGnB;AAED,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,eAAe,EAAE,SAAS,EAC1B,UAAU,CAAC,EAAE,KAAK,QAGnB;AAID,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,QAEvF;AAED,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,GAAG,MAAM,GAAG,SAAS,CAI/F;AAID,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,QAEvF;AAED,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,GAAG,MAAM,GAAG,SAAS,CAI/F;AAED,wBAAgB,uBAAuB,CACrC,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,KAAK,EACb,cAAc,EAAE,KAAK,QAsBtB;AAED,wBAAgB,+BAA+B,CAC7C,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,KAAK,EACb,cAAc,EAAE,KAAK,QAuBtB;AAID,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,YAAY,EAAE,IAAI,QAkDnB;AAED,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,GAAG,OAAO,CAEnF;AAID,wBAAgB,WAAW,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,SAAS,QAMvE;AAED,wBAAgB,cAAc,CAAC,OAAO,EAAE,OAAO,QAa9C;AAID,wBAAgB,WAAW,CACzB,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,YAAY,EAAE,MAAM,EACpB,IAAI,EAAE,MAAM,QAGb;AAED,wBAAgB,eAAe,CAAC,OAAO,EAAE,OAAO,QAkB/C"}
+{"version":3,"file":"decorators.d.ts","sourceRoot":"","sources":["../../src/decorators.ts"],"names":[],"mappings":"AAGA,OAAO,EAEL,gBAAgB,EAChB,UAAU,EACV,IAAI,EACJ,UAAU,EAOV,KAAK,EACL,aAAa,EACb,SAAS,EACT,OAAO,EACP,MAAM,EAEN,aAAa,EACb,IAAI,EACL,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAuB,WAAW,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAE7E,eAAO,MAAM,SAAS,eAAe,CAAC;AAKtC,eAAO,MAAM,mBAAmB,EAAE,MAAkB,CAAC;AACrD,eAAO,MAAM,iBAAiB,UAAU,CAAC;AAMzC,wBAAgB,MAAM,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,IAAI,QAE7D;AAED,wBAAgB,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,GAAG,OAAO,CAE/D;AAMD,wBAAgB,YAAY,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,QAEpE;AAED,MAAM,WAAW,mBAAmB;IAClC,SAAS,EAAE,KAAK,CAAC;IACjB,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,qCAAqC;IACrC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,kCAAkC;IAClC,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,gBAAgB,CAAC,EAAE,aAAa,CAAC;IACjC,uCAAuC;IACvC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,sCAAsC;IACtC,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,iBAAiB,CAAC,EAAE,SAAS,CAAC;CAC/B;AA6DD;;;;GAIG;AACH,wBAAgB,cAAc,CAC5B,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,KAAK,GAAG,SAAS,GACxB,mBAAmB,GAAG,SAAS,CA0EjC;AAID,wBAAgB,MAAM,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,aAAa,QAEtE;AAED;;GAEG;AACH,wBAAgB,QAAQ,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,GAAG,OAAO,GAAG,SAAS,CAE5E;AAID,wBAAgB,SAAS,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,aAAa,QAEzE;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,GAAG,OAAO,GAAG,SAAS,CAE/E;AAED;;;IAGI;AACJ,MAAM,WAAW,iBAAiB;IAEhC,cAAc,EAAE,MAAM,EAAE,CAAC;IAGzB,WAAW,EAAE,MAAM,EAAE,CAAC;IAGtB,aAAa,EAAE,MAAM,EAAE,CAAC;IAGxB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAID,wBAAgB,UAAU,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,IAAI,GAAG,aAAa,QAIjF;AA+BD,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,IAAI,GACX,CAAC,iBAAiB,GAAG,SAAS,EAAE,SAAS,UAAU,EAAE,CAAC,CAwExD;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,IAAI,GAAG,KAAK,GAAG,MAAM,GAAG,aAAa,GAC5C,iBAAiB,GAAG,SAAS,CAU/B;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,GAAG,aAAa,GAAG,SAAS,CAkB/F;AAMD,wBAAgB,UAAU,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,aAAa,QAY1E;AAED,wBAAgB,YAAY,CAC1B,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,KAAK,GAAG,aAAa,EAC7B,UAAU,GAAE,OAAc,GACzB,aAAa,GAAG,SAAS,CAc3B;AAKD,wBAAgB,eAAe,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,aAAa,QAY/E;AAED,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,KAAK,GAAG,aAAa,EAC7B,UAAU,GAAE,OAAc,GACzB,aAAa,GAAG,SAAS,CAc3B;AAID,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,aAAa,EACrB,MAAM,CAAC,EAAE,aAAa,GAAG,aAAa,QAevC;AAED,wBAAgB,4BAA4B,CAC1C,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,aAAa,GACpB,MAAM,GAAG,aAAa,GAAG,SAAS,CAEpC;AAMD,wBAAgB,aAAa,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,UAAU,QAE1E;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,WAEvE;AAMD,wBAAgB,YAAY,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,UAAU,QAEzE;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,WAEtE;AAMD,wBAAgB,UAAU,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,UAAU,QAEvE;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,GAAG,OAAO,CAE9E;AAMD,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,aAAa,QAEhF;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,GAAG,OAAO,CAElF;AAMD,wBAAgB,cAAc,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,aAAa,QAE9E;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,GAAG,OAAO,CAEhF;AAED,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,QAO1E;AAED,MAAM,WAAW,qBAAqB;IACpC,UAAU,CAAC,EAAE,IAAI,CAAC;IAClB,eAAe,EAAE,SAAS,CAAC;IAC3B,QAAQ,EAAE,MAAM,CAAC;IAEjB,IAAI,CAAC,EAAE,aAAa,CAAC;IACrB,YAAY,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IACxC,MAAM,CAAC,EAAE,UAAU,CAAC;CACrB;AAID,wBAAgB,cAAc,CAC5B,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,eAAe,EAAE,SAAS,EAC1B,QAAQ,EAAE,MAAM,EAChB,UAAU,CAAC,EAAE,KAAK,QA8BnB;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,SAAS,EACjB,QAAQ,EAAE,MAAM,GACf,qBAAqB,GAAG,SAAS,CASnC;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,SAAS,GAChB,GAAG,CAAC,MAAM,EAAE,qBAAqB,CAAC,GAAG,SAAS,CAEhD;AAED,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,eAAe,EAAE,SAAS,EAC1B,UAAU,CAAC,EAAE,KAAK,QAqDnB;AAED,wBAAgB,eAAe,CAC7B,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,eAAe,EAAE,SAAS,EAC1B,UAAU,CAAC,EAAE,KAAK,QAYnB;AAED,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,eAAe,EAAE,SAAS,EAC1B,UAAU,CAAC,EAAE,KAAK,QAGnB;AAID,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,QAEvF;AAED,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,GAAG,MAAM,GAAG,SAAS,CAI/F;AAID,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,QAEvF;AAED,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,GAAG,MAAM,GAAG,SAAS,CAI/F;AAED,wBAAgB,uBAAuB,CACrC,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,KAAK,EACb,cAAc,EAAE,KAAK,QAsBtB;AAED,wBAAgB,+BAA+B,CAC7C,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,KAAK,EACb,cAAc,EAAE,KAAK,QAuBtB;AAID,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,YAAY,EAAE,IAAI,QAkDnB;AAED,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,GAAG,OAAO,CAEnF;AAID,wBAAgB,WAAW,CAAC,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,SAAS,QAMvE;AAED,wBAAgB,cAAc,CAAC,OAAO,EAAE,OAAO,QAa9C;AAID,wBAAgB,WAAW,CACzB,OAAO,EAAE,gBAAgB,EACzB,MAAM,EAAE,SAAS,EACjB,YAAY,EAAE,MAAM,EACpB,IAAI,EAAE,MAAM,QAGb;AAED,wBAAgB,eAAe,CAAC,OAAO,EAAE,OAAO,QAkB/C"}

package/dist/src/decorators.js

@@ -1,7 +1,9 @@
 import { createDiagnostic, createStateSymbol, reportDiagnostic } from "./lib.js";
-import { createDiagnosticCollector, getKnownValues, getNamespaceFullName, getTypeName, isErrorModel, isKey, isStringType, isTemplateDeclarationOrInstance, setTypeSpecNamespace, } from "@typespec/compiler";
-import { getHttpOperation, getOperationParameters, getResponsesForOperation, getRoutePath, } from "@typespec/http";
+import { filterModelProperties, getAllProperties } from "./utils.js";
+import { createDiagnosticCollector, getKnownValues, getNamespaceFullName, getTypeName, isKey, isStringType, isTemplateDeclarationOrInstance, setTypeSpecNamespace, } from "@typespec/compiler";
+import { getHttpOperation, getRoutePath } from "@typespec/http";
 import { getResourceTypeKey, getSegment, isAutoRoute } from "@typespec/rest";
+import { getLroOperationInfo } from "./lro-info.js";
 export const namespace = "Azure.Core";
 /*
  * Constants for polling and final operation links
@@ -21,17 +23,20 @@ const pagedResultsKey = createStateSymbol("pagedResult");
 export function $pagedResult(context, entity) {
     context.program.stateMap(pagedResultsKey).set(entity, true);
 }
-function _getItems(program, entity, prefix = "") {
+function findPathToProperty(program, entity, condition, current = []) {
     for (const prop of entity.properties.values()) {
-        const hasItems = getItems(program, prop);
-        if (hasItems !== undefined) {
-            const currentPrefix = prefix !== "" ? `${prefix}.${prop.name}` : `${prop.name}`;
-            return [prop, currentPrefix];
+        const match = condition(prop);
+        if (match) {
+            const segments = [...current, prop.name];
+            return {
+                property: prop,
+                path: segments.join("."),
+                segments,
+            };
         }
         else {
             if (prop.type.kind === "Model") {
-                const currentPrefix = prefix !== "" ? `${prefix}.${prop.name}` : `${prop.name}`;
-                const items = _getItems(program, prop.type, currentPrefix);
+                const items = findPathToProperty(program, prop.type, condition, [...current, prop.name]);
                 if (items !== undefined) {
                     return items;
                 }
@@ -40,24 +45,11 @@ function _getItems(program, entity, prefix = "") {
     }
     return undefined;
 }
-function _getNextLink(program, entity, prefix = "") {
-    for (const prop of entity.properties.values()) {
-        const hasNextLink = getNextLink(program, prop);
-        if (hasNextLink !== undefined) {
-            const currentPrefix = prefix !== "" ? `${prefix}.${prop.name}` : `${prop.name}`;
-            return [prop, currentPrefix];
-        }
-        else {
-            if (prop.type.kind === "Model") {
-                const currentPrefix = prefix !== "" ? `${prefix}.${prop.name}` : `${prop.name}`;
-                const items = _getNextLink(program, prop.type, currentPrefix);
-                if (items !== undefined) {
-                    return items;
-                }
-            }
-        }
-    }
-    return undefined;
+function _getItems(program, entity) {
+    return findPathToProperty(program, entity, (prop) => getItems(program, prop) !== undefined);
+}
+function _getNextLink(program, entity) {
+    return findPathToProperty(program, entity, (prop) => getNextLink(program, prop) !== undefined);
 }
 /**
  * Find all named models that could have been the source of the given
@@ -90,13 +82,17 @@ export function getPagedResult(program, entity) {
                 metadata = { modelType: entity };
                 const items = _getItems(program, entity);
                 if (items !== undefined) {
-                    metadata.itemsProperty = items[0];
-                    metadata.itemsPath = items[1];
+                    metadata.itemsProperty = items.property;
+                    // eslint-disable-next-line deprecation/deprecation
+                    metadata.itemsPath = items.path;
+                    metadata.itemsSegments = items.segments;
                 }
                 const nextLink = _getNextLink(program, entity);
                 if (nextLink !== undefined) {
-                    metadata.nextLinkProperty = nextLink[0];
-                    metadata.nextLinkPath = nextLink[1];
+                    metadata.nextLinkProperty = nextLink.property;
+                    // eslint-disable-next-line deprecation/deprecation
+                    metadata.nextLinkPath = nextLink.path;
+                    metadata.nextLinkSegments = nextLink.segments;
                 }
                 return metadata;
             }
@@ -184,9 +180,9 @@ export function $lroStatus(context, entity) {
 }
 function storeLroState(program, states, name, enumMember) {
     const expectedStates = [
-        ["Succeeded", isLroSucceededState, () => (states.succeededState = name)],
-        ["Failed", isLroFailedState, () => (states.failedState = name)],
-        ["Canceled", isLroCanceledState, () => (states.canceledState = name)],
+        ["Succeeded", isLroSucceededState, () => states.succeededState.push(name)],
+        ["Failed", isLroFailedState, () => states.failedState.push(name)],
+        ["Canceled", isLroCanceledState, () => states.canceledState.push(name)],
     ];
     states.states.push(name);
     for (const [knownState, stateTest, setter] of expectedStates) {
@@ -197,7 +193,12 @@ function storeLroState(program, states, name, enumMember) {
     }
 }
 export function extractLroStates(program, entity) {
-    const result = { states: [] };
+    const result = {
+        states: [],
+        succeededState: [],
+        failedState: [],
+        canceledState: [],
+    };
     const diagnostics = createDiagnosticCollector();
     if (entity.kind === "ModelProperty") {
         // Call the function recursively on the property type
@@ -236,10 +237,11 @@ export function extractLroStates(program, entity) {
     }
     // Make sure all terminal states have been identified
     const missingStates = [];
-    for (const expected of ["Succeeded", "Failed", "Canceled"]) {
-        if (!(`${expected.toLowerCase()}State` in result)) {
-            missingStates.push(expected);
-        }
+    if (result.succeededState.length < 1) {
+        missingStates.push("Succeeded");
+    }
+    if (result.failedState.length < 1) {
+        missingStates.push("Failed");
     }
     if (missingStates.length > 0) {
         diagnostics.add(createDiagnostic({
@@ -266,6 +268,113 @@ export function getLongRunningStates(program, entity) {
     // Otherwise just check the type itself
     return program.stateMap(lroStatusKey).get(entity);
 }
+/**
+ * Return the property that contains the lro status
+ * @param program The program to process
+ * @param target The model to check for lro status
+ */
+export function getLroStatusProperty(program, target) {
+    const props = getAllProperties(target);
+    for (const [_, prop] of props.entries()) {
+        let values = program.stateMap(lroStatusKey).get(prop);
+        if (values !== undefined)
+            return prop;
+        if (prop.type.kind === "Enum" || prop.type.kind === "Union") {
+            values = program.stateMap(lroStatusKey).get(prop.type);
+            if (values !== undefined)
+                return prop;
+        }
+    }
+    const statusProp = props.get("status");
+    if (statusProp) {
+        const [states, _] = extractLroStates(program, statusProp);
+        if (states !== undefined)
+            return statusProp;
+    }
+    return undefined;
+}
+//@lroResult
+const lroResultKey = createStateSymbol("lroResult");
+export function $lroResult(context, entity) {
+    const { program } = context;
+    if (entity.model) {
+        if (getLroResult(program, entity.model, false) !== undefined) {
+            reportDiagnostic(program, {
+                code: "lro-status-monitor-invalid-result-property",
+                target: entity.model,
+                format: { resultType: "result", decorator: "@lroResult" },
+            });
+        }
+    }
+    program.stateMap(lroResultKey).set(entity, true);
+}
+export function getLroResult(program, entity, useDefault = true) {
+    switch (entity.kind) {
+        case "ModelProperty":
+            return program.stateMap(lroResultKey).has(entity) ? entity : undefined;
+        case "Model":
+            let props = filterModelProperties(entity, (e) => getLroResult(program, e) !== undefined);
+            if (props.length > 0)
+                return props[0];
+            if (useDefault) {
+                props = filterModelProperties(entity, (p) => p.name.toLowerCase() === "result");
+                if (props.length === 1)
+                    return props[0];
+            }
+    }
+    return undefined;
+}
+//@lroErrorResult
+const lroErrorResultKey = createStateSymbol("lroErrorResult");
+export function $lroErrorResult(context, entity) {
+    const { program } = context;
+    if (entity.model) {
+        if (getLroErrorResult(program, entity.model, false) !== undefined) {
+            reportDiagnostic(program, {
+                code: "lro-status-monitor-invalid-result-property",
+                target: entity.model,
+                format: { resultType: "error", decorator: "@lroErrorResult" },
+            });
+        }
+    }
+    program.stateMap(lroErrorResultKey).set(entity, true);
+}
+export function getLroErrorResult(program, entity, useDefault = true) {
+    switch (entity.kind) {
+        case "ModelProperty":
+            return program.stateMap(lroErrorResultKey).has(entity) ? entity : undefined;
+        case "Model":
+            let props = filterModelProperties(entity, (e) => getLroErrorResult(program, e) !== undefined);
+            if (props.length > 0)
+                return props[0];
+            if (useDefault) {
+                props = filterModelProperties(entity, (p) => p.name.toLowerCase() === "error");
+                if (props.length === 1)
+                    return props[0];
+            }
+    }
+    return undefined;
+}
+//@pollingOperationParameter
+const pollingParameterKey = createStateSymbol("pollingOperationParameter");
+export function $pollingOperationParameter(context, entity, target) {
+    const { program } = context;
+    let storedValue;
+    switch (target === null || target === void 0 ? void 0 : target.kind) {
+        case "ModelProperty":
+            storedValue = target;
+            break;
+        case "String":
+            storedValue = target.value;
+            break;
+        default:
+            storedValue = undefined;
+    }
+    program.stateMap(pollingParameterKey).set(entity, storedValue !== null && storedValue !== void 0 ? storedValue : entity.name);
+}
+export function getPollingOperationParameter(program, entity) {
+    return program.stateMap(pollingParameterKey).get(entity);
+}
 // @lroSucceeded
 const lroSucceededKey = createStateSymbol("lroSucceeded");
 export function $lroSucceeded(context, entity) {
@@ -331,77 +440,11 @@ export function $omitKeyProperties(context, entity) {
 }
 const operationLinkKey = createStateSymbol("operationLink");
 export function $operationLink(context, entity, linkedOperation, linkType, parameters) {
-    var _a, _b;
-    const properties = parameters === null || parameters === void 0 ? void 0 : parameters.properties;
-    if (properties !== undefined) {
-        for (const data of properties.values()) {
-            const typeName = data.type.name;
-            const namespace = (_a = data.type.namespace) === null || _a === void 0 ? void 0 : _a.name;
-            const valid = namespace === "Core" &&
-                (typeName === "RequestParameter" || typeName === "ResponseProperty");
-            if (!valid) {
-                reportDiagnostic(context.program, {
-                    code: "operation-link-parameter-invalid",
-                    target: context.decoratorTarget,
-                    format: {},
-                });
-                return;
-            }
-            const targetProperty = data.type.templateMapper.args[0];
-            switch (targetProperty.kind) {
-                case "String":
-                    const targetPropertyName = targetProperty.value;
-                    if (typeName === "RequestParameter") {
-                        const bodyType = getOperationParameters(context.program, entity)[0].body.type;
-                        switch (bodyType.kind) {
-                            case "Model":
-                                if (!bodyType.properties.get(targetPropertyName)) {
-                                    reportDiagnostic(context.program, {
-                                        code: "request-parameter-invalid",
-                                        target: context.decoratorTarget,
-                                        format: { name: targetPropertyName },
-                                    });
-                                }
-                                break;
-                            default:
-                                reportDiagnostic(context.program, {
-                                    code: "request-parameter-invalid",
-                                    target: context.decoratorTarget,
-                                    format: { name: targetPropertyName },
-                                });
-                                break;
-                        }
-                    }
-                    else if (typeName === "ResponseProperty") {
-                        const responses = getResponsesForOperation(context.program, entity)[0];
-                        for (const response of responses) {
-                            switch (response.type.kind) {
-                                case "Model":
-                                    // error responses don't need the property
-                                    if (isErrorModel(context.program, response.type)) {
-                                        continue;
-                                    }
-                                    // however ALL non-error responses do need the property!
-                                    if (!((_b = response.type) === null || _b === void 0 ? void 0 : _b.properties.get(targetPropertyName))) {
-                                        reportDiagnostic(context.program, {
-                                            code: "response-property-invalid",
-                                            target: context.decoratorTarget,
-                                            format: { name: targetPropertyName },
-                                        });
-                                    }
-                                    break;
-                                default:
-                                    reportDiagnostic(context.program, {
-                                        code: "response-property-invalid",
-                                        target: context.decoratorTarget,
-                                        format: { name: targetPropertyName },
-                                    });
-                                    break;
-                            }
-                        }
-                    }
-            }
-        }
+    var _a;
+    const { program } = context;
+    const [operationInfo, diagnostics] = getLroOperationInfo(program, entity, linkedOperation, parameters);
+    if (diagnostics.length > 0) {
+        program.reportDiagnostics(diagnostics);
     }
     // An operation may have many operationLinks, so treat them as a collection
     let items = context.program.stateMap(operationLinkKey).get(entity);
@@ -412,6 +455,9 @@ export function $operationLink(context, entity, linkedOperation, linkType, param
         parameters: parameters,
         linkedOperation: linkedOperation,
         linkType: linkType,
+        link: operationInfo === null || operationInfo === void 0 ? void 0 : operationInfo.getOperationLink(),
+        parameterMap: (_a = operationInfo === null || operationInfo === void 0 ? void 0 : operationInfo.getInvocationInfo()) === null || _a === void 0 ? void 0 : _a.parameterMap,
+        result: operationInfo === null || operationInfo === void 0 ? void 0 : operationInfo.getResultInfo(),
     });
     context.program.stateMap(operationLinkKey).set(entity, items);
 }
@@ -432,21 +478,62 @@ export function getOperationLinks(program, entity) {
     return program.stateMap(operationLinkKey).get(entity);
 }
 export function $pollingOperation(context, target, linkedOperation, parameters) {
+    const { program } = context;
     const isValidReturnType = target.returnType.kind === "Model" ||
         (target.returnType.kind === "Union" &&
             [...target.returnType.variants.values()].every((x) => x.type.kind === "Model"));
-    if (isValidReturnType) {
-        context.call($operationLink, target, linkedOperation, PollingOperationKey, parameters);
+    if (!isValidReturnType) {
+        reportDiagnostic(context.program, {
+            code: "polling-operation-return-model",
+            target: target,
+        });
+        return;
     }
-    else {
+    context.call($operationLink, target, linkedOperation, PollingOperationKey, parameters);
+    const operationDetails = getOperationLink(program, target, PollingOperationKey);
+    if (operationDetails === undefined || operationDetails.result === undefined) {
         reportDiagnostic(context.program, {
             code: "polling-operation-return-model",
             target: target,
         });
+        return;
+    }
+    if (operationDetails.result.statusMonitor === undefined) {
+        reportDiagnostic(context.program, {
+            code: "polling-operation-no-status-monitor",
+            target: linkedOperation,
+        });
+        return;
+    }
+    if (operationDetails.result.statusMonitor.terminationInfo.succeededState.length < 1) {
+        reportDiagnostic(context.program, {
+            code: "polling-operation-no-lro-success",
+            target: operationDetails.result.statusMonitor.type,
+        });
+    }
+    if (operationDetails.result.statusMonitor.terminationInfo.failedState.length < 1) {
+        reportDiagnostic(context.program, {
+            code: "polling-operation-no-lro-failure",
+            target: operationDetails.result.statusMonitor.type,
+        });
+    }
+    if (operationDetails.parameterMap === undefined && operationDetails.link === undefined) {
+        reportDiagnostic(context.program, {
+            code: "polling-operation-no-ref-or-link",
+            target: target,
+        });
     }
 }
 export function $finalOperation(context, entity, linkedOperation, parameters) {
+    const { program } = context;
     context.call($operationLink, entity, linkedOperation, FinalOperationKey, parameters);
+    const operationDetails = getOperationLink(program, entity, FinalOperationKey);
+    if (operationDetails === undefined || operationDetails.result === undefined) {
+        reportDiagnostic(context.program, {
+            code: "invalid-final-operation",
+            target: entity,
+        });
+    }
 }
 export function $nextPageOperation(context, entity, linkedOperation, parameters) {
     context.call($operationLink, entity, linkedOperation, "nextPage", parameters);