pi-cc-extensions 0.1.2 → 0.1.5

package/README.md

@@ -1,162 +1,162 @@
-<p align="center">
-  <img src="./assets/readme/hero.svg" width="100%" alt="pi-cc-extensions：为 Pi 提供 Claude Code 风格界面、上下文洞察和历史 Session 引用">
-</p>
-
-<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="#兼容性"><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>
-
-<p align="center">
-  为 Pi 补充一套更接近 Claude Code 的终端工作流 · 上下文洞察 · 历史 Session 引用
-</p>
-
----
-
-## 截图
-
-<table>
-  <tr>
-    <td colspan="2" align="center">
-      <img src="./assets/readme/welcome.webp" width="100%" alt="Pi 的 Welcome 界面，展示 Tips、已加载上下文、扩展和最近会话">
-      <br>
-      <sub><b>Welcome 界面</b><br>启动时查看已加载上下文、扩展与最近会话</sub>
-    </td>
-  </tr>
-  <tr>
-    <td width="50%" align="center">
-      <img src="./assets/readme/showcase.webp" width="100%" alt="Pi 终端中的 Claude Code 风格扩展界面，包含工具调用摘要、Diff 预览和 Powerline 状态栏">
-      <br>
-      <sub><b>Claude Code 风格界面</b><br>工具调用摘要、Diff 预览与 Powerline 状态栏</sub>
-    </td>
-    <td width="50%" align="center">
-      <img src="./assets/readme/session-reference.webp" width="100%" alt="在 Pi 中通过 @ 补全选择历史 Session">
-      <br>
-      <sub><b>历史 Session 引用</b><br>通过 @ 补全快速找到并引用历史会话</sub>
-    </td>
-  </tr>
-  <tr>
-    <td width="50%" align="center">
-      <img src="./assets/readme/context-usage.webp" width="100%" alt="Pi 上下文窗口用量查看器">
-      <br>
-      <sub><b>上下文用量</b><br>查看 System prompt、Tools、Skills 等占用分布</sub>
-    </td>
-    <td width="50%" align="center">
-      <img src="./assets/readme/context-preview.webp" width="100%" alt="Pi 中展开查看 System Prompt 内容">
-      <br>
-      <sub><b>上下文预览</b><br>按类别展开查看实际注入的上下文内容</sub>
-    </td>
-  </tr>
-</table>
-
-## 功能
-
-### Claude Code 风格界面
-
-`extensions/claude-code-style.ts` 提供：
-
-- Claude Code 风格的工具调用行与结果摘要
-- Powerline 状态栏和工作状态提示
-- Edit / Write 的 Diff 预览
-- 工具结果的折叠与展开
-- `/ccstyle` 配置命令和 `Ctrl+Shift+O` 快捷键
-
-常用命令：
-
-```text
-/ccstyle             # 切换开关
-/ccstyle status      # 查看状态
-/ccstyle compact     # 紧凑摘要
-/ccstyle minimal     # 最小化输出
-/ccstyle powerline   # 查看 Powerline 预设
-```
-
-### 上下文窗口查看
-
-`extensions/context.ts` 注册 `/context`，展示当前上下文窗口的使用分布，并可进一步预览：
-
-- System prompt
-- Tools
-- Context files
-- Skills
-- User / assistant messages
-- Tool results
-- Compaction summaries
-
-### 历史 Session 引用
-
-`extensions/session-reference/index.ts` 将历史 Session 接入 `@` 补全：
-
-1. 在提示词中输入 `@`。
-2. 从 `[Session] ...` 或 `[SubAgent] ...` 补全项中选择要引用的上下文。
-3. 提交时以 `@session:<session-id>` 引用其当前有效上下文。
-
-Session 模糊查询默认显示 5 个候选；若已加载 `pi-subagents`，也可直接引用现有 SubAgent。一次提示词可以引用多个 Session；扩展会自动去重，并限制注入规模以避免上下文无限膨胀。更多细节见 [`extensions/session-reference/README.md`](./extensions/session-reference/README.md)。
-
-## 安装
-
-从 Git 安装指定版本：
-
-```bash
-pi install git:github.com/minuque/pi-cc-extensions@v0.1.0
-```
-
-安装后启动 Pi，即可使用上述扩展。更新到主分支版本：
-
-```bash
-pi install git:github.com/minuque/pi-cc-extensions
-```
-
-安装后可以先运行：
-
-```text
-/context
-/ccstyle status
-```
-
-## 本地开发
-
-```bash
-npm test
-pi -e .
-```
-
-也可以把当前仓库作为本地 Pi 包安装：
-
-```bash
-pi install /absolute/path/to/pi-cc-extensions
-```
-
-修改扩展后，在 Pi 中执行：
-
-```text
-/reload
-```
-
-## 兼容性
-
-- Node.js `>=22.19.0`
-- 作为 Pi package 加载，入口由根目录 `package.json` 的 `pi.extensions` 显式声明
-
-## 发布
-
-发布前检查：
-
-```bash
-npm test
-npm pack --dry-run
-```
-
-使用 npm 发布。未指定 tag 时使用 `latest`：
-
-```bash
-npm run release
-npm run release -- --tag next
-npm run release -- --tag beta
-```
-
-## 致谢
-
-本项目中的 Powerline 状态栏、Welcome 界面、Working Vibes 等功能基于 [nicobailon/pi-powerline-footer](https://github.com/nicobailon/pi-powerline-footer) 二次开发，感谢原作者的开源贡献。
+<p align="center">
+  <img src="./assets/readme/hero.svg" width="100%" alt="pi-cc-extensions：为 Pi 提供 Claude Code 风格界面、上下文洞察和历史 Session 引用">
+</p>
+
+<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.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>
+
+<p align="center">
+  为 Pi 补充一套更接近 Claude Code 的终端工作流 · 上下文洞察 · 历史 Session 引用
+</p>
+
+---
+
+## 截图
+
+<table>
+  <tr>
+    <td colspan="2" align="center">
+      <img src="./assets/readme/welcome.webp" width="100%" alt="Pi 的 Welcome 界面，展示 Tips、已加载上下文、扩展和最近会话">
+      <br>
+      <sub><b>Welcome 界面</b><br>启动时查看已加载上下文、扩展与最近会话</sub>
+    </td>
+  </tr>
+  <tr>
+    <td width="50%" align="center">
+      <img src="./assets/readme/showcase.webp" width="100%" alt="Pi 终端中的 Claude Code 风格扩展界面，包含工具调用摘要、Diff 预览和 Powerline 状态栏">
+      <br>
+      <sub><b>Claude Code 风格界面</b><br>工具调用摘要、Diff 预览与 Powerline 状态栏</sub>
+    </td>
+    <td width="50%" align="center">
+      <img src="./assets/readme/session-reference.webp" width="100%" alt="在 Pi 中通过 @ 补全选择历史 Session">
+      <br>
+      <sub><b>历史 Session 引用</b><br>通过 @ 补全快速找到并引用历史会话</sub>
+    </td>
+  </tr>
+  <tr>
+    <td width="50%" align="center">
+      <img src="./assets/readme/context-usage.webp" width="100%" alt="Pi 上下文窗口用量查看器">
+      <br>
+      <sub><b>上下文用量</b><br>查看 System prompt、Tools、Skills 等占用分布</sub>
+    </td>
+    <td width="50%" align="center">
+      <img src="./assets/readme/context-preview.webp" width="100%" alt="Pi 中展开查看 System Prompt 内容">
+      <br>
+      <sub><b>上下文预览</b><br>按类别展开查看实际注入的上下文内容</sub>
+    </td>
+  </tr>
+</table>
+
+## 功能
+
+### Claude Code 风格界面
+
+`extensions/claude-code-style.ts` 提供：
+
+- Claude Code 风格的工具调用行与结果摘要
+- Powerline 状态栏和工作状态提示
+- Edit / Write 的 Diff 预览
+- 工具结果的折叠与展开
+- `/ccstyle` 配置命令和 `Ctrl+Shift+O` 快捷键
+
+常用命令：
+
+```text
+/ccstyle             # 切换开关
+/ccstyle status      # 查看状态
+/ccstyle compact     # 紧凑摘要
+/ccstyle minimal     # 最小化输出
+/ccstyle powerline   # 查看 Powerline 预设
+```
+
+### 上下文窗口查看
+
+`extensions/context.ts` 注册 `/context`，展示当前上下文窗口的使用分布，并可进一步预览：
+
+- System prompt
+- Tools
+- Context files
+- Skills
+- User / assistant messages
+- Tool results
+- Compaction summaries
+
+### 历史 Session 引用
+
+`extensions/session-reference/index.ts` 将历史 Session 接入 `@` 补全：
+
+1. 在提示词中输入 `@`。
+2. 从 `[Session] ...` 或 `[SubAgent] ...` 补全项中选择要引用的上下文。
+3. 提交时以 `@session:<session-id>` 引用其当前有效上下文。
+
+Session 模糊查询默认显示 5 个候选；若已加载 `pi-subagents`，也可直接引用现有 SubAgent。一次提示词可以引用多个 Session；扩展会自动去重，并限制注入规模以避免上下文无限膨胀。更多细节见 [`extensions/session-reference/README.md`](./extensions/session-reference/README.md)。
+
+## 安装
+
+从 npm 安装（推荐）：
+
+```bash
+pi install npm:pi-cc-extensions
+```
+
+或从 Git 安装：
+
+```bash
+pi install git:github.com/minuque/pi-cc-extensions
+```
+
+安装后可以先运行：
+
+```text
+/context
+/ccstyle status
+```
+
+## 本地开发
+
+```bash
+npm test
+pi -e .
+```
+
+也可以把当前仓库作为本地 Pi 包安装：
+
+```bash
+pi install /absolute/path/to/pi-cc-extensions
+```
+
+修改扩展后，在 Pi 中执行：
+
+```text
+/reload
+```
+
+## 兼容性
+
+- Node.js `>=22.19.0`
+- 作为 Pi package 加载，入口由根目录 `package.json` 的 `pi.extensions` 显式声明
+
+## 发布
+
+发布前检查：
+
+```bash
+npm test
+npm pack --dry-run
+```
+
+使用 npm 发布。未指定 tag 时使用 `latest`：
+
+```bash
+npm run release
+npm run release -- --tag next
+npm run release -- --tag beta
+```
+
+## 致谢
+
+本项目中的 Powerline 状态栏、Welcome 界面、Working Vibes 等功能基于 [nicobailon/pi-powerline-footer](https://github.com/nicobailon/pi-powerline-footer) 二次开发，感谢原作者的开源贡献。

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".`,
+		};
+	});
+}