Autocomplete path and support absolute and home directory paths

NOTE: These have been slopped out by clankers. Use caution.

Author: NonlinearFruit

pi/save-last-extension.ts

@@ -1,10 +1,61 @@
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { writeFile } from "node:fs/promises";
-import { join } from "node:path";
+import type { AutocompleteItem } from "@mariozechner/pi-tui";
+import { writeFile, readdir, stat } from "node:fs/promises";
+import { join, isAbsolute, dirname, basename } from "node:path";
+import { homedir } from "node:os";
 
 export default function (pi: ExtensionAPI) {
+  const getFileCompletions = async (prefix: string): Promise<AutocompleteItem[] | null> => {
+    try {
+      // Resolve the directory and file prefix
+      let dir = "";
+      let filePrefix = prefix;
+
+      if (prefix.includes("/")) {
+        dir = dirname(prefix);
+        filePrefix = basename(prefix);
+      } else if (prefix.startsWith("~")) {
+        const home = homedir();
+        const rest = prefix.slice(1);
+        if (rest.includes("/")) {
+          dir = join(home, dirname(rest));
+          filePrefix = basename(rest);
+        } else {
+          dir = home;
+          filePrefix = rest;
+        }
+      } else {
+        dir = ".";
+      }
+
+      // Read directory contents
+      const entries = await readdir(dir, { withFileTypes: true });
+      const completions: AutocompleteItem[] = [];
+
+      for (const entry of entries) {
+        if (entry.name.startsWith(filePrefix)) {
+          const fullPath = join(dir, entry.name);
+          let displayPath = fullPath;
+          if (prefix.startsWith("~")) {
+            displayPath = "~" + fullPath.slice(homedir().length);
+          }
+
+          completions.push({
+            value: displayPath + (entry.isDirectory() ? "/" : ""),
+            label: entry.name + (entry.isDirectory() ? "/" : ""),
+          });
+        }
+      }
+
+      return completions.length > 0 ? completions : null;
+    } catch {
+      return null;
+    }
+  };
+
   pi.registerCommand("save-last", {
     description: "Export the most recent AI response to a txt file",
+    getArgumentCompletions: (prefix: string) => getFileCompletions(prefix),
     handler: async (args, ctx) => {
       ctx.ui.notify(`Export started`, "success");
       const entries = ctx.sessionManager.getBranch();
@@ -30,7 +81,15 @@ export default function (pi: ExtensionAPI) {
 
       const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
       const filename = args.trim() || `ai-response-${timestamp}.txt`;
-      const filepath = join(ctx.cwd, filename);
+
+      let filepath: string;
+      if (isAbsolute(filename)) {
+        filepath = filename;
+      } else if (filename.startsWith("~")) {
+        filepath = join(homedir(), filename.slice(1));
+      } else {
+        filepath = join(ctx.cwd, filename);
+      }
 
       await writeFile(filepath, lastAssistantText, "utf8");
       ctx.ui.notify(`Exported to ${filename}`, "success");