@tomsun28/pizza 0.0.9 → 0.1.1

package/dist/modes/interactive/components/tool-execution.d.ts

@@ -0,0 +1,64 @@
+import { Container, type TUI } from "@earendil-works/pi-tui";
+import type { ToolDefinition } from "../../../core/extensions/types.js";
+export interface ToolExecutionOptions {
+    showImages?: boolean;
+    imageWidthCells?: number;
+}
+export declare class ToolExecutionComponent extends Container {
+    private contentBox;
+    private contentText;
+    private selfRenderContainer;
+    private callRendererComponent?;
+    private resultRendererComponent?;
+    private rendererState;
+    private imageComponents;
+    private imageSpacers;
+    private toolName;
+    private toolCallId;
+    private args;
+    private expanded;
+    private showImages;
+    private imageWidthCells;
+    private isPartial;
+    private toolDefinition?;
+    private builtInToolDefinition?;
+    private ui;
+    private cwd;
+    private executionStarted;
+    private argsComplete;
+    private result?;
+    private convertedImages;
+    private hideComponent;
+    constructor(toolName: string, toolCallId: string, args: any, options: ToolExecutionOptions | undefined, toolDefinition: ToolDefinition<any, any> | undefined, ui: TUI, cwd: string);
+    private attachRenderShell;
+    private getCallRenderer;
+    private getResultRenderer;
+    private hasRendererDefinition;
+    private getRenderShell;
+    private getRenderContext;
+    private createCallFallback;
+    private createResultFallback;
+    updateArgs(args: any): void;
+    updateToolCall(toolName: string, args: any, toolDefinition: ToolDefinition<any, any> | undefined): void;
+    markExecutionStarted(): void;
+    setArgsComplete(): void;
+    updateResult(result: {
+        content: Array<{
+            type: string;
+            text?: string;
+            data?: string;
+            mimeType?: string;
+        }>;
+        details?: any;
+        isError: boolean;
+    }, isPartial?: boolean): void;
+    private maybeConvertImagesForKitty;
+    setExpanded(expanded: boolean): void;
+    setShowImages(show: boolean): void;
+    setImageWidthCells(width: number): void;
+    invalidate(): void;
+    render(width: number): string[];
+    private updateDisplay;
+    private getTextOutput;
+    private formatToolExecution;
+}

package/dist/modes/interactive/components/tool-execution.js

@@ -0,0 +1,305 @@
+import { Box, Container, getCapabilities, Image, Spacer, Text } from "@earendil-works/pi-tui";
+import { createAllToolDefinitions } from "../../../core/tools/index.js";
+import { getTextOutput as getRenderedTextOutput } from "../../../core/tools/render-utils.js";
+import { convertToPng } from "../../../utils/image-convert.js";
+import { theme } from "../theme/theme.js";
+export class ToolExecutionComponent extends Container {
+    constructor(toolName, toolCallId, args, options = {}, toolDefinition, ui, cwd) {
+        super();
+        this.rendererState = {};
+        this.imageComponents = [];
+        this.imageSpacers = [];
+        this.expanded = false;
+        this.isPartial = true;
+        this.executionStarted = false;
+        this.argsComplete = false;
+        this.convertedImages = new Map();
+        this.hideComponent = false;
+        this.toolName = toolName;
+        this.toolCallId = toolCallId;
+        this.args = args;
+        this.toolDefinition = toolDefinition;
+        this.builtInToolDefinition = createAllToolDefinitions(cwd)[toolName];
+        this.showImages = options.showImages ?? true;
+        this.imageWidthCells = options.imageWidthCells ?? 60;
+        this.ui = ui;
+        this.cwd = cwd;
+        this.addChild(new Spacer(1));
+        // Always create all shell variants. contentBox is used for default renderer-based composition.
+        // selfRenderContainer is used when the tool renders its own framing.
+        // contentText is reserved for generic fallback rendering when no tool definition exists.
+        this.contentBox = new Box(1, 1, (text) => theme.bg("toolPendingBg", text));
+        this.contentText = new Text("", 1, 1, (text) => theme.bg("toolPendingBg", text));
+        this.selfRenderContainer = new Container();
+        this.attachRenderShell();
+        this.updateDisplay();
+    }
+    attachRenderShell() {
+        this.clear();
+        this.addChild(new Spacer(1));
+        if (this.hasRendererDefinition()) {
+            this.addChild(this.getRenderShell() === "self" ? this.selfRenderContainer : this.contentBox);
+        }
+        else {
+            this.addChild(this.contentText);
+        }
+    }
+    getCallRenderer() {
+        if (!this.builtInToolDefinition) {
+            return this.toolDefinition?.renderCall;
+        }
+        if (!this.toolDefinition) {
+            return this.builtInToolDefinition.renderCall;
+        }
+        return this.toolDefinition.renderCall ?? this.builtInToolDefinition.renderCall;
+    }
+    getResultRenderer() {
+        if (!this.builtInToolDefinition) {
+            return this.toolDefinition?.renderResult;
+        }
+        if (!this.toolDefinition) {
+            return this.builtInToolDefinition.renderResult;
+        }
+        return this.toolDefinition.renderResult ?? this.builtInToolDefinition.renderResult;
+    }
+    hasRendererDefinition() {
+        return this.builtInToolDefinition !== undefined || this.toolDefinition !== undefined;
+    }
+    getRenderShell() {
+        if (!this.builtInToolDefinition) {
+            return this.toolDefinition?.renderShell ?? "default";
+        }
+        if (!this.toolDefinition) {
+            return this.builtInToolDefinition.renderShell ?? "default";
+        }
+        return this.toolDefinition.renderShell ?? this.builtInToolDefinition.renderShell ?? "default";
+    }
+    getRenderContext(lastComponent) {
+        return {
+            args: this.args,
+            toolCallId: this.toolCallId,
+            invalidate: () => {
+                this.invalidate();
+                this.ui.requestRender();
+            },
+            lastComponent,
+            state: this.rendererState,
+            cwd: this.cwd,
+            executionStarted: this.executionStarted,
+            argsComplete: this.argsComplete,
+            isPartial: this.isPartial,
+            expanded: this.expanded,
+            showImages: this.showImages,
+            isError: this.result?.isError ?? false,
+        };
+    }
+    createCallFallback() {
+        return new Text(theme.fg("toolTitle", theme.bold(this.toolName)), 0, 0);
+    }
+    createResultFallback() {
+        const output = this.getTextOutput();
+        if (!output) {
+            return undefined;
+        }
+        return new Text(theme.fg("toolOutput", output), 0, 0);
+    }
+    updateArgs(args) {
+        this.args = args;
+        this.updateDisplay();
+    }
+    updateToolCall(toolName, args, toolDefinition) {
+        const previousToolName = this.toolName;
+        const previousToolDefinition = this.toolDefinition;
+        const hadRendererDefinition = this.hasRendererDefinition();
+        const previousRenderShell = hadRendererDefinition ? this.getRenderShell() : undefined;
+        this.toolName = toolName;
+        this.args = args;
+        this.toolDefinition = toolDefinition;
+        this.builtInToolDefinition = createAllToolDefinitions(this.cwd)[toolName];
+        if (previousToolName !== toolName || previousToolDefinition !== toolDefinition) {
+            this.callRendererComponent = undefined;
+            this.resultRendererComponent = undefined;
+            this.rendererState = {};
+        }
+        const hasRendererDefinition = this.hasRendererDefinition();
+        const renderShell = hasRendererDefinition ? this.getRenderShell() : undefined;
+        if (hadRendererDefinition !== hasRendererDefinition || previousRenderShell !== renderShell) {
+            this.attachRenderShell();
+        }
+        this.updateDisplay();
+    }
+    markExecutionStarted() {
+        this.executionStarted = true;
+        this.updateDisplay();
+        this.ui.requestRender();
+    }
+    setArgsComplete() {
+        this.argsComplete = true;
+        this.updateDisplay();
+        this.ui.requestRender();
+    }
+    updateResult(result, isPartial = false) {
+        this.result = result;
+        this.isPartial = isPartial;
+        this.updateDisplay();
+        this.maybeConvertImagesForKitty();
+    }
+    maybeConvertImagesForKitty() {
+        const caps = getCapabilities();
+        if (caps.images !== "kitty")
+            return;
+        if (!this.result)
+            return;
+        const imageBlocks = this.result.content.filter((c) => c.type === "image");
+        for (let i = 0; i < imageBlocks.length; i++) {
+            const img = imageBlocks[i];
+            if (!img.data || !img.mimeType)
+                continue;
+            if (img.mimeType === "image/png")
+                continue;
+            if (this.convertedImages.has(i))
+                continue;
+            const index = i;
+            convertToPng(img.data, img.mimeType).then((converted) => {
+                if (converted) {
+                    this.convertedImages.set(index, converted);
+                    this.updateDisplay();
+                    this.ui.requestRender();
+                }
+            });
+        }
+    }
+    setExpanded(expanded) {
+        this.expanded = expanded;
+        this.updateDisplay();
+    }
+    setShowImages(show) {
+        this.showImages = show;
+        this.updateDisplay();
+    }
+    setImageWidthCells(width) {
+        this.imageWidthCells = Math.max(1, Math.floor(width));
+        this.updateDisplay();
+    }
+    invalidate() {
+        super.invalidate();
+        this.updateDisplay();
+    }
+    render(width) {
+        if (this.hideComponent) {
+            return [];
+        }
+        return super.render(width);
+    }
+    updateDisplay() {
+        const bgFn = this.isPartial
+            ? (text) => theme.bg("toolPendingBg", text)
+            : this.result?.isError
+                ? (text) => theme.bg("toolErrorBg", text)
+                : (text) => theme.bg("toolSuccessBg", text);
+        let hasContent = false;
+        this.hideComponent = false;
+        if (this.hasRendererDefinition()) {
+            const renderContainer = this.getRenderShell() === "self" ? this.selfRenderContainer : this.contentBox;
+            if (renderContainer instanceof Box) {
+                renderContainer.setBgFn(bgFn);
+            }
+            renderContainer.clear();
+            const callRenderer = this.getCallRenderer();
+            if (!callRenderer) {
+                renderContainer.addChild(this.createCallFallback());
+                hasContent = true;
+            }
+            else {
+                try {
+                    const component = callRenderer(this.args, theme, this.getRenderContext(this.callRendererComponent));
+                    this.callRendererComponent = component;
+                    renderContainer.addChild(component);
+                    hasContent = true;
+                }
+                catch {
+                    this.callRendererComponent = undefined;
+                    renderContainer.addChild(this.createCallFallback());
+                    hasContent = true;
+                }
+            }
+            if (this.result) {
+                const resultRenderer = this.getResultRenderer();
+                if (!resultRenderer) {
+                    const component = this.createResultFallback();
+                    if (component) {
+                        renderContainer.addChild(component);
+                        hasContent = true;
+                    }
+                }
+                else {
+                    try {
+                        const component = resultRenderer({ content: this.result.content, details: this.result.details }, { expanded: this.expanded, isPartial: this.isPartial }, theme, this.getRenderContext(this.resultRendererComponent));
+                        this.resultRendererComponent = component;
+                        renderContainer.addChild(component);
+                        hasContent = true;
+                    }
+                    catch {
+                        this.resultRendererComponent = undefined;
+                        const component = this.createResultFallback();
+                        if (component) {
+                            renderContainer.addChild(component);
+                            hasContent = true;
+                        }
+                    }
+                }
+            }
+        }
+        else {
+            this.contentText.setCustomBgFn(bgFn);
+            this.contentText.setText(this.formatToolExecution());
+            hasContent = true;
+        }
+        for (const img of this.imageComponents) {
+            this.removeChild(img);
+        }
+        this.imageComponents = [];
+        for (const spacer of this.imageSpacers) {
+            this.removeChild(spacer);
+        }
+        this.imageSpacers = [];
+        if (this.result) {
+            const imageBlocks = this.result.content.filter((c) => c.type === "image");
+            const caps = getCapabilities();
+            for (let i = 0; i < imageBlocks.length; i++) {
+                const img = imageBlocks[i];
+                if (caps.images && this.showImages && img.data && img.mimeType) {
+                    const converted = this.convertedImages.get(i);
+                    const imageData = converted?.data ?? img.data;
+                    const imageMimeType = converted?.mimeType ?? img.mimeType;
+                    if (caps.images === "kitty" && imageMimeType !== "image/png")
+                        continue;
+                    const spacer = new Spacer(1);
+                    this.addChild(spacer);
+                    this.imageSpacers.push(spacer);
+                    const imageComponent = new Image(imageData, imageMimeType, { fallbackColor: (s) => theme.fg("toolOutput", s) }, { maxWidthCells: this.imageWidthCells });
+                    this.imageComponents.push(imageComponent);
+                    this.addChild(imageComponent);
+                }
+            }
+        }
+        if (this.hasRendererDefinition() && !hasContent && this.imageComponents.length === 0) {
+            this.hideComponent = true;
+        }
+    }
+    getTextOutput() {
+        return getRenderedTextOutput(this.result, this.showImages);
+    }
+    formatToolExecution() {
+        let text = theme.fg("toolTitle", theme.bold(this.toolName));
+        const content = JSON.stringify(this.args, null, 2);
+        if (content) {
+            text += `\n\n${content}`;
+        }
+        const output = this.getTextOutput();
+        if (output) {
+            text += `\n${output}`;
+        }
+        return text;
+    }
+}

package/dist/modes/interactive/components/tree-selector.d.ts

@@ -0,0 +1,88 @@
+import { type Component, Container, type Focusable } from "@earendil-works/pi-tui";
+import type { SessionTreeNode } from "../../../core/types/session-types.js";
+/** Filter mode for tree display */
+export type FilterMode = "default" | "no-tools" | "user-only" | "labeled-only" | "all";
+declare class TreeList implements Component {
+    private flatNodes;
+    private filteredNodes;
+    private selectedIndex;
+    private currentLeafId;
+    private maxVisibleLines;
+    private filterMode;
+    private searchQuery;
+    private toolCallMap;
+    private multipleRoots;
+    private showLabelTimestamps;
+    private activePathIds;
+    private visibleParentMap;
+    private visibleChildrenMap;
+    private lastSelectedId;
+    private foldedNodes;
+    onSelect?: (entryId: string) => void;
+    onCancel?: () => void;
+    onLabelEdit?: (entryId: string, currentLabel: string | undefined) => void;
+    constructor(tree: SessionTreeNode[], currentLeafId: string | null, maxVisibleLines: number, initialSelectedId?: string, initialFilterMode?: FilterMode);
+    /**
+     * Find the index of the nearest visible entry, walking up the parent chain if needed.
+     * Returns the index in filteredNodes, or the last index as fallback.
+     */
+    private findNearestVisibleIndex;
+    /** Build the set of entry IDs on the path from root to current leaf */
+    private buildActivePath;
+    private flattenTree;
+    private applyFilter;
+    /**
+     * Recompute indentation/connectors for the filtered view
+     *
+     * Filtering can hide intermediate entries; descendants attach to the nearest visible ancestor.
+     * Keep indentation semantics aligned with flattenTree() so single-child chains don't drift right.
+     */
+    private recalculateVisualStructure;
+    /** Get searchable text content from a node */
+    private getSearchableText;
+    invalidate(): void;
+    getSearchQuery(): string;
+    getSelectedNode(): SessionTreeNode | undefined;
+    updateNodeLabel(entryId: string, label: string | undefined, labelTimestamp?: string): void;
+    private getStatusLabels;
+    render(width: number): string[];
+    private getEntryDisplayText;
+    private formatLabelTimestamp;
+    private extractContent;
+    private hasTextContent;
+    private formatToolCall;
+    handleInput(keyData: string): void;
+    /**
+     * Whether a node can be folded. A node is foldable if it has visible children
+     * and is either a root (no visible parent) or a segment start (visible parent
+     * has multiple visible children).
+     */
+    private isFoldable;
+    /**
+     * Find the index of the next branch segment start in the given direction.
+     * A segment start is the first child of a branch point.
+     *
+     * "up" walks the visible parent chain; "down" walks visible children
+     * (always following the first child).
+     */
+    private findBranchSegmentStart;
+}
+/**
+ * Component that renders a session tree selector for navigation
+ */
+export declare class TreeSelectorComponent extends Container implements Focusable {
+    private treeList;
+    private labelInput;
+    private labelInputContainer;
+    private treeContainer;
+    private onLabelChangeCallback?;
+    private _focused;
+    get focused(): boolean;
+    set focused(value: boolean);
+    constructor(tree: SessionTreeNode[], currentLeafId: string | null, terminalHeight: number, onSelect: (entryId: string) => void, onCancel: () => void, onLabelChange?: (entryId: string, label: string | undefined) => void, initialSelectedId?: string, initialFilterMode?: FilterMode);
+    private showLabelInput;
+    private hideLabelInput;
+    handleInput(keyData: string): void;
+    getTreeList(): TreeList;
+}
+export {};