@henryqw/pi-add-dir 0.1.11 → 0.1.13

package/README.md

@@ -23,7 +23,7 @@ pi install npm:@henryqw/pi-add-dir
 | `add_directory` | tool | Add a directory. |
 | `search_external_files` | tool | Glob-search added directories. |
 
-Added directories return root `AGENTS.md`, `CLAUDE.md`, `.pi/AGENTS.md`, and `.pi/CLAUDE.md`, and inject them into later prompts. Skills load from `.pi/skills`, `.agents/skills`, and `.claude/skills`. `/dir-add` reloads when it finds skills; `add_directory` reports when a reload is needed.
+Added directories return root `AGENTS.md`, `CLAUDE.md`, `.pi/AGENTS.md`, and `.pi/CLAUDE.md`, and inject them into later prompts. Skills load from `.pi/skills`, `.agents/skills`, and `.claude/skills`. Their files also appear in the editor's `@` autocomplete with absolute paths. `/dir-add` reloads when it finds skills; `add_directory` reports when a reload is needed.
 
 Search uses Node filesystem traversal, skips `.git` and `node_modules`, supports basename and relative-path globs, and caps results at 1,000 per call.
 

package/extensions/add-dir.ts

@@ -1,5 +1,11 @@
-import { basename, isAbsolute, relative, sep } from "node:path";
-import { Text, truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+import { basename, isAbsolute, join, relative, sep } from "node:path";
+import {
+	Text,
+	truncateToWidth,
+	visibleWidth,
+	type AutocompleteProvider,
+	type AutocompleteSuggestions,
+} from "@earendil-works/pi-tui";
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { Type } from "typebox";
 import {
@@ -85,6 +91,67 @@ function contextSummary(dirCtx: DirContext): string {
 	return found.length > 0 ? ` Found: ${found.join(", ")}.` : " No context files found.";
 }
 
+function extractAtToken(textBeforeCursor: string): string | undefined {
+	return textBeforeCursor.match(/(?:^|[ \t])(@(?:"[^"]*|[^\s"]*))$/)?.[1];
+}
+
+export function createExternalAutocompleteProvider(
+	current: AutocompleteProvider,
+	getAddedDirs: () => AddedDir[],
+): AutocompleteProvider {
+	return {
+		async getSuggestions(lines, cursorLine, cursorCol, options): Promise<AutocompleteSuggestions | null> {
+			const line = lines[cursorLine] ?? "";
+			const token = extractAtToken(line.slice(0, cursorCol));
+			if (!token) return current.getSuggestions(lines, cursorLine, cursorCol, options);
+
+			const quoted = token.startsWith('@"');
+			const query = token.slice(quoted ? 2 : 1);
+			if (isAbsolute(query) || query.startsWith("~/") || query.startsWith("./") || query.startsWith("../")) {
+				return current.getSuggestions(lines, cursorLine, cursorCol, options);
+			}
+
+			const tokenStart = cursorCol - token.length;
+			const sources = await Promise.all([
+				current.getSuggestions(lines, cursorLine, cursorCol, options),
+				...getAddedDirs().map(async (dir) => {
+					const externalPath = join(dir.absolutePath, query);
+					const externalToken = `${quoted || externalPath.includes(" ") ? '@"' : "@"}${externalPath}`;
+					const externalLines = [...lines];
+					externalLines[cursorLine] = `${line.slice(0, tokenStart)}${externalToken}${line.slice(cursorCol)}`;
+					const suggestions = await current.getSuggestions(
+						externalLines,
+						cursorLine,
+						tokenStart + externalToken.length,
+						options,
+					);
+					return suggestions
+						? {
+							...suggestions,
+							items: suggestions.items.map((item) => ({
+								...item,
+								description: `${dir.label}: ${item.description ?? item.value}`,
+							})),
+						}
+						: null;
+				}),
+			]);
+
+			const itemCount = Math.max(...sources.map((source) => source?.items.length ?? 0));
+			const items = Array.from({ length: itemCount }, (_, index) =>
+				sources.flatMap((source) => (source?.items[index] ? [source.items[index]] : [])),
+			).flat();
+			return items.length > 0 ? { prefix: token, items } : null;
+		},
+		applyCompletion(lines, cursorLine, cursorCol, item, prefix) {
+			return current.applyCompletion(lines, cursorLine, cursorCol, item, prefix);
+		},
+		shouldTriggerFileCompletion(lines, cursorLine, cursorCol) {
+			return current.shouldTriggerFileCompletion?.(lines, cursorLine, cursorCol) ?? true;
+		},
+	};
+}
+
 export default function addDirExtension(pi: ExtensionAPI): void {
 	let addedDirs: AddedDir[] = [];
 	let currentCwd = "";
@@ -212,6 +279,7 @@ export default function addDirExtension(pi: ExtensionAPI): void {
 
 	pi.on("session_start", async (_event, ctx) => {
 		reconstructState(ctx);
+		ctx.ui.addAutocompleteProvider((current) => createExternalAutocompleteProvider(current, () => addedDirs));
 	});
 	pi.on("session_tree", async (_event, ctx) => {
 		if (reconstructState(ctx)) pi.sendUserMessage("/dir-reload", { expandPromptTemplates: true });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@henryqw/pi-add-dir",
-  "version": "0.1.11",
+  "version": "0.1.13",
   "description": "Add external directories to a Pi session with context, skills, and file search.",
   "keywords": [
     "pi-package",
@@ -25,8 +25,8 @@
     "pack:check": "npm pack --dry-run"
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": ">=0.84.1",
-    "@earendil-works/pi-tui": ">=0.84.1",
+    "@earendil-works/pi-coding-agent": "^0.84.4",
+    "@earendil-works/pi-tui": "^0.84.4",
     "typebox": "^1.3.15"
   },
   "repository": {