@typespec/mutator-framework 0.12.0

package/src/mutation-node/union.ts

@@ -0,0 +1,33 @@
+import type { Union, UnionVariant } from "@typespec/compiler";
+import { MutationEdge } from "./mutation-edge.js";
+import { MutationNode } from "./mutation-node.js";
+
+export class UnionMutationNode extends MutationNode<Union> {
+  readonly kind = "Union";
+
+  traverse(): void {
+    for (const variant of this.sourceType.variants.values()) {
+      const variantNode = this.subgraph.getNode(variant);
+      this.connectVariant(variantNode, variant.name);
+    }
+  }
+
+  connectVariant(variantNode: MutationNode<UnionVariant>, sourcePropName: string | symbol) {
+    MutationEdge.create(this, variantNode, {
+      onTailMutation: () => {
+        this.mutatedType.variants.delete(sourcePropName);
+        this.mutatedType.variants.set(variantNode.mutatedType.name, variantNode.mutatedType);
+      },
+      onTailDeletion: () => {
+        this.mutatedType.variants.delete(sourcePropName);
+      },
+      onTailReplaced: (newTail) => {
+        if (newTail.mutatedType.kind !== "UnionVariant") {
+          throw new Error("Cannot replace union variant with non-union variant type");
+        }
+        this.mutatedType.variants.delete(sourcePropName);
+        this.mutatedType.variants.set(newTail.mutatedType.name, newTail.mutatedType);
+      },
+    });
+  }
+}

package/test/test-host.ts

@@ -0,0 +1,6 @@
+import { resolvePath } from "@typespec/compiler";
+import { createTester } from "@typespec/compiler/testing";
+
+export const Tester = createTester(resolvePath((import.meta as any).dirname, ".."), {
+  libraries: [],
+});

package/test/utils.ts

@@ -0,0 +1,9 @@
+import type { Program } from "@typespec/compiler";
+import { $ } from "@typespec/compiler/typekit";
+import { MutationEngine, MutationSubgraph } from "../src/index.js";
+
+export function getSubgraph(program: Program) {
+  const tk = $(program);
+  const engine = new MutationEngine(tk, {});
+  return new MutationSubgraph(engine);
+}

package/tsconfig.json

@@ -0,0 +1,19 @@
+{
+  "compilerOptions": {
+    "lib": ["es2023", "DOM"],
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "target": "es2022",
+    "strict": true,
+    "skipLibCheck": true,
+    "isolatedModules": true,
+    "declaration": true,
+    "sourceMap": true,
+    "declarationMap": true,
+    "outDir": "dist",
+    "rootDir": "./",
+    "verbatimModuleSyntax": true
+  },
+  "include": ["src/**/*.ts", "src/**/*.tsx", "test/**/*.ts", "test/**/*.tsx"],
+  "exclude": ["node_modules", "dist"]
+}

package/vitest.config.ts

@@ -0,0 +1,4 @@
+import { defineConfig, mergeConfig } from "vitest/config";
+import { defaultTypeSpecVitestConfig } from "../../vitest.config.js";
+
+export default mergeConfig(defaultTypeSpecVitestConfig, defineConfig({}));