@typespec/mutator-framework 0.12.0

package/dist/src/mutation-node/factory.js

@@ -0,0 +1,45 @@
+import { EnumMemberMutationNode } from "./enum-member.js";
+import { EnumMutationNode } from "./enum.js";
+import { InterfaceMutationNode } from "./interface.js";
+import { IntrinsicMutationNode } from "./intrinsic.js";
+import { LiteralMutationNode } from "./literal.js";
+import { ModelPropertyMutationNode } from "./model-property.js";
+import { ModelMutationNode } from "./model.js";
+import { OperationMutationNode } from "./operation.js";
+import { ScalarMutationNode } from "./scalar.js";
+import { TupleMutationNode } from "./tuple.js";
+import { UnionVariantMutationNode } from "./union-variant.js";
+import { UnionMutationNode } from "./union.js";
+export function mutationNodeFor(subgraph, sourceType) {
+    switch (sourceType.kind) {
+        case "Operation":
+            return new OperationMutationNode(subgraph, sourceType);
+        case "Interface":
+            return new InterfaceMutationNode(subgraph, sourceType);
+        case "Model":
+            return new ModelMutationNode(subgraph, sourceType);
+        case "ModelProperty":
+            return new ModelPropertyMutationNode(subgraph, sourceType);
+        case "Scalar":
+            return new ScalarMutationNode(subgraph, sourceType);
+        case "Tuple":
+            return new TupleMutationNode(subgraph, sourceType);
+        case "Union":
+            return new UnionMutationNode(subgraph, sourceType);
+        case "UnionVariant":
+            return new UnionVariantMutationNode(subgraph, sourceType);
+        case "Enum":
+            return new EnumMutationNode(subgraph, sourceType);
+        case "EnumMember":
+            return new EnumMemberMutationNode(subgraph, sourceType);
+        case "String":
+        case "Number":
+        case "Boolean":
+            return new LiteralMutationNode(subgraph, sourceType);
+        case "Intrinsic":
+            return new IntrinsicMutationNode(subgraph, sourceType);
+        default:
+            throw new Error("Unsupported type kind: " + sourceType.kind);
+    }
+}
+//# sourceMappingURL=factory.js.map

package/dist/src/mutation-node/factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.js","sourceRoot":"","sources":["../../../src/mutation-node/factory.ts"],"names":[],"mappings":"AAiBA,OAAO,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AACvD,OAAO,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AACvD,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EAAE,yBAAyB,EAAE,MAAM,qBAAqB,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE/C,OAAO,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAC/C,OAAO,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE/C,MAAM,UAAU,eAAe,CAC7B,QAA0B,EAC1B,UAAa;IAEb,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;QACxB,KAAK,WAAW;YACd,OAAO,IAAI,qBAAqB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QACnF,KAAK,WAAW;YACd,OAAO,IAAI,qBAAqB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QACnF,KAAK,OAAO;YACV,OAAO,IAAI,iBAAiB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QAC/E,KAAK,eAAe;YAClB,OAAO,IAAI,yBAAyB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QACvF,KAAK,QAAQ;YACX,OAAO,IAAI,kBAAkB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QAChF,KAAK,OAAO;YACV,OAAO,IAAI,iBAAiB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QAC/E,KAAK,OAAO;YACV,OAAO,IAAI,iBAAiB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QAC/E,KAAK,cAAc;YACjB,OAAO,IAAI,wBAAwB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QACtF,KAAK,MAAM;YACT,OAAO,IAAI,gBAAgB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QAC9E,KAAK,YAAY;YACf,OAAO,IAAI,sBAAsB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QACpF,KAAK,QAAQ,CAAC;QACd,KAAK,QAAQ,CAAC;QACd,KAAK,SAAS;YACZ,OAAO,IAAI,mBAAmB,CAC5B,QAAQ,EACR,UAA6D,CACpC,CAAC;QAC9B,KAAK,WAAW;YACd,OAAO,IAAI,qBAAqB,CAAC,QAAQ,EAAE,UAAU,CAA2B,CAAC;QACnF;YACE,MAAM,IAAI,KAAK,CAAC,yBAAyB,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;IACjE,CAAC;AACH,CAAC"}

package/dist/src/mutation-node/index.d.ts

