oh-pi 0.1.72 → 0.1.74

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "oh-pi",
-  "version": "0.1.72",
+  "version": "0.1.74",
   "description": "One-click setup for pi-coding-agent. Like oh-my-zsh for pi.",
   "type": "module",
   "bin": {

package/pi-package/extensions/ant-colony/index.ts

@@ -51,7 +51,6 @@ export default function antColonyExtension(pi: ExtensionAPI) {
 
   // 当前运行中的后台蚁群（同时只允许一个）
   let activeColony: BackgroundColony | null = null;
-  let uiListenersRegistered = false;
 
   // ─── Status 渲染 ───
 
@@ -63,12 +62,18 @@ export default function antColonyExtension(pi: ExtensionAPI) {
     pi.events.emit("ant-colony:render");
   };
 
-  // 监听事件来更新 UI（确保在有 ctx 的上下文中）
+  // 每次 session_start 重新绑定事件，确保 ctx 始终是最新的
+  let renderHandler: (() => void) | null = null;
+  let clearHandler: (() => void) | null = null;
+  let notifyHandler: ((data: { msg: string; level: "info" | "success" | "warning" | "error" }) => void) | null = null;
+
   pi.on("session_start", async (_event, ctx) => {
-    if (uiListenersRegistered) return;
-    uiListenersRegistered = true;
+    // 移除旧监听器（session 重启 / /reload 时 ctx 已失效）
+    if (renderHandler) pi.events.off("ant-colony:render", renderHandler);
+    if (clearHandler) pi.events.off("ant-colony:clear-ui", clearHandler);
+    if (notifyHandler) pi.events.off("ant-colony:notify", notifyHandler);
 
-    pi.events.on("ant-colony:render", () => {
+    renderHandler = () => {
       if (!activeColony) return;
       const { state } = activeColony;
       const elapsed = state ? formatDuration(Date.now() - state.createdAt) : "0s";
@@ -81,14 +86,17 @@ export default function antColonyExtension(pi: ExtensionAPI) {
       parts.push(elapsed);
 
       ctx.ui.setStatus("ant-colony", parts.join(" │ "));
-    });
-
-    pi.events.on("ant-colony:clear-ui", () => {
+    };
+    clearHandler = () => {
       ctx.ui.setStatus("ant-colony", undefined);
-    });
-    pi.events.on("ant-colony:notify", (data: { msg: string; level: "info" | "success" | "warning" | "error" }) => {
+    };
+    notifyHandler = (data) => {
       ctx.ui.notify(data.msg, data.level);
-    });
+    };
+
+    pi.events.on("ant-colony:render", renderHandler);
+    pi.events.on("ant-colony:clear-ui", clearHandler);
+    pi.events.on("ant-colony:notify", notifyHandler);
   });
 
   // ─── 同步模式（print mode）：阻塞等待蚁群完成 ───
@@ -162,14 +170,14 @@ export default function antColonyExtension(pi: ExtensionAPI) {
     const callbacks: QueenCallbacks = {
       onSignal(signal) {
         colony.phase = signal.message;
-        // 阶段切换时注入消息到主进程对话流
+        // 阶段切换时注入消息到主进程对话流（display: true 让 LLM 下次可见，无需轮询）
         if (signal.phase !== lastPhase) {
           lastPhase = signal.phase;
           const pct = Math.round(signal.progress * 100);
           pi.sendMessage({
             customType: "ant-colony-progress",
             content: `[COLONY_SIGNAL:${signal.phase.toUpperCase()}] 🐜 ${signal.message} (${pct}%, ${formatCost(signal.cost)})`,
-            display: false,
+            display: true,
           }, { triggerTurn: false, deliverAs: "followUp" });
         }
         throttledRender();
@@ -194,7 +202,7 @@ export default function antColonyExtension(pi: ExtensionAPI) {
         pi.sendMessage({
           customType: "ant-colony-progress",
           content: `[COLONY_SIGNAL:TASK_DONE] 🐜 ${icon} ${task.title.slice(0, 60)} (${progress}, ${cost})`,
-          display: false,
+          display: true,
         }, { triggerTurn: false, deliverAs: "followUp" });
         throttledRender();
       },
@@ -367,12 +375,14 @@ export default function antColonyExtension(pi: ExtensionAPI) {
         };
 
         // 定时刷新
-        const timer = setInterval(() => {
+        let timer: ReturnType<typeof setInterval> | null = setInterval(() => {
           cachedWidth = undefined;
           cachedLines = undefined;
           tui.requestRender();
         }, 1000);
 
+        const cleanup = () => { if (timer) { clearInterval(timer); timer = null; } };
+
         return {
           render(width: number): string[] {
             if (cachedLines && cachedWidth === width) return cachedLines;
@@ -380,10 +390,10 @@ export default function antColonyExtension(pi: ExtensionAPI) {
             cachedWidth = width;
             return cachedLines;
           },
-          invalidate() { cachedWidth = undefined; cachedLines = undefined; },
+          invalidate() { cachedWidth = undefined; cachedLines = undefined; cleanup(); },
           handleInput(data: string) {
             if (matchesKey(data, "escape")) {
-              clearInterval(timer);
+              cleanup();
               done(undefined);
             }
           },

package/pi-package/extensions/ant-colony/nest.test.ts

@@ -0,0 +1,130 @@
+import { describe, it, expect, beforeEach, afterEach } from "vitest";
+import * as fs from "node:fs";
+import * as path from "node:path";
+import * as os from "node:os";
+import { Nest } from "./nest.js";
+import type { ColonyState, Pheromone } from "./types.js";
+
+const mkState = (overrides: Partial<ColonyState> = {}): ColonyState => ({
+  id: "test-colony", goal: "test", status: "working",
+  tasks: [], ants: [], pheromones: [],
+  concurrency: { current: 2, min: 1, max: 4, optimal: 3, history: [] },
+  metrics: { tasksTotal: 0, tasksDone: 0, tasksFailed: 0, antsSpawned: 0, totalCost: 0, totalTokens: 0, startTime: Date.now(), throughputHistory: [] },
+  maxCost: null, modelOverrides: {}, createdAt: Date.now(), finishedAt: null,
+  ...overrides,
+});
+
+const mkPheromone = (overrides: Partial<Pheromone> = {}): Pheromone => ({
+  id: `p-${Math.random().toString(36).slice(2)}`, type: "warning", antId: "ant-1", antCaste: "worker",
+  taskId: "t-1", content: "test", files: ["a.ts"], strength: 1.0, createdAt: Date.now(),
+  ...overrides,
+});
+
+let tmpDir: string;
+let nest: Nest;
+
+beforeEach(() => {
+  tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "nest-test-"));
+  nest = new Nest(tmpDir, "test-colony");
+  nest.init(mkState());
+});
+
+afterEach(() => {
+  try { fs.rmSync(tmpDir, { recursive: true, force: true }); } catch { /* ignore */ }
+});
+
+describe("getStateLight", () => {
+  it("returns state without triggering pheromone read", () => {
+    nest.dropPheromone(mkPheromone());
+    const light = nest.getStateLight();
+    expect(light.id).toBe("test-colony");
+    expect(light.tasks).toEqual([]);
+    // pheromones should not be populated by getStateLight
+    // (it returns stateCache which has empty pheromones from init)
+  });
+
+  it("includes tasks from cache", () => {
+    nest.writeTask({
+      id: "t-1", parentId: null, title: "Test", description: "desc",
+      caste: "worker", status: "pending", priority: 3, files: [],
+      claimedBy: null, result: null, error: null, spawnedTasks: [],
+      createdAt: Date.now(), startedAt: null, finishedAt: null,
+    });
+    const light = nest.getStateLight();
+    expect(light.tasks).toHaveLength(1);
+    expect(light.tasks[0].id).toBe("t-1");
+  });
+});
+
+describe("countWarnings", () => {
+  it("returns 0 when no pheromones", () => {
+    expect(nest.countWarnings(["a.ts"])).toBe(0);
+  });
+
+  it("counts warning pheromones for matching files", () => {
+    nest.dropPheromone(mkPheromone({ type: "warning", files: ["a.ts"] }));
+    nest.dropPheromone(mkPheromone({ type: "warning", files: ["a.ts"] }));
+    nest.dropPheromone(mkPheromone({ type: "completion", files: ["a.ts"] }));
+    expect(nest.countWarnings(["a.ts"])).toBe(2);
+  });
+
+  it("counts repellent pheromones", () => {
+    nest.dropPheromone(mkPheromone({ type: "repellent", files: ["b.ts"] }));
+    expect(nest.countWarnings(["b.ts"])).toBe(1);
+  });
+
+  it("returns 0 for unrelated files", () => {
+    nest.dropPheromone(mkPheromone({ type: "warning", files: ["a.ts"] }));
+    expect(nest.countWarnings(["c.ts"])).toBe(0);
+  });
+});
+
+describe("pheromone dirty flag", () => {
+  it("rebuilds index after dropPheromone", () => {
+    nest.dropPheromone(mkPheromone({ type: "discovery", files: ["x.ts"] }));
+    const pheromones = nest.getAllPheromones();
+    expect(pheromones.length).toBe(1);
+    expect(pheromones[0].type).toBe("discovery");
+  });
+
+  it("does not rebuild index when nothing changed", () => {
+    nest.dropPheromone(mkPheromone({ files: ["x.ts"] }));
+    nest.getAllPheromones(); // builds index, clears dirty
+    // Second call should use cached index (no new data, no GC)
+    const p2 = nest.getAllPheromones();
+    expect(p2.length).toBe(1);
+  });
+});
+
+describe("claimNextTask", () => {
+  it("claims highest scored pending task", () => {
+    nest.writeTask({
+      id: "t-low", parentId: null, title: "Low", description: "",
+      caste: "worker", status: "pending", priority: 5, files: [],
+      claimedBy: null, result: null, error: null, spawnedTasks: [],
+      createdAt: Date.now(), startedAt: null, finishedAt: null,
+    });
+    nest.writeTask({
+      id: "t-high", parentId: null, title: "High", description: "",
+      caste: "worker", status: "pending", priority: 1, files: [],
+      claimedBy: null, result: null, error: null, spawnedTasks: [],
+      createdAt: Date.now(), startedAt: null, finishedAt: null,
+    });
+    const claimed = nest.claimNextTask("worker", "ant-1");
+    expect(claimed).not.toBeNull();
+    expect(claimed!.id).toBe("t-high");
+    expect(claimed!.status).toBe("claimed");
+  });
+
+  it("returns null when no pending tasks", () => {
+    expect(nest.claimNextTask("worker", "ant-1")).toBeNull();
+  });
+});
+
+describe("withStateLock spin", () => {
+  it("updateState works under normal conditions", () => {
+    nest.updateState({ status: "reviewing" });
+    const state = nest.getStateLight();
+    expect(state.status).toBe("reviewing");
+  });
+});

package/pi-package/extensions/ant-colony/prompts.test.ts

@@ -54,4 +54,46 @@ describe("buildPrompt", () => {
     const r = buildPrompt(mkTask({ description: "修复登录问题" }), "", "System");
     expect(r).toContain("Chinese");
   });
+
+  // Bio 3: 串联觅食 — tandem context
+  it("includes tandem parent result when provided", () => {
+    const r = buildPrompt(mkTask(), "", "System", 10, { parentResult: "Parent found auth module at src/auth.ts" });
+    expect(r).toContain("Tandem Context");
+    expect(r).toContain("Parent found auth module");
+  });
+
+  it("includes tandem prior error when provided", () => {
+    const r = buildPrompt(mkTask(), "", "System", 10, { priorError: "TypeError: cannot read property" });
+    expect(r).toContain("Prior Attempt Failed");
+    expect(r).toContain("TypeError: cannot read property");
+  });
+
+  it("includes both tandem fields when provided", () => {
+    const r = buildPrompt(mkTask(), "", "System", 10, {
+      parentResult: "Scout found files",
+      priorError: "Build failed",
+    });
+    expect(r).toContain("Tandem Context");
+    expect(r).toContain("Prior Attempt Failed");
+  });
+
+  it("omits tandem sections when not provided", () => {
+    const r = buildPrompt(mkTask(), "", "System", 10);
+    expect(r).not.toContain("Tandem Context");
+    expect(r).not.toContain("Prior Attempt Failed");
+  });
+
+  it("omits tandem sections when empty object", () => {
+    const r = buildPrompt(mkTask(), "", "System", 10, {});
+    expect(r).not.toContain("Tandem Context");
+    expect(r).not.toContain("Prior Attempt Failed");
+  });
+
+  it("truncates long parent result", () => {
+    const longResult = "x".repeat(5000);
+    const r = buildPrompt(mkTask(), "", "System", 10, { parentResult: longResult });
+    expect(r).toContain("Tandem Context");
+    // Should be truncated to 3000 chars
+    expect(r.length).toBeLessThan(longResult.length);
+  });
 });

package/pi-package/extensions/ant-colony/queen.test.ts

@@ -0,0 +1,193 @@
+import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
+import * as fs from "node:fs";
+import * as path from "node:path";
+import * as os from "node:os";
+
+vi.mock("@mariozechner/pi-coding-agent", () => ({
+  AuthStorage: class {},
+  createAgentSession: vi.fn(),
+  createReadTool: vi.fn(), createBashTool: vi.fn(), createEditTool: vi.fn(),
+  createWriteTool: vi.fn(), createGrepTool: vi.fn(), createFindTool: vi.fn(),
+  createLsTool: vi.fn(), ModelRegistry: class {}, SessionManager: { inMemory: vi.fn() },
+  SettingsManager: { inMemory: vi.fn() }, createExtensionRuntime: vi.fn(),
+}));
+vi.mock("@mariozechner/pi-ai", () => ({ getModel: vi.fn() }));
+
+import { classifyError, quorumMergeTasks } from "./queen.js";
+import { Nest } from "./nest.js";
+import type { ColonyState, Task } from "./types.js";
+
+// ═══ classifyError ═══
+
+describe("classifyError", () => {
+  it("classifies TypeError", () => {
+    expect(classifyError("TypeError: cannot read property 'x'")).toBe("type_error");
+  });
+
+  it("classifies TS errors", () => {
+    expect(classifyError("TS2345: Argument of type")).toBe("type_error");
+  });
+
+  it("classifies permission errors", () => {
+    expect(classifyError("EACCES: permission denied")).toBe("permission");
+  });
+
+  it("classifies 401", () => {
+    expect(classifyError("Error: 401 Unauthorized")).toBe("permission");
+  });
+
+  it("classifies timeout", () => {
+    expect(classifyError("Error: Timeout after 5000ms")).toBe("timeout");
+  });
+
+  it("classifies ETIMEDOUT", () => {
+    expect(classifyError("connect ETIMEDOUT 1.2.3.4")).toBe("timeout");
+  });
+
+  it("classifies ENOENT", () => {
+    expect(classifyError("ENOENT: no such file or directory")).toBe("not_found");
+  });
+
+  it("classifies Cannot find module", () => {
+    expect(classifyError("Cannot find module './foo'")).toBe("not_found");
+  });
+
+  it("classifies syntax errors", () => {
+    expect(classifyError("SyntaxError: Unexpected token")).toBe("syntax");
+  });
+
+  it("classifies rate limit", () => {
+    expect(classifyError("Error: 429 Too Many Requests")).toBe("rate_limit");
+  });
+
+  it("returns unknown for unrecognized errors", () => {
+    expect(classifyError("Something completely different")).toBe("unknown");
+  });
+
+  it("handles empty string", () => {
+    expect(classifyError("")).toBe("unknown");
+  });
+});
+
+// ═══ quorumMergeTasks ═══
+
+const mkState = (overrides: Partial<ColonyState> = {}): ColonyState => ({
+  id: "test-colony", goal: "test", status: "working",
+  tasks: [], ants: [], pheromones: [],
+  concurrency: { current: 2, min: 1, max: 4, optimal: 3, history: [] },
+  metrics: { tasksTotal: 0, tasksDone: 0, tasksFailed: 0, antsSpawned: 0, totalCost: 0, totalTokens: 0, startTime: Date.now(), throughputHistory: [] },
+  maxCost: null, modelOverrides: {}, createdAt: Date.now(), finishedAt: null,
+  ...overrides,
+});
+
+const mkTask = (overrides: Partial<Task> = {}): Task => ({
+  id: `t-${Math.random().toString(36).slice(2)}`, parentId: null,
+  title: "Test task", description: "Do something",
+  caste: "worker", status: "pending", priority: 3, files: [],
+  claimedBy: null, result: null, error: null, spawnedTasks: [],
+  createdAt: Date.now(), startedAt: null, finishedAt: null,
+  ...overrides,
+});
+
+let tmpDir: string;
+let nest: Nest;
+
+beforeEach(() => {
+  tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "queen-test-"));
+  nest = new Nest(tmpDir, "test-colony");
+  nest.init(mkState());
+});
+
+afterEach(() => {
+  try { fs.rmSync(tmpDir, { recursive: true, force: true }); } catch { /* ignore */ }
+});
+
+describe("quorumMergeTasks", () => {
+  it("does nothing with 0-1 tasks", () => {
+    const t1 = mkTask({ files: ["a.ts"], priority: 3 });
+    nest.writeTask(t1);
+    quorumMergeTasks(nest);
+    const tasks = nest.getAllTasks().filter(t => t.status === "pending");
+    expect(tasks).toHaveLength(1);
+  });
+
+  it("merges duplicate tasks with same files", () => {
+    const t1 = mkTask({ id: "t-1", title: "Fix auth", files: ["a.ts", "b.ts"], priority: 3 });
+    const t2 = mkTask({ id: "t-2", title: "Fix auth v2", files: ["a.ts", "b.ts"], priority: 3 });
+    nest.writeTask(t1);
+    nest.writeTask(t2);
+    quorumMergeTasks(nest);
+    const pending = nest.getAllTasks().filter(t => t.status === "pending");
+    const done = nest.getAllTasks().filter(t => t.status === "done");
+    expect(pending).toHaveLength(1);
+    expect(done).toHaveLength(1);
+    expect(done[0].result).toContain("quorum");
+  });
+
+  it("boosts priority of merged task", () => {
+    const t1 = mkTask({ id: "t-1", files: ["x.ts"], priority: 3 });
+    const t2 = mkTask({ id: "t-2", files: ["x.ts"], priority: 3 });
+    nest.writeTask(t1);
+    nest.writeTask(t2);
+    quorumMergeTasks(nest);
+    const pending = nest.getAllTasks().filter(t => t.status === "pending");
+    expect(pending[0].priority).toBe(2); // boosted from 3 to 2
+  });
+
+  it("does not merge tasks with different files", () => {
+    const t1 = mkTask({ id: "t-1", files: ["a.ts"] });
+    const t2 = mkTask({ id: "t-2", files: ["b.ts"] });
+    nest.writeTask(t1);
+    nest.writeTask(t2);
+    quorumMergeTasks(nest);
+    const pending = nest.getAllTasks().filter(t => t.status === "pending");
+    expect(pending).toHaveLength(2);
+  });
+
+  it("merges context from duplicate tasks", () => {
+    const t1 = mkTask({ id: "t-1", files: ["a.ts"], context: "context A" });
+    const t2 = mkTask({ id: "t-2", files: ["a.ts"], context: "context B" });
+    nest.writeTask(t1);
+    nest.writeTask(t2);
+    quorumMergeTasks(nest);
+    const pending = nest.getAllTasks().filter(t => t.status === "pending");
+    expect(pending[0].context).toContain("context A");
+    expect(pending[0].context).toContain("context B");
+  });
+
+  it("handles three duplicates", () => {
+    const t1 = mkTask({ id: "t-1", files: ["a.ts"], priority: 4 });
+    const t2 = mkTask({ id: "t-2", files: ["a.ts"], priority: 4 });
+    const t3 = mkTask({ id: "t-3", files: ["a.ts"], priority: 4 });
+    nest.writeTask(t1);
+    nest.writeTask(t2);
+    nest.writeTask(t3);
+    quorumMergeTasks(nest);
+    const pending = nest.getAllTasks().filter(t => t.status === "pending");
+    const done = nest.getAllTasks().filter(t => t.status === "done");
+    expect(pending).toHaveLength(1);
+    expect(done).toHaveLength(2);
+    expect(pending[0].priority).toBe(3); // boosted from 4 to 3
+  });
+
+  it("skips non-pending tasks", () => {
+    const t1 = mkTask({ id: "t-1", files: ["a.ts"], status: "done" });
+    const t2 = mkTask({ id: "t-2", files: ["a.ts"], status: "pending" });
+    nest.writeTask(t1);
+    nest.writeTask(t2);
+    quorumMergeTasks(nest);
+    // Only 1 pending, so no merge
+    const pending = nest.getAllTasks().filter(t => t.status === "pending");
+    expect(pending).toHaveLength(1);
+  });
+
+  it("does not boost priority below 1", () => {
+    const t1 = mkTask({ id: "t-1", files: ["a.ts"], priority: 1 });
+    const t2 = mkTask({ id: "t-2", files: ["a.ts"], priority: 1 });
+    nest.writeTask(t1);
+    nest.writeTask(t2);
+    quorumMergeTasks(nest);
+    const pending = nest.getAllTasks().filter(t => t.status === "pending");
+    expect(pending[0].priority).toBe(1); // stays at 1
+  });
+});

package/pi-package/extensions/ant-colony/queen.ts

@@ -102,7 +102,7 @@ function childTaskFromParsed(
  * Bio 5: 蚁群投票 — 合并多 Scout 产生的重复任务
  * 相同文件集合的任务合并，被多 Scout 提及的任务 priority 提升
  */
-function quorumMergeTasks(nest: Nest): void {
+export function quorumMergeTasks(nest: Nest): void {
   const tasks = nest.getAllTasks().filter(t =>
     (t.caste === "worker" || t.caste === "drone") && t.status === "pending"
   );
@@ -198,7 +198,7 @@ interface WaveOptions {
 /**
  * Bio 6: 尸体清理 — 错误模式分类
  */
-function classifyError(errStr: string): string {
+export function classifyError(errStr: string): string {
   if (errStr.includes("TypeError") || errStr.includes("type") || errStr.includes("TS")) return "type_error";
   if (errStr.includes("permission") || errStr.includes("401") || errStr.includes("EACCES")) return "permission";
   if (errStr.includes("timeout") || errStr.includes("Timeout") || errStr.includes("ETIMEDOUT")) return "timeout";

package/pi-package/extensions/ant-colony/ui.ts

@@ -33,7 +33,8 @@ export function buildReport(state: ColonyState): string {
   return [
     `## 🐜 Ant Colony Report`,
     `**Goal:** ${state.goal}`,
-    `**Status:** ${statusIcon(state.status)} ${state.status} │ ${elapsed} │ ${formatCost(m.totalCost)}`,
+    `**Status:** ${statusIcon(state.status)} ${state.status} │ ${formatCost(m.totalCost)}`,
+    `**Duration:** ${elapsed}`,
     `**Tasks:** ${m.tasksDone}/${m.tasksTotal} done${m.tasksFailed > 0 ? `, ${m.tasksFailed} failed` : ""}`,
     ``,
     ...state.tasks.filter(t => t.status === "done").map(t =>

package/pi-package/extensions/bg-process.ts

@@ -2,13 +2,14 @@
  * oh-pi Background Process Extension
  *
  * 任何 bash 命令超时未完成时，自动送到后台执行。
+ * 进程完成后自动通过 sendMessage 通知 LLM，无需轮询。
  * 提供 bg_status 工具让 LLM 查看/停止后台进程。
  */
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { Type } from "@sinclair/typebox";
 import { StringEnum } from "@mariozechner/pi-ai";
 import { spawn, execSync } from "node:child_process";
-import { writeFileSync, readFileSync, existsSync } from "node:fs";
+import { writeFileSync, readFileSync, appendFileSync, existsSync } from "node:fs";
 
 /** 超时阈值（毫秒），超过此时间自动后台化 */
 const BG_TIMEOUT_MS = 10_000;
@@ -18,6 +19,8 @@ interface BgProcess {
   command: string;
   logFile: string;
   startedAt: number;
+  finished: boolean;
+  exitCode: number | null;
 }
 
 export default function (pi: ExtensionAPI) {
@@ -41,6 +44,7 @@ export default function (pi: ExtensionAPI) {
         let stdout = "";
         let stderr = "";
         let settled = false;
+        let backgrounded = false;
 
         const child = spawn("bash", ["-c", command], {
           cwd: process.cwd(),
@@ -48,36 +52,58 @@ export default function (pi: ExtensionAPI) {
           stdio: ["ignore", "pipe", "pipe"],
         });
 
-        child.stdout?.on("data", (d: Buffer) => { stdout += d.toString(); });
-        child.stderr?.on("data", (d: Buffer) => { stderr += d.toString(); });
+        child.stdout?.on("data", (d: Buffer) => {
+          const chunk = d.toString();
+          stdout += chunk;
+          // 后台化后追加写入日志
+          if (backgrounded) {
+            try { appendFileSync(bgProcesses.get(child.pid!)?.logFile ?? "", chunk); } catch {}
+          }
+        });
+        child.stderr?.on("data", (d: Buffer) => {
+          const chunk = d.toString();
+          stderr += chunk;
+          if (backgrounded) {
+            try { appendFileSync(bgProcesses.get(child.pid!)?.logFile ?? "", chunk); } catch {}
+          }
+        });
 
-        // 超时处理：分离进程，送到后台
+        // 超时处理：保持管道，标记为后台
         const timer = setTimeout(() => {
           if (settled) return;
           settled = true;
+          backgrounded = true;
 
-          // 分离子进程，让它继续运行
-          child.stdout?.removeAllListeners();
-          child.stderr?.removeAllListeners();
-          child.removeAllListeners();
           child.unref();
 
           const logFile = `/tmp/oh-pi-bg-${Date.now()}.log`;
           const pid = child.pid!;
 
-          // 启动一个 tail 进程把后续输出写入日志
-          try {
-            const tailCmd = `(echo ${JSON.stringify(stdout + stderr)}; tail --pid=${pid} -f /proc/${pid}/fd/1 2>/dev/null) > ${logFile} 2>&1 &`;
-            spawn("bash", ["-c", tailCmd], { detached: true, stdio: "ignore" }).unref();
-          } catch {
-            // fallback: 至少把已有输出写入日志
-            writeFileSync(logFile, stdout + stderr);
-          }
-
-          bgProcesses.set(pid, { pid, command, logFile, startedAt: Date.now() });
+          // 把已有输出写入日志
+          writeFileSync(logFile, stdout + stderr);
+
+          const proc: BgProcess = { pid, command, logFile, startedAt: Date.now(), finished: false, exitCode: null };
+          bgProcesses.set(pid, proc);
+
+          // 监听完成事件，自动通知 LLM
+          child.on("close", (code) => {
+            proc.finished = true;
+            proc.exitCode = code;
+            const tail = (stdout + stderr).slice(-3000);
+            const truncated = (stdout + stderr).length > 3000 ? "[...truncated]\n" + tail : tail;
+            // 最终输出写入日志
+            try { writeFileSync(logFile, stdout + stderr); } catch {}
+
+            pi.sendMessage({
+              content: `[BG_PROCESS_DONE] PID ${pid} finished (exit ${code ?? "?"})\nCommand: ${command}\n\nOutput (last 3000 chars):\n${truncated}`,
+              display: true,
+              triggerTurn: true,
+              deliverAs: "followUp",
+            });
+          });
 
           const preview = (stdout + stderr).slice(0, 500);
-          const text = `Command still running after ${effectiveTimeout / 1000}s, moved to background.\nPID: ${pid}\nLog: ${logFile}\nView output: tail -f ${logFile}\nStop: kill ${pid}\n\nOutput so far:\n${preview}`;
+          const text = `Command still running after ${effectiveTimeout / 1000}s, moved to background.\nPID: ${pid}\nLog: ${logFile}\nStop: kill ${pid}\n\nOutput so far:\n${preview}\n\n⏳ You will be notified automatically when it finishes. No need to poll.`;
 
           resolve({
             content: [{ type: "text", text }],
@@ -85,7 +111,7 @@ export default function (pi: ExtensionAPI) {
           });
         }, effectiveTimeout);
 
-        // 正常结束
+        // 正常结束（超时前）
         child.on("close", (code) => {
           if (settled) return;
           settled = true;
@@ -146,8 +172,7 @@ export default function (pi: ExtensionAPI) {
           return { content: [{ type: "text", text: "No background processes." }], details: {} };
         }
         const lines = [...bgProcesses.values()].map((p) => {
-          const alive = isAlive(p.pid);
-          const status = alive ? "🟢 running" : "⚪ stopped";
+          const status = p.finished ? `⚪ stopped (exit ${p.exitCode ?? "?"})` : (isAlive(p.pid) ? "🟢 running" : "⚪ stopped");
           return `PID: ${p.pid} | ${status} | Log: ${p.logFile}\n  Cmd: ${p.command}`;
         });
         return { content: [{ type: "text", text: lines.join("\n\n") }], details: {} };
@@ -171,13 +196,7 @@ export default function (pi: ExtensionAPI) {
             return { content: [{ type: "text", text: `Error reading log: ${e.message}` }], details: {}, isError: true };
           }
         }
-        // fallback: 直接读 /proc
-        try {
-          const out = execSync(`tail -20 /proc/${pid}/fd/1 2>/dev/null || echo "(cannot read output)"`, { timeout: 3000 }).toString();
-          return { content: [{ type: "text", text: out }], details: {} };
-        } catch {
-          return { content: [{ type: "text", text: "No log available for this PID." }], details: {} };
-        }
+        return { content: [{ type: "text", text: "No log available for this PID." }], details: {} };
       }
 
       if (action === "stop") {
@@ -197,8 +216,10 @@ export default function (pi: ExtensionAPI) {
 
   // 清理：退出时杀掉所有后台进程
   pi.on("session_shutdown", async () => {
-    for (const [pid] of bgProcesses) {
-      try { process.kill(pid, "SIGTERM"); } catch {}
+    for (const [pid, proc] of bgProcesses) {
+      if (!proc.finished) {
+        try { process.kill(pid, "SIGTERM"); } catch {}
+      }
     }
     bgProcesses.clear();
   });