pi-shepherd 0.1.2 → 0.2.0

package/shepherd/git.ts

@@ -0,0 +1,64 @@
+import { execSync } from "node:child_process";
+
+function parseGitStatus(): { dirty: string[]; untracked: string[] } {
+	const result = { dirty: [] as string[], untracked: [] as string[] };
+	try {
+		const status = execSync("git status --porcelain", {
+			timeout: 5000,
+			stdio: ["pipe", "pipe", "pipe"],
+			cwd: process.cwd(),
+		})
+			.toString()
+			.trim();
+		if (!status) return result;
+		for (const line of status.split("\n")) {
+			if (!line) continue;
+			if (line.startsWith("??")) {
+				result.untracked.push(line);
+			} else {
+				result.dirty.push(line);
+			}
+		}
+	} catch {
+		/* not a git repo or git not available */
+	}
+	return result;
+}
+
+/** 检测 git 工作区是否有已跟踪文件的未提交改动（M/A/D/R） */
+export function isGitDirty(): boolean {
+	return parseGitStatus().dirty.length > 0;
+}
+
+/** 检测 git 工作区是否有未跟踪文件（??） */
+export function hasGitUntracked(): boolean {
+	return parseGitStatus().untracked.length > 0;
+}
+
+/** 检测 git 工作区是否有已跟踪改动或未跟踪文件 */
+export function isGitDirtyOrUntracked(): boolean {
+	const { dirty, untracked } = parseGitStatus();
+	return dirty.length > 0 || untracked.length > 0;
+}
+
+/** 向后兼容：原 hasGitUncommittedChanges 只看已跟踪文件 */
+export function hasGitUncommittedChanges(): boolean {
+	return isGitDirty();
+}
+
+/** 当前是否在 worktree 中 */
+export function isInWorktree(): boolean {
+	try {
+		const cwd = process.cwd();
+		if (/\/\.worktrees\/[^/]+/.test(cwd)) return true;
+		const gitDir = execSync("git rev-parse --git-dir", {
+			timeout: 5000,
+			stdio: ["pipe", "pipe", "pipe"],
+		})
+			.toString()
+			.trim();
+		return gitDir !== ".git";
+	} catch {
+		return false;
+	}
+}

package/shepherd/index.ts

@@ -3,6 +3,7 @@
  */
 
 export { hasWarnings, notifySummary, pushWarning } from "./ephemeral";
