More browser extension work. Old interface fully ported. Direct transport. Small UX fixes.

packages/browser-extension/src/tools/index.ts

@@ -0,0 +1,38 @@
+import { html, type TemplateResult } from "@mariozechner/mini-lit";
+import type { ToolResultMessage } from "@mariozechner/pi-ai";
+import { getToolRenderer, registerToolRenderer } from "./renderer-registry.js";
+import { BashRenderer } from "./renderers/BashRenderer.js";
+import { CalculateRenderer } from "./renderers/CalculateRenderer.js";
+import { DefaultRenderer } from "./renderers/DefaultRenderer.js";
+import { GetCurrentTimeRenderer } from "./renderers/GetCurrentTimeRenderer.js";
+
+// Register all built-in tool renderers
+registerToolRenderer("calculate", new CalculateRenderer());
+registerToolRenderer("get_current_time", new GetCurrentTimeRenderer());
+registerToolRenderer("bash", new BashRenderer());
+
+const defaultRenderer = new DefaultRenderer();
+
+/**
+ * Render tool call parameters
+ */
+export function renderToolParams(toolName: string, params: any, isStreaming?: boolean): TemplateResult {
+	const renderer = getToolRenderer(toolName);
+	if (renderer) {
+		return renderer.renderParams(params, isStreaming);
+	}
+	return defaultRenderer.renderParams(params, isStreaming);
+}
+
+/**
+ * Render tool result
+ */
+export function renderToolResult(toolName: string, params: any, result: ToolResultMessage): TemplateResult {
+	const renderer = getToolRenderer(toolName);
+	if (renderer) {
+		return renderer.renderResult(params, result);
+	}
+	return defaultRenderer.renderResult(params, result);
+}
+
+export { registerToolRenderer, getToolRenderer };

packages/browser-extension/src/tools/renderer-registry.ts

@@ -0,0 +1,18 @@
+import type { ToolRenderer } from "./types.js";
+
+// Registry of tool renderers
+export const toolRenderers = new Map<string, ToolRenderer>();
+
+/**
+ * Register a custom tool renderer
+ */
+export function registerToolRenderer(toolName: string, renderer: ToolRenderer): void {
+	toolRenderers.set(toolName, renderer);
+}
+
+/**
+ * Get a tool renderer by name
+ */
+export function getToolRenderer(toolName: string): ToolRenderer | undefined {
+	return toolRenderers.get(toolName);
+}

packages/browser-extension/src/tools/renderers/BashRenderer.ts

@@ -0,0 +1,45 @@
+import { html, type TemplateResult } from "@mariozechner/mini-lit";
+import type { ToolResultMessage } from "@mariozechner/pi-ai";
+import { i18n } from "../../utils/i18n.js";
+import type { ToolRenderer } from "../types.js";
+
+interface BashParams {
+	command: string;
+}
+
+// Bash tool has undefined details (only uses output)
+export class BashRenderer implements ToolRenderer<BashParams, undefined> {
+	renderParams(params: BashParams, isStreaming?: boolean): TemplateResult {
+		if (isStreaming && (!params.command || params.command.length === 0)) {
+			return html`<div class="text-sm text-muted-foreground">${i18n("Writing command...")}</div>`;
+		}
+
+		return html`
+			<div class="text-sm text-muted-foreground">
+				<span>${i18n("Running command:")}</span>
+				<code class="ml-1 px-1.5 py-0.5 bg-muted rounded text-xs font-mono">${params.command}</code>
+			</div>
+		`;
+	}
+
+	renderResult(_params: BashParams, result: ToolResultMessage<undefined>): TemplateResult {
+		const output = result.output || "";
+		const isError = result.isError === true;
+
+		if (isError) {
+			return html`
+				<div class="text-sm">
+					<div class="text-destructive font-medium mb-1">${i18n("Command failed:")}</div>
+					<pre class="text-xs font-mono text-destructive bg-destructive/10 p-2 rounded overflow-x-auto">${output}</pre>
+				</div>
+			`;
+		}
+
+		// Display the command output
+		return html`
+			<div class="text-sm">
+				<pre class="text-xs font-mono text-foreground bg-muted/50 p-2 rounded overflow-x-auto">${output}</pre>
+			</div>
+		`;
+	}
+}

packages/browser-extension/src/tools/renderers/CalculateRenderer.ts

