@astralyn/pi 1.4.1 → 1.4.3

package/dist/modes/interactive/interactive-mode.js

@@ -1,5 +1,3 @@
-import { confirmStopBranchTasks, hotkeysTaskSection } from "./tasks/host-hooks.js";
-import { bindTasksUI } from "./tasks/index.js";
 /**
  * Interactive mode for the coding agent.
  * Handles TUI rendering and user interaction, delegating business logic to AgentSession.
@@ -22,6 +20,7 @@ import { findExtensionStackMatches, recordCrash, takeUnnotifiedCrash } from "../
 import { DEFAULT_THINKING_LEVEL, THINKING_LEVEL_OPTIONS } from "../../core/defaults.js";
 import { FooterDataProvider } from "../../core/footer-data-provider.js";
 import { configureHttpDispatcher, formatHttpIdleTimeoutMs } from "../../core/http-dispatcher.js";
+import { getRegisteredKeybindingHelpSections } from "../../core/keybinding-registry.js";
 import { KeybindingsManager } from "../../core/keybindings.js";
 import { createCompactionSummaryMessage, createCustomMessage } from "../../core/messages.js";
 import { defaultModelPerProvider, findExactModelReferenceMatch, resolveModelScopeFromModels, } from "../../core/model-resolver.js";
@@ -50,7 +49,7 @@ import { checkForNewPiVersion } from "../../utils/version-check.js";
 import { reportBug } from "./bug-report.js";
 import { createChatViewport } from "./chat-viewport.js";
 import { ArminComponent } from "./components/armin.js";
-import { AssistantMessageComponent } from "./components/assistant-message.js";
+import { AssistantMessageComponent, DEFAULT_ACTIVE_THINKING_LABEL } from "./components/assistant-message.js";
 import { BashExecutionComponent } from "./components/bash-execution.js";
 import { BranchSummaryMessageComponent } from "./components/branch-summary-message.js";
 import { CompactionSummaryMessageComponent } from "./components/compaction-summary-message.js";
@@ -78,6 +77,7 @@ import { SkillInvocationMessageComponent } from "./components/skill-invocation-m
 import { BranchSummaryStatusIndicator, CompactionStatusIndicator, IdleStatus, RetryStatusIndicator, WorkingStatusIndicator, } from "./components/status-indicator.js";
 import { ThemedText } from "./components/themed-text.js";
 import { ThinkingSelectorComponent } from "./components/thinking-selector.js";
+import { ToolExecutionComponent } from "./components/tool-execution.js";
 import { TreeSelectorComponent } from "./components/tree-selector.js";
 import { TrustSelectorComponent } from "./components/trust-selector.js";
 import { UserMessageComponent } from "./components/user-message.js";
@@ -91,7 +91,6 @@ import { shareSession } from "./session-share.js";
 import { getAvailableThemes, getAvailableThemesWithPaths, getEditorTheme, getMarkdownTheme, getThemeByName, onThemeChange, SYSTEM_THEME_NAME, setRegisteredThemes, stopThemeWatcher, Theme, theme, } from "./theme/theme.js";
 import { InteractiveThemeController } from "./theme/theme-controller.js";
 import { disposeChatRows, PendingToolMap, ToolChatContainer } from "./tool-view/chat.js";
-import { ToolExecutionComponent } from "./tool-view/tool-execution.js";
 import { createInteractiveTui, createInteractiveTuiReference } from "./tui-renderer.js";
 export { createInteractiveTui, createInteractiveTuiReference } from "./tui-renderer.js";
 function isEmbeddedStatus(editor) {
@@ -259,8 +258,7 @@ export class InteractiveMode {
         this.workingVisible = true;
         this.workingIndicatorOptions = undefined;
         this.defaultWorkingMessage = "Working";
-        this.defaultHiddenThinkingLabel = "Thinking...";
-        this.hiddenThinkingLabel = this.defaultHiddenThinkingLabel;
+        this.activeThinkingLabel = DEFAULT_ACTIVE_THINKING_LABEL;
         this.lastSigintTime = 0;
         this.lastEscapeTime = 0;
         this.changelogMarkdown = undefined;
@@ -345,7 +343,6 @@ export class InteractiveMode {
         this.options = { ...options, tuiMode };
         this.autoTrustOnReloadCwd = options.autoTrustOnReloadCwd;
         this.runtimeHost.setBeforeSessionInvalidate(() => {
-            this.tasksUI?.dispose();
             this.resetExtensionUI();
         });
         this.runtimeHost.setRebindSession(async () => {
@@ -1456,7 +1453,6 @@ export class InteractiveMode {
      * Initialize the extension system with TUI-based UI context.
      */
     async bindCurrentSessionExtensions() {
-        this.tasksUI?.dispose();
         const uiContext = this.createExtensionUIContext();
         await this.session.bindExtensions({
             uiContext,
@@ -1525,7 +1521,6 @@ export class InteractiveMode {
                 this.showExtensionError(error.extensionPath, error.error, error.stack);
             },
         });
-        this.tasksUI = bindTasksUI({ tasks: this.session.tasks, ui: uiContext });
         setRegisteredThemes(this.session.resourceLoader.getThemes().themes);
         this.setupAutocompleteProvider();
         const extensionRunner = this.session.extensionRunner;
@@ -1787,14 +1782,14 @@ export class InteractiveMode {
         this.ui.requestRender();
     }
     setHiddenThinkingLabel(label) {
-        this.hiddenThinkingLabel = label ?? this.defaultHiddenThinkingLabel;
+        this.activeThinkingLabel = label ?? DEFAULT_ACTIVE_THINKING_LABEL;
         for (const child of this.chatContainer.children) {
             if (child instanceof AssistantMessageComponent) {
-                child.setHiddenThinkingLabel(this.hiddenThinkingLabel);
+                child.setHiddenThinkingLabel(this.activeThinkingLabel);
             }
         }
         if (this.streamingComponent) {
-            this.streamingComponent.setHiddenThinkingLabel(this.hiddenThinkingLabel);
+            this.streamingComponent.setHiddenThinkingLabel(this.activeThinkingLabel);
         }
         this.ui.requestRender();
     }
@@ -2463,11 +2458,6 @@ export class InteractiveMode {
             if (this.editorHost.intercept(text, "steer"))
                 return;
             // Handle commands
-            if (text === "/tasks") {
-                this.editor.setText("");
-                await this.tasksUI?.open();
-                return;
-            }
             if (text === "/settings") {
                 this.showSettingsSelector();
                 this.editor.setText("");
@@ -2758,7 +2748,7 @@ export class InteractiveMode {
                     this.ui.requestRender();
                 }
                 else if (event.message.role === "assistant") {
-                    this.streamingComponent = new AssistantMessageComponent(undefined, this.hideThinkingBlock, this.getMarkdownThemeWithSettings(), this.hiddenThinkingLabel, this.outputPad, this.getMarkdownTransformers());
+                    this.streamingComponent = new AssistantMessageComponent(undefined, this.hideThinkingBlock, this.getMarkdownThemeWithSettings(), this.activeThinkingLabel, this.outputPad, this.getMarkdownTransformers());
                     this.streamingMessage = event.message;
                     this.chatContainer.addChild(this.streamingComponent);
                     this.streamingComponent.updateContent(this.streamingMessage, true);
@@ -3139,7 +3129,7 @@ export class InteractiveMode {
                 break;
             }
             case "assistant": {
-                const assistantComponent = new AssistantMessageComponent(message, this.hideThinkingBlock, this.getMarkdownThemeWithSettings(), this.hiddenThinkingLabel, this.outputPad, this.getMarkdownTransformers());
+                const assistantComponent = new AssistantMessageComponent(message, this.hideThinkingBlock, this.getMarkdownThemeWithSettings(), this.activeThinkingLabel, this.outputPad, this.getMarkdownTransformers());
                 this.chatContainer.addChild(assistantComponent);
                 break;
             }
@@ -3403,7 +3393,6 @@ export class InteractiveMode {
         if (this.isShuttingDown)
             return;
         this.isShuttingDown = true;
-        this.tasksUI?.dispose();
         this.session.tasks.close();
         // Keep signal handlers registered until terminal cleanup has completed.
         // `signal-exit` checks the listener list during the same SIGTERM/SIGHUP
@@ -4586,11 +4575,6 @@ export class InteractiveMode {
                 }
                 // Ask about summarization
                 done(); // Close selector first
-                const stopped = this.session.tasksStoppedByTreeNavigation(entryId);
-                if (!(await confirmStopBranchTasks(stopped, (title, message) => this.showExtensionConfirm(title, message)))) {
-                    this.showTreeSelector(entryId);
-                    return;
-                }
                 // Loop until user makes a complete choice or cancels to tree
                 let wantsSummary = false;
                 let customInstructions;
@@ -5342,9 +5326,7 @@ export class InteractiveMode {
             chatRestoredBeforeSessionStart = true;
         };
         try {
-            this.tasksUI?.dispose();
             await this.session.reload({ beforeSessionStart: restoreChatBeforeSessionStart });
-            this.tasksUI = bindTasksUI({ tasks: this.session.tasks, ui: this.createExtensionUIContext() });
             restoreChatBeforeSessionStart();
             this.keybindings.reload();
             const activeHeader = this.customHeader ?? this.builtInHeader;
@@ -5726,8 +5708,11 @@ export class InteractiveMode {
 | \`!\` | Run bash command |
 | \`!!\` | Run bash command (excluded from context) |
 `;
-        // Task keybindings act outside the editor contexts this table was built from.
-        hotkeys += hotkeysTaskSection((action) => this.getAppKeyDisplay(action));
+        for (const section of getRegisteredKeybindingHelpSections()) {
+            hotkeys += `\n**${section.title}**\n| Key | Action |\n|-----|--------|\n`;
+            for (const row of section.rows)
+                hotkeys += `| \`${this.getEditorKeyDisplay(row.action)}\` | ${row.description} |\n`;
+        }
         // Add extension-registered shortcuts
         const extensionRunner = this.session.extensionRunner;
         const shortcuts = extensionRunner.getShortcuts(this.keybindings.getEffectiveConfig());