@gotgenes/pi-autoformat 0.1.0

package/test/formatter-executor.test.ts

@@ -0,0 +1,82 @@
+import { describe, expect, it } from "vitest";
+
+import {
+  type CommandRunner,
+  executeFormatterChain,
+} from "../src/formatter-executor.js";
+import type { ResolvedFormatter } from "../src/formatter-registry.js";
+
+describe("executeFormatterChain", () => {
+  const chain: ResolvedFormatter[] = [
+    {
+      name: "prettier",
+      command: ["prettier", "--write", "/repo/docs/readme.md"],
+      environment: {
+        PRETTIERD_DEFAULT_CONFIG: "./.prettierrc",
+      },
+    },
+    {
+      name: "markdownlint",
+      command: ["markdownlint-cli2", "--fix", "/repo/docs/readme.md"],
+    },
+  ];
+
+  it("executes formatters in order", async () => {
+    const calls: string[] = [];
+    const runner: CommandRunner = async (command, args) => {
+      calls.push([command, ...args].join(" "));
+      return { exitCode: 0 };
+    };
+
+    const result = await executeFormatterChain(chain, runner);
+
+    expect(calls).toEqual([
+      "prettier --write /repo/docs/readme.md",
+      "markdownlint-cli2 --fix /repo/docs/readme.md",
+    ]);
+    expect(result.every((entry) => entry.success)).toBe(true);
+  });
+
+  it("continues running remaining formatters after a failure", async () => {
+    const calls: string[] = [];
+    const runner: CommandRunner = async (command, _args) => {
+      calls.push(command);
+      if (command === "prettier") {
+        return {
+          exitCode: 2,
+          stderr: "syntax error",
+        };
+      }
+
+      return { exitCode: 0 };
+    };
+
+    const result = await executeFormatterChain(chain, runner);
+
+    expect(calls).toEqual(["prettier", "markdownlint-cli2"]);
+    expect(result[0]).toMatchObject({
+      formatterName: "prettier",
+      success: false,
+      exitCode: 2,
+    });
+    expect(result[1]).toMatchObject({
+      formatterName: "markdownlint",
+      success: true,
+      exitCode: 0,
+    });
+  });
+
+  it("passes formatter environment overrides to the runner", async () => {
+    let capturedEnv: Record<string, string> | undefined;
+    const runner: CommandRunner = async (_command, _args, options) => {
+      capturedEnv = options?.env;
+      return { exitCode: 0 };
+    };
+
+    await executeFormatterChain([chain[0]], runner);
+
+    expect(capturedEnv).toMatchObject({
+      PRETTIERD_DEFAULT_CONFIG: "./.prettierrc",
+    });
+  });
+});

package/test/formatter-registry.test.ts

@@ -0,0 +1,75 @@
+import { describe, expect, it } from "vitest";
+
+import {
+  type FormatterConfig,
+  resolveFormatterChainForFile,
+} from "../src/formatter-registry.js";
+
+describe("resolveFormatterChainForFile", () => {
+  const config: FormatterConfig = {
+    formatters: {
+      prettier: {
+        command: ["prettier", "--write", "$FILE"],
+        extensions: [".ts", ".md"],
+      },
+      markdownlint: {
+        command: ["markdownlint-cli2", "--fix", "$FILE"],
+        extensions: [".md"],
+      },
+    },
+    chains: {
+      ".md": ["prettier", "markdownlint"],
+    },
+  };
+
+  it("resolves explicit chains in declared order", () => {
+    const chain = resolveFormatterChainForFile("/repo/docs/readme.md", config);
+
+    expect(chain.map((entry) => entry.name)).toEqual([
+      "prettier",
+      "markdownlint",
+    ]);
+  });
+
+  it("returns an empty chain when no explicit chain exists for the extension", () => {
+    const chain = resolveFormatterChainForFile("/repo/src/index.ts", config);
+
+    expect(chain).toEqual([]);
+  });
+
+  it("substitutes $FILE in formatter commands", () => {
+    const chain = resolveFormatterChainForFile("/repo/docs/readme.md", config);
+
+    expect(chain[0]?.command).toEqual([
+      "prettier",
+      "--write",
+      "/repo/docs/readme.md",
+    ]);
+  });
+
+  it("skips disabled formatters", () => {
+    const withDisabled: FormatterConfig = {
+      ...config,
+      formatters: {
+        ...config.formatters,
+        markdownlint: {
+          ...config.formatters.markdownlint,
+          disabled: true,
+        },
+      },
+    };
+
+    const chain = resolveFormatterChainForFile(
+      "/repo/docs/readme.md",
+      withDisabled,
+    );
+
+    expect(chain.map((entry) => entry.name)).toEqual(["prettier"]);
+  });
+
+  it("returns an empty chain when no formatter matches", () => {
+    const chain = resolveFormatterChainForFile("/repo/assets/logo.png", config);
+
+    expect(chain).toEqual([]);
+  });
+});

package/test/prompt-autoformatter.test.ts

