pi-cc-extensions 0.1.3 → 0.2.0

package/README.md

@@ -4,7 +4,7 @@
 
 <p align="center">
   <a href="#安装"><img alt="Pi package" src="https://img.shields.io/badge/Pi-package-58B7FF?style=flat-square"></a>
-  <a href="./package.json"><img alt="Version 0.1.1" src="https://img.shields.io/badge/version-0.1.1-66E3C4?style=flat-square"></a>
+  <a href="./package.json"><img alt="Version 0.1.3" src="https://img.shields.io/badge/version-0.1.3-66E3C4?style=flat-square"></a>
   <a href="#兼容性"><img alt="Node.js 22.19 or newer" src="https://img.shields.io/badge/Node.js-%E2%89%A522.19-66E3C4?style=flat-square"></a>
   <a href="./extensions"><img alt="TypeScript extensions" src="https://img.shields.io/badge/TypeScript-extensions-3178C6?style=flat-square"></a>
 </p>
@@ -97,7 +97,13 @@ Session 模糊查询默认显示 5 个候选；若已加载 `pi-subagents`，也
 
 ## 安装
 
-从 Git 安装：
+从 npm 安装（推荐）：
+
+```bash
+pi install npm:pi-cc-extensions
+```
+
+或从 Git 安装：
 
 ```bash
 pi install git:github.com/minuque/pi-cc-extensions

package/extensions/agent-autocomplete.ts

@@ -0,0 +1,175 @@
+import { existsSync, readdirSync, readFileSync } from "node:fs";
+import { join } from "node:path";
+import { homedir } from "node:os";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import {
+	type AutocompleteItem,
+	type AutocompleteProvider,
+	type AutocompleteSuggestions,
+	fuzzyFilter,
+} from "@earendil-works/pi-tui";
+
+type AgentInfo = {
+	name: string;
+	displayName: string;
+	description: string;
+	model?: string;
+	filePath: string;
+};
+
+const MAX_SUGGESTIONS = 10;
+// Match `@name` but NOT `@session:` (reserved by session-reference extension).
+const AGENT_NAME_PATTERN = /(?:^|[\t ])@([^\s:@][^\s:]*)$/;
+
+function getAgentDir(): string {
+	const override = process.env.PI_CODING_AGENT_DIR;
+	return join(override ? override : homedir(), ".pi", "agent", "agents");
+}
+
+function parseFrontmatter(content: string): Record<string, string> {
+	const match = content.match(/^---\r?\n([\s\S]*?)\r?\n---/);
+	if (!match) return {};
+	const result: Record<string, string> = {};
+	for (const line of match[1]!.split("\n")) {
+		const kv = line.match(/^(\w+):\s*(.*)$/);
+		if (kv) result[kv[1]!] = kv[2]!.trim();
+	}
+	return result;
+}
+
+function loadAgents(): AgentInfo[] {
+	const dir = getAgentDir();
+	if (!existsSync(dir)) return [];
+
+	return readdirSync(dir)
+		.filter((f) => f.endsWith(".md"))
+		.map((file) => {
+			const filePath = join(dir, file);
+			const content = readFileSync(filePath, "utf-8");
+			const fm = parseFrontmatter(content);
+			const name = file.replace(/\.md$/, "");
+			return {
+				name,
+				displayName: fm.display_name || name,
+				description: fm.description || "",
+				model: fm.model,
+				filePath,
+			};
+		});
+}
+
+function createAgentAutocompleteProvider(
+	current: AutocompleteProvider,
+	getAgents: () => AgentInfo[],
+): AutocompleteProvider {
+	return {
+		async getSuggestions(
+			lines: string[],
+			cursorLine: number,
+			cursorCol: number,
+			options: { signal: AbortSignal; force?: boolean },
+		): Promise<AutocompleteSuggestions | null> {
+			const currentLine = lines[cursorLine] ?? "";
+			const textBeforeCursor = currentLine.slice(0, cursorCol);
+			const match = textBeforeCursor.match(AGENT_NAME_PATTERN);
+
+			if (!match) {
+				return current.getSuggestions(lines, cursorLine, cursorCol, options);
+			}
+
+			const query = match[1]!;
+			const agents = getAgents();
+
+			if (options.signal.aborted || agents.length === 0) {
+				return current.getSuggestions(lines, cursorLine, cursorCol, options);
+			}
+
+			const matches = query.trim()
+				? fuzzyFilter(agents, query, (a) => `${a.name} ${a.displayName} ${a.description}`)
+				: agents;
+
+			const items: AutocompleteItem[] = matches
+				.slice(0, MAX_SUGGESTIONS)
+				.map((agent) => ({
+					value: `@${agent.name}`,
+					label: `[SubAgent] ${agent.displayName}`,
+					description: agent.description
+						+ (agent.model ? ` · ${agent.model}` : ""),
+				}));
+
+			if (items.length === 0) {
+				return current.getSuggestions(lines, cursorLine, cursorCol, options);
+			}
+
+			return { items, prefix: `@${query}` };
+		},
+
+		applyCompletion(
+			lines: string[],
+			cursorLine: number,
+			cursorCol: number,
+			item: AutocompleteItem,
+			prefix: string,
+		) {
+			return current.applyCompletion(lines, cursorLine, cursorCol, item, prefix);
+		},
+
+		shouldTriggerFileCompletion(lines: string[], cursorLine: number, cursorCol: number) {
+			return current.shouldTriggerFileCompletion?.(lines, cursorLine, cursorCol) ?? true;
+		},
+	};
+}
+
+export default function agentAutocompleteExtension(pi: ExtensionAPI): void {
+	let cachedAgents: AgentInfo[] | undefined;
+
+	const getAgents = (): AgentInfo[] => {
+		if (!cachedAgents) {
+			cachedAgents = loadAgents();
+		}
+		return cachedAgents;
+	};
+
+	// Reload agents on /reload
+	pi.on("resources_discover", () => {
+		cachedAgents = undefined;
+	});
+
+	pi.on("session_start", (_event, ctx) => {
+		const agents = getAgents();
+		if (agents.length === 0 || ctx.mode !== "tui") return;
+		ctx.ui.addAutocompleteProvider((current) =>
+			createAgentAutocompleteProvider(current, getAgents),
+		);
+	});
+
+	// Inject instruction when user types @agent-name (not @session:) in prompt.
+	// Supports multiple different subagents in a single prompt.
+	const AGENT_PROMPT_PATTERN = /(?:^|[\s])@([^\s:@][^\s:]*)/g;
+	pi.on("before_agent_start", async (event, _ctx) => {
+		const agents = getAgents();
+		if (agents.length === 0) return;
+
+		const mentions: string[] = [];
+		for (const m of event.prompt.matchAll(AGENT_PROMPT_PATTERN)) {
+			const name = m[1]!;
+			if (agents.some((a) => a.name === name) && !mentions.includes(name)) {
+				mentions.push(name);
+			}
+		}
+		if (mentions.length === 0) return;
+
+		const agentMap = new Map(agents.map((a) => [a.name, a]));
+		const agentList = mentions.map((n) => `"${n}" (${agentMap.get(n)!.displayName})`).join(", ");
+		const agentTypes = mentions.map((n) => `"${n}"`).join(" | ");
+
+		return {
+			systemPrompt:
+				event.systemPrompt
+				+ `\n\nThe user's prompt references these subagent types: ${agentList}. `
+				+ `You MUST use the Agent tool for EACH mentioned subagent to delegate the relevant parts of the request. `
+				+ `Handle different subagents separately — do NOT merge their tasks into a single Agent call. `
+				+ `For example, if the user mentions @coder and @explore, make two separate Agent tool calls, one with subagent_type="coder" and another with subagent_type="explore".`,
+		};
+	});
+}

