getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-21 21:03:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(coding-agent): highlight active path with ● marker in tree selector

Browse source
This commit is contained in:
Mario Zechner 2025-12-29 15:18:02 +01:00
parent 8d3b4dd762
commit 96c071b4c4
1 changed files with 28 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

29
packages/coding-agent/src/modes/interactive/components/tree-selector.ts
View file

@ -55,6 +55,7 @@ class TreeList implements Component {
private searchQuery = ""; private searchQuery = "";
private toolCallMap: Map<string, ToolCallInfo> = new Map(); private toolCallMap: Map<string, ToolCallInfo> = new Map();
private multipleRoots = false; private multipleRoots = false;
private activePathIds: Set<string> = new Set();
public onSelect?: (entryId: string) => void; public onSelect?: (entryId: string) => void;
public onCancel?: () => void; public onCancel?: () => void;
@ -65,6 +66,7 @@ class TreeList implements Component {
this.maxVisibleLines = maxVisibleLines; this.maxVisibleLines = maxVisibleLines;
this.multipleRoots = tree.length > 1; this.multipleRoots = tree.length > 1;
this.flatNodes = this.flattenTree(tree); this.flatNodes = this.flattenTree(tree);
this.buildActivePath();
this.applyFilter(); this.applyFilter();
// Start with current leaf selected // Start with current leaf selected
@ -76,6 +78,27 @@ class TreeList implements Component {
} }
} }
/** Build the set of entry IDs on the path from root to current leaf */
private buildActivePath(): void {
this.activePathIds.clear();
if (!this.currentLeafId) return;
// Build a map of id -> entry for parent lookup
const entryMap = new Map<string, FlatNode>();
for (const flatNode of this.flatNodes) {
entryMap.set(flatNode.node.entry.id, flatNode);
}
// Walk from leaf to root
let currentId: string | null = this.currentLeafId;
while (currentId) {
this.activePathIds.add(currentId);
const node = entryMap.get(currentId);
if (!node) break;
currentId = node.node.entry.parentId ?? null;
}
}
private flattenTree(roots: SessionTreeNode[]): FlatNode[] { private flattenTree(roots: SessionTreeNode[]): FlatNode[] {
const result: FlatNode[] = []; const result: FlatNode[] = [];
this.toolCallMap.clear(); this.toolCallMap.clear();
@ -340,11 +363,15 @@ class TreeList implements Component {
const extraIndent = " ".repeat(Math.max(0, displayIndent - prefixLevels)); const extraIndent = " ".repeat(Math.max(0, displayIndent - prefixLevels));
const prefix = gutterStr + connector + extraIndent; const prefix = gutterStr + connector + extraIndent;
// Active path marker
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 label = flatNode.node.label ? theme.fg("warning", `[${flatNode.node.label}] `) : "";
const content = this.getEntryDisplayText(flatNode.node, isSelected); const content = this.getEntryDisplayText(flatNode.node, isSelected);
const suffix = isCurrentLeaf ? theme.fg("accent", " *") : ""; const suffix = isCurrentLeaf ? theme.fg("accent", " *") : "";
const line = cursor + theme.fg("dim", prefix) + label + content + suffix; const line = cursor + pathMarker + theme.fg("dim", prefix) + label + content + suffix;
lines.push(truncateToWidth(line, width)); lines.push(truncateToWidth(line, width));
} }