Ast-Guided Llm Edits: LLMs Propose Intents and Codemods Apply Them

{
  "$id": "https://acme.dev/codemod-intent.json",
  "type": "object",
  "required": ["kind", "target"],
  "properties": {
    "kind": { "enum": ["rename_symbol", "replace_import", "wrap_call", "add_jsdoc"] },
    "target": {
      "type": "object",
      "required": ["path", "exportName"],
      "properties": {
        "path": { "type": "string" },
        "exportName": { "type": "string" }
      }
    },
    "payload": { "type": "object" }
  }
}

// intents/rename.ts
import { z } from "zod";

export const RenameIntent = z.object({
  kind: z.literal("rename_symbol"),
  target: z.object({ path: z.string(), exportName: z.string() }),
  payload: z.object({ to: z.string().regex(/^[A-Za-z_][A-Za-z0-9_]*$/) }),
});

import { Project } from "ts-morph";
import { RenameIntent } from "./intents/rename";

export function applyRename(raw: unknown) {
  const intent = RenameIntent.parse(raw); // ❌ never apply unparsed model JSON
  const project = new Project({ tsConfigFilePath: "tsconfig.json" });
  const file = project.getSourceFileOrThrow(intent.target.path);
  const exported = file.getExportedDeclarations().get(intent.target.exportName);
  if (!exported?.length) throw new Error("symbol_not_found");

  for (const decl of exported) {
    // ts-morph rename updates references in the project graph
    if ("rename" in decl) (decl as any).rename(intent.payload.to);
  }
  project.saveSync();
}

// jscodeshift example for import replacement
export default function transformer(file, api, opts) {
  const j = api.jscodeshift;
  const root = j(file.source);
  root.find(j.ImportDeclaration, { source: { value: opts.from } })
    .forEach((p) => { p.value.source.value = opts.to; });
  return root.toSource();
}