package/extensions/session-reference/core.ts

@@ -1,5 +1,3 @@
-import { Buffer } from "node:buffer";
-
 export const SESSION_REFERENCE_CUSTOM_TYPE = "session-reference";
 export const SESSION_REFERENCE_PREFIX = "@session:";
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-cc-extensions",
-  "version": "0.1.3",
+  "version": "0.2.0",
   "description": "Personal extensions for the Pi coding agent",
   "type": "module",
   "keywords": [
@@ -16,15 +16,18 @@
     "extensions": [
       "./extensions/claude-code-style.ts",
       "./extensions/context.ts",
-      "./extensions/session-reference/index.ts"
+      "./extensions/session-reference/index.ts",
+      "./extensions/agent-autocomplete.ts"
     ]
   },
   "scripts": {
     "test": "node --test tests/**/*.test.ts",
     "typecheck": "node scripts/typecheck.mjs",
-    "prepublishOnly": "npm run typecheck && npm test",
-    "postversion": "git push && git push --tags",
-    "release": "npm publish"
+    "preversion": "npm run typecheck && npm test",
+    "release": "npm version patch && npm publish",
+    "release:minor": "npm version minor && npm publish",
+    "release:major": "npm version major && npm publish",
+    "postpublish": "git push && git push --tags"
   },
   "peerDependencies": {
     "@earendil-works/pi-ai": "*",