@@ -0,0 +1,93 @@
+import { describe, expect, it } from "vitest";
+import type { CommandRunner } from "../src/formatter-executor.js";
+import type { FormatterConfig } from "../src/formatter-registry.js";
+import {
+  PromptAutoformatter,
+  type PromptAutoformatterResult,
+} from "../src/prompt-autoformatter.js";
+
+describe("PromptAutoformatter", () => {
+  const config: FormatterConfig = {
+    formatters: {
+      prettier: {
+        command: ["prettier", "--write", "$FILE"],
+        extensions: [".ts", ".md"],
+      },
+      markdownlint: {
+        command: ["markdownlint-cli2", "--fix", "$FILE"],
+        extensions: [".md"],
+      },
+    },
+    chains: {
+      ".md": ["prettier", "markdownlint"],
+      ".ts": ["prettier"],
+    },
+  };
+
+  it("is a no-op when no formatter matches touched files", async () => {
+    const calls: string[] = [];
+    const runner: CommandRunner = async (command, args) => {
+      calls.push([command, ...args].join(" "));
+      return { exitCode: 0 };
+    };
+
+    const formatter = new PromptAutoformatter("/repo", config, runner);
+    formatter.recordToolResult("write", { path: "assets/logo.png" });
+
+    const result = await formatter.flushPrompt();
+
+    expect(calls).toEqual([]);
+    expect(result).toEqual({ files: [] });
+  });
+
+  it("dedupes touched files across prompt tool results", async () => {
+    const calls: string[] = [];
+    const runner: CommandRunner = async (command, args) => {
+      calls.push([command, ...args].join(" "));
+      return { exitCode: 0 };
+    };
+
+    const formatter = new PromptAutoformatter("/repo", config, runner);
+    formatter.recordToolResult("write", { path: "docs/readme.md" });
+    formatter.recordToolResult("edit", { path: "./docs/readme.md" });
+
+    const result = await formatter.flushPrompt();
+
+    expect(calls).toEqual([
+      "prettier --write /repo/docs/readme.md",
+      "markdownlint-cli2 --fix /repo/docs/readme.md",
+    ]);
+    expect(result.files).toHaveLength(1);
+  });
+
+  it("returns formatter failures without throwing", async () => {
+    const runner: CommandRunner = async (command) => {
+      if (command === "prettier") {
+        return {
+          exitCode: 2,
+          stderr: "parse error",
+        };
+      }
+      return { exitCode: 0 };
+    };
+
+    const formatter = new PromptAutoformatter("/repo", config, runner);
+    formatter.recordToolResult("write", { path: "docs/readme.md" });
+
+    const result = await formatter.flushPrompt();
+    const firstFile = result
+      .files[0] as PromptAutoformatterResult["files"][number];
+
+    expect(firstFile.path).toBe("/repo/docs/readme.md");
+    expect(firstFile.runs[0]).toMatchObject({
+      formatterName: "prettier",
+      success: false,
+      exitCode: 2,
+    });
+    expect(firstFile.runs[1]).toMatchObject({
+      formatterName: "markdownlint",
+      success: true,
+      exitCode: 0,
+    });
+  });
+});

package/test/smoke.test.ts

@@ -0,0 +1,9 @@
+import { describe, expect, it } from "vitest";
+
+import { extensionName } from "../src/index.js";
+
+describe("smoke", () => {
+  it("exports extension name", () => {
+    expect(extensionName).toBe("pi-autoformat");
+  });
+});

package/test/touched-files-queue.test.ts

@@ -0,0 +1,46 @@
+import { describe, expect, it } from "vitest";
+
+import { TouchedFilesQueue } from "../src/touched-files-queue.js";
+
+describe("TouchedFilesQueue", () => {
+  it("collects paths from write and edit tool results", () => {
+    const queue = new TouchedFilesQueue("/repo");
+
+    queue.recordToolResult("write", { path: "src/index.ts" });
+    queue.recordToolResult("edit", { path: "docs/readme.md" });
+
+    expect(queue.flush()).toEqual([
+      "/repo/src/index.ts",
+      "/repo/docs/readme.md",
+    ]);
+  });
+
+  it("dedupes repeated file touches in a prompt", () => {
+    const queue = new TouchedFilesQueue("/repo");
+
+    queue.recordToolResult("write", { path: "src/index.ts" });
+    queue.recordToolResult("edit", { path: "./src/index.ts" });
+    queue.recordToolResult("edit", { path: "/repo/src/index.ts" });
+
+    expect(queue.flush()).toEqual(["/repo/src/index.ts"]);
+  });
+
+  it("ignores non-mutation tools and invalid payloads", () => {
+    const queue = new TouchedFilesQueue("/repo");
+
+    queue.recordToolResult("bash", { path: "src/index.ts" });
+    queue.recordToolResult("write", { foo: "bar" });
+    queue.recordToolResult("edit", null);
+
+    expect(queue.flush()).toEqual([]);
+  });
+
+  it("clears collected state after flush", () => {
+    const queue = new TouchedFilesQueue("/repo");
+
+    queue.recordToolResult("write", { path: "src/index.ts" });
+
+    expect(queue.flush()).toEqual(["/repo/src/index.ts"]);
+    expect(queue.flush()).toEqual([]);
+  });
+});

package/tsconfig.json

@@ -0,0 +1,13 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "strict": true,
+    "esModuleInterop": true,
+    "forceConsistentCasingInFileNames": true,
+    "skipLibCheck": true,
+    "types": ["node", "vitest/globals"]
+  },
+  "include": ["src", "test", "vitest.config.ts"]
+}

package/vitest.config.ts

@@ -0,0 +1,10 @@
+import { defineConfig } from "vitest/config";
+
+export default defineConfig({
+  test: {
+    include: ["test/**/*.test.ts"],
+    coverage: {
+      reporter: ["text", "html"],
+    },
+  },
+});