@apholdings/jensen-code 0.0.3 → 0.0.5

package/examples/extensions/summarize.ts

@@ -1,196 +0,0 @@
-import { complete, getModel } from "@apholdings/jensen-ai";
-import type { ExtensionAPI, ExtensionCommandContext } from "@apholdings/jensen-code";
-import { DynamicBorder, getMarkdownTheme } from "@apholdings/jensen-code";
-import { Container, Markdown, matchesKey, Text } from "@apholdings/jensen-tui";
-
-type ContentBlock = {
-	type?: string;
-	text?: string;
-	name?: string;
-	arguments?: Record<string, unknown>;
-};
-
-type SessionEntry = {
-	type: string;
-	message?: {
-		role?: string;
-		content?: unknown;
-	};
-};
-
-const extractTextParts = (content: unknown): string[] => {
-	if (typeof content === "string") {
-		return [content];
-	}
-
-	if (!Array.isArray(content)) {
-		return [];
-	}
-
-	const textParts: string[] = [];
-	for (const part of content) {
-		if (!part || typeof part !== "object") {
-			continue;
-		}
-
-		const block = part as ContentBlock;
-		if (block.type === "text" && typeof block.text === "string") {
-			textParts.push(block.text);
-		}
-	}
-
-	return textParts;
-};
-
-const extractToolCallLines = (content: unknown): string[] => {
-	if (!Array.isArray(content)) {
-		return [];
-	}
-
-	const toolCalls: string[] = [];
-	for (const part of content) {
-		if (!part || typeof part !== "object") {
-			continue;
-		}
-
-		const block = part as ContentBlock;
-		if (block.type !== "toolCall" || typeof block.name !== "string") {
-			continue;
-		}
-
-		const args = block.arguments ?? {};
-		toolCalls.push(`Tool ${block.name} was called with args ${JSON.stringify(args)}`);
-	}
-
-	return toolCalls;
-};
-
-const buildConversationText = (entries: SessionEntry[]): string => {
-	const sections: string[] = [];
-
-	for (const entry of entries) {
-		if (entry.type !== "message" || !entry.message?.role) {
-			continue;
-		}
-
-		const role = entry.message.role;
-		const isUser = role === "user";
-		const isAssistant = role === "assistant";
-
-		if (!isUser && !isAssistant) {
-			continue;
-		}
-
-		const entryLines: string[] = [];
-		const textParts = extractTextParts(entry.message.content);
-		if (textParts.length > 0) {
-			const roleLabel = isUser ? "User" : "Assistant";
-			const messageText = textParts.join("\n").trim();
-			if (messageText.length > 0) {
-				entryLines.push(`${roleLabel}: ${messageText}`);
-			}
-		}
-
-		if (isAssistant) {
-			entryLines.push(...extractToolCallLines(entry.message.content));
-		}
-
-		if (entryLines.length > 0) {
-			sections.push(entryLines.join("\n"));
-		}
-	}
-
-	return sections.join("\n\n");
-};
-
-const buildSummaryPrompt = (conversationText: string): string =>
-	[
-		"Summarize this conversation so I can resume it later.",
-		"Include goals, key decisions, progress, open questions, and next steps.",
-		"Keep it concise and structured with headings.",
-		"",
-		"<conversation>",
-		conversationText,
-		"</conversation>",
-	].join("\n");
-
-const showSummaryUi = async (summary: string, ctx: ExtensionCommandContext) => {
-	if (!ctx.hasUI) {
-		return;
-	}
-
-	await ctx.ui.custom((_tui, theme, _kb, done) => {
-		const container = new Container();
-		const border = new DynamicBorder((s: string) => theme.fg("accent", s));
-		const mdTheme = getMarkdownTheme();
-
-		container.addChild(border);
-		container.addChild(new Text(theme.fg("accent", theme.bold("Conversation Summary")), 1, 0));
-		container.addChild(new Markdown(summary, 1, 1, mdTheme));
-		container.addChild(new Text(theme.fg("dim", "Press Enter or Esc to close"), 1, 0));
-		container.addChild(border);
-
-		return {
-			render: (width: number) => container.render(width),
-			invalidate: () => container.invalidate(),
-			handleInput: (data: string) => {
-				if (matchesKey(data, "enter") || matchesKey(data, "escape")) {
-					done(undefined);
-				}
-			},
-		};
-	});
-};
-
-export default function (pi: ExtensionAPI) {
-	pi.registerCommand("summarize", {
-		description: "Summarize the current conversation in a custom UI",
-		handler: async (_args, ctx) => {
-			const branch = ctx.sessionManager.getBranch();
-			const conversationText = buildConversationText(branch);
-
-			if (!conversationText.trim()) {
-				if (ctx.hasUI) {
-					ctx.ui.notify("No conversation text found", "warning");
-				}
-				return;
-			}
-
-			if (ctx.hasUI) {
-				ctx.ui.notify("Preparing summary...", "info");
-			}
-
-			const model = getModel("openai", "gpt-5.2");
-			if (!model && ctx.hasUI) {
-				ctx.ui.notify("Model openai/gpt-5.2 not found", "warning");
-			}
-
-			const apiKey = model ? await ctx.modelRegistry.getApiKey(model) : undefined;
-			if (!apiKey && ctx.hasUI) {
-				ctx.ui.notify("No API key for openai/gpt-5.2", "warning");
-			}
-
-			if (!model || !apiKey) {
-				return;
-			}
-
-			const summaryMessages = [
-				{
-					role: "user" as const,
-					content: [{ type: "text" as const, text: buildSummaryPrompt(conversationText) }],
-					timestamp: Date.now(),
-				},
-			];
-
-			const response = await complete(model, { messages: summaryMessages }, { apiKey, reasoningEffort: "high" });
-
-			const summary = response.content
-				.filter((c): c is { type: "text"; text: string } => c.type === "text")
-				.map((c) => c.text)
-				.join("\n");
-
-			await showSummaryUi(summary, ctx);
-		},
-	});
-}
-

