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

Preserve tree selection when returning from summary selector or abort

Browse source
This commit is contained in:
Mario Zechner 2026-01-12 00:15:12 +01:00
parent a0311fd5b9
commit c16e21331a
2 changed files with 19 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -62,7 +62,12 @@ class TreeList implements Component {
public onCancel?: () => void;
public onLabelEdit?: (entryId: string, currentLabel: string | undefined) => void;
constructor(tree: SessionTreeNode[], currentLeafId: string | null, maxVisibleLines: number) {
constructor(
tree: SessionTreeNode[],
currentLeafId: string | null,
maxVisibleLines: number,
initialSelectedId?: string,
) {
this.currentLeafId = currentLeafId;
this.maxVisibleLines = maxVisibleLines;
this.multipleRoots = tree.length > 1;
@ -70,10 +75,11 @@ class TreeList implements Component {
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;
// Start with initialSelectedId if provided, otherwise current leaf
const targetId = initialSelectedId ?? currentLeafId;
const targetIndex = this.filteredNodes.findIndex((n) => n.node.entry.id === targetId);
if (targetIndex !== -1) {
this.selectedIndex = targetIndex;
} else {
this.selectedIndex = Math.max(0, this.filteredNodes.length - 1);
}
@ -788,13 +794,14 @@ export class TreeSelectorComponent extends Container {
onSelect: (entryId: string) => void,
onCancel: () => void,
onLabelChange?: (entryId: string, label: string | undefined) => void,
initialSelectedId?: string,
) {
super();
this.onLabelChangeCallback = onLabelChange;
const maxVisibleLines = Math.max(5, Math.floor(terminalHeight / 2));
this.treeList = new TreeList(tree, currentLeafId, maxVisibleLines);
this.treeList = new TreeList(tree, currentLeafId, maxVisibleLines, initialSelectedId);
this.treeList.onSelect = onSelect;
this.treeList.onCancel = onCancel;
this.treeList.onLabelEdit = (entryId, currentLabel) => this.showLabelInput(entryId, currentLabel);

11
packages/coding-agent/src/modes/interactive/interactive-mode.ts
View file

@ -2773,7 +2773,7 @@ export class InteractiveMode {
});
}
private showTreeSelector(): void {
private showTreeSelector(initialSelectedId?: string): void {
const tree = this.sessionManager.getTree();
const realLeafId = this.sessionManager.getLeafId();
@ -2819,8 +2819,8 @@ export class InteractiveMode {
]);
if (summaryChoice === undefined) {
// User pressed escape - re-show tree selector
this.showTreeSelector();
// User pressed escape - re-show tree selector with same selection
this.showTreeSelector(entryId);
return;
}
@ -2864,9 +2864,9 @@ export class InteractiveMode {
});
if (result.aborted) {
// Summarization aborted - re-show tree selector
// Summarization aborted - re-show tree selector with same selection
this.showStatus("Branch summarization cancelled");
this.showTreeSelector();
this.showTreeSelector(entryId);
return;
}
if (result.cancelled) {
@ -2899,6 +2899,7 @@ export class InteractiveMode {
this.sessionManager.appendLabelChange(entryId, label);
this.ui.requestRender();
},
initialSelectedId,
);
return { component: selector, focus: selector };
});