oh-pi 0.1.70 → 0.1.72

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

@@ -0,0 +1,66 @@
+import { describe, it, expect } from "vitest";
+import { formatDuration, formatCost, formatTokens, statusIcon, casteIcon, buildReport } from "./ui.js";
+import type { ColonyState } from "./types.js";
+
+describe("formatDuration", () => {
+  it("0ms", () => expect(formatDuration(0)).toBe("0s"));
+  it("5000ms", () => expect(formatDuration(5000)).toBe("5s"));
+  it("59000ms", () => expect(formatDuration(59000)).toBe("59s"));
+  it("60000ms", () => expect(formatDuration(60000)).toBe("1m0s"));
+  it("90000ms", () => expect(formatDuration(90000)).toBe("1m30s"));
+});
+
+describe("formatCost", () => {
+  it("0.001", () => expect(formatCost(0.001)).toBe("$0.0010"));
+  it("0.009", () => expect(formatCost(0.009)).toBe("$0.0090"));
+  it("0.01", () => expect(formatCost(0.01)).toBe("$0.01"));
+  it("1.5", () => expect(formatCost(1.5)).toBe("$1.50"));
+});
+
+describe("formatTokens", () => {
+  it("500", () => expect(formatTokens(500)).toBe("500"));
+  it("999", () => expect(formatTokens(999)).toBe("999"));
+  it("1500", () => expect(formatTokens(1500)).toBe("1.5k"));
+  it("1500000", () => expect(formatTokens(1500000)).toBe("1.5M"));
+});
+
+describe("statusIcon", () => {
+  it("scouting", () => expect(statusIcon("scouting")).toBe("🔍"));
+  it("working", () => expect(statusIcon("working")).toBe("⚒️"));
+  it("reviewing", () => expect(statusIcon("reviewing")).toBe("🛡️"));
+  it("done", () => expect(statusIcon("done")).toBe("✅"));
+  it("failed", () => expect(statusIcon("failed")).toBe("❌"));
+  it("budget_exceeded", () => expect(statusIcon("budget_exceeded")).toBe("💰"));
+  it("unknown", () => expect(statusIcon("xyz")).toBe("🐜"));
+});
+
+describe("casteIcon", () => {
+  it("scout", () => expect(casteIcon("scout")).toBe("🔍"));
+  it("soldier", () => expect(casteIcon("soldier")).toBe("🛡️"));
+  it("drone", () => expect(casteIcon("drone")).toBe("⚙️"));
+  it("worker", () => expect(casteIcon("worker")).toBe("⚒️"));
+  it("unknown", () => expect(casteIcon("xyz")).toBe("⚒️"));
+});
+
+describe("buildReport", () => {
+  it("builds report with goal, status, cost, tasks", () => {
+    const state: ColonyState = {
+      id: "c-1", goal: "Test goal", status: "done",
+      tasks: [
+        { id: "t1", parentId: null, title: "Task A", description: "", caste: "worker", status: "done", priority: 3, files: [], claimedBy: null, result: null, error: null, spawnedTasks: [], createdAt: 0, startedAt: 0, finishedAt: 1000 },
+        { id: "t2", parentId: null, title: "Task B", description: "", caste: "worker", status: "failed", priority: 3, files: [], claimedBy: null, result: null, error: "some error", spawnedTasks: [], createdAt: 0, startedAt: 0, finishedAt: 1000 },
+      ],
+      ants: [], pheromones: [],
+      concurrency: { current: 2, min: 1, max: 4, optimal: 3, history: [] },
+      metrics: { tasksTotal: 2, tasksDone: 1, tasksFailed: 1, antsSpawned: 2, totalCost: 0.05, totalTokens: 1000, startTime: 0, throughputHistory: [] },
+      maxCost: null, modelOverrides: {}, createdAt: 0, finishedAt: 5000,
+    };
+    const report = buildReport(state);
+    expect(report).toContain("Test goal");
+    expect(report).toContain("✅");
+    expect(report).toContain("$0.05");
+    expect(report).toContain("Task A");
+    expect(report).toContain("Task B");
+    expect(report).toContain("some error");
+  });
+});