+export { registerSessionCompact } from "./compaction";
 export {
 	drainHints,
 	hasHints,
@@ -12,15 +13,26 @@ export {
 } from "./ephemeral-shared";
 export { checkLineCount } from "./line-count";
 export {
+	extractAssistantText,
+	registerMessageEnd,
+	resetMessageEndState,
+} from "./message-end";
+export {
+	type BuiltinContext,
 	CODE_EXT_RE,
 	type Condition,
+	type ConditionBuiltin,
 	compileRules,
 	getMatchTargets,
 	hasGitUncommittedChanges,
+	hasGitUntracked,
+	isGitDirty,
+	isGitDirtyOrUntracked,
 	isInWorktree,
 	isSubagent,
 	loadRules,
 	loadRulesFromFile,
+	matchBuiltinCondition,
 	type Rule,
 	ruleMatches,
 	toolMatches,
@@ -38,3 +50,4 @@ export {
 	toolsAvailable,
 } from "./tool-hooks";
 export { checkWorktrees } from "./worktree-check";
+export type { ToolEvent } from "./tool-event-types";

package/shepherd/line-count.ts

@@ -1,86 +1,86 @@
-/**
- * 文件行数检查
- * edit/write 后自动检测文件行数，超阈值时注入 ephemeral 提醒
- */
-
-import * as fs from "node:fs";
-import * as path from "node:path";
-import { pushWarning } from "./ephemeral.js";
-
-// ── 代码文件阈值 ─────────────────────────────────────────────
-const LINE_WARN = 200;
-const LINE_MUST = 300;
-const LINE_BAN = 500;
-
-// ── 记忆文件阈值 ─────────────────────────────────────────────
-const MEMORY_LINE_LIMIT = 200;
-
-const CHECKED_EXTENSIONS = new Set([".ts", ".tsx", ".js", ".py", ".rs", ".go"]);
-
-/** 判断路径是否为记忆文件（memory/ 或 .pi/memory/ 下的 .md，排除索引文件 MEMORY.md） */
-function isMemoryFile(filePath: string): boolean {
-	if (!filePath.endsWith(".md")) return false;
-	const name = path.basename(filePath);
-	if (name === "MEMORY.md") return false;
-	const normalized = filePath.replace(/\\/g, "/");
-	return /\/memory\//.test(normalized);
-}
-
-export function checkLineCount(filePath: string): void {
-	const ext = path.extname(filePath);
-
-	// 记忆文件检查
-	if (isMemoryFile(filePath)) {
-		checkMemoryFile(filePath);
-		return;
-	}
-
-	// 代码文件检查
-	if (!CHECKED_EXTENSIONS.has(ext)) return;
-
-	let lines: number;
-	try {
-		const content = fs.readFileSync(filePath, "utf-8");
-		lines = content.split("\n").length;
-	} catch {
-		return;
-	}
-
-	const name = path.basename(filePath);
-
-	if (lines >= LINE_BAN) {
-		pushWarning(
-			`❌ 严禁: ${name} 已有 ${lines} 行（阈值 ${LINE_BAN} 行）。\n` +
-				`拆分建议：提取公共函数/工具类到独立文件，目标每个文件 ≤ 200 行。`,
-		);
-	} else if (lines >= LINE_MUST) {
-		pushWarning(
-			`🔴 必须拆分: ${name} 已有 ${lines} 行（阈值 ${LINE_MUST} 行）。\n` +
-				`拆分建议：提取公共函数/工具类到独立文件，目标每个文件 ≤ 200 行。`,
-		);
-	} else if (lines >= LINE_WARN) {
-		pushWarning(
-			`⚠️ 应主动检查: ${name} 已有 ${lines} 行（阈值 ${LINE_WARN} 行）。\n` +
-				`拆分建议：提取公共函数/工具类到独立文件，目标每个文件 ≤ 200 行。`,
-		);
-	}
-}
-
-/** 记忆文件行数检查：超过 200 行必须拆分 */
-function checkMemoryFile(filePath: string): void {
-	let lines: number;
-	try {
-		const content = fs.readFileSync(filePath, "utf-8");
-		lines = content.split("\n").length;
-	} catch {
-		return;
-	}
-
-	if (lines >= MEMORY_LINE_LIMIT) {
-		const name = path.basename(filePath);
-		pushWarning(
-			`📝 记忆文件过长: ${name} 已有 ${lines} 行（上限 ${MEMORY_LINE_LIMIT} 行）。\n` +
-				`必须拆分：每个记忆文件一个主题，拆分后用 memory_update 或手动更新 MEMORY.md 索引。`,
-		);
-	}
-}
+/**
+ * 文件行数检查
+ * edit/write 后自动检测文件行数，超阈值时注入 ephemeral 提醒
+ */
+
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { pushWarning } from "./ephemeral.js";
+
+// ── 代码文件阈值 ─────────────────────────────────────────────
+const LINE_WARN = 200;
+const LINE_MUST = 300;
+const LINE_BAN = 500;
+
+// ── 记忆文件阈值 ─────────────────────────────────────────────
+const MEMORY_LINE_LIMIT = 200;
+
+const CHECKED_EXTENSIONS = new Set([".ts", ".tsx", ".js", ".py", ".rs", ".go"]);
+
+/** 判断路径是否为记忆文件（memory/ 或 .pi/memory/ 下的 .md，排除索引文件 MEMORY.md） */
+function isMemoryFile(filePath: string): boolean {
+	if (!filePath.endsWith(".md")) return false;
+	const name = path.basename(filePath);
+	if (name === "MEMORY.md") return false;
+	const normalized = filePath.replace(/\\/g, "/");
+	return /\/memory\//.test(normalized);
+}
+
+export function checkLineCount(filePath: string): void {
+	const ext = path.extname(filePath);
+
+	// 记忆文件检查
+	if (isMemoryFile(filePath)) {
+		checkMemoryFile(filePath);
+		return;
+	}
+
+	// 代码文件检查
+	if (!CHECKED_EXTENSIONS.has(ext)) return;
+
+	let lines: number;
+	try {
+		const content = fs.readFileSync(filePath, "utf-8");
+		lines = content.split("\n").length;
+	} catch {
+		return;
+	}
+
+	const name = path.basename(filePath);
+
+	if (lines >= LINE_BAN) {
+		pushWarning(
+			`❌ 严禁: ${name} 已有 ${lines} 行（阈值 ${LINE_BAN} 行）。\n` +
+				`拆分建议：提取公共函数/工具类到独立文件，目标每个文件 ≤ 200 行。`,
+		);
+	} else if (lines >= LINE_MUST) {
+		pushWarning(
+			`🔴 必须拆分: ${name} 已有 ${lines} 行（阈值 ${LINE_MUST} 行）。\n` +
+				`拆分建议：提取公共函数/工具类到独立文件，目标每个文件 ≤ 200 行。`,
+		);
+	} else if (lines >= LINE_WARN) {
+		pushWarning(
+			`⚠️ 应主动检查: ${name} 已有 ${lines} 行（阈值 ${LINE_WARN} 行）。\n` +
+				`拆分建议：提取公共函数/工具类到独立文件，目标每个文件 ≤ 200 行。`,
+		);
+	}
+}
+
+/** 记忆文件行数检查：超过 200 行必须拆分 */
+function checkMemoryFile(filePath: string): void {
+	let lines: number;
+	try {
+		const content = fs.readFileSync(filePath, "utf-8");
+		lines = content.split("\n").length;
+	} catch {
+		return;
+	}
+
+	if (lines >= MEMORY_LINE_LIMIT) {
+		const name = path.basename(filePath);
+		pushWarning(
+			`📝 记忆文件过长: ${name} 已有 ${lines} 行（上限 ${MEMORY_LINE_LIMIT} 行）。\n` +
+				`必须拆分：每个记忆文件一个主题，拆分后用 memory_update 或手动更新 MEMORY.md 索引。`,
+		);
+	}
+}

package/shepherd/message-end.ts

@@ -0,0 +1,120 @@
+/**
+ * message_end hook 处理逻辑
+ *
+ * 在 AI 助手回复完成后，对回复文本进行正则匹配，
+ * 触发 notify（弹通知）或 steer（注入下一轮）。
+ */
+
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { pushWarning } from "./ephemeral.js";
+import {
+	isSubagent,
+	type LoadRulesOptions,
+	loadRules,
+	ruleMatches,
+} from "./rules.js";
+import type { ToolState } from "./tool-hooks.js";
+import { toolsAvailable } from "./tool-hooks.js";
+
+// ── 文本提取 ──────────────────────────────────────────────────
+
+/** assistant 消息中的 content block 类型 */
+interface TextBlock {
+	type: "text";
+	text: string;
+}
+
+/**
+ * 从 assistant 消息中提取所有文本 content blocks 并拼接。
+ * 安全处理 null/undefined/非数组 content。
+ */
+export function extractAssistantText(
+	message: { role: string; content?: unknown } | null | undefined,
+): string {
+	if (!message?.content || !Array.isArray(message.content)) return "";
+	return (message.content as Array<unknown>)
+		.filter(
+			(block): block is TextBlock =>
+				!!block &&
+				typeof block === "object" &&
+				(block as any).type === "text" &&
+				typeof (block as any).text === "string",
+		)
+		.map((block) => block.text)
+		.join("\n");
+}
+
+// ── message_end hook 注册 ─────────────────────────────────────
+
+/** 防重复触发的 rule comment 集合 */
+const _messageEndFired = new Set<string>();
+
+/** 重置防重复标记（agent_start 时调用） */
+export function resetMessageEndState(): void {
+	_messageEndFired.clear();
+}
+
+/** 注册 message_end hook */
+export function registerMessageEnd(
+	pi: ExtensionAPI,
+	state: ToolState,
+	rulesDir?: string,
+	rulesOptions?: LoadRulesOptions,
+): void {
+	pi.on("message_end", async (event, _ctx) => {
+		// 只处理 assistant 消息
+		const message = event.message;
+		if (!message || (message as any).role !== "assistant") return;
+
+		const rules = loadRules(rulesDir, rulesOptions).filter(
+			(r) => r.hook === "message_end",
+		);
+		if (rules.length === 0) return;
+
+		// 提取 AI 回复的纯文本
+		const text = extractAssistantText(message as any);
+		if (!text) return;
+
+		// 构造匹配目标：只有 text 字段
+		const targets = { text, path: "", command: "", glob: "" };
+
+		for (const rule of rules) {
+			// 跳过禁用规则
+			if (rule.enabled === false) continue;
+			// 子代理控制
+			if (isSubagent() && rule.subagent === false) continue;
+			// 工具依赖检查
+			if (!toolsAvailable(rule, pi, state)) continue;
+			// 防重复触发
+			if (_messageEndFired.has(rule.comment)) continue;
+
+			// 条件匹配
+			if (rule.conditions && rule.conditions.length > 0) {
+				if (!ruleMatches(rule, targets, undefined, "")) continue;
+			} else if (rule.pattern) {
+				// 单条件模式：pattern 匹配 text
+				const re = rule._compiled ?? new RegExp(rule.pattern, rule.flags || "");
+				if (!re.test(text)) continue;
+			} else {
+				// 无条件：无条件匹配（总是触发）
+			}
+
+			// 标记已触发
+			_messageEndFired.add(rule.comment);
+
+			// 执行 action
+			if (rule.action === "steer") {
+				pushWarning(rule.reason, rule.comment);
+			} else if (rule.action === "notify" || !rule.action) {
+				pushWarning(rule.reason, rule.comment);
+			}
+		}
+
+		// 不再在此处发送 sendMessage(triggerTurn)。
+		// agent_end 会统一检查 hasWarnings() 并发送唯一的 triggerTurn，
+		// 避免 message_end 和 agent_end 各发一个 sendMessage 导致第二个 drainHints 空 → 空回复。
+		//
+		// 时序：message_end → turn_end → agent_end
+		// message_end 推入的 warning 会在 agent_end 的 sendMessage 触发后被 drainHints 消费。
+	});
+}