@@ -0,0 +1,49 @@
+import { html, type TemplateResult } from "@mariozechner/mini-lit";
+import type { ToolResultMessage } from "@mariozechner/pi-ai";
+import { i18n } from "../../utils/i18n.js";
+import type { ToolRenderer } from "../types.js";
+
+interface CalculateParams {
+	expression: string;
+}
+
+// Calculate tool has undefined details (only uses output)
+export class CalculateRenderer implements ToolRenderer<CalculateParams, undefined> {
+	renderParams(params: CalculateParams, isStreaming?: boolean): TemplateResult {
+		if (isStreaming && !params.expression) {
+			return html`<div class="text-sm text-muted-foreground">${i18n("Writing expression...")}</div>`;
+		}
+
+		return html`
+			<div class="text-sm text-muted-foreground">
+				<span>${i18n("Calculating")}</span>
+				<code class="mx-1 px-1.5 py-0.5 bg-muted rounded text-xs font-mono">${params.expression}</code>
+			</div>
+		`;
+	}
+
+	renderResult(_params: CalculateParams, result: ToolResultMessage<undefined>): TemplateResult {
+		// Parse the output to make it look nicer
+		const output = result.output || "";
+		const isError = result.isError === true;
+
+		if (isError) {
+			return html`<div class="text-sm text-destructive">${output}</div>`;
+		}
+
+		// Try to split on = to show expression and result separately
+		const parts = output.split(" = ");
+		if (parts.length === 2) {
+			return html`
+				<div class="text-sm font-mono">
+					<span class="text-muted-foreground">${parts[0]}</span>
+					<span class="text-muted-foreground mx-1">=</span>
+					<span class="text-foreground font-semibold">${parts[1]}</span>
+				</div>
+			`;
+		}
+
+		// Fallback to showing the whole output
+		return html`<div class="text-sm font-mono text-foreground">${output}</div>`;
+	}
+}

packages/browser-extension/src/tools/renderers/DefaultRenderer.ts

@@ -0,0 +1,36 @@
+import { html, type TemplateResult } from "@mariozechner/mini-lit";
+import type { ToolResultMessage } from "@mariozechner/pi-ai";
+import { i18n } from "../../utils/i18n.js";
+import type { ToolRenderer } from "../types.js";
+
+export class DefaultRenderer implements ToolRenderer {
+	renderParams(params: any, isStreaming?: boolean): TemplateResult {
+		let text: string;
+		let isJson = false;
+
+		try {
+			text = JSON.stringify(JSON.parse(params), null, 2);
+			isJson = true;
+		} catch {
+			try {
+				text = JSON.stringify(params, null, 2);
+				isJson = true;
+			} catch {
+				text = String(params);
+			}
+		}
+
+		if (isStreaming && (!text || text === "{}" || text === "null")) {
+			return html`<div class="text-sm text-muted-foreground">${i18n("Preparing tool parameters...")}</div>`;
+		}
+
+		return html`<console-block .content=${text}></console-block>`;
+	}
+
+	renderResult(_params: any, result: ToolResultMessage): TemplateResult {
+		// Just show the output field - that's what was sent to the LLM
+		const text = result.output || i18n("(no output)");
+
+		return html`<div class="text-sm text-muted-foreground whitespace-pre-wrap font-mono">${text}</div>`;
+	}
+}

packages/browser-extension/src/tools/renderers/GetCurrentTimeRenderer.ts

@@ -0,0 +1,39 @@
+import { html, type TemplateResult } from "@mariozechner/mini-lit";
+import type { ToolResultMessage } from "@mariozechner/pi-ai";
+import { i18n } from "../../utils/i18n.js";
+import type { ToolRenderer } from "../types.js";
+
+interface GetCurrentTimeParams {
+	timezone?: string;
+}
+
+// GetCurrentTime tool has undefined details (only uses output)
+export class GetCurrentTimeRenderer implements ToolRenderer<GetCurrentTimeParams, undefined> {
+	renderParams(params: GetCurrentTimeParams, isStreaming?: boolean): TemplateResult {
+		if (params.timezone) {
+			return html`
+				<div class="text-sm text-muted-foreground">
+					<span>${i18n("Getting current time in")}</span>
+					<code class="mx-1 px-1.5 py-0.5 bg-muted rounded text-xs font-mono">${params.timezone}</code>
+				</div>
+			`;
+		}
+		return html`
+			<div class="text-sm text-muted-foreground">
+				<span>${i18n("Getting current date and time")}${isStreaming ? "..." : ""}</span>
+			</div>
+		`;
+	}
+
+	renderResult(_params: GetCurrentTimeParams, result: ToolResultMessage<undefined>): TemplateResult {
+		const output = result.output || "";
+		const isError = result.isError === true;
+
+		if (isError) {
+			return html`<div class="text-sm text-destructive">${output}</div>`;
+		}
+
+		// Display the date/time result
+		return html`<div class="text-sm font-mono text-foreground">${output}</div>`;
+	}
+}

packages/browser-extension/src/tools/types.ts

@@ -0,0 +1,7 @@
+import type { ToolResultMessage } from "@mariozechner/pi-ai";
+import type { TemplateResult } from "lit";
+
+export interface ToolRenderer<TParams = any, TDetails = any> {
+	renderParams(params: TParams, isStreaming?: boolean): TemplateResult;
+	renderResult(params: TParams, result: ToolResultMessage<TDetails>): TemplateResult;
+}