package/examples/extensions/system-prompt-header.ts

@@ -1,18 +0,0 @@
-/**
- * Displays a status widget showing the system prompt length.
- *
- * Demonstrates ctx.getSystemPrompt() for accessing the effective system prompt.
- */
-import type { ExtensionAPI } from "@apholdings/jensen-code";
-
-export default function (pi: ExtensionAPI) {
-	pi.on("agent_start", (_event, ctx) => {
-		const prompt = ctx.getSystemPrompt();
-		ctx.ui.setStatus("system-prompt", `System: ${prompt.length} chars`);
-	});
-
-	pi.on("session_shutdown", (_event, ctx) => {
-		ctx.ui.setStatus("system-prompt", undefined);
-	});
-}
-

package/examples/extensions/timed-confirm.ts

@@ -1,71 +0,0 @@
-/**
- * Example extension demonstrating timed dialogs with live countdown.
- *
- * Commands:
- * - /timed - Shows confirm dialog that auto-cancels after 5 seconds with countdown
- * - /timed-select - Shows select dialog that auto-cancels after 10 seconds with countdown
- * - /timed-signal - Shows confirm using AbortSignal (manual approach)
- */
-
-import type { ExtensionAPI } from "@apholdings/jensen-code";
-
-export default function (pi: ExtensionAPI) {
-	// Simple approach: use timeout option (recommended)
-	pi.registerCommand("timed", {
-		description: "Show a timed confirmation dialog (auto-cancels in 5s with countdown)",
-		handler: async (_args, ctx) => {
-			const confirmed = await ctx.ui.confirm(
-				"Timed Confirmation",
-				"This dialog will auto-cancel in 5 seconds. Confirm?",
-				{ timeout: 5000 },
-			);
-
-			if (confirmed) {
-				ctx.ui.notify("Confirmed by user!", "info");
-			} else {
-				ctx.ui.notify("Cancelled or timed out", "info");
-			}
-		},
-	});
-
-	pi.registerCommand("timed-select", {
-		description: "Show a timed select dialog (auto-cancels in 10s with countdown)",
-		handler: async (_args, ctx) => {
-			const choice = await ctx.ui.select("Pick an option", ["Option A", "Option B", "Option C"], { timeout: 10000 });
-
-			if (choice) {
-				ctx.ui.notify(`Selected: ${choice}`, "info");
-			} else {
-				ctx.ui.notify("Selection cancelled or timed out", "info");
-			}
-		},
-	});
-
-	// Manual approach: use AbortSignal for more control
-	pi.registerCommand("timed-signal", {
-		description: "Show a timed confirm using AbortSignal (manual approach)",
-		handler: async (_args, ctx) => {
-			const controller = new AbortController();
-			const timeoutId = setTimeout(() => controller.abort(), 5000);
-
-			ctx.ui.notify("Dialog will auto-cancel in 5 seconds...", "info");
-
-			const confirmed = await ctx.ui.confirm(
-				"Timed Confirmation",
-				"This dialog will auto-cancel in 5 seconds. Confirm?",
-				{ signal: controller.signal },
-			);
-
-			clearTimeout(timeoutId);
-
-			if (confirmed) {
-				ctx.ui.notify("Confirmed by user!", "info");
-			} else if (controller.signal.aborted) {
-				ctx.ui.notify("Dialog timed out (auto-cancelled)", "warning");
-			} else {
-				ctx.ui.notify("Cancelled by user", "info");
-			}
-		},
-	});
-}
-

