oh-pi 0.1.72 → 0.1.73

package/package.json

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

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";