@@ -0,0 +1,15 @@
+export * from "./mutation-node.js";
+export * from "./enum.js";
+export * from "./interface.js";
+export * from "./intrinsic.js";
+export * from "./literal.js";
+export * from "./model-property.js";
+export * from "./model.js";
+export * from "./mutation-edge.js";
+export * from "./mutation-subgraph.js";
+export * from "./operation.js";
+export * from "./scalar.js";
+export * from "./tuple.js";
+export * from "./union-variant.js";
+export * from "./union.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/src/mutation-node/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/mutation-node/index.ts"],"names":[],"mappings":"AAAA,cAAc,oBAAoB,CAAC;AAEnC,cAAc,WAAW,CAAC;AAC1B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,cAAc,CAAC;AAC7B,cAAc,qBAAqB,CAAC;AACpC,cAAc,YAAY,CAAC;AAC3B,cAAc,oBAAoB,CAAC;AACnC,cAAc,wBAAwB,CAAC;AACvC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,aAAa,CAAC;AAC5B,cAAc,YAAY,CAAC;AAC3B,cAAc,oBAAoB,CAAC;AACnC,cAAc,YAAY,CAAC"}

package/dist/src/mutation-node/index.js

@@ -0,0 +1,16 @@
+export * from "./mutation-node.js";
+export * from "./enum.js";
+export * from "./interface.js";
+export * from "./intrinsic.js";
+export * from "./literal.js";
+export * from "./model-property.js";
+export * from "./model.js";
+export * from "./mutation-edge.js";
+export * from "./mutation-subgraph.js";
+export * from "./operation.js";
+export * from "./scalar.js";
+export * from "./tuple.js";
+export * from "./union-variant.js";
+export * from "./union.js";
+//export * from "./enum-member.js";
+//# sourceMappingURL=index.js.map

package/dist/src/mutation-node/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/mutation-node/index.ts"],"names":[],"mappings":"AAAA,cAAc,oBAAoB,CAAC;AAEnC,cAAc,WAAW,CAAC;AAC1B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,cAAc,CAAC;AAC7B,cAAc,qBAAqB,CAAC;AACpC,cAAc,YAAY,CAAC;AAC3B,cAAc,oBAAoB,CAAC;AACnC,cAAc,wBAAwB,CAAC;AACvC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,aAAa,CAAC;AAC5B,cAAc,YAAY,CAAC;AAC3B,cAAc,oBAAoB,CAAC;AACnC,cAAc,YAAY,CAAC;AAC3B,mCAAmC"}

package/dist/src/mutation-node/interface.d.ts

@@ -0,0 +1,8 @@
+import type { Interface, Operation } from "@typespec/compiler";
+import { MutationNode } from "./mutation-node.js";
+export declare class InterfaceMutationNode extends MutationNode<Interface> {
+    readonly kind = "Interface";
+    traverse(): void;
+    connectOperation(opNode: MutationNode<Operation>, opName: string): void;
+}
+//# sourceMappingURL=interface.d.ts.map

package/dist/src/mutation-node/interface.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"interface.d.ts","sourceRoot":"","sources":["../../../src/mutation-node/interface.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAE/D,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,qBAAa,qBAAsB,SAAQ,YAAY,CAAC,SAAS,CAAC;IAChE,QAAQ,CAAC,IAAI,eAAe;IAE5B,QAAQ;IAOR,gBAAgB,CAAC,MAAM,EAAE,YAAY,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,MAAM;CAkBjE"}

package/dist/src/mutation-node/interface.js

@@ -0,0 +1,30 @@
+import { MutationEdge } from "./mutation-edge.js";
+import { MutationNode } from "./mutation-node.js";
+export class InterfaceMutationNode extends MutationNode {
+    kind = "Interface";
+    traverse() {
+        for (const [opName, op] of this.sourceType.operations) {
+            const opNode = this.subgraph.getNode(op);
+            this.connectOperation(opNode, opName);
+        }
+    }
+    connectOperation(opNode, opName) {
+        MutationEdge.create(this, opNode, {
+            onTailMutation: () => {
+                this.mutatedType.operations.delete(opName);
+                this.mutatedType.operations.set(opNode.mutatedType.name, opNode.mutatedType);
+            },
+            onTailDeletion: () => {
+                this.mutatedType.operations.delete(opName);
+            },
+            onTailReplaced: (newTail) => {
+                if (newTail.mutatedType.kind !== "Operation") {
+                    throw new Error("Cannot replace operation with non-operation type");
+                }
+                this.mutatedType.operations.delete(opName);
+                this.mutatedType.operations.set(newTail.mutatedType.name, newTail.mutatedType);
+            },
+        });
+    }
+}
+//# sourceMappingURL=interface.js.map

