@zhushanwen/pi-goal 0.4.0 → 0.4.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhushanwen/pi-goal",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "description": "Codex-style /goal command for Pi — persistent goal-driven autonomous loop with evidence-based completion, token/time budgets, blocked detection, and steering templates.",
   "type": "module",
   "main": "src/index.ts",

package/src/__tests__/event-adapter.test.ts

@@ -10,7 +10,7 @@
  * 用 fake pi + fake ctx（不 import Pi SDK）。handler 签名 (pi, session, ctx) → void/result。
  */
 import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
-import { describe, expect, it } from "vitest";
+import { describe, expect, it, vi } from "vitest";
 
 import { handleAgentEnd } from "../adapters/event-handlers/agent-end";
 import { handleBeforeAgentStart } from "../adapters/event-handlers/before-agent-start";
@@ -336,4 +336,75 @@ describe("handleBeforeAgentStart", () => {
 
 		expect(result).toBeUndefined();
 	});
+
+	// ── pending-notifications（W2 跨扩展集成）───────────────────
+
+	it("有活跃 pending operations → 注入提示含 pending 数量", async () => {
+		const { pi } = makeFakePi();
+		// 创建带 pending:register entries 的 ctx
+		const entries = [
+			{ customType: "pending:register", data: { id: "wf-1", type: "workflow", name: "test-wf" } },
+			{ customType: "pending:register", data: { id: "bg-1", type: "subagent", name: "worker" } },
+		];
+		const ctx = {
+			hasUI: true,
+			signal: { aborted: false } as AbortSignal,
+			getContextUsage: () => undefined,
+			ui: {
+				notify: vi.fn(),
+				setStatus: vi.fn(),
+				setWidget: vi.fn(),
+				theme: { fg: (_c: string, t: string) => t, bold: (t: string) => t },
+			},
+			sessionManager: { getEntries: () => entries, getBranch: () => undefined },
+		} as unknown as ExtensionContext;
+		const session = createGoalSession();
+		session.state = makeRunningState();
+
+		const result = await handleBeforeAgentStart(pi, session, ctx);
+
+		expect(result).toBeDefined();
+		expect(result!.message.customType).toBe("goal-context");
+		expect(result!.message.content).toContain("2 pending async operation(s)");
+	});
+
+	it("有 pending register 但已 unregister → 不注入提示", async () => {
+		const { pi } = makeFakePi();
+		const entries = [
+			{ customType: "pending:register", data: { id: "wf-1", type: "workflow", name: "test-wf" } },
+			{ customType: "pending:unregister", data: { id: "wf-1" } },
+		];
+		const ctx = {
+			hasUI: true,
+			signal: { aborted: false } as AbortSignal,
+			getContextUsage: () => undefined,
+			ui: {
+				notify: vi.fn(),
+				setStatus: vi.fn(),
+				setWidget: vi.fn(),
+				theme: { fg: (_c: string, t: string) => t, bold: (t: string) => t },
+			},
+			sessionManager: { getEntries: () => entries, getBranch: () => undefined },
+		} as unknown as ExtensionContext;
+		const session = createGoalSession();
+		session.state = makeRunningState();
+
+		const result = await handleBeforeAgentStart(pi, session, ctx);
+
+		expect(result).toBeDefined();
+		expect(result!.message.customType).toBe("goal-context");
+		expect(result!.message.content).not.toContain("pending async operation");
+	});
+
+	it("无 pending entries → 正常注入（不含 pending 提示）", async () => {
+		const { pi } = makeFakePi();
+		const { ctx } = makeFakeCtx(); // 默认 getEntries: () => []
+		const session = createGoalSession();
+		session.state = makeRunningState();
+
+		const result = await handleBeforeAgentStart(pi, session, ctx);
+
+		expect(result).toBeDefined();
+		expect(result!.message.content).not.toContain("pending async operation");
+	});
 });

package/src/adapters/event-handlers/before-agent-start.ts

@@ -40,6 +40,14 @@ export async function handleBeforeAgentStart(
 ): Promise<BeforeAgentStartResult | undefined> {
 	if (!session.state) return;
 
+	pi.appendEntry("goal:log", {
+		timestamp: Date.now(),
+		level: "debug",
+		component: "goal:before-agent-start",
+		message: "handleBeforeAgentStart invoked",
+		data: { status: session.state.status },
+	});
+
 	// 终态处理
 	if (isTerminalStatus(session.state.status)) {
 		handleTerminalStateBeforeAgent(pi, session, ctx);
@@ -53,10 +61,44 @@ export async function handleBeforeAgentStart(
 
 	// 正常 context injection。
 	// 全解耦：planAvailable 恒 true（contextInjectionPrompt 恒定建议 plan mode，AI 自行决定）。
+	// pending-notifications：若有活跃的异步操作，注入等待提示。
+	// 读取 pending:register/unregister entry 算差集，并校验 expiresAt（TTL 过期的视为非活跃）。
+	// pending-notifications 的 rebuildFromEntries 在其 session_start 已 flush 跨 session/过期残留，
+	// 但同一 session 内超时未 unregister 的 entry（如崩溃恢复遗漏）需此处完补以防误报 active。
+	const now = Date.now();
+	const entries = ctx.sessionManager.getEntries();
+	const pendingUnregisters = new Set(
+		entries.filter((e) => e.customType === "pending:unregister").map((e) => (e.data as Record<string, unknown>)?.id),
+	);
+	const activePending = entries.filter((e) => {
+		if (e.customType !== "pending:register") return false;
+		const id = (e.data as Record<string, unknown>)?.id;
+		if (pendingUnregisters.has(id)) return false;
+		// TTL 过期视为非活跃（expiresAt 缺失或类型错误时宽容处理为活跃）
+		const expiresAt = (e.data as Record<string, unknown>)?.expiresAt;
+		if (typeof expiresAt === "number" && expiresAt <= now) return false;
+		return true;
+	});
+	const pendingHint = activePending.length > 0
+		? `\nNote: There are ${activePending.length} pending async operation(s) running. Consider waiting for them to complete before starting new work.`
+		: "";
+
+	pi.appendEntry("goal:log", {
+		timestamp: Date.now(),
+		level: "debug",
+		component: "goal:before-agent-start",
+		message: "pending entries computed",
+		data: {
+			activePending: activePending.length,
+			injectHint: activePending.length > 0,
+			pendingIds: activePending.map((e) => (e.data as Record<string, unknown>)?.id),
+		},
+	});
+
 	return {
 		message: {
 			customType: "goal-context",
-			content: contextInjectionPrompt(session.state, session.state.timeUsedSeconds, true),
+			content: contextInjectionPrompt(session.state, session.state.timeUsedSeconds, true) + pendingHint,
 			display: false,
 		},
 	};