omegon 0.6.0

package/extensions/cleave/worktree.ts

@@ -0,0 +1,243 @@
+/**
+ * cleave/worktree — Git worktree operations for child isolation.
+ *
+ * Each child gets its own git worktree on a separate branch,
+ * preventing file conflicts during parallel execution.
+ * Completed branches are merged back to base after harvesting.
+ */
+
+import { mkdirSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import type { ExtensionAPI } from "@cwilson613/pi-coding-agent";
+
+/** Base directory for cleave worktrees. */
+const WORKTREE_HOME = join(homedir(), ".pi", "cleave", "wt");
+
+export interface WorktreeInfo {
+	path: string;
+	branch: string;
+}
+
+/**
+ * Get the current git branch name.
+ */
+export async function getCurrentBranch(
+	pi: ExtensionAPI,
+	repoPath: string,
+): Promise<string> {
+	const result = await pi.exec("git", ["branch", "--show-current"], {
+		cwd: repoPath,
+		timeout: 5_000,
+	});
+	const branch = result.stdout.trim();
+	if (!branch) {
+		// Detached HEAD — get the SHA
+		const sha = await pi.exec("git", ["rev-parse", "--short", "HEAD"], {
+			cwd: repoPath,
+			timeout: 5_000,
+		});
+		return sha.stdout.trim();
+	}
+	return branch;
+}
+
+/**
+ * Ensure the working tree is clean (no uncommitted changes).
+ * Throws if dirty.
+ */
+export async function ensureCleanWorktree(
+	pi: ExtensionAPI,
+	repoPath: string,
+): Promise<void> {
+	const result = await pi.exec("git", ["status", "--porcelain"], {
+		cwd: repoPath,
+		timeout: 5_000,
+	});
+	if (result.stdout.trim()) {
+		throw new Error(
+			"Working tree has uncommitted changes. Commit or stash before cleaving.\n" +
+			result.stdout.trim(),
+		);
+	}
+}
+
+/**
+ * Create a git worktree for a child task.
+ *
+ * Creates a new branch from baseBranch and checks it out in a
+ * worktree directory adjacent to the repo.
+ */
+export async function createWorktree(
+	pi: ExtensionAPI,
+	repoPath: string,
+	childLabel: string,
+	childId: number,
+	baseBranch: string,
+): Promise<WorktreeInfo> {
+	// Sanitize childLabel to prevent path traversal via / or ..
+	const safeLabel = childLabel.replace(/[^a-zA-Z0-9_-]/g, "-").replace(/^\.+/, "") || "task";
+	const branch = `cleave/${childId}-${safeLabel}`;
+	// Worktree goes in ~/.pi/cleave/wt/ to avoid polluting the repo or its parent
+	mkdirSync(WORKTREE_HOME, { recursive: true });
+	const worktreePath = join(WORKTREE_HOME, `${childId}-${safeLabel}`);
+
+	// Delete branch if it already exists (leftover from a previous run)
+	await pi.exec("git", ["branch", "-D", branch], {
+		cwd: repoPath,
+		timeout: 5_000,
+	}).catch(() => {}); // ignore if it doesn't exist
+
+	// Remove stale worktree path if it exists
+	await pi.exec("rm", ["-rf", worktreePath], { timeout: 5_000 }).catch(() => {});
+
+	// Create worktree with new branch from base
+	const result = await pi.exec(
+		"git",
+		["worktree", "add", "-b", branch, worktreePath, baseBranch],
+		{ cwd: repoPath, timeout: 30_000 },
+	);
+
+	if (result.code !== 0) {
+		throw new Error(`Failed to create worktree: ${result.stderr}`);
+	}
+
+	return { path: worktreePath, branch };
+}
+
+/**
+ * Merge a child's branch back into the base branch.
+ *
+ * Returns { success, conflictFiles } — does NOT abort on conflict,
+ * leaving the merge state for manual resolution.
+ */
+export async function mergeBranch(
+	pi: ExtensionAPI,
+	repoPath: string,
+	childBranch: string,
+	baseBranch: string,
+): Promise<{ success: boolean; conflictFiles: string[]; error?: string }> {
+	// Checkout base branch
+	let result = await pi.exec("git", ["checkout", baseBranch], {
+		cwd: repoPath,
+		timeout: 10_000,
+	});
+	if (result.code !== 0) {
+		return { success: false, conflictFiles: [], error: `Failed to checkout ${baseBranch}: ${result.stderr}` };
+	}
+
+	// Attempt merge
+	result = await pi.exec("git", ["merge", "--no-ff", childBranch, "-m", `merge: cleave child ${childBranch}`], {
+		cwd: repoPath,
+		timeout: 30_000,
+	});
+
+	if (result.code === 0) {
+		return { success: true, conflictFiles: [] };
+	}
+
+	// Merge conflict — detect which files
+	const statusResult = await pi.exec("git", ["diff", "--name-only", "--diff-filter=U"], {
+		cwd: repoPath,
+		timeout: 5_000,
+	});
+	const conflictFiles = statusResult.stdout.trim().split("\n").filter(Boolean);
+
+	// Abort the merge to leave repo in clean state
+	await pi.exec("git", ["merge", "--abort"], { cwd: repoPath, timeout: 5_000 }).catch(() => {});
+
+	return {
+		success: false,
+		conflictFiles,
+		error: `Merge conflict in ${conflictFiles.length} file(s)`,
+	};
+}
+
+/**
+ * Remove cleave worktree directories without deleting branches.
+ * Used on merge failure to free disk space while preserving branches
+ * for manual conflict resolution.
+ */
+export async function pruneWorktreeDirs(
+	pi: ExtensionAPI,
+	repoPath: string,
+): Promise<void> {
+	const result = await pi.exec("git", ["worktree", "list", "--porcelain"], {
+		cwd: repoPath,
+		timeout: 5_000,
+	});
+
+	for (const line of result.stdout.split("\n")) {
+		if (line.startsWith("worktree ")) {
+			const wtPath = line.replace("worktree ", "").trim();
+			if (wtPath.startsWith(WORKTREE_HOME) || wtPath.includes(".cleave-wt-")) {
+				await pi.exec("git", ["worktree", "remove", "--force", wtPath], {
+					cwd: repoPath,
+					timeout: 10_000,
+				}).catch(() => {});
+			}
+		}
+	}
+
+	await pi.exec("git", ["worktree", "prune"], {
+		cwd: repoPath,
+		timeout: 5_000,
+	}).catch(() => {});
+}
+
+/**
+ * Clean up all cleave worktrees and their branches.
+ */
+export async function cleanupWorktrees(
+	pi: ExtensionAPI,
+	repoPath: string,
+): Promise<void> {
+	// List worktrees
+	const result = await pi.exec("git", ["worktree", "list", "--porcelain"], {
+		cwd: repoPath,
+		timeout: 5_000,
+	});
+
+	const lines = result.stdout.split("\n");
+	const worktreePaths: string[] = [];
+
+	for (const line of lines) {
+		if (line.startsWith("worktree ")) {
+			const wtPath = line.replace("worktree ", "").trim();
+			// Match worktrees in the cleave wt directory or legacy .cleave-wt- locations
+			if (wtPath.startsWith(WORKTREE_HOME) || wtPath.includes(".cleave-wt-")) {
+				worktreePaths.push(wtPath);
+			}
+		}
+	}
+
+	// Remove each worktree
+	for (const wtPath of worktreePaths) {
+		await pi.exec("git", ["worktree", "remove", "--force", wtPath], {
+			cwd: repoPath,
+			timeout: 10_000,
+		}).catch(() => {});
+	}
+
+	// Prune stale worktree references
+	await pi.exec("git", ["worktree", "prune"], {
+		cwd: repoPath,
+		timeout: 5_000,
+	}).catch(() => {});
+
+	// Delete orphaned cleave/* branches
+	const branchResult = await pi.exec(
+		"git", ["branch", "--list", "cleave/*"],
+		{ cwd: repoPath, timeout: 5_000 },
+	);
+	const branches = branchResult.stdout
+		.split("\n")
+		.map((b) => b.trim().replace(/^\*\s*/, ""))
+		.filter(Boolean);
+	for (const branch of branches) {
+		await pi.exec("git", ["branch", "-D", branch], {
+			cwd: repoPath,
+			timeout: 5_000,
+		}).catch(() => {});
+	}
+}

package/extensions/core-renderers.ts

@@ -0,0 +1,253 @@
+/**
+ * Core Tool Renderers — Sci-UI rendering for pi built-in tools.
+ *
+ * Uses registerToolRenderer() to attach renderCall/renderResult
+ * to built-in tools (bash, read, edit, write) without replacing them.
+ *
+ * The built-in renderers handle syntax highlighting, diffs, and streaming —
+ * we only override the COLLAPSED view to match the Sci-UI visual language.
+ * Expanded views fall through to the built-in renderer.
+ */
+import type { ExtensionAPI } from "@cwilson613/pi-coding-agent";
+import { sciCall, sciOk, sciErr, sciLoading } from "./sci-ui.ts";
+
+/** Shorten a file path for display — keep last 2-3 segments. */
+function shortenPath(p: string | null | undefined, maxLen = 55): string {
+	if (!p) return "…";
+	if (p.length <= maxLen) return p;
+	const parts = p.split("/");
+	// Show last 3 segments at most
+	const tail = parts.slice(-3).join("/");
+	return tail.length <= maxLen ? tail : "…" + p.slice(-(maxLen - 1));
+}
+
+export default function coreRenderers(pi: ExtensionAPI): void {
+	// ── Read ──────────────────────────────────────────────────────────────
+	pi.registerToolRenderer("read", {
+		renderCall(args: any, theme: any) {
+			const p = shortenPath(args?.file_path ?? args?.path);
+			let range = "";
+			if (args?.offset != null || args?.limit != null) {
+				const start = args.offset ?? 1;
+				const end = args.limit != null ? start + args.limit - 1 : "";
+				range = `:${start}${end ? `-${end}` : ""}`;
+			}
+			return sciCall("read", `${p}${range}`, theme);
+		},
+		// renderResult omitted — built-in handles syntax highlighting + truncation
+	});
+
+	// ── Edit ──────────────────────────────────────────────────────────────
+	pi.registerToolRenderer("edit", {
+		renderCall(args: any, theme: any) {
+			const p = shortenPath(args?.file_path ?? args?.path);
+			// Show the size of the change: lines changed
+			const oldLines = (args?.old_text ?? args?.oldText ?? "").split("\n").length;
+			const newLines = (args?.new_text ?? args?.newText ?? "").split("\n").length;
+			const delta = newLines - oldLines;
+			const deltaStr = delta === 0 ? `${oldLines}L` : delta > 0 ? `+${delta}L` : `${delta}L`;
+			return sciCall("edit", `${p} (${deltaStr})`, theme);
+		},
+		// renderResult omitted — built-in handles diff rendering
+	});
+
+	// ── Write ─────────────────────────────────────────────────────────────
+	pi.registerToolRenderer("write", {
+		renderCall(args: any, theme: any) {
+			const p = shortenPath(args?.file_path ?? args?.path);
+			const content = args?.content ?? "";
+			const lines = content.split("\n").length;
+			return sciCall("write", `${p} (${lines}L)`, theme);
+		},
+		// renderResult omitted — built-in handles syntax highlighting
+	});
+
+	// ── Bash ──────────────────────────────────────────────────────────────
+	pi.registerToolRenderer("bash", {
+		renderCall(args: any, theme: any) {
+			const cmd = args?.command ?? "";
+			// Truncate long commands
+			const display = cmd.length > 70 ? cmd.slice(0, 67) + "…" : cmd;
+			return sciCall("bash", display, theme);
+		},
+		// renderResult omitted — built-in handles output display + truncation
+	});
+
+	// ── Grep ──────────────────────────────────────────────────────────────
+	pi.registerToolRenderer("grep", {
+		renderCall(args: any, theme: any) {
+			const pattern = args?.pattern ?? "";
+			const p = shortenPath(args?.path);
+			const glob = args?.glob ? ` (${args.glob})` : "";
+			return sciCall("grep", `/${pattern}/ in ${p}${glob}`, theme);
+		},
+	});
+
+	// ── Find ──────────────────────────────────────────────────────────────
+	pi.registerToolRenderer("find", {
+		renderCall(args: any, theme: any) {
+			const pattern = args?.pattern ?? "";
+			const p = shortenPath(args?.path);
+			return sciCall("find", `${pattern} in ${p}`, theme);
+		},
+	});
+
+	// ── Ls ─────────────────────────────────────────────────────────────────
+	pi.registerToolRenderer("ls", {
+		renderCall(args: any, theme: any) {
+			const p = shortenPath(args?.path || ".");
+			return sciCall("ls", p, theme);
+		},
+	});
+
+	// ── View ──────────────────────────────────────────────────────────────
+	pi.registerToolRenderer("view", {
+		renderCall(args: any, theme: any) {
+			const p = shortenPath(args?.path);
+			const page = args?.page ? ` p${args.page}` : "";
+			return sciCall("view", `${p}${page}`, theme);
+		},
+	});
+
+	// ── Web Search ────────────────────────────────────────────────────────
+	pi.registerToolRenderer("web_search", {
+		renderCall(args: any, theme: any) {
+			const query = args?.query ?? "";
+			const mode = args?.mode ?? "quick";
+			const display = query.length > 55 ? query.slice(0, 52) + "…" : query;
+			const modeTag = mode !== "quick" ? ` [${mode}]` : "";
+			return sciCall("web_search", `${display}${modeTag}`, theme);
+		},
+	});
+
+	// ── Chronos ───────────────────────────────────────────────────────────
+	pi.registerToolRenderer("chronos", {
+		renderCall(args: any, theme: any) {
+			const sub = args?.subcommand ?? "week";
+			const expr = args?.expression ? ` "${args.expression}"` : "";
+			return sciCall("chronos", `${sub}${expr}`, theme);
+		},
+	});
+
+	// ── Render Diagram (D2) ───────────────────────────────────────────────
+	pi.registerToolRenderer("render_diagram", {
+		renderCall(args: any, theme: any) {
+			const title = args?.title ?? "diagram";
+			return sciCall("render_diagram", title, theme);
+		},
+	});
+
+	// ── Render Native Diagram ─────────────────────────────────────────────
+	pi.registerToolRenderer("render_native_diagram", {
+		renderCall(args: any, theme: any) {
+			const title = args?.title ?? "diagram";
+			return sciCall("render_native_diagram", title, theme);
+		},
+	});
+
+	// ── Render Excalidraw ─────────────────────────────────────────────────
+	pi.registerToolRenderer("render_excalidraw", {
+		renderCall(args: any, theme: any) {
+			const p = shortenPath(args?.path);
+			return sciCall("render_excalidraw", p, theme);
+		},
+	});
+
+	// ── Generate Image Local ──────────────────────────────────────────────
+	pi.registerToolRenderer("generate_image_local", {
+		renderCall(args: any, theme: any) {
+			const prompt = args?.prompt ?? "";
+			const preset = args?.preset ?? "schnell";
+			const display = prompt.length > 50 ? prompt.slice(0, 47) + "…" : prompt;
+			return sciCall("generate_image_local", `${display} [${preset}]`, theme);
+		},
+	});
+
+	// ── Render Composition Still ──────────────────────────────────────────
+	pi.registerToolRenderer("render_composition_still", {
+		renderCall(args: any, theme: any) {
+			const p = shortenPath(args?.composition_path);
+			const frame = args?.frame != null ? ` f${args.frame}` : "";
+			return sciCall("render_composition_still", `${p}${frame}`, theme);
+		},
+	});
+
+	// ── Render Composition Video ──────────────────────────────────────────
+	pi.registerToolRenderer("render_composition_video", {
+		renderCall(args: any, theme: any) {
+			const p = shortenPath(args?.composition_path);
+			const frames = args?.duration_in_frames ?? "?";
+			const fmt = args?.format ?? "gif";
+			return sciCall("render_composition_video", `${p} (${frames}f, ${fmt})`, theme);
+		},
+	});
+
+	// ── Model Tier ────────────────────────────────────────────────────────
+	pi.registerToolRenderer("set_model_tier", {
+		renderCall(args: any, theme: any) {
+			const tier = args?.tier ?? "?";
+			return sciCall("set_model_tier", `→ ${tier}`, theme);
+		},
+	});
+
+	// ── Thinking Level ────────────────────────────────────────────────────
+	pi.registerToolRenderer("set_thinking_level", {
+		renderCall(args: any, theme: any) {
+			const level = args?.level ?? "?";
+			return sciCall("set_thinking_level", `→ ${level}`, theme);
+		},
+	});
+
+	// ── Ask Local Model ───────────────────────────────────────────────────
+	pi.registerToolRenderer("ask_local_model", {
+		renderCall(args: any, theme: any) {
+			const model = args?.model ?? "auto";
+			const prompt = args?.prompt ?? "";
+			const display = prompt.length > 45 ? prompt.slice(0, 42) + "…" : prompt;
+			return sciCall("ask_local_model", `[${model}] ${display}`, theme);
+		},
+	});
+
+	// ── Manage Ollama ─────────────────────────────────────────────────────
+	pi.registerToolRenderer("manage_ollama", {
+		renderCall(args: any, theme: any) {
+			const action = args?.action ?? "?";
+			const model = args?.model ? ` ${args.model}` : "";
+			return sciCall("manage_ollama", `${action}${model}`, theme);
+		},
+	});
+
+	// ── List Local Models ─────────────────────────────────────────────────
+	pi.registerToolRenderer("list_local_models", {
+		renderCall(_args: any, theme: any) {
+			return sciCall("list_local_models", "inventory", theme);
+		},
+	});
+
+	// ── Switch Offline Driver ─────────────────────────────────────────────
+	pi.registerToolRenderer("switch_to_offline_driver", {
+		renderCall(args: any, theme: any) {
+			const reason = args?.reason ?? "";
+			const display = reason.length > 50 ? reason.slice(0, 47) + "…" : reason;
+			return sciCall("switch_to_offline_driver", display, theme);
+		},
+	});
+
+	// ── Manage Tools ──────────────────────────────────────────────────────
+	pi.registerToolRenderer("manage_tools", {
+		renderCall(args: any, theme: any) {
+			const action = args?.action ?? "list";
+			const tools = args?.tools?.join(", ") ?? "";
+			const profile = args?.profile ?? "";
+			const detail = tools || profile || "";
+			return sciCall("manage_tools", `${action}${detail ? ` ${detail}` : ""}`, theme);
+		},
+	});
+
+	// ── Whoami ────────────────────────────────────────────────────────────
+	pi.registerToolRenderer("whoami", {
+		renderCall(_args: any, theme: any) {
+			return sciCall("whoami", "check auth", theme);
+		},
+	});
+}

package/extensions/dashboard/context-gauge.ts

@@ -0,0 +1,58 @@
+import type { MemoryInjectionMetrics } from "../project-memory/injection-metrics.ts";
+
+export interface ContextGaugeInput {
+  percent: number | null | undefined;
+  contextWindow: number;
+  memoryTokenEstimate: number;
+  turns: number;
+}
+
+export interface ContextGaugeModel {
+  state: "known" | "unknown";
+  turns: number;
+  contextWindow: number;
+  percent: number | null;
+  memoryPercent: number;
+  otherPercent: number;
+  memoryBlocks: number;
+  otherBlocks: number;
+  freeBlocks: number;
+}
+
+export function buildContextGaugeModel(input: ContextGaugeInput, barWidth: number): ContextGaugeModel {
+  const contextWindow = input.contextWindow;
+  const pct = input.percent ?? null;
+
+  if (pct === null) {
+    return {
+      state: "unknown",
+      turns: input.turns,
+      contextWindow,
+      percent: null,
+      memoryPercent: 0,
+      otherPercent: 0,
+      memoryBlocks: 0,
+      otherBlocks: 0,
+      freeBlocks: barWidth,
+    };
+  }
+
+  const memoryPercent = contextWindow > 0 ? (input.memoryTokenEstimate / contextWindow) * 100 : 0;
+  const otherPercent = Math.max(0, pct - memoryPercent);
+  const memoryBlocks = memoryPercent > 0 ? Math.ceil((memoryPercent / 100) * barWidth) : 0;
+  const otherBlocks = otherPercent > 0 ? Math.ceil((otherPercent / 100) * barWidth) : 0;
+  const totalFilled = Math.min(memoryBlocks + otherBlocks, barWidth);
+  const freeBlocks = barWidth - totalFilled;
+
+  return {
+    state: "known",
+    turns: input.turns,
+    contextWindow,
+    percent: pct,
+    memoryPercent,
+    otherPercent,
+    memoryBlocks,
+    otherBlocks,
+    freeBlocks,
+  };
+}

package/extensions/dashboard/file-watch.ts

@@ -0,0 +1,14 @@
+import * as path from "node:path";
+
+export function shouldRefreshDesignTreeForPath(filePath: string, docsDir: string): boolean {
+  const rel = path.relative(docsDir, filePath);
+  if (!rel || rel.startsWith("..") || path.isAbsolute(rel)) return false;
+  return rel.endsWith(".md");
+}
+
+export function shouldRefreshOpenSpecForPath(filePath: string, repoRoot: string): boolean {
+  const openspecDir = path.join(repoRoot, "openspec");
+  const rel = path.relative(openspecDir, filePath);
+  if (!rel || rel.startsWith("..") || path.isAbsolute(rel)) return false;
+  return rel.endsWith(".md");
+}