@henryqw/pi-add-dir 0.1.12 → 0.1.14

package/README.md

@@ -1,11 +1,11 @@
 # `@henryqw/pi-add-dir`
 
-Add external directories to the current Pi session: inject context files, load skills, and search those trees.
+Add external directories to the current Pi session. Pi can use their context files and skills, and search their files.
 
 ## Why
 
-- **Created for**: Extending Pi context and skill discovery beyond the current working directory.
-- **Advantage**: Inject files and load skills from external trees without copying them into the repo.
+- **Created for**: Use context and skills outside the current working directory.
+- **Advantage**: Use external trees without copying them into the repository.
 
 ## Install
 
@@ -23,10 +23,16 @@ 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 give Pi these resources:
 
-Search uses Node filesystem traversal, skips `.git` and `node_modules`, supports basename and relative-path globs, and caps results at 1,000 per call.
+- Root `AGENTS.md`, `CLAUDE.md`, `.pi/AGENTS.md`, and `.pi/CLAUDE.md` files. Pi injects them into later prompts.
+- Skills that Pi loads from `.pi/skills`, `.agents/skills`, and `.claude/skills`.
+- Files in the editor's `@` autocomplete, with absolute paths.
+
+`/dir-add` reloads when it finds skills. `add_directory` reports when a reload is needed.
+
+Search supports basename and relative-path globs. It skips `.git` and `node_modules`. It uses Node filesystem traversal and returns at most 1,000 results per call.
 
 ## State
 
-Pi session entry `add-dir:state` stores added directories; package-managed, do not edit. Tree navigation restores the active branch's directories and reloads resources when that set changes.
+Pi session entry `add-dir:state` stores added directories. It is package-managed; do not edit it. Tree navigation restores the active branch's directories and reloads resources when that set changes.

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.12",
+  "version": "0.1.14",
   "description": "Add external directories to a Pi session with context, skills, and file search.",
   "keywords": [
     "pi-package",
@@ -15,9 +15,9 @@
   },
   "license": "MIT",
   "files": [
+    "LICENSE",
     "extensions",
-    "README.md",
-    "LICENSE"
+    "README.md"
   ],
   "scripts": {
     "test": "node --test test/*.test.ts",