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

Add global debug key (Shift+Ctrl+D), iterative tree sorting to avoid stack overflow

Browse source
This commit is contained in:
Mario Zechner 2025-12-29 04:50:15 +01:00
parent 1f4594598b
commit 544814875e
5 changed files with 30 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
packages/coding-agent/src/core/session-manager.ts
View file

@ -789,7 +789,7 @@ export class SessionManager {
// Build tree
for (const entry of entries) {
const node = nodeMap.get(entry.id)!;
if (entry.parentId === null) {
if (entry.parentId === null || entry.parentId === entry.id) {
roots.push(node);
} else {
const parent = nodeMap.get(entry.parentId);
@ -803,11 +803,13 @@ export class SessionManager {
}
// Sort children by timestamp (oldest first, newest at bottom)
const sortChildren = (node: SessionTreeNode): void => {
// Use iterative approach to avoid stack overflow on deep trees
const stack: SessionTreeNode[] = [...roots];
while (stack.length > 0) {
const node = stack.pop()!;
node.children.sort((a, b) => new Date(a.entry.timestamp).getTime() - new Date(b.entry.timestamp).getTime());
node.children.forEach(sortChildren);
};
roots.forEach(sortChildren);
stack.push(...node.children);
}
return roots;
}

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

@ -621,6 +621,9 @@ export class InteractiveMode {
this.editor.onShiftTab = () => this.cycleThinkingLevel();
this.editor.onCtrlP = () => this.cycleModel("forward");
this.editor.onShiftCtrlP = () => this.cycleModel("backward");
// Global debug handler on TUI (works regardless of focus)
this.ui.onDebug = () => this.handleDebugCommand();
this.editor.onCtrlL = () => this.showModelSelector();
this.editor.onCtrlO = () => this.toggleToolOutputExpansion();
this.editor.onCtrlT = () => this.toggleThinkingBlockVisibility();
@ -1645,7 +1648,7 @@ export class InteractiveMode {
this.ui.requestRender();
},
);
return { component: selector, focus: selector.getTreeList() };
return { component: selector, focus: selector };
});
}