package/examples/extensions/titlebar-spinner.ts

@@ -1,59 +0,0 @@
-/**
- * Titlebar Spinner Extension
- *
- * Shows a braille spinner animation in the terminal title while the agent is working.
- * Uses `ctx.ui.setTitle()` to update the terminal title via the extension API.
- *
- * Usage:
- *   pi --extension examples/extensions/titlebar-spinner.ts
- */
-
-import path from "node:path";
-import type { ExtensionAPI, ExtensionContext } from "@apholdings/jensen-code";
-
-const BRAILLE_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
-
-function getBaseTitle(pi: ExtensionAPI): string {
-	const cwd = path.basename(process.cwd());
-	const session = pi.getSessionName();
-	return session ? `π - ${session} - ${cwd}` : `π - ${cwd}`;
-}
-
-export default function (pi: ExtensionAPI) {
-	let timer: ReturnType<typeof setInterval> | null = null;
-	let frameIndex = 0;
-
-	function stopAnimation(ctx: ExtensionContext) {
-		if (timer) {
-			clearInterval(timer);
-			timer = null;
-		}
-		frameIndex = 0;
-		ctx.ui.setTitle(getBaseTitle(pi));
-	}
-
-	function startAnimation(ctx: ExtensionContext) {
-		stopAnimation(ctx);
-		timer = setInterval(() => {
-			const frame = BRAILLE_FRAMES[frameIndex % BRAILLE_FRAMES.length];
-			const cwd = path.basename(process.cwd());
-			const session = pi.getSessionName();
-			const title = session ? `${frame} π - ${session} - ${cwd}` : `${frame} π - ${cwd}`;
-			ctx.ui.setTitle(title);
-			frameIndex++;
-		}, 80);
-	}
-
-	pi.on("agent_start", async (_event, ctx) => {
-		startAnimation(ctx);
-	});
-
-	pi.on("agent_end", async (_event, ctx) => {
-		stopAnimation(ctx);
-	});
-
-	pi.on("session_shutdown", async (_event, ctx) => {
-		stopAnimation(ctx);
-	});
-}
-

package/examples/extensions/todo.ts