package/pi-package/extensions/auto-update.test.ts

@@ -0,0 +1,15 @@
+import { describe, it, expect, vi } from "vitest";
+
+vi.mock("@mariozechner/pi-coding-agent", () => ({}));
+
+import { isNewer } from "./auto-update";
+
+describe("isNewer", () => {
+  it("patch newer", () => expect(isNewer("1.0.1", "1.0.0")).toBe(true));
+  it("minor newer", () => expect(isNewer("1.1.0", "1.0.0")).toBe(true));
+  it("major newer", () => expect(isNewer("2.0.0", "1.9.9")).toBe(true));
+  it("equal", () => expect(isNewer("1.0.0", "1.0.0")).toBe(false));
+  it("older", () => expect(isNewer("1.0.0", "1.0.1")).toBe(false));
+  it("0.1.70 > 0.1.69", () => expect(isNewer("0.1.70", "0.1.69")).toBe(true));
+  it("0.1.69 < 0.1.70", () => expect(isNewer("0.1.69", "0.1.70")).toBe(false));
+});

package/pi-package/extensions/auto-update.ts

@@ -39,7 +39,7 @@ function getCurrentVersion(): string | null {
   } catch { return null; }
 }
 
-function isNewer(latest: string, current: string): boolean {
+export function isNewer(latest: string, current: string): boolean {
   const a = latest.split(".").map(Number);
   const b = current.split(".").map(Number);
   for (let i = 0; i < 3; i++) {

package/pi-package/extensions/safe-guard.test.ts

@@ -0,0 +1,26 @@
+import { describe, it, expect } from "vitest";
+import { DANGEROUS_PATTERNS, PROTECTED_PATHS } from "./safe-guard";
+
+describe("DANGEROUS_PATTERNS", () => {
+  const matches = (cmd: string) => DANGEROUS_PATTERNS.some((p) => p.test(cmd));
+
+  it("matches rm -rf /", () => expect(matches("rm -rf /")).toBe(true));
+  it("matches rm -f file.txt", () => expect(matches("rm -f file.txt")).toBe(true));
+  it("matches rm --force file", () => expect(matches("rm --force file")).toBe(true));
+  it("matches sudo rm file", () => expect(matches("sudo rm file")).toBe(true));
+  it("matches DROP TABLE users", () => expect(matches("DROP TABLE users")).toBe(true));
+  it("matches TRUNCATE TABLE foo", () => expect(matches("TRUNCATE TABLE foo")).toBe(true));
+  it("matches DELETE FROM users", () => expect(matches("DELETE FROM users")).toBe(true));
+  it("matches chmod 777 /tmp", () => expect(matches("chmod 777 /tmp")).toBe(true));
+  it("matches mkfs /dev/sda", () => expect(matches("mkfs /dev/sda")).toBe(true));
+  it("matches dd if=/dev/zero", () => expect(matches("dd if=/dev/zero")).toBe(true));
+  it("does not match ls -la", () => expect(matches("ls -la")).toBe(false));
+  it("does not match echo hello", () => expect(matches("echo hello")).toBe(false));
+});
+
+describe("PROTECTED_PATHS", () => {
+  it("contains all expected paths", () => {
+    expect(PROTECTED_PATHS).toEqual([".env", ".git/", "node_modules/", ".pi/", "id_rsa", ".ssh/"]);
+  });
+  it("has length 6", () => expect(PROTECTED_PATHS).toHaveLength(6));
+});

package/pi-package/extensions/safe-guard.ts

@@ -5,7 +5,7 @@
  */
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 
-const DANGEROUS_PATTERNS = [
+export const DANGEROUS_PATTERNS = [
   /\brm\s+(-[a-zA-Z]*f[a-zA-Z]*\s+|.*-rf\b|.*--force\b)/,
   /\bsudo\s+rm\b/,
   /\b(DROP|TRUNCATE|DELETE\s+FROM)\b/i,
@@ -15,7 +15,7 @@ const DANGEROUS_PATTERNS = [
   />\s*\/dev\/sd[a-z]/,
 ];
 
-const PROTECTED_PATHS = [".env", ".git/", "node_modules/", ".pi/", "id_rsa", ".ssh/"];
+export const PROTECTED_PATHS = [".env", ".git/", "node_modules/", ".pi/", "id_rsa", ".ssh/"];
 
 export default function (pi: ExtensionAPI) {
   pi.on("tool_call", async (event, ctx) => {

package/pi-package/extensions/smart-compact.test.ts

@@ -0,0 +1,64 @@
+import { describe, it, expect } from "vitest";
+import { truncateText, compactContent } from "./smart-compact";
+
+const longMultiline = (lines: number, lineLen = 100) =>
+  Array.from({ length: lines }, (_, i) => "x".repeat(lineLen) + i).join("\n");
+
+describe("truncateText", () => {
+  it("short text returns unchanged", () => {
+    expect(truncateText("hello", 8000)).toBe("hello");
+  });
+
+  it("few lines but long chars returns unchanged", () => {
+    const text = "x".repeat(9000) + "\n" + "y".repeat(9000);
+    expect(truncateText(text, 8000)).toBe(text);
+  });
+
+  it("long text with many lines gets truncated", () => {
+    const text = longMultiline(200);
+    const result = truncateText(text, 8000);
+    expect(result).toContain("[...truncated");
+  });
+
+  it("preserves head and tail", () => {
+    const text = longMultiline(200);
+    const result = truncateText(text, 8000);
+    expect(result.startsWith(text.slice(0, 1500))).toBe(true);
+    expect(result.endsWith(text.slice(-2500))).toBe(true);
+  });
+
+  it("custom head/tail params work", () => {
+    const text = longMultiline(200);
+    const result = truncateText(text, 100, 50, 50);
+    expect(result).toContain("[...truncated");
+    expect(result.startsWith(text.slice(0, 50))).toBe(true);
+    expect(result.endsWith(text.slice(-50))).toBe(true);
+  });
+});
+
+describe("compactContent", () => {
+  it("short string returns unchanged", () => {
+    expect(compactContent("short")).toBe("short");
+  });
+
+  it("long string gets truncated", () => {
+    const text = longMultiline(200);
+    expect(compactContent(text)).toContain("[...truncated");
+  });
+
+  it("array text block gets truncated", () => {
+    const text = longMultiline(200);
+    const result = compactContent([{ type: "text", text }]);
+    expect(result[0].text).toContain("[...truncated");
+  });
+
+  it("array non-text block returned unchanged", () => {
+    const block = { type: "image", url: "x" };
+    expect(compactContent([block])).toEqual([block]);
+  });
+
+  it("non-array non-string returned as-is", () => {
+    expect(compactContent(42)).toBe(42);
+    expect(compactContent({ a: 1 })).toEqual({ a: 1 });
+  });
+});

package/pi-package/extensions/smart-compact.ts

@@ -12,7 +12,7 @@ const MAX_USER_BLOCK_CHARS = 12000;
 const KEEP_HEAD = 1500;
 const KEEP_TAIL = 2500;
 
-function truncateText(text: string, max: number, head = KEEP_HEAD, tail = KEEP_TAIL): string {
+export function truncateText(text: string, max: number, head = KEEP_HEAD, tail = KEEP_TAIL): string {
   if (text.length <= max) return text;
   const lines = text.split("\n");
   if (lines.length <= 10) return text; // 短文本不裁
@@ -22,7 +22,7 @@ function truncateText(text: string, max: number, head = KEEP_HEAD, tail = KEEP_T
   return `${headText}\n\n[...truncated ${removedLines} lines...]\n\n${tailText}`;
 }
 
-function compactContent(content: any): any {
+export function compactContent(content: any): any {
   if (typeof content === "string") {
     return truncateText(content, MAX_TOOL_OUTPUT_CHARS);
   }