apex-code 0.5.1 → 0.6.0

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

@@ -12,7 +12,7 @@ import { CombinedAutocompleteProvider, Container, fuzzyFilter, getCapabilities,
 import chalk from "chalk";
 import { spawn, spawnSync } from "child_process";
 import { APP_NAME, APP_TITLE, CONFIG_DIR_NAME, getAgentDir, getAuthPath, getDebugLogPath, getDocsPath, getShareViewerUrl, VERSION, } from "../../config.js";
-import { parseSkillBlock } from "../../core/agent-session.js";
+import { parseSkillBlock, } from "../../core/agent-session.js";
 import { SessionImportFileNotFoundError } from "../../core/agent-session-runtime.js";
 import { CACHE_TTL_MS, collectCacheMisses, computeCacheWaste, detectCacheMiss, } from "../../core/cache-stats.js";
 import { formatCacheWarmingStatus, formatCacheWarmingUsage } from "../../core/cache-warmer.js";
@@ -81,10 +81,12 @@ import { PERMISSION_MODE_OVERRIDE_HINTS, SettingsSelectorComponent } from "./com
 import { SkillInvocationMessageComponent } from "./components/skill-invocation-message.js";
 import { ApexSplashHeader } from "./components/splash-header.js";
 import { BranchSummaryStatusIndicator, CompactionStatusIndicator, IdleStatus, RetryStatusIndicator, WorkingStatusIndicator, } from "./components/status-indicator.js";
+import { TaskPanelComponent } from "./components/task-panel.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 { TurnSummaryComponent } from "./components/turn-summary.js";
 import { UserMessageComponent } from "./components/user-message.js";
 import { UserMessageSelectorComponent } from "./components/user-message-selector.js";
 import { editInExternalEditor } from "./external-editor.js";
@@ -273,6 +275,8 @@ export class InteractiveMode {
     runtimeHost;
     /** Settled once the most recent /settings permission-mode write has hit disk. */
     pendingPermissionModeWrite;
+    pendingPermissionModeCycle;
+    permissionModeCycleGeneration = 0;
     renderer;
     ui;
     mainScreenRenderState;
@@ -372,6 +376,7 @@ export class InteractiveMode {
     retriedAttempt;
     /** Set when the user presses Escape during a retry wait, which is the only way to cancel one. */
     retryCancelled = false;
+    turnStartedAt;
     // Messages queued while compaction is running
     compactionQueuedMessages = [];
     // Shutdown state
@@ -387,6 +392,8 @@ export class InteractiveMode {
     extensionWidgetsAbove = new Map();
     extensionWidgetsBelow = new Map();
     widgetContainerAbove;
+    extensionWidgetContainerAbove;
+    taskPanel;
     widgetContainerBelow;
     // Custom footer from extension (undefined = use built-in footer)
     customFooter = undefined;
@@ -415,13 +422,13 @@ export class InteractiveMode {
     get settingsManager() {
         return this.session.settingsManager;
     }
-    offerFirstUseHint(id) {
+    offerFirstUseHint(id, text) {
         this.firstUseHints ??= new FirstUseHints(this.settingsManager.getFirstUseHints());
         const hint = this.firstUseHints.offer(id);
         if (!hint)
             return;
         this.settingsManager.setFirstUseHints(this.firstUseHints.getSeen());
-        this.chatContainer.addChild(new Text(theme.fg("dim", hint), this.outputPad, 0));
+        this.chatContainer.addChild(new Text(theme.fg("dim", text ?? hint), this.outputPad, 0));
     }
     constructor(runtimeHost, options = {}) {
         this.runtimeHost = runtimeHost;
@@ -457,6 +464,13 @@ export class InteractiveMode {
         this.pendingMessagesContainer = new Container();
         this.statusContainer = new Container();
         this.widgetContainerAbove = new Container();
+        this.extensionWidgetContainerAbove = new Container();
+        this.taskPanel = new TaskPanelComponent({
+            getSession: () => this.session,
+            getSettings: () => this.settingsManager,
+        });
+        this.widgetContainerAbove.addChild(this.taskPanel);
+        this.widgetContainerAbove.addChild(this.extensionWidgetContainerAbove);
         this.widgetContainerBelow = new Container();
         this.keybindings = KeybindingsManager.create();
         setKeybindings(this.keybindings);
@@ -1619,6 +1633,7 @@ export class InteractiveMode {
                         customInstructions: options?.customInstructions,
                         replaceInstructions: options?.replaceInstructions,
                         label: options?.label,
+                        workspacePolicy: options?.workspacePolicy,
                     });
                     if (result.cancelled) {
                         return { cancelled: true };
@@ -1688,10 +1703,13 @@ export class InteractiveMode {
         }
     }
     async rebindCurrentSession(options = {}) {
+        this.permissionModeCycleGeneration = (this.permissionModeCycleGeneration ?? 0) + 1;
+        this.pendingPermissionModeCycle = undefined;
         const session = this.session;
         this.unsubscribe?.();
         this.unsubscribe = undefined;
         this.applyRuntimeSettings();
+        this.taskPanel?.rebind();
         if (options.renderBeforeBind) {
             this.renderCurrentSessionState();
             this.subscribeToAgent();
@@ -2027,13 +2045,17 @@ export class InteractiveMode {
     }
     // Maximum total widget lines to prevent viewport overflow
     static MAX_WIDGET_LINES = 10;
+    toggleTaskPanel() {
+        this.taskPanel.toggle();
+        this.ui.requestRender();
+    }
     /**
      * Render all extension widgets to the widget container.
      */
     renderWidgets() {
         if (!this.widgetContainerAbove || !this.widgetContainerBelow)
             return;
-        this.renderWidgetContainer(this.widgetContainerAbove, this.extensionWidgetsAbove, true, true);
+        this.renderWidgetContainer(this.extensionWidgetContainerAbove ?? this.widgetContainerAbove, this.extensionWidgetsAbove, true, true);
         this.renderWidgetContainer(this.widgetContainerBelow, this.extensionWidgetsBelow, false, false);
         this.ui.requestRender();
     }
@@ -2599,12 +2621,14 @@ export class InteractiveMode {
         this.defaultEditor.onCtrlD = () => this.handleCtrlD();
         this.defaultEditor.onAction("app.suspend", () => this.handleCtrlZ());
         this.defaultEditor.onAction("app.thinking.cycle", () => this.cycleThinkingLevel());
+        this.defaultEditor.onAction("app.permissionMode.cycle", () => this.cyclePermissionMode());
         this.defaultEditor.onAction("app.model.cycleForward", () => this.cycleModel("forward"));
         this.defaultEditor.onAction("app.model.cycleBackward", () => this.cycleModel("backward"));
         // Global debug handler on TUI (works regardless of focus)
         this.ui.onDebug = () => this.handleDebugCommand();
         this.defaultEditor.onAction("app.model.select", () => this.showModelSelector());
         this.defaultEditor.onAction("app.tools.expand", () => this.cycleChatDetail());
+        this.defaultEditor.onAction("app.tasks.toggle", () => this.toggleTaskPanel());
         this.defaultEditor.onAction("app.thinking.toggle", () => this.toggleThinkingBlockVisibility());
         this.defaultEditor.onAction("app.editor.external", () => void this.handleOpenExternalEditor());
         this.defaultEditor.onAction("app.message.copy", () => void this.handleCopyCommand({ flashConfirmation: true, preferSelection: true }));
@@ -2690,6 +2714,11 @@ export class InteractiveMode {
                 await this.showConfigurationIndex();
                 return;
             }
+            if (text === "/tasks") {
+                this.editor.setText("");
+                this.toggleTaskPanel();
+                return;
+            }
             if (text === "/settings") {
                 this.editor.setText("");
                 await this.showSettingsSelector();
@@ -2907,6 +2936,7 @@ export class InteractiveMode {
             await this.init();
         }
         this.footer.invalidate();
+        this.taskPanel?.handleEvent(event);
         switch (event.type) {
             case "agent_start":
                 this.clearPendingTools();
@@ -2921,6 +2951,7 @@ export class InteractiveMode {
                 }
                 break;
             case "turn_start":
+                this.turnStartedAt ??= Date.now();
                 if (this.settingsManager.getShowTerminalProgress()) {
                     this.ui.terminal.setProgress(true);
                 }
@@ -3108,6 +3139,8 @@ export class InteractiveMode {
                 break;
             }
             case "tool_execution_end": {
+                if (event.toolName === "plan_present" && !event.isError)
+                    await this.refreshFooterPermissionMode();
                 const component = this.pendingTools.get(event.toolCallId);
                 if (component) {
                     component.updateResult({ ...event.result, isError: event.isError });
@@ -3127,9 +3160,20 @@ export class InteractiveMode {
                     this.streamingMessage = undefined;
                 }
                 this.clearPendingTools();
+                if (!event.willRetry) {
+                    const kind = event.stopReason?.kind ??
+                        [...event.messages].reverse().find((message) => message.role === "assistant")?.stopReason;
+                    this.finishTurnSummary(kind === "aborted"
+                        ? "aborted"
+                        : kind === "completed" || kind === "stop" || kind === undefined
+                            ? "completed"
+                            : undefined);
+                }
                 this.ui.requestRender();
                 break;
             case "agent_settled":
+                this.turnStartedAt = undefined;
+                this.ui.requestRender();
                 await this.checkShutdownRequested();
                 break;
             case "compaction_start": {
@@ -3229,6 +3273,8 @@ export class InteractiveMode {
                 }
                 this.retriedAttempt = undefined;
                 this.retryCancelled = false;
+                if (event.aborted)
+                    this.finishTurnSummary("aborted");
                 this.ui.requestRender();
                 break;
             }
@@ -3256,6 +3302,16 @@ export class InteractiveMode {
             }
         }
     }
+    finishTurnSummary(outcome) {
+        const startedAt = this.turnStartedAt;
+        this.turnStartedAt = undefined;
+        if (startedAt === undefined || outcome === undefined)
+            return;
+        const elapsedMs = Date.now() - startedAt;
+        if (elapsedMs < 1000)
+            return;
+        this.chatContainer.addChild(new TurnSummaryComponent(elapsedMs, outcome));
+    }
     /** Extract text content from a user message */
     getUserMessageText(message) {
         if (message.role !== "user")
@@ -4278,25 +4334,61 @@ export class InteractiveMode {
      * would under-report bypassPermissions, which is the one direction that matters.
      */
     async refreshFooterPermissionMode() {
-        const resolution = await this.session.getPermissionMode();
-        this.footer.setPermissionMode(resolution?.mode ?? "default");
+        const session = this.session;
+        const resolution = await session.getPermissionMode();
+        if (this.session !== session)
+            return;
+        this.footer.setPermissionMode(resolution?.mode ?? "default", resolution?.origin);
         if (this.isInitialized) {
             this.ui.requestRender();
         }
     }
+    cyclePermissionMode() {
+        const session = this.session;
+        const generation = this.permissionModeCycleGeneration ?? 0;
+        const isCurrent = () => this.session === session && (this.permissionModeCycleGeneration ?? 0) === generation;
+        const cycle = (this.pendingPermissionModeCycle ?? Promise.resolve())
+            .then(async () => {
+            if (!isCurrent())
+                return;
+            const modes = await session.getInteractivePermissionModeCycle();
+            if (!isCurrent() || modes.length === 0)
+                return;
+            const current = await session.getPermissionMode();
+            if (!isCurrent())
+                return;
+            const index = modes.indexOf(current?.mode ?? "default");
+            const resolution = await session.setInteractivePermissionMode(modes[(index + 1) % modes.length]);
+            if (!isCurrent() || !resolution)
+                return;
+            this.footer.setPermissionMode(resolution.mode, resolution.origin);
+            this.offerFirstUseHint("permission-mode-cycle", `${keyDisplayText("app.permissionMode.cycle")} now cycles permission modes for this session; ${keyDisplayText("app.thinking.cycle")} cycles thinking level.`);
+            this.ui.requestRender();
+        })
+            .catch((error) => {
+            if (!isCurrent())
+                return;
+            this.showError(`Permission mode could not be changed: ${error instanceof Error ? error.message : String(error)}`);
+        });
+        this.pendingPermissionModeCycle = cycle;
+        return cycle;
+    }
     /**
      * Writes the mode to user scope, then reports what is actually in force. The
      * write can be outranked by `--permission-mode` or a project file, and a
      * settings row that silently does nothing is worse than no row at all.
      */
     async applyPermissionMode(mode) {
+        const session = this.session;
         let resolution;
         try {
-            const write = this.session.setPermissionMode(mode);
+            const write = session.setPermissionMode(mode);
             this.pendingPermissionModeWrite = write.then(() => { }, () => { });
             resolution = await write;
         }
         catch (error) {
+            if (this.session !== session)
+                return;
             // Never swallow this. A permission mode the user believes they set, and
             // did not, is the one outcome worse than the prompt they were escaping.
             const detail = error instanceof Error ? error.message : String(error);
@@ -4304,17 +4396,27 @@ export class InteractiveMode {
             this.ui.requestRender();
             return;
         }
-        if (!resolution)
+        if (this.session !== session || !resolution)
             return;
-        const outranked = resolution.mode !== mode;
+        const outranked = resolution.mode !== mode || resolution.origin === "interactive";
         const source = PERMISSION_MODE_OVERRIDE_HINTS[resolution.origin] ?? resolution.origin;
         const line = outranked
             ? `Permission mode saved as ${mode}, but ${source} sets ${resolution.mode} for this session.`
             : `Permission mode set to ${mode}.`;
         this.chatContainer.addChild(new Text(theme.fg(outranked ? "warning" : "muted", line), 1, 0));
-        this.footer.setPermissionMode(resolution.mode);
+        this.footer.setPermissionMode(resolution.mode, resolution.origin);
         this.ui.requestRender();
     }
+    setTaskListToolDefault(enabled) {
+        if (this.settingsManager.getProjectSettings().defaultTools !== undefined)
+            return;
+        const names = this.settingsManager.getDefaultTools() ?? this.session.getImplicitDefaultToolNames();
+        this.settingsManager.setDefaultTools(enabled
+            ? names.includes("todo_write")
+                ? names
+                : [...names, "todo_write"]
+            : names.filter((name) => name !== "todo_write"));
+    }
     async showSettingsSelector() {
         const permissionMode = await this.session.getPermissionMode();
         this.showSelector((done) => {
@@ -4324,6 +4426,10 @@ export class InteractiveMode {
             const defaultModel = defaultProvider && defaultModelId ? `${defaultProvider}/${defaultModelId}` : "not set";
             selector = new SettingsSelectorComponent({
                 autoCompact: this.session.autoCompactionEnabled,
+                taskListTool: {
+                    enabled: (this.settingsManager.getDefaultTools() ?? this.session.getImplicitDefaultToolNames()).includes("todo_write"),
+                    projectControlled: this.settingsManager.getProjectSettings().defaultTools !== undefined,
+                },
                 defaultModel,
                 currentModel: this.session.model,
                 availableDefaultModels: this.session.modelRuntime.getAvailableSnapshot(),
@@ -4366,6 +4472,7 @@ export class InteractiveMode {
                 fullscreenCopyOnSelect: this.settingsManager.getFullscreenCopyOnSelect(),
                 warnings: this.settingsManager.getWarnings(),
             }, {
+                onTaskListToolChange: (enabled) => this.setTaskListToolDefault(enabled),
                 onAutoCompactChange: (enabled) => {
                     this.session.setAutoCompactionEnabled(enabled);
                     this.footer.setAutoCompactEnabled(enabled);
@@ -4982,6 +5089,29 @@ export class InteractiveMode {
                     this.showError("Wait for the current compaction or tree navigation to finish before navigating the session tree.");
                     return;
                 }
+                const workspacePreview = await this.session.previewTreeWorkspace(entryId);
+                if (this.session.isCompacting) {
+                    this.showError("Wait for the current compaction or tree navigation to finish before navigating the session tree.");
+                    return;
+                }
+                let workspacePolicy = "keep";
+                if (workspacePreview.state === "differs") {
+                    const choice = await this.showExtensionSelector("Restore workspace files?", [
+                        "Keep current files",
+                        "Restore files to this point",
+                        "Cancel",
+                    ]);
+                    if (choice === undefined || choice === "Cancel") {
+                        this.showStatus("Navigation cancelled");
+                        return;
+                    }
+                    if (choice === "Restore files to this point")
+                        workspacePolicy = "restore";
+                }
+                if (this.session.isCompacting) {
+                    this.showError("Wait for the current compaction or tree navigation to finish before navigating the session tree.");
+                    return;
+                }
                 // Set up escape handler and status indicator if summarizing
                 let showingSummaryIndicator = false;
                 const originalOnEscape = this.defaultEditor.onEscape;
@@ -4998,6 +5128,7 @@ export class InteractiveMode {
                     const result = await this.session.navigateTree(entryId, {
                         summarize: wantsSummary,
                         customInstructions,
+                        workspacePolicy,
                     });
                     if (result.aborted) {
                         // Summarization aborted - re-show tree selector with same selection
@@ -5015,7 +5146,22 @@ export class InteractiveMode {
                     if (result.editorText && !this.editor.getText().trim()) {
                         this.editor.setText(result.editorText);
                     }
-                    this.showStatus("Navigated to selected point");
+                    const workspace = result.workspace;
+                    if (workspace?.outcome === "restored") {
+                        this.showStatus(workspace.preRestoreCheckpoint
+                            ? "Restored files · pre-restore checkpoint saved"
+                            : "Restored files");
+                    }
+                    else if (workspace?.outcome === "failed") {
+                        this.showStatus(workspace.warnings.join(" · ") || "Workspace restore failed");
+                    }
+                    else if (workspace?.outcome === "missing-checkpoint" ||
+                        workspacePreview.state === "no-checkpoint") {
+                        this.showStatus("Files unchanged · no checkpoint for this point");
+                    }
+                    else {
+                        this.showStatus("Files unchanged");
+                    }
                     void this.flushCompactionQueue({ willRetry: false });
                 }
                 catch (error) {
@@ -5600,6 +5746,8 @@ export class InteractiveMode {
             this.showWarning("Wait for compaction to finish before reloading.");
             return;
         }
+        this.permissionModeCycleGeneration = (this.permissionModeCycleGeneration ?? 0) + 1;
+        this.pendingPermissionModeCycle = undefined;
         this.resetExtensionUI();
         const reloadBox = new Container();
         reloadBox.addChild(new DynamicBorder());
@@ -5665,6 +5813,7 @@ export class InteractiveMode {
                 dismissReloadBox(previousEditor);
             }
             this.showError(`Reload failed: ${error instanceof Error ? error.message : String(error)}`);
+            await this.refreshFooterPermissionMode();
         }
     }
     async handleExportCommand(text) {
@@ -6030,6 +6179,7 @@ export class InteractiveMode {
         const exit = this.getAppKeyDisplay("app.exit");
         const suspend = this.getAppKeyDisplay("app.suspend");
         const cycleThinkingLevel = this.getAppKeyDisplay("app.thinking.cycle");
+        const cyclePermissionMode = this.getAppKeyDisplay("app.permissionMode.cycle");
         const cycleModelForward = this.getAppKeyDisplay("app.model.cycleForward");
         const selectModel = this.getAppKeyDisplay("app.model.select");
         const expandTools = this.getAppKeyDisplay("app.tools.expand");
@@ -6074,6 +6224,7 @@ export class InteractiveMode {
 | \`${exit}\` | Exit (when editor is empty) |
 | \`${suspend}\` | Suspend to background |
 | \`${cycleThinkingLevel}\` | Cycle thinking level |
+| \`${cyclePermissionMode}\` | Cycle permission mode for this session |
 | \`${cycleModelForward}\` / \`${cycleModelBackward}\` | Cycle models |
 | \`${selectModel}\` | Open model selector |
 | \`${expandTools}\` | Cycle conversation detail (overview / details / all) |