@@ -1,300 +0,0 @@
-/**
- * Todo Extension - Demonstrates state management via session entries
- *
- * This extension:
- * - Registers a `todo` tool for the LLM to manage todos
- * - Registers a `/todos` command for users to view the list
- *
- * State is stored in tool result details (not external files), which allows
- * proper branching - when you branch, the todo state is automatically
- * correct for that point in history.
- */
-
-import { StringEnum } from "@apholdings/jensen-ai";
-import type { ExtensionAPI, ExtensionContext, Theme } from "@apholdings/jensen-code";
-import { matchesKey, Text, truncateToWidth } from "@apholdings/jensen-tui";
-import { Type } from "@sinclair/typebox";
-
-interface Todo {
-	id: number;
-	text: string;
-	done: boolean;
-}
-
-interface TodoDetails {
-	action: "list" | "add" | "toggle" | "clear";
-	todos: Todo[];
-	nextId: number;
-	error?: string;
-}
-
-const TodoParams = Type.Object({
-	action: StringEnum(["list", "add", "toggle", "clear"] as const),
-	text: Type.Optional(Type.String({ description: "Todo text (for add)" })),
-	id: Type.Optional(Type.Number({ description: "Todo ID (for toggle)" })),
-});
-
-/**
- * UI component for the /todos command
- */
-class TodoListComponent {
-	private todos: Todo[];
-	private theme: Theme;
-	private onClose: () => void;
-	private cachedWidth?: number;
-	private cachedLines?: string[];
-
-	constructor(todos: Todo[], theme: Theme, onClose: () => void) {
-		this.todos = todos;
-		this.theme = theme;
-		this.onClose = onClose;
-	}
-
-	handleInput(data: string): void {
-		if (matchesKey(data, "escape") || matchesKey(data, "ctrl+c")) {
-			this.onClose();
-		}
-	}
-
-	render(width: number): string[] {
-		if (this.cachedLines && this.cachedWidth === width) {
-			return this.cachedLines;
-		}
-
-		const lines: string[] = [];
-		const th = this.theme;
-
-		lines.push("");
-		const title = th.fg("accent", " Todos ");
-		const headerLine =
-			th.fg("borderMuted", "─".repeat(3)) + title + th.fg("borderMuted", "─".repeat(Math.max(0, width - 10)));
-		lines.push(truncateToWidth(headerLine, width));
-		lines.push("");
-
-		if (this.todos.length === 0) {
-			lines.push(truncateToWidth(`  ${th.fg("dim", "No todos yet. Ask the agent to add some!")}`, width));
-		} else {
-			const done = this.todos.filter((t) => t.done).length;
-			const total = this.todos.length;
-			lines.push(truncateToWidth(`  ${th.fg("muted", `${done}/${total} completed`)}`, width));
-			lines.push("");
-
-			for (const todo of this.todos) {
-				const check = todo.done ? th.fg("success", "✓") : th.fg("dim", "○");
-				const id = th.fg("accent", `#${todo.id}`);
-				const text = todo.done ? th.fg("dim", todo.text) : th.fg("text", todo.text);
-				lines.push(truncateToWidth(`  ${check} ${id} ${text}`, width));
-			}
-		}
-
-		lines.push("");
-		lines.push(truncateToWidth(`  ${th.fg("dim", "Press Escape to close")}`, width));
-		lines.push("");
-
-		this.cachedWidth = width;
-		this.cachedLines = lines;
-		return lines;
-	}
-
-	invalidate(): void {
-		this.cachedWidth = undefined;
-		this.cachedLines = undefined;
-	}
-}
-
-export default function (pi: ExtensionAPI) {
-	// In-memory state (reconstructed from session on load)
-	let todos: Todo[] = [];
-	let nextId = 1;
-
-	/**
-	 * Reconstruct state from session entries.
-	 * Scans tool results for this tool and applies them in order.
-	 */
-	const reconstructState = (ctx: ExtensionContext) => {
-		todos = [];
-		nextId = 1;
-
-		for (const entry of ctx.sessionManager.getBranch()) {
-			if (entry.type !== "message") continue;
-			const msg = entry.message;
-			if (msg.role !== "toolResult" || msg.toolName !== "todo") continue;
-
-			const details = msg.details as TodoDetails | undefined;
-			if (details) {
-				todos = details.todos;
-				nextId = details.nextId;
-			}
-		}
-	};
-
-	// Reconstruct state on session events
-	pi.on("session_start", async (_event, ctx) => reconstructState(ctx));
-	pi.on("session_switch", async (_event, ctx) => reconstructState(ctx));
-	pi.on("session_fork", async (_event, ctx) => reconstructState(ctx));
-	pi.on("session_tree", async (_event, ctx) => reconstructState(ctx));
-
-	// Register the todo tool for the LLM
-	pi.registerTool({
-		name: "todo",
-		label: "Todo",
-		description: "Manage a todo list. Actions: list, add (text), toggle (id), clear",
-		parameters: TodoParams,
-
-		async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
-			switch (params.action) {
-				case "list":
-					return {
-						content: [
-							{
-								type: "text",
-								text: todos.length
-									? todos.map((t) => `[${t.done ? "x" : " "}] #${t.id}: ${t.text}`).join("\n")
-									: "No todos",
-							},
-						],
-						details: { action: "list", todos: [...todos], nextId } as TodoDetails,
-					};
-
-				case "add": {
-					if (!params.text) {
-						return {
-							content: [{ type: "text", text: "Error: text required for add" }],
-							details: { action: "add", todos: [...todos], nextId, error: "text required" } as TodoDetails,
-						};
-					}
-					const newTodo: Todo = { id: nextId++, text: params.text, done: false };
-					todos.push(newTodo);
-					return {
-						content: [{ type: "text", text: `Added todo #${newTodo.id}: ${newTodo.text}` }],
-						details: { action: "add", todos: [...todos], nextId } as TodoDetails,
-					};
-				}
-
-				case "toggle": {
-					if (params.id === undefined) {
-						return {
-							content: [{ type: "text", text: "Error: id required for toggle" }],
-							details: { action: "toggle", todos: [...todos], nextId, error: "id required" } as TodoDetails,
-						};
-					}
-					const todo = todos.find((t) => t.id === params.id);
-					if (!todo) {
-						return {
-							content: [{ type: "text", text: `Todo #${params.id} not found` }],
-							details: {
-								action: "toggle",
-								todos: [...todos],
-								nextId,
-								error: `#${params.id} not found`,
-							} as TodoDetails,
-						};
-					}
-					todo.done = !todo.done;
-					return {
-						content: [{ type: "text", text: `Todo #${todo.id} ${todo.done ? "completed" : "uncompleted"}` }],
-						details: { action: "toggle", todos: [...todos], nextId } as TodoDetails,
-					};
-				}
-
-				case "clear": {
-					const count = todos.length;
-					todos = [];
-					nextId = 1;
-					return {
-						content: [{ type: "text", text: `Cleared ${count} todos` }],
-						details: { action: "clear", todos: [], nextId: 1 } as TodoDetails,
-					};
-				}
-
-				default:
-					return {
-						content: [{ type: "text", text: `Unknown action: ${params.action}` }],
-						details: {
-							action: "list",
-							todos: [...todos],
-							nextId,
-							error: `unknown action: ${params.action}`,
-						} as TodoDetails,
-					};
-			}
-		},
-
-		renderCall(args, theme) {
-			let text = theme.fg("toolTitle", theme.bold("todo ")) + theme.fg("muted", args.action);
-			if (args.text) text += ` ${theme.fg("dim", `"${args.text}"`)}`;
-			if (args.id !== undefined) text += ` ${theme.fg("accent", `#${args.id}`)}`;
-			return new Text(text, 0, 0);
-		},
-
-		renderResult(result, { expanded }, theme) {
-			const details = result.details as TodoDetails | undefined;
-			if (!details) {
-				const text = result.content[0];
-				return new Text(text?.type === "text" ? text.text : "", 0, 0);
-			}
-
-			if (details.error) {
-				return new Text(theme.fg("error", `Error: ${details.error}`), 0, 0);
-			}
-
-			const todoList = details.todos;
-
-			switch (details.action) {
-				case "list": {
-					if (todoList.length === 0) {
-						return new Text(theme.fg("dim", "No todos"), 0, 0);
-					}
-					let listText = theme.fg("muted", `${todoList.length} todo(s):`);
-					const display = expanded ? todoList : todoList.slice(0, 5);
-					for (const t of display) {
-						const check = t.done ? theme.fg("success", "✓") : theme.fg("dim", "○");
-						const itemText = t.done ? theme.fg("dim", t.text) : theme.fg("muted", t.text);
-						listText += `\n${check} ${theme.fg("accent", `#${t.id}`)} ${itemText}`;
-					}
-					if (!expanded && todoList.length > 5) {
-						listText += `\n${theme.fg("dim", `... ${todoList.length - 5} more`)}`;
-					}
-					return new Text(listText, 0, 0);
-				}
-
-				case "add": {
-					const added = todoList[todoList.length - 1];
-					return new Text(
-						theme.fg("success", "✓ Added ") +
-							theme.fg("accent", `#${added.id}`) +
-							" " +
-							theme.fg("muted", added.text),
-						0,
-						0,
-					);
-				}
-
-				case "toggle": {
-					const text = result.content[0];
-					const msg = text?.type === "text" ? text.text : "";
-					return new Text(theme.fg("success", "✓ ") + theme.fg("muted", msg), 0, 0);
-				}
-
-				case "clear":
-					return new Text(theme.fg("success", "✓ ") + theme.fg("muted", "Cleared all todos"), 0, 0);
-			}
-		},
-	});
-
-	// Register the /todos command for users
-	pi.registerCommand("todos", {
-		description: "Show all todos on the current branch",
-		handler: async (_args, ctx) => {
-			if (!ctx.hasUI) {
-				ctx.ui.notify("/todos requires interactive mode", "error");
-				return;
-			}
-
-			await ctx.ui.custom<void>((_tui, theme, _kb, done) => {
-				return new TodoListComponent(todos, theme, () => done());
-			});
-		},
-	});
-}
-