Add extension to format on edit

2026-05-04 21:41:01 +01:00
parent dc3b31378d
commit 65366738a7
3 changed files with 58 additions and 0 deletions
@@ -0,0 +1,53 @@
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { resolve } from "node:path";
+
+const ELIXIR_EXTENSIONS = [".ex", ".exs", ".heex"];
+
+function isPiTsFile(path: string, cwd: string): boolean {
+  if (!path.endsWith(".ts")) return false;
+  const piDir = resolve(cwd, ".pi");
+  return resolve(cwd, path).startsWith(piDir);
+}
+
+async function formatElixir(
+  pi: ExtensionAPI,
+  path: string,
+  signal: AbortSignal | undefined,
+) {
+  await pi.exec("mix", ["format", path], { signal, timeout: 10_000 });
+}
+
+async function formatTypeScript(
+  pi: ExtensionAPI,
+  path: string,
+  signal: AbortSignal | undefined,
+) {
+  await pi.exec("prettier", ["--write", path], { signal, timeout: 10_000 });
+}
+
+export default function (pi: ExtensionAPI) {
+  pi.on("tool_result", async (event, ctx) => {
+    // Only hook into file-modifying tools, and only on success
+    if (event.toolName !== "edit" && event.toolName !== "write") return;
+    if (event.isError) return;
+
+    // Get the file path from the tool input
+    const path = event.input?.path;
+    if (typeof path !== "string") return;
+
+    // Determine formatter based on file type
+    if (ELIXIR_EXTENSIONS.some((ext) => path.endsWith(ext))) {
+      try {
+        await formatElixir(pi, path, ctx.signal);
+      } catch {
+        /* ignore */
+      }
+    } else if (isPiTsFile(path, ctx.cwd)) {
+      try {
+        await formatTypeScript(pi, path, ctx.signal);
+      } catch {
+        /* ignore */
+      }
+    }
+  });
+}