package/dist/src/mutation-node/interface.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"interface.js","sourceRoot":"","sources":["../../../src/mutation-node/interface.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,MAAM,OAAO,qBAAsB,SAAQ,YAAuB;IACvD,IAAI,GAAG,WAAW,CAAC;IAE5B,QAAQ;QACN,KAAK,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,CAAC;YACtD,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACzC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED,gBAAgB,CAAC,MAA+B,EAAE,MAAc;QAC9D,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE;YAChC,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBAC3C,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;YAC/E,CAAC;YACD,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC7C,CAAC;YACD,cAAc,EAAE,CAAC,OAAO,EAAE,EAAE;gBAC1B,IAAI,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC7C,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;gBACtE,CAAC;gBACD,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBAC3C,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;YACjF,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/src/mutation-node/intrinsic.d.ts

@@ -0,0 +1,7 @@
+import type { IntrinsicType } from "@typespec/compiler";
+import { MutationNode } from "./mutation-node.js";
+export declare class IntrinsicMutationNode extends MutationNode<IntrinsicType> {
+    readonly kind = "Intrinsic";
+    traverse(): void;
+}
+//# sourceMappingURL=intrinsic.d.ts.map

package/dist/src/mutation-node/intrinsic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"intrinsic.d.ts","sourceRoot":"","sources":["../../../src/mutation-node/intrinsic.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,qBAAa,qBAAsB,SAAQ,YAAY,CAAC,aAAa,CAAC;IACpE,QAAQ,CAAC,IAAI,eAAe;IAE5B,QAAQ;CACT"}

package/dist/src/mutation-node/intrinsic.js

@@ -0,0 +1,6 @@
+import { MutationNode } from "./mutation-node.js";
+export class IntrinsicMutationNode extends MutationNode {
+    kind = "Intrinsic";
+    traverse() { }
+}
+//# sourceMappingURL=intrinsic.js.map

package/dist/src/mutation-node/intrinsic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"intrinsic.js","sourceRoot":"","sources":["../../../src/mutation-node/intrinsic.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,MAAM,OAAO,qBAAsB,SAAQ,YAA2B;IAC3D,IAAI,GAAG,WAAW,CAAC;IAE5B,QAAQ,KAAI,CAAC;CACd"}

package/dist/src/mutation-node/literal.d.ts

@@ -0,0 +1,7 @@
+import type { BooleanLiteral, NumericLiteral, StringLiteral } from "@typespec/compiler";
+import { MutationNode } from "./mutation-node.js";
+export declare class LiteralMutationNode extends MutationNode<StringLiteral | NumericLiteral | BooleanLiteral> {
+    readonly kind = "Literal";
+    traverse(): void;
+}
+//# sourceMappingURL=literal.d.ts.map

package/dist/src/mutation-node/literal.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"literal.d.ts","sourceRoot":"","sources":["../../../src/mutation-node/literal.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACxF,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,qBAAa,mBAAoB,SAAQ,YAAY,CACnD,aAAa,GAAG,cAAc,GAAG,cAAc,CAChD;IACC,QAAQ,CAAC,IAAI,aAAa;IAE1B,QAAQ;CACT"}

package/dist/src/mutation-node/literal.js

@@ -0,0 +1,6 @@
+import { MutationNode } from "./mutation-node.js";
+export class LiteralMutationNode extends MutationNode {
+    kind = "Literal";
+    traverse() { }
+}
+//# sourceMappingURL=literal.js.map

package/dist/src/mutation-node/literal.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"literal.js","sourceRoot":"","sources":["../../../src/mutation-node/literal.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,MAAM,OAAO,mBAAoB,SAAQ,YAExC;IACU,IAAI,GAAG,SAAS,CAAC;IAE1B,QAAQ,KAAI,CAAC;CACd"}

package/dist/src/mutation-node/model-property.d.ts

@@ -0,0 +1,10 @@
+import type { ModelProperty, Type } from "@typespec/compiler";
+import { MutationNode } from "./mutation-node.js";
+export declare class ModelPropertyMutationNode extends MutationNode<ModelProperty> {
+    #private;
+    readonly kind = "ModelProperty";
+    traverse(): void;
+    connectReference(referenceNode: MutationNode<Type>): void;
+    connectType(typeNode: MutationNode<Type>): void;
+}
+//# sourceMappingURL=model-property.d.ts.map

package/dist/src/mutation-node/model-property.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-property.d.ts","sourceRoot":"","sources":["../../../src/mutation-node/model-property.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAE9D,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,qBAAa,yBAA0B,SAAQ,YAAY,CAAC,aAAa,CAAC;;IACxE,QAAQ,CAAC,IAAI,mBAAmB;IAGhC,QAAQ;IAQR,gBAAgB,CAAC,aAAa,EAAE,YAAY,CAAC,IAAI,CAAC;IAiBlD,WAAW,CAAC,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC;CAmBzC"}

package/dist/src/mutation-node/model-property.js

@@ -0,0 +1,48 @@
+import { MutationEdge } from "./mutation-edge.js";
+import { MutationNode } from "./mutation-node.js";
+export class ModelPropertyMutationNode extends MutationNode {
+    kind = "ModelProperty";
+    #referenceMutated = false;
+    traverse() {
+        const typeNode = this.subgraph.getNode(this.sourceType.type);
+        const referenceNode = this.subgraph.getReferenceNode(this.sourceType);
+        this.connectType(typeNode);
+        this.connectReference(referenceNode);
+    }
+    connectReference(referenceNode) {
+        MutationEdge.create(this, referenceNode, {
+            onTailMutation: () => {
+                this.#referenceMutated = true;
+                this.mutatedType.type = referenceNode.mutatedType;
+            },
+            onTailDeletion: () => {
+                this.#referenceMutated = true;
+                this.mutatedType.type = this.$.intrinsic.any;
+            },
+            onTailReplaced: (newTail) => {
+                this.#referenceMutated = true;
+                this.mutatedType.type = newTail.mutatedType;
+            },
+        });
+    }
+    connectType(typeNode) {
+        MutationEdge.create(this, typeNode, {
+            onTailMutation: () => {
+                if (this.#referenceMutated) {
+                    return;
+                }
+                this.mutatedType.type = typeNode.mutatedType;
+            },
+            onTailDeletion: () => {
+                if (this.#referenceMutated) {
+                    return;
+                }
+                this.mutatedType.type = this.$.intrinsic.any;
+            },
+            onTailReplaced: (newTail) => {
+                this.mutatedType.type = newTail.mutatedType;
+            },
+        });
+    }
+}
+//# sourceMappingURL=model-property.js.map

package/dist/src/mutation-node/model-property.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-property.js","sourceRoot":"","sources":["../../../src/mutation-node/model-property.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,MAAM,OAAO,yBAA0B,SAAQ,YAA2B;IAC/D,IAAI,GAAG,eAAe,CAAC;IAChC,iBAAiB,GAAG,KAAK,CAAC;IAE1B,QAAQ;QACN,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAC7D,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAEtE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC3B,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;IACvC,CAAC;IAED,gBAAgB,CAAC,aAAiC;QAChD,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,aAAa,EAAE;YACvC,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;gBAC9B,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,aAAa,CAAC,WAAW,CAAC;YACpD,CAAC;YACD,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;gBAC9B,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC;YAC/C,CAAC;YACD,cAAc,EAAE,CAAC,OAAO,EAAE,EAAE;gBAC1B,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;gBAC9B,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,OAAO,CAAC,WAAW,CAAC;YAC9C,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,WAAW,CAAC,QAA4B;QACtC,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE;YAClC,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;oBAC3B,OAAO;gBACT,CAAC;gBACD,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,QAAQ,CAAC,WAAW,CAAC;YAC/C,CAAC;YACD,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;oBAC3B,OAAO;gBACT,CAAC;gBACD,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC;YAC/C,CAAC;YACD,cAAc,EAAE,CAAC,OAAO,EAAE,EAAE;gBAC1B,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,OAAO,CAAC,WAAW,CAAC;YAC9C,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/src/mutation-node/model-property.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=model-property.test.d.ts.map

package/dist/src/mutation-node/model-property.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-property.test.d.ts","sourceRoot":"","sources":["../../../src/mutation-node/model-property.test.ts"],"names":[],"mappings":""}

package/dist/src/mutation-node/model-property.test.js

@@ -0,0 +1,108 @@
+import { t } from "@typespec/compiler/testing";
+import { $ } from "@typespec/compiler/typekit";
+import { beforeEach, expect, it } from "vitest";
+import { Tester } from "../../test/test-host.js";
+import { getSubgraph } from "../../test/utils.js";
+let runner;
+beforeEach(async () => {
+    runner = await Tester.createInstance();
+});
+it("handles mutation of property types", async () => {
+    const { prop, program } = await runner.compile(t.code `
+      model Foo {
+        ${t.modelProperty("prop")}: string;
+      }
+    `);
+    const subgraph = getSubgraph(program);
+    const propNode = subgraph.getNode(prop);
+    const stringNode = subgraph.getNode($(program).builtin.string);
+    stringNode.mutate();
+    expect(propNode.isMutated).toBe(true);
+    expect(propNode.mutatedType.type === stringNode.mutatedType).toBe(true);
+});
+it("handles mutating a reference", async () => {
+    const { Foo, Bar, prop, program } = await runner.compile(t.code `
+      model ${t.model("Foo")} {
+        ${t.modelProperty("prop")}: Bar;
+      };
+      model ${t.model("Bar")} {}
+    `);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const propNode = subgraph.getNode(prop);
+    const barPrime = subgraph.getReferenceNode(prop);
+    // initially the source type is just Bar.
+    expect(barPrime.sourceType === Bar).toBe(true);
+    barPrime.mutate();
+    expect(fooNode.isMutated).toBe(true);
+    expect(propNode.isMutated).toBe(true);
+    expect(barPrime.isMutated).toBe(true);
+    expect(fooNode.mutatedType.properties.get("prop").type === barPrime.mutatedType).toBeTruthy();
+    const barNode = subgraph.getNode(Bar);
+    barNode.mutate();
+    expect(barNode.isMutated).toBe(true);
+    expect(barPrime.isMutated).toBe(true);
+    // the mutated type doesn't change here.
+    expect(fooNode.mutatedType.properties.get("prop").type === barPrime.mutatedType).toBeTruthy();
+});
+it("handles replacing the model reference", async () => {
+    const { Foo, Bar, prop, program } = await runner.compile(t.code `
+      model ${t.model("Foo")} {
+        ${t.modelProperty("prop")}: Bar;
+      };
+      model ${t.model("Bar")} {}
+    `);
+    const tk = $(program);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const propNode = subgraph.getNode(prop);
+    const barPrime = subgraph.getReferenceNode(prop);
+    const unionType = tk.union.create({
+        variants: [
+            tk.unionVariant.create({ type: tk.builtin.string }),
+            tk.unionVariant.create({ type: Bar }),
+        ],
+    });
+    const replacedBarPrime = barPrime.replace(unionType);
+    // the subgraph now returns the new reference node
+    expect(subgraph.getReferenceNode(prop) === replacedBarPrime).toBe(true);
+    // foo and prop are marked mutated, barPrime is replaced
+    expect(fooNode.isMutated).toBe(true);
+    expect(propNode.isMutated).toBe(true);
+    expect(barPrime.isReplaced).toBe(true);
+    // prop's type is the replaced type
+    expect(tk.union.is(propNode.mutatedType.type)).toBe(true);
+    expect(fooNode.mutatedType.properties.get("prop").type === replacedBarPrime.mutatedType).toBeTruthy();
+});
+it("handles mutating a reference to a reference", async () => {
+    const { myString, Foo, fprop, Bar, program } = await runner.compile(t.code `
+      scalar ${t.scalar("myString")} extends string;
+      model ${t.model("Foo")} {
+        ${t.modelProperty("fprop")}: myString;
+      };
+      model ${t.model("Bar")} {
+        bprop: Foo.fprop;
+      }
+    `);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const barNode = subgraph.getNode(Bar);
+    const myStringNode = subgraph.getNode(myString);
+    myStringNode.mutate();
+    expect(myStringNode.isMutated).toBe(true);
+    expect(fooNode.isMutated).toBe(true);
+    expect(barNode.isMutated).toBe(true);
+    // Foo.prop's type is the mutated myString
+    expect(fooNode.mutatedType.properties.get("fprop").type === myStringNode.mutatedType).toBeTruthy();
+    // Bar.prop's type is the mutated Foo.prop
+    expect(barNode.mutatedType.properties.get("bprop").type ===
+        fooNode.mutatedType.properties.get("fprop")).toBeTruthy();
+    const fpropRefNode = subgraph.getReferenceNode(fprop);
+    fpropRefNode.mutate();
+    expect(fpropRefNode.isMutated).toBe(true);
+    expect(fooNode.mutatedType.properties.get("fprop").type === fpropRefNode.mutatedType).toBeTruthy();
+    // Bar.bprop references the mutated type (though is the same reference since fprop was already mutated)
+    expect(barNode.mutatedType.properties.get("bprop").type ===
+        fooNode.mutatedType.properties.get("fprop")).toBeTruthy();
+});
+//# sourceMappingURL=model-property.test.js.map

package/dist/src/mutation-node/model-property.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-property.test.js","sourceRoot":"","sources":["../../../src/mutation-node/model-property.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAuB,MAAM,4BAA4B,CAAC;AACpE,OAAO,EAAE,CAAC,EAAE,MAAM,4BAA4B,CAAC;AAC/C,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAChD,OAAO,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAElD,IAAI,MAAsB,CAAC;AAC3B,UAAU,CAAC,KAAK,IAAI,EAAE;IACpB,MAAM,GAAG,MAAM,MAAM,CAAC,cAAc,EAAE,CAAC;AACzC,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;IAClD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAA;;UAE7C,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC;;KAE5B,CAAC,CAAC;IACL,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC/D,UAAU,CAAC,MAAM,EAAE,CAAC;IACpB,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,KAAK,UAAU,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1E,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;IAC5C,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAA;cACnD,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;UAClB,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC;;cAEnB,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;KACvB,CAAC,CAAC;IAEL,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACtC,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,QAAQ,GAAG,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAEjD,yCAAyC;IACzC,MAAM,CAAC,QAAQ,CAAC,UAAU,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAE/C,QAAQ,CAAC,MAAM,EAAE,CAAC;IAClB,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC,IAAI,KAAK,QAAQ,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;IAE/F,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACtC,OAAO,CAAC,MAAM,EAAE,CAAC;IACjB,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtC,wCAAwC;IACxC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC,IAAI,KAAK,QAAQ,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;AACjG,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;IACrD,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAA;cACnD,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;UAClB,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC;;cAEnB,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;KACvB,CAAC,CAAC;IACL,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC;IACtB,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACtC,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,QAAQ,GAAG,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;IACjD,MAAM,SAAS,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC;QAChC,QAAQ,EAAE;YACR,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACnD,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;SACtC;KACF,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAG,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAErD,kDAAkD;IAClD,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,KAAK,gBAAgB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAExE,wDAAwD;IACxD,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEvC,mCAAmC;IACnC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1D,MAAM,CACJ,OAAO,CAAC,WAAY,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC,IAAI,KAAK,gBAAgB,CAAC,WAAW,CACnF,CAAC,UAAU,EAAE,CAAC;AACjB,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;IAC3D,MAAM,EAAE,QAAQ,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAA;eAC7D,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC;cACrB,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;UAClB,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC;;cAEpB,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC;;;KAGvB,CAAC,CAAC;IAEL,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACtC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACtC,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAEhD,YAAY,CAAC,MAAM,EAAE,CAAC;IACtB,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAErC,0CAA0C;IAC1C,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC,IAAI,KAAK,YAAY,CAAC,WAAW,CAC/E,CAAC,UAAU,EAAE,CAAC;IAEf,0CAA0C;IAC1C,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC,IAAI;QAC/C,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAE,CAC/C,CAAC,UAAU,EAAE,CAAC;IAEf,MAAM,YAAY,GAAG,QAAQ,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACtD,YAAY,CAAC,MAAM,EAAE,CAAC;IACtB,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC,IAAI,KAAK,YAAY,CAAC,WAAW,CAC/E,CAAC,UAAU,EAAE,CAAC;IAEf,uGAAuG;IACvG,MAAM,CACJ,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC,IAAI;QAC/C,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAE,CAC/C,CAAC,UAAU,EAAE,CAAC;AACjB,CAAC,CAAC,CAAC"}

package/dist/src/mutation-node/model.d.ts

@@ -0,0 +1,10 @@
+import type { Model, ModelProperty, Type } from "@typespec/compiler";
+import { MutationNode } from "./mutation-node.js";
+export declare class ModelMutationNode extends MutationNode<Model> {
+    readonly kind = "Model";
+    traverse(): void;
+    connectToBase(baseNode: MutationNode<Model>): void;
+    connectProperty(propNode: MutationNode<ModelProperty>, sourcePropName: string): void;
+    connectIndexerValue(indexerNode: MutationNode<Type>): void;
+}
+//# sourceMappingURL=model.d.ts.map

package/dist/src/mutation-node/model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../../src/mutation-node/model.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAErE,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,qBAAa,iBAAkB,SAAQ,YAAY,CAAC,KAAK,CAAC;IACxD,QAAQ,CAAC,IAAI,WAAW;IAExB,QAAQ;IAiBR,aAAa,CAAC,QAAQ,EAAE,YAAY,CAAC,KAAK,CAAC;IAiB3C,eAAe,CAAC,QAAQ,EAAE,YAAY,CAAC,aAAa,CAAC,EAAE,cAAc,EAAE,MAAM;IAmB7E,mBAAmB,CAAC,WAAW,EAAE,YAAY,CAAC,IAAI,CAAC;CA4BpD"}

package/dist/src/mutation-node/model.js

@@ -0,0 +1,82 @@
+import { MutationEdge } from "./mutation-edge.js";
+import { MutationNode } from "./mutation-node.js";
+export class ModelMutationNode extends MutationNode {
+    kind = "Model";
+    traverse() {
+        if (this.sourceType.baseModel) {
+            const baseNode = this.subgraph.getNode(this.sourceType.baseModel);
+            this.connectToBase(baseNode);
+        }
+        for (const [propName, prop] of this.sourceType.properties) {
+            const propNode = this.subgraph.getNode(prop);
+            this.connectProperty(propNode, propName);
+        }
+        if (this.sourceType.indexer) {
+            const indexerNode = this.subgraph.getNode(this.sourceType.indexer.value);
+            this.connectIndexerValue(indexerNode);
+        }
+    }
+    connectToBase(baseNode) {
+        MutationEdge.create(this, baseNode, {
+            onTailMutation: () => {
+                this.mutatedType.baseModel = baseNode.mutatedType;
+            },
+            onTailDeletion: () => {
+                this.mutatedType.baseModel = undefined;
+            },
+            onTailReplaced: (newTail) => {
+                if (newTail.mutatedType.kind !== "Model") {
+                    throw new Error("Cannot replace base model with non-model type");
+                }
+                this.mutatedType.baseModel = newTail.mutatedType;
+            },
+        });
+    }
+    connectProperty(propNode, sourcePropName) {
+        MutationEdge.create(this, propNode, {
+            onTailMutation: () => {
+                this.mutatedType.properties.delete(sourcePropName);
+                this.mutatedType.properties.set(propNode.mutatedType.name, propNode.mutatedType);
+            },
+            onTailDeletion: () => {
+                this.mutatedType.properties.delete(sourcePropName);
+            },
+            onTailReplaced: (newTail) => {
+                if (newTail.mutatedType.kind !== "ModelProperty") {
+                    throw new Error("Cannot replace model property with non-model property type");
+                }
+                this.mutatedType.properties.delete(sourcePropName);
+                this.mutatedType.properties.set(newTail.mutatedType.name, newTail.mutatedType);
+            },
+        });
+    }
+    connectIndexerValue(indexerNode) {
+        MutationEdge.create(this, indexerNode, {
+            onTailMutation: () => {
+                if (this.mutatedType.indexer) {
+                    this.mutatedType.indexer = {
+                        key: this.mutatedType.indexer.key,
+                        value: indexerNode.mutatedType,
+                    };
+                }
+            },
+            onTailDeletion: () => {
+                if (this.mutatedType.indexer) {
+                    this.mutatedType.indexer = {
+                        key: this.mutatedType.indexer.key,
+                        value: this.$.intrinsic.any,
+                    };
+                }
+            },
+            onTailReplaced: (newTail) => {
+                if (this.mutatedType.indexer) {
+                    this.mutatedType.indexer = {
+                        key: this.mutatedType.indexer.key,
+                        value: newTail.mutatedType,
+                    };
+                }
+            },
+        });
+    }
+}
+//# sourceMappingURL=model.js.map

package/dist/src/mutation-node/model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.js","sourceRoot":"","sources":["../../../src/mutation-node/model.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAElD,MAAM,OAAO,iBAAkB,SAAQ,YAAmB;IAC/C,IAAI,GAAG,OAAO,CAAC;IAExB,QAAQ;QACN,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC;YAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YAClE,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;QAED,KAAK,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,CAAC;YAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC7C,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YAC5B,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACzE,IAAI,CAAC,mBAAmB,CAAC,WAAW,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED,aAAa,CAAC,QAA6B;QACzC,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE;YAClC,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,CAAC,WAAY,CAAC,SAAS,GAAG,QAAQ,CAAC,WAAW,CAAC;YACrD,CAAC;YACD,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,CAAC,WAAW,CAAC,SAAS,GAAG,SAAS,CAAC;YACzC,CAAC;YACD,cAAc,EAAE,CAAC,OAAO,EAAE,EAAE;gBAC1B,IAAI,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBACzC,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;gBACnE,CAAC;gBACD,IAAI,CAAC,WAAW,CAAC,SAAS,GAAG,OAAO,CAAC,WAAW,CAAC;YACnD,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,eAAe,CAAC,QAAqC,EAAE,cAAsB;QAC3E,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE;YAClC,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;gBACnD,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC,WAAW,CAAC,CAAC;YACnF,CAAC;YACD,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YACrD,CAAC;YACD,cAAc,EAAE,CAAC,OAAO,EAAE,EAAE;gBAC1B,IAAI,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;oBACjD,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;gBAChF,CAAC;gBACD,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;gBACnD,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;YACjF,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,mBAAmB,CAAC,WAA+B;QACjD,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,EAAE;YACrC,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;oBAC7B,IAAI,CAAC,WAAW,CAAC,OAAO,GAAG;wBACzB,GAAG,EAAE,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG;wBACjC,KAAK,EAAE,WAAW,CAAC,WAAW;qBAC/B,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,cAAc,EAAE,GAAG,EAAE;gBACnB,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;oBAC7B,IAAI,CAAC,WAAW,CAAC,OAAO,GAAG;wBACzB,GAAG,EAAE,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG;wBACjC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG;qBAC5B,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,cAAc,EAAE,CAAC,OAAO,EAAE,EAAE;gBAC1B,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;oBAC7B,IAAI,CAAC,WAAW,CAAC,OAAO,GAAG;wBACzB,GAAG,EAAE,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG;wBACjC,KAAK,EAAE,OAAO,CAAC,WAAW;qBAC3B,CAAC;gBACJ,CAAC;YACH,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/src/mutation-node/model.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=model.test.d.ts.map

package/dist/src/mutation-node/model.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.test.d.ts","sourceRoot":"","sources":["../../../src/mutation-node/model.test.ts"],"names":[],"mappings":""}

package/dist/src/mutation-node/model.test.js

@@ -0,0 +1,133 @@
+import { t } from "@typespec/compiler/testing";
+import { beforeEach, expect, it } from "vitest";
+import { Tester } from "../../test/test-host.js";
+import { getSubgraph } from "../../test/utils.js";
+let runner;
+beforeEach(async () => {
+    runner = await Tester.createInstance();
+});
+it("handles mutation of properties", async () => {
+    const { Foo, program } = await runner.compile(t.code `
+      model ${t.model("Foo")} {
+        prop: string;
+      }
+    `);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const propNode = subgraph.getNode(Foo.properties.get("prop"));
+    propNode.mutate();
+    expect(fooNode.isMutated).toBe(true);
+    expect(fooNode.mutatedType.properties.get("prop") === propNode.mutatedType).toBe(true);
+});
+it("handles deletion of properties", async () => {
+    const { Foo, program } = await runner.compile(t.code `
+      model ${t.model("Foo")} {
+        prop: string;
+      }
+    `);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const propNode = subgraph.getNode(Foo.properties.get("prop"));
+    propNode.delete();
+    expect(fooNode.isMutated).toBe(true);
+    expect(fooNode.mutatedType.properties.get("prop")).toBeUndefined();
+});
+it("handles mutation of properties with name change", async () => {
+    const { Foo, program } = await runner.compile(t.code `
+      model ${t.model("Foo")} {
+        prop: string;
+      }
+    `);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const propNode = subgraph.getNode(Foo.properties.get("prop"));
+    propNode.mutate((clone) => (clone.name = "propRenamed"));
+    expect(fooNode.isMutated).toBe(true);
+    expect(fooNode.mutatedType.properties.get("prop") === undefined).toBe(true);
+    expect(fooNode.mutatedType.properties.get("propRenamed") === propNode.mutatedType).toBe(true);
+});
+it("handles mutation of base models", async () => {
+    const { Foo, Bar, program } = await runner.compile(t.code `
+      model ${t.model("Foo")} extends Bar {
+        barProp: string;
+      }
+
+      model ${t.model("Bar")} {
+        bazProp: string;
+      }
+    `);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const barNode = subgraph.getNode(Bar);
+    barNode.mutate();
+    expect(barNode.isMutated).toBe(true);
+    expect(fooNode.isMutated).toBe(true);
+    expect(fooNode.mutatedType.baseModel === barNode.mutatedType).toBeTruthy();
+});
+it("handles deletion of base models", async () => {
+    const { Foo, Bar, program } = await runner.compile(t.code `
+      model ${t.model("Foo")} extends Bar {
+        barProp: string;
+      }
+
+      model ${t.model("Bar")} {
+        bazProp: string;
+      }
+    `);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const barNode = subgraph.getNode(Bar);
+    barNode.delete();
+    expect(barNode.isDeleted).toBe(true);
+    expect(fooNode.isMutated).toBe(true);
+    expect(fooNode.mutatedType.baseModel).toBeUndefined();
+});
+it("handles mutation of indexers", async () => {
+    const { Foo, Bar, program } = await runner.compile(t.code `
+      model ${t.model("Foo")} is Record<Bar> {};
+      model ${t.model("Bar")} {
+        bazProp: string;
+      }
+    `);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const barNode = subgraph.getNode(Bar);
+    barNode.mutate();
+    expect(barNode.isMutated).toBe(true);
+    expect(fooNode.isMutated).toBe(true);
+    expect(fooNode.mutatedType.indexer?.value === barNode.mutatedType).toBeTruthy();
+});
+it("handles mutation of arrays", async () => {
+    const { Foo, Bar, bazProp, program } = await runner.compile(t.code `
+      model ${t.model("Foo")} {};
+      model ${t.model("Bar")} {
+        ${t.modelProperty("bazProp")}: Foo[];
+      }
+    `);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const barNode = subgraph.getNode(Bar);
+    const bazPropNode = subgraph.getNode(bazProp);
+    fooNode.mutate();
+    expect(fooNode.isMutated).toBe(true);
+    expect(barNode.isMutated).toBe(true);
+    expect(bazPropNode.isMutated).toBe(true);
+    expect(bazPropNode.mutatedType.type.indexer.value === fooNode.mutatedType).toBeTruthy();
+});
+it("handles circular models", async () => {
+    const { Foo, Bar, program } = await runner.compile(t.code `
+      model ${t.model("Foo")} {
+        bar: Bar;
+      };
+      model ${t.model("Bar")} {
+        foo: Foo;
+      }
+    `);
+    const subgraph = getSubgraph(program);
+    const fooNode = subgraph.getNode(Foo);
+    const barNode = subgraph.getNode(Bar);
+    fooNode.mutate();
+    expect(fooNode.isMutated).toBe(true);
+    expect(barNode.isMutated).toBe(true);
+});
+//# sourceMappingURL=model.test.js.map