@mariozechner/pi-coding-agent 0.30.2 → 0.31.0

package/dist/modes/interactive/components/tree-selector.js

@@ -0,0 +1,745 @@
+import { Container, Input, isArrowDown, isArrowLeft, isArrowRight, isArrowUp, isBackspace, isCtrlC, isCtrlO, isEnter, isEscape, isShiftCtrlO, Spacer, Text, TruncatedText, truncateToWidth, } from "@mariozechner/pi-tui";
+import { theme } from "../theme/theme.js";
+import { DynamicBorder } from "./dynamic-border.js";
+class TreeList {
+    flatNodes = [];
+    filteredNodes = [];
+    selectedIndex = 0;
+    currentLeafId;
+    maxVisibleLines;
+    filterMode = "default";
+    searchQuery = "";
+    toolCallMap = new Map();
+    multipleRoots = false;
+    activePathIds = new Set();
+    onSelect;
+    onCancel;
+    onLabelEdit;
+    constructor(tree, currentLeafId, maxVisibleLines) {
+        this.currentLeafId = currentLeafId;
+        this.maxVisibleLines = maxVisibleLines;
+        this.multipleRoots = tree.length > 1;
+        this.flatNodes = this.flattenTree(tree);
+        this.buildActivePath();
+        this.applyFilter();
+        // Start with current leaf selected
+        const leafIndex = this.filteredNodes.findIndex((n) => n.node.entry.id === currentLeafId);
+        if (leafIndex !== -1) {
+            this.selectedIndex = leafIndex;
+        }
+        else {
+            this.selectedIndex = Math.max(0, this.filteredNodes.length - 1);
+        }
+    }
+    /** Build the set of entry IDs on the path from root to current leaf */
+    buildActivePath() {
+        this.activePathIds.clear();
+        if (!this.currentLeafId)
+            return;
+        // Build a map of id -> entry for parent lookup
+        const entryMap = new Map();
+        for (const flatNode of this.flatNodes) {
+            entryMap.set(flatNode.node.entry.id, flatNode);
+        }
+        // Walk from leaf to root
+        let currentId = this.currentLeafId;
+        while (currentId) {
+            this.activePathIds.add(currentId);
+            const node = entryMap.get(currentId);
+            if (!node)
+                break;
+            currentId = node.node.entry.parentId ?? null;
+        }
+    }
+    flattenTree(roots) {
+        const result = [];
+        this.toolCallMap.clear();
+        const stack = [];
+        // Determine which subtrees contain the active leaf (to sort current branch first)
+        // Use iterative post-order traversal to avoid stack overflow
+        const containsActive = new Map();
+        const leafId = this.currentLeafId;
+        {
+            // Build list in pre-order, then process in reverse for post-order effect
+            const allNodes = [];
+            const preOrderStack = [...roots];
+            while (preOrderStack.length > 0) {
+                const node = preOrderStack.pop();
+                allNodes.push(node);
+                // Push children in reverse so they're processed left-to-right
+                for (let i = node.children.length - 1; i >= 0; i--) {
+                    preOrderStack.push(node.children[i]);
+                }
+            }
+            // Process in reverse (post-order): children before parents
+            for (let i = allNodes.length - 1; i >= 0; i--) {
+                const node = allNodes[i];
+                let has = leafId !== null && node.entry.id === leafId;
+                for (const child of node.children) {
+                    if (containsActive.get(child)) {
+                        has = true;
+                    }
+                }
+                containsActive.set(node, has);
+            }
+        }
+        // Add roots in reverse order, prioritizing the one containing the active leaf
+        // If multiple roots, treat them as children of a virtual root that branches
+        const multipleRoots = roots.length > 1;
+        const orderedRoots = [...roots].sort((a, b) => Number(containsActive.get(b)) - Number(containsActive.get(a)));
+        for (let i = orderedRoots.length - 1; i >= 0; i--) {
+            const isLast = i === orderedRoots.length - 1;
+            stack.push([orderedRoots[i], multipleRoots ? 1 : 0, multipleRoots, multipleRoots, isLast, [], multipleRoots]);
+        }
+        while (stack.length > 0) {
+            const [node, indent, justBranched, showConnector, isLast, gutters, isVirtualRootChild] = stack.pop();
+            // Extract tool calls from assistant messages for later lookup
+            const entry = node.entry;
+            if (entry.type === "message" && entry.message.role === "assistant") {
+                const content = entry.message.content;
+                if (Array.isArray(content)) {
+                    for (const block of content) {
+                        if (typeof block === "object" && block !== null && "type" in block && block.type === "toolCall") {
+                            const tc = block;
+                            this.toolCallMap.set(tc.id, { name: tc.name, arguments: tc.arguments });
+                        }
+                    }
+                }
+            }
+            result.push({ node, indent, showConnector, isLast, gutters, isVirtualRootChild });
+            const children = node.children;
+            const multipleChildren = children.length > 1;
+            // Order children so the branch containing the active leaf comes first
+            const orderedChildren = (() => {
+                const prioritized = [];
+                const rest = [];
+                for (const child of children) {
+                    if (containsActive.get(child)) {
+                        prioritized.push(child);
+                    }
+                    else {
+                        rest.push(child);
+                    }
+                }
+                return [...prioritized, ...rest];
+            })();
+            // Calculate child indent
+            let childIndent;
+            if (multipleChildren) {
+                // Parent branches: children get +1
+                childIndent = indent + 1;
+            }
+            else if (justBranched && indent > 0) {
+                // First generation after a branch: +1 for visual grouping
+                childIndent = indent + 1;
+            }
+            else {
+                // Single-child chain: stay flat
+                childIndent = indent;
+            }
+            // Build gutters for children
+            // If this node showed a connector, add a gutter entry for descendants
+            // Only add gutter if connector is actually displayed (not suppressed for virtual root children)
+            const connectorDisplayed = showConnector && !isVirtualRootChild;
+            // When connector is displayed, add a gutter entry at the connector's position
+            // Connector is at position (displayIndent - 1), so gutter should be there too
+            const currentDisplayIndent = this.multipleRoots ? Math.max(0, indent - 1) : indent;
+            const connectorPosition = Math.max(0, currentDisplayIndent - 1);
+            const childGutters = connectorDisplayed
+                ? [...gutters, { position: connectorPosition, show: !isLast }]
+                : gutters;
+            // Add children in reverse order
+            for (let i = orderedChildren.length - 1; i >= 0; i--) {
+                const childIsLast = i === orderedChildren.length - 1;
+                stack.push([
+                    orderedChildren[i],
+                    childIndent,
+                    multipleChildren,
+                    multipleChildren,
+                    childIsLast,
+                    childGutters,
+                    false,
+                ]);
+            }
+        }
+        return result;
+    }
+    applyFilter() {
+        // Remember currently selected node to preserve cursor position
+        const previouslySelectedId = this.filteredNodes[this.selectedIndex]?.node.entry.id;
+        const searchTokens = this.searchQuery.toLowerCase().split(/\s+/).filter(Boolean);
+        this.filteredNodes = this.flatNodes.filter((flatNode) => {
+            const entry = flatNode.node.entry;
+            const isCurrentLeaf = entry.id === this.currentLeafId;
+            // Skip assistant messages with only tool calls (no text) unless error/aborted
+            // Always show current leaf so active position is visible
+            if (entry.type === "message" && entry.message.role === "assistant" && !isCurrentLeaf) {
+                const msg = entry.message;
+                const hasText = this.hasTextContent(msg.content);
+                const isErrorOrAborted = msg.stopReason && msg.stopReason !== "stop" && msg.stopReason !== "toolUse";
+                // Only hide if no text AND not an error/aborted message
+                if (!hasText && !isErrorOrAborted) {
+                    return false;
+                }
+            }
+            // Apply filter mode
+            let passesFilter = true;
+            // Entry types hidden in default view (settings/bookkeeping)
+            const isSettingsEntry = entry.type === "label" ||
+                entry.type === "custom" ||
+                entry.type === "model_change" ||
+                entry.type === "thinking_level_change";
+            switch (this.filterMode) {
+                case "user-only":
+                    // Just user messages
+                    passesFilter = entry.type === "message" && entry.message.role === "user";
+                    break;
+                case "no-tools":
+                    // Default minus tool results
+                    passesFilter = !isSettingsEntry && !(entry.type === "message" && entry.message.role === "toolResult");
+                    break;
+                case "labeled-only":
+                    // Just labeled entries
+                    passesFilter = flatNode.node.label !== undefined;
+                    break;
+                case "all":
+                    // Show everything
+                    passesFilter = true;
+                    break;
+                default:
+                    // Default mode: hide settings/bookkeeping entries
+                    passesFilter = !isSettingsEntry;
+                    break;
+            }
+            if (!passesFilter)
+                return false;
+            // Apply search filter
+            if (searchTokens.length > 0) {
+                const nodeText = this.getSearchableText(flatNode.node).toLowerCase();
+                return searchTokens.every((token) => nodeText.includes(token));
+            }
+            return true;
+        });
+        // Try to preserve cursor on the same node after filtering
+        if (previouslySelectedId) {
+            const newIndex = this.filteredNodes.findIndex((n) => n.node.entry.id === previouslySelectedId);
+            if (newIndex !== -1) {
+                this.selectedIndex = newIndex;
+                return;
+            }
+        }
+        // Fall back: clamp index if out of bounds
+        if (this.selectedIndex >= this.filteredNodes.length) {
+            this.selectedIndex = Math.max(0, this.filteredNodes.length - 1);
+        }
+    }
+    /** Get searchable text content from a node */
+    getSearchableText(node) {
+        const entry = node.entry;
+        const parts = [];
+        if (node.label) {
+            parts.push(node.label);
+        }
+        switch (entry.type) {
+            case "message": {
+                const msg = entry.message;
+                parts.push(msg.role);
+                if ("content" in msg && msg.content) {
+                    parts.push(this.extractContent(msg.content));
+                }
+                if (msg.role === "bashExecution") {
+                    const bashMsg = msg;
+                    if (bashMsg.command)
+                        parts.push(bashMsg.command);
+                }
+                break;
+            }
+            case "custom_message": {
+                parts.push(entry.customType);
+                if (typeof entry.content === "string") {
+                    parts.push(entry.content);
+                }
+                else {
+                    parts.push(this.extractContent(entry.content));
+                }
+                break;
+            }
+            case "compaction":
+                parts.push("compaction");
+                break;
+            case "branch_summary":
+                parts.push("branch summary", entry.summary);
+                break;
+            case "model_change":
+                parts.push("model", entry.modelId);
+                break;
+            case "thinking_level_change":
+                parts.push("thinking", entry.thinkingLevel);
+                break;
+            case "custom":
+                parts.push("custom", entry.customType);
+                break;
+            case "label":
+                parts.push("label", entry.label ?? "");
+                break;
+        }
+        return parts.join(" ");
+    }
+    invalidate() { }
+    getSearchQuery() {
+        return this.searchQuery;
+    }
+    getSelectedNode() {
+        return this.filteredNodes[this.selectedIndex]?.node;
+    }
+    updateNodeLabel(entryId, label) {
+        for (const flatNode of this.flatNodes) {
+            if (flatNode.node.entry.id === entryId) {
+                flatNode.node.label = label;
+                break;
+            }
+        }
+    }
+    getFilterLabel() {
+        switch (this.filterMode) {
+            case "no-tools":
+                return " [no-tools]";
+            case "user-only":
+                return " [user]";
+            case "labeled-only":
+                return " [labeled]";
+            case "all":
+                return " [all]";
+            default:
+                return "";
+        }
+    }
+    render(width) {
+        const lines = [];
+        if (this.filteredNodes.length === 0) {
+            lines.push(truncateToWidth(theme.fg("muted", "  No entries found"), width));
+            lines.push(truncateToWidth(theme.fg("muted", `  (0/0)${this.getFilterLabel()}`), width));
+            return lines;
+        }
+        const startIndex = Math.max(0, Math.min(this.selectedIndex - Math.floor(this.maxVisibleLines / 2), this.filteredNodes.length - this.maxVisibleLines));
+        const endIndex = Math.min(startIndex + this.maxVisibleLines, this.filteredNodes.length);
+        for (let i = startIndex; i < endIndex; i++) {
+            const flatNode = this.filteredNodes[i];
+            const entry = flatNode.node.entry;
+            const isSelected = i === this.selectedIndex;
+            // Build line: cursor + prefix + path marker + label + content
+            const cursor = isSelected ? theme.fg("accent", "› ") : "  ";
+            // If multiple roots, shift display (roots at 0, not 1)
+            const displayIndent = this.multipleRoots ? Math.max(0, flatNode.indent - 1) : flatNode.indent;
+            // Build prefix with gutters at their correct positions
+            // Each gutter has a position (displayIndent where its connector was shown)
+            const connector = flatNode.showConnector && !flatNode.isVirtualRootChild ? (flatNode.isLast ? "└─ " : "├─ ") : "";
+            const connectorPosition = connector ? displayIndent - 1 : -1;
+            // Build prefix char by char, placing gutters and connector at their positions
+            const totalChars = displayIndent * 3;
+            const prefixChars = [];
+            for (let i = 0; i < totalChars; i++) {
+                const level = Math.floor(i / 3);
+                const posInLevel = i % 3;
+                // Check if there's a gutter at this level
+                const gutter = flatNode.gutters.find((g) => g.position === level);
+                if (gutter) {
+                    if (posInLevel === 0) {
+                        prefixChars.push(gutter.show ? "│" : " ");
+                    }
+                    else {
+                        prefixChars.push(" ");
+                    }
+                }
+                else if (connector && level === connectorPosition) {
+                    // Connector at this level
+                    if (posInLevel === 0) {
+                        prefixChars.push(flatNode.isLast ? "└" : "├");
+                    }
+                    else if (posInLevel === 1) {
+                        prefixChars.push("─");
+                    }
+                    else {
+                        prefixChars.push(" ");
+                    }
+                }
+                else {
+                    prefixChars.push(" ");
+                }
+            }
+            const prefix = prefixChars.join("");
+            // Active path marker - shown right before the entry text
+            const isOnActivePath = this.activePathIds.has(entry.id);
+            const pathMarker = isOnActivePath ? theme.fg("accent", "• ") : "";
+            const label = flatNode.node.label ? theme.fg("warning", `[${flatNode.node.label}] `) : "";
+            const content = this.getEntryDisplayText(flatNode.node, isSelected);
+            let line = cursor + theme.fg("dim", prefix) + pathMarker + label + content;
+            if (isSelected) {
+                line = theme.bg("selectedBg", line);
+            }
+            lines.push(truncateToWidth(line, width));
+        }
+        lines.push(truncateToWidth(theme.fg("muted", `  (${this.selectedIndex + 1}/${this.filteredNodes.length})${this.getFilterLabel()}`), width));
+        return lines;
+    }
+    getEntryDisplayText(node, isSelected) {
+        const entry = node.entry;
+        let result;
+        const normalize = (s) => s.replace(/[\n\t]/g, " ").trim();
+        switch (entry.type) {
+            case "message": {
+                const msg = entry.message;
+                const role = msg.role;
+                if (role === "user") {
+                    const msgWithContent = msg;
+                    const content = normalize(this.extractContent(msgWithContent.content));
+                    result = theme.fg("accent", "user: ") + content;
+                }
+                else if (role === "assistant") {
+                    const msgWithContent = msg;
+                    const textContent = normalize(this.extractContent(msgWithContent.content));
+                    if (textContent) {
+                        result = theme.fg("success", "assistant: ") + textContent;
+                    }
+                    else if (msgWithContent.stopReason === "aborted") {
+                        result = theme.fg("success", "assistant: ") + theme.fg("muted", "(aborted)");
+                    }
+                    else if (msgWithContent.errorMessage) {
+                        const errMsg = normalize(msgWithContent.errorMessage).slice(0, 80);
+                        result = theme.fg("success", "assistant: ") + theme.fg("error", errMsg);
+                    }
+                    else {
+                        result = theme.fg("success", "assistant: ") + theme.fg("muted", "(no content)");
+                    }
+                }
+                else if (role === "toolResult") {
+                    const toolMsg = msg;
+                    const toolCall = toolMsg.toolCallId ? this.toolCallMap.get(toolMsg.toolCallId) : undefined;
+                    if (toolCall) {
+                        result = theme.fg("muted", this.formatToolCall(toolCall.name, toolCall.arguments));
+                    }
+                    else {
+                        result = theme.fg("muted", `[${toolMsg.toolName ?? "tool"}]`);
+                    }
+                }
+                else if (role === "bashExecution") {
+                    const bashMsg = msg;
+                    result = theme.fg("dim", `[bash]: ${normalize(bashMsg.command ?? "")}`);
+                }
+                else {
+                    result = theme.fg("dim", `[${role}]`);
+                }
+                break;
+            }
+            case "custom_message": {
+                const content = typeof entry.content === "string"
+                    ? entry.content
+                    : entry.content
+                        .filter((c) => c.type === "text")
+                        .map((c) => c.text)
+                        .join("");
+                result = theme.fg("customMessageLabel", `[${entry.customType}]: `) + normalize(content);
+                break;
+            }
+            case "compaction": {
+                const tokens = Math.round(entry.tokensBefore / 1000);
+                result = theme.fg("borderAccent", `[compaction: ${tokens}k tokens]`);
+                break;
+            }
+            case "branch_summary":
+                result = theme.fg("warning", `[branch summary]: `) + normalize(entry.summary);
+                break;
+            case "model_change":
+                result = theme.fg("dim", `[model: ${entry.modelId}]`);
+                break;
+            case "thinking_level_change":
+                result = theme.fg("dim", `[thinking: ${entry.thinkingLevel}]`);
+                break;
+            case "custom":
+                result = theme.fg("dim", `[custom: ${entry.customType}]`);
+                break;
+            case "label":
+                result = theme.fg("dim", `[label: ${entry.label ?? "(cleared)"}]`);
+                break;
+            default:
+                result = "";
+        }
+        return isSelected ? theme.bold(result) : result;
+    }
+    extractContent(content) {
+        const maxLen = 200;
+        if (typeof content === "string")
+            return content.slice(0, maxLen);
+        if (Array.isArray(content)) {
+            let result = "";
+            for (const c of content) {
+                if (typeof c === "object" && c !== null && "type" in c && c.type === "text") {
+                    result += c.text;
+                    if (result.length >= maxLen)
+                        return result.slice(0, maxLen);
+                }
+            }
+            return result;
+        }
+        return "";
+    }
+    hasTextContent(content) {
+        if (typeof content === "string")
+            return content.trim().length > 0;
+        if (Array.isArray(content)) {
+            for (const c of content) {
+                if (typeof c === "object" && c !== null && "type" in c && c.type === "text") {
+                    const text = c.text;
+                    if (text && text.trim().length > 0)
+                        return true;
+                }
+            }
+        }
+        return false;
+    }
+    formatToolCall(name, args) {
+        const shortenPath = (p) => {
+            const home = process.env.HOME || process.env.USERPROFILE || "";
+            if (home && p.startsWith(home))
+                return `~${p.slice(home.length)}`;
+            return p;
+        };
+        switch (name) {
+            case "read": {
+                const path = shortenPath(String(args.path || args.file_path || ""));
+                const offset = args.offset;
+                const limit = args.limit;
+                let display = path;
+                if (offset !== undefined || limit !== undefined) {
+                    const start = offset ?? 1;
+                    const end = limit !== undefined ? start + limit - 1 : "";
+                    display += `:${start}${end ? `-${end}` : ""}`;
+                }
+                return `[read: ${display}]`;
+            }
+            case "write": {
+                const path = shortenPath(String(args.path || args.file_path || ""));
+                return `[write: ${path}]`;
+            }
+            case "edit": {
+                const path = shortenPath(String(args.path || args.file_path || ""));
+                return `[edit: ${path}]`;
+            }
+            case "bash": {
+                const rawCmd = String(args.command || "");
+                const cmd = rawCmd
+                    .replace(/[\n\t]/g, " ")
+                    .trim()
+                    .slice(0, 50);
+                return `[bash: ${cmd}${rawCmd.length > 50 ? "..." : ""}]`;
+            }
+            case "grep": {
+                const pattern = String(args.pattern || "");
+                const path = shortenPath(String(args.path || "."));
+                return `[grep: /${pattern}/ in ${path}]`;
+            }
+            case "find": {
+                const pattern = String(args.pattern || "");
+                const path = shortenPath(String(args.path || "."));
+                return `[find: ${pattern} in ${path}]`;
+            }
+            case "ls": {
+                const path = shortenPath(String(args.path || "."));
+                return `[ls: ${path}]`;
+            }
+            default: {
+                // Custom tool - show name and truncated JSON args
+                const argsStr = JSON.stringify(args).slice(0, 40);
+                return `[${name}: ${argsStr}${JSON.stringify(args).length > 40 ? "..." : ""}]`;
+            }
+        }
+    }
+    handleInput(keyData) {
+        if (isArrowUp(keyData)) {
+            this.selectedIndex = this.selectedIndex === 0 ? this.filteredNodes.length - 1 : this.selectedIndex - 1;
+        }
+        else if (isArrowDown(keyData)) {
+            this.selectedIndex = this.selectedIndex === this.filteredNodes.length - 1 ? 0 : this.selectedIndex + 1;
+        }
+        else if (isArrowLeft(keyData)) {
+            // Page up
+            this.selectedIndex = Math.max(0, this.selectedIndex - this.maxVisibleLines);
+        }
+        else if (isArrowRight(keyData)) {
+            // Page down
+            this.selectedIndex = Math.min(this.filteredNodes.length - 1, this.selectedIndex + this.maxVisibleLines);
+        }
+        else if (isEnter(keyData)) {
+            const selected = this.filteredNodes[this.selectedIndex];
+            if (selected && this.onSelect) {
+                this.onSelect(selected.node.entry.id);
+            }
+        }
+        else if (isEscape(keyData)) {
+            if (this.searchQuery) {
+                this.searchQuery = "";
+                this.applyFilter();
+            }
+            else {
+                this.onCancel?.();
+            }
+        }
+        else if (isCtrlC(keyData)) {
+            this.onCancel?.();
+        }
+        else if (isShiftCtrlO(keyData)) {
+            // Cycle filter backwards
+            const modes = ["default", "no-tools", "user-only", "labeled-only", "all"];
+            const currentIndex = modes.indexOf(this.filterMode);
+            this.filterMode = modes[(currentIndex - 1 + modes.length) % modes.length];
+            this.applyFilter();
+        }
+        else if (isCtrlO(keyData)) {
+            // Cycle filter forwards: default → no-tools → user-only → labeled-only → all → default
+            const modes = ["default", "no-tools", "user-only", "labeled-only", "all"];
+            const currentIndex = modes.indexOf(this.filterMode);
+            this.filterMode = modes[(currentIndex + 1) % modes.length];
+            this.applyFilter();
+        }
+        else if (isBackspace(keyData)) {
+            if (this.searchQuery.length > 0) {
+                this.searchQuery = this.searchQuery.slice(0, -1);
+                this.applyFilter();
+            }
+        }
+        else if (keyData === "l" && !this.searchQuery) {
+            const selected = this.filteredNodes[this.selectedIndex];
+            if (selected && this.onLabelEdit) {
+                this.onLabelEdit(selected.node.entry.id, selected.node.label);
+            }
+        }
+        else {
+            const hasControlChars = [...keyData].some((ch) => {
+                const code = ch.charCodeAt(0);
+                return code < 32 || code === 0x7f || (code >= 0x80 && code <= 0x9f);
+            });
+            if (!hasControlChars && keyData.length > 0) {
+                this.searchQuery += keyData;
+                this.applyFilter();
+            }
+        }
+    }
+}
+/** Component that displays the current search query */
+class SearchLine {
+    treeList;
+    constructor(treeList) {
+        this.treeList = treeList;
+    }
+    invalidate() { }
+    render(width) {
+        const query = this.treeList.getSearchQuery();
+        if (query) {
+            return [truncateToWidth(`  ${theme.fg("muted", "Search:")} ${theme.fg("accent", query)}`, width)];
+        }
+        return [truncateToWidth(`  ${theme.fg("muted", "Search:")}`, width)];
+    }
+    handleInput(_keyData) { }
+}
+/** Label input component shown when editing a label */
+class LabelInput {
+    input;
+    entryId;
+    onSubmit;
+    onCancel;
+    constructor(entryId, currentLabel) {
+        this.entryId = entryId;
+        this.input = new Input();
+        if (currentLabel) {
+            this.input.setValue(currentLabel);
+        }
+    }
+    invalidate() { }
+    render(width) {
+        const lines = [];
+        const indent = "  ";
+        const availableWidth = width - indent.length;
+        lines.push(truncateToWidth(`${indent}${theme.fg("muted", "Label (empty to remove):")}`, width));
+        lines.push(...this.input.render(availableWidth).map((line) => truncateToWidth(`${indent}${line}`, width)));
+        lines.push(truncateToWidth(`${indent}${theme.fg("dim", "enter: save  esc: cancel")}`, width));
+        return lines;
+    }
+    handleInput(keyData) {
+        if (isEnter(keyData)) {
+            const value = this.input.getValue().trim();
+            this.onSubmit?.(this.entryId, value || undefined);
+        }
+        else if (isEscape(keyData)) {
+            this.onCancel?.();
+        }
+        else {
+            this.input.handleInput(keyData);
+        }
+    }
+}
+/**
+ * Component that renders a session tree selector for navigation
+ */
+export class TreeSelectorComponent extends Container {
+    treeList;
+    labelInput = null;
+    labelInputContainer;
+    treeContainer;
+    onLabelChangeCallback;
+    constructor(tree, currentLeafId, terminalHeight, onSelect, onCancel, onLabelChange) {
+        super();
+        this.onLabelChangeCallback = onLabelChange;
+        const maxVisibleLines = Math.max(5, Math.floor(terminalHeight / 2));
+        this.treeList = new TreeList(tree, currentLeafId, maxVisibleLines);
+        this.treeList.onSelect = onSelect;
+        this.treeList.onCancel = onCancel;
+        this.treeList.onLabelEdit = (entryId, currentLabel) => this.showLabelInput(entryId, currentLabel);
+        this.treeContainer = new Container();
+        this.treeContainer.addChild(this.treeList);
+        this.labelInputContainer = new Container();
+        this.addChild(new Spacer(1));
+        this.addChild(new DynamicBorder());
+        this.addChild(new Text(theme.bold("  Session Tree"), 1, 0));
+        this.addChild(new TruncatedText(theme.fg("muted", "  ↑/↓: move. ←/→: page. l: label. ^O/⇧^O: filter. Type to search"), 0, 0));
+        this.addChild(new SearchLine(this.treeList));
+        this.addChild(new DynamicBorder());
+        this.addChild(new Spacer(1));
+        this.addChild(this.treeContainer);
+        this.addChild(this.labelInputContainer);
+        this.addChild(new Spacer(1));
+        this.addChild(new DynamicBorder());
+        if (tree.length === 0) {
+            setTimeout(() => onCancel(), 100);
+        }
+    }
+    showLabelInput(entryId, currentLabel) {
+        this.labelInput = new LabelInput(entryId, currentLabel);
+        this.labelInput.onSubmit = (id, label) => {
+            this.treeList.updateNodeLabel(id, label);
+            this.onLabelChangeCallback?.(id, label);
+            this.hideLabelInput();
+        };
+        this.labelInput.onCancel = () => this.hideLabelInput();
+        this.treeContainer.clear();
+        this.labelInputContainer.clear();
+        this.labelInputContainer.addChild(this.labelInput);
+    }
+    hideLabelInput() {
+        this.labelInput = null;
+        this.labelInputContainer.clear();
+        this.treeContainer.clear();
+        this.treeContainer.addChild(this.treeList);
+    }
+    handleInput(keyData) {
+        if (this.labelInput) {
+            this.labelInput.handleInput(keyData);
+        }
+        else {
+            this.treeList.handleInput(keyData);
+        }
+    }
+    getTreeList() {
+        return this.treeList;
+    }
+}
+//# sourceMappingURL=tree-selector.js.map