@bacnh85/pi-subagent 0.5.0 → 0.7.0

package/extensions/thread-viewer.ts

@@ -7,134 +7,10 @@
 
 import { getMarkdownTheme } from "@earendil-works/pi-coding-agent";
 import { matchesKey, Key, truncateToWidth, Markdown } from "@earendil-works/pi-tui";
-import type { Message } from "@earendil-works/pi-ai";
 
-import { type SubAgentResult, isFailedResult, getResultOutput, getFinalOutput } from "./runner.ts";
-import { formatUsageStats } from "./render.ts";
+import { isFailedResult, getFinalOutput } from "./runner.ts";
+import { formatToolCall, getDisplayItems, formatUsageStats } from "./render.ts";
 import type { SubagentThread } from "./threads.ts";
-import * as os from "node:os";
-
-// ---------------------------------------------------------------------------
-// Safe type guards for tool-call arguments
-// ---------------------------------------------------------------------------
-
-function asString(value: unknown, fallback = "..."): string {
-	return typeof value === "string" ? value : fallback;
-}
-
-function asNumber(value: unknown): number | undefined {
-	return typeof value === "number" && Number.isFinite(value) ? value : undefined;
-}
-
-function asRecord(value: unknown): Record<string, unknown> {
-	return value && typeof value === "object" && !Array.isArray(value)
-		? (value as Record<string, unknown>)
-		: {};
-}
-
-// ---------------------------------------------------------------------------
-// Tool-call formatting (same as render.ts)
-// ---------------------------------------------------------------------------
-
-function formatToolCall(
-	toolName: string,
-	args: Record<string, unknown>,
-	themeFg: (color: string, text: string) => string,
-): string {
-	const shortenPath = (p: string) => {
-		const home = os.homedir();
-		return p.startsWith(home) ? `~${p.slice(home.length)}` : p;
-	};
-
-	switch (toolName) {
-		case "bash": {
-			const command = asString(args.command);
-			const preview = command.length > 60 ? `${command.slice(0, 60)}...` : command;
-			return themeFg("muted", "$ ") + themeFg("toolOutput", preview);
-		}
-		case "read": {
-			const rawPath = asString(args.file_path ?? args.path);
-			const filePath = shortenPath(rawPath);
-			const offset = asNumber(args.offset);
-			const limit = asNumber(args.limit);
-			let text = themeFg("accent", filePath);
-			if (offset !== undefined || limit !== undefined) {
-				const startLine = offset ?? 1;
-				const endLine = limit !== undefined ? startLine + limit - 1 : "";
-				text += themeFg("warning", `:${startLine}${endLine ? `-${endLine}` : ""}`);
-			}
-			return themeFg("muted", "read ") + text;
-		}
-		case "write": {
-			const rawPath = asString(args.file_path ?? args.path);
-			const filePath = shortenPath(rawPath);
-			const content = asString(args.content, "");
-			const lines = content.split("\n").length;
-			let text = themeFg("muted", "write ") + themeFg("accent", filePath);
-			if (lines > 1) text += themeFg("dim", ` (${lines} lines)`);
-			return text;
-		}
-		case "edit": {
-			const rawPath = asString(args.file_path ?? args.path);
-			return themeFg("muted", "edit ") + themeFg("accent", shortenPath(rawPath));
-		}
-		case "ls": {
-			const rawPath = asString(args.path, ".");
-			return themeFg("muted", "ls ") + themeFg("accent", shortenPath(rawPath));
-		}
-		case "find": {
-			const pattern = asString(args.pattern, "*");
-			const rawPath = asString(args.path, ".");
-			return (
-				themeFg("muted", "find ") +
-				themeFg("accent", pattern) +
-				themeFg("dim", ` in ${shortenPath(rawPath)}`)
-			);
-		}
-		case "grep": {
-			const pattern = asString(args.pattern);
-			const rawPath = asString(args.path, ".");
-			return (
-				themeFg("muted", "grep ") +
-				themeFg("accent", `/${pattern}/`) +
-				themeFg("dim", ` in ${shortenPath(rawPath)}`)
-			);
-		}
-		default: {
-			const argsStr = JSON.stringify(args);
-			const preview = argsStr.length > 50 ? `${argsStr.slice(0, 50)}...` : argsStr;
-			return themeFg("accent", toolName) + themeFg("dim", ` ${preview}`);
-		}
-	}
-}
-
-// ---------------------------------------------------------------------------
-// Display helpers
-// ---------------------------------------------------------------------------
-
-type DisplayItem =
-	| { type: "text"; text: string }
-	| { type: "toolCall"; name: string; args: Record<string, unknown> };
-
-function getDisplayItems(messages: Message[]): DisplayItem[] {
-	const items: DisplayItem[] = [];
-	for (const msg of messages) {
-		if (msg.role === "assistant") {
-			for (const part of msg.content) {
-				if (part.type === "text" && part.text.trim()) {
-					items.push({ type: "text", text: part.text });
-				} else if (part.type === "toolCall") {
-					items.push({
-						type: "toolCall",
-						name: part.name,
-						args: asRecord(part.arguments),
-					});
-				}
-			}
-		}
-	}
-	return items;
-}
 
 // ---------------------------------------------------------------------------
 // Theme types
@@ -247,7 +123,8 @@ export class ThreadViewer {
 		else if (this.thread.mode === "chain-step") modeLabel = t.fg("muted", " [chain]");
 
 		// Header
-		let header = `${icon} ${t.fg("toolTitle", t.bold(this.thread.agentName))}${modeLabel}`;
+		const agentColor = this.thread.color ?? "accent";
+		let header = `${icon} ${t.fg(agentColor, t.bold(this.thread.agentName))}${modeLabel}`;
 		if (status === "running") header += ` ${t.fg("warning", "(running...)")}`;
 		else if (status === "aborted") header += ` ${t.fg("error", "[aborted]")}`;
 		if (result && isErr && result.stopReason && result.stopReason !== "error" && result.stopReason !== "aborted") {

package/extensions/threads.ts

@@ -23,6 +23,7 @@ export interface SubagentThread {
 	status: ThreadStatus;
 	result?: SubAgentResult;
 	toolCallId?: string;
+	color?: string;
 	createdAt: number;
 	updatedAt: number;
 }
@@ -53,6 +54,7 @@ export class ThreadStore {
 		task: string;
 		mode: ThreadMode;
 		toolCallId?: string;
+		color?: string;
 	}): SubagentThread {
 		const id = cryptoGenId();
 		const now = Date.now();
@@ -63,6 +65,7 @@ export class ThreadStore {
 			mode: params.mode,
 			status: "running",
 			toolCallId: params.toolCallId,
+			color: params.color,
 			createdAt: now,
 			updatedAt: now,
 		};
@@ -103,14 +106,4 @@ export const threadStore = new ThreadStore();
 // ID generation (UUID v4 style, good enough for in-memory use)
 // ---------------------------------------------------------------------------
 
-function cryptoGenId(): string {
-	if (typeof crypto !== "undefined" && crypto.randomUUID) {
-		return crypto.randomUUID();
-	}
-	// Fallback for environments without crypto (very unlikely in Node)
-	return "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, (c) => {
-		const r = (Math.random() * 16) | 0;
-		const v = c === "x" ? r : (r & 0x3) | 0x8;
-		return v.toString(16);
-	});
-}
+const cryptoGenId = () => crypto.randomUUID();

package/package.json

@@ -1,21 +1,31 @@
 {
   "name": "@bacnh85/pi-subagent",
-  "version": "0.5.0",
-  "description": "Minimal-overhead sub-agents for Pi with isolated SDK sessions, parallel and chained delegation, and inspectable threads.",
+  "version": "0.7.0",
+  "description": "In-process subagents for Pi with isolated SDK sessions, parallel and chained delegation, and inspectable threads.",
   "type": "module",
   "license": "MIT",
   "publishConfig": { "access": "public" },
-  "homepage": "https://github.com/bacnh85/pi-extensions#readme",
+  "homepage": "https://github.com/bacnh85/pi-extensions/tree/main/pi-subagent",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/bacnh85/pi-extensions.git",
     "directory": "pi-subagent"
   },
   "bugs": { "url": "https://github.com/bacnh85/pi-extensions/issues" },
-  "keywords": ["pi-package", "pi-extension", "subagent", "sub-agent", "delegation", "parallel"],
+  "keywords": [
+    "pi-package",
+    "pi-extension",
+    "subagent",
+    "sub-agent",
+    "delegation",
+    "parallel",
+    "chain",
+    "security"
+  ],
   "files": [
     "README.md",
     "agent-format.md",
+    "CHANGELOG.md",
     "agents/",
     "extensions/index.ts",
     "extensions/agents.ts",
@@ -25,19 +35,24 @@
     "extensions/render.ts",
     "extensions/threads.ts",
     "extensions/thread-viewer.ts",
+    "extensions/security.ts",
     "extensions/package.json"
   ],
   "pi": { "extensions": ["./extensions/index.ts"] },
+  "engines": {
+    "node": ">=20.18"
+  },
   "scripts": {
-    "test": "cd extensions && npx mocha",
-    "typecheck": "tsc --noEmit"
+    "test": "cd extensions && mocha",
+    "typecheck": "tsc --noEmit",
+    "check": "npm run typecheck && npm test"
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "*",
-    "@earendil-works/pi-ai": "*",
-    "@earendil-works/pi-agent-core": "*",
-    "@earendil-works/pi-tui": "*",
-    "typebox": "*"
+    "@earendil-works/pi-coding-agent": ">=0.80.0 <0.81.0",
+    "@earendil-works/pi-ai": ">=0.80.0 <0.81.0",
+    "@earendil-works/pi-agent-core": ">=0.80.0 <0.81.0",
+    "@earendil-works/pi-tui": ">=0.80.0 <0.81.0",
+    "typebox": ">=1.3.0 <2.0.0"
   },
   "devDependencies": {
     "@earendil-works/pi-agent-core": "^0.80.2",