@bastani/atomic 0.8.21-0 → 0.8.22-0

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

@@ -8,7 +8,7 @@ import * as path from "node:path";
 import { getProviders, } from "@earendil-works/pi-ai";
 import { CombinedAutocompleteProvider, Container, fuzzyFilter, getCapabilities, hyperlink, Loader, Markdown, matchesKey, ProcessTerminal, Spacer, setKeybindings, Text, TruncatedText, TUI, visibleWidth, } from "@earendil-works/pi-tui";
 import { spawn, spawnSync } from "child_process";
-import { APP_NAME, APP_TITLE, CHANGELOG_URL, ENV_OFFLINE, getEnvValue, getAgentDir, getAuthPath, getDebugLogPath, getDocsPath, getShareViewerUrl, VERSION, } from "../../config.js";
+import { APP_NAME, APP_TITLE, CHANGELOG_URL, ENV_OFFLINE, getEnvValue, setCodexFastModeEnvironmentSettings, getAgentDir, getAuthPath, getDebugLogPath, getDocsPath, getShareViewerUrl, VERSION, } from "../../config.js";
 import { parseSkillBlock, } from "../../core/agent-session.js";
 import { pickWhimsicalWorkingMessage } from "./whimsical-messages.js";
 import { SessionImportFileNotFoundError, } from "../../core/agent-session-runtime.js";
@@ -16,6 +16,7 @@ import { FooterDataProvider, } from "../../core/footer-data-provider.js";
 import { KeybindingsManager, } from "../../core/keybindings.js";
 import { createCompactionSummaryMessage } from "../../core/messages.js";
 import { defaultModelPerProvider, findExactModelReferenceMatch, resolveModelScope, } from "../../core/model-resolver.js";
+import { formatCodexFastModeModelLabel, hasSupportedCodexFastModeModel, shouldApplyCodexFastMode, } from "../../core/codex-fast-mode.js";
 import { configureHttpDispatcher } from "../../core/http-dispatcher.js";
 import { DefaultPackageManager } from "../../core/package-manager.js";
 import { BUILT_IN_PROVIDER_DISPLAY_NAMES } from "../../core/provider-display-names.js";
@@ -50,6 +51,7 @@ import { EarendilAnnouncementComponent } from "./components/earendil-announcemen
 import { ExtensionEditorComponent } from "./components/extension-editor.js";
 import { ExtensionInputComponent } from "./components/extension-input.js";
 import { ExtensionSelectorComponent } from "./components/extension-selector.js";
+import { FastModeSelectorComponent } from "./components/fast-mode-selector.js";
 import { FooterComponent, UsageMeterComponent } from "./components/footer.js";
 import { formatKeyText, keyDisplayText, keyText, } from "./components/keybinding-hints.js";
 import { LoginDialogComponent } from "./components/login-dialog.js";
@@ -94,6 +96,7 @@ function isDeadTerminalError(error) {
     return code !== undefined && DEAD_TERMINAL_ERROR_CODES.has(code);
 }
 const ANTHROPIC_SUBSCRIPTION_AUTH_WARNING = "Anthropic subscription auth is active. Third-party harness usage draws from extra usage and is billed per token, not your Claude plan limits. Manage extra usage at https://claude.ai/settings/usage.";
+const BUILTIN_SLASH_COMMAND_NAMES = new Set(BUILTIN_SLASH_COMMANDS.map((command) => command.name));
 function isAnthropicSubscriptionAuthKey(apiKey) {
     return typeof apiKey === "string" && apiKey.startsWith("sk-ant-oat");
 }
@@ -265,10 +268,9 @@ export class InteractiveMode {
         return description ? `[${sourceTag}] ${description}` : `[${sourceTag}]`;
     }
     getBuiltInCommandConflictDiagnostics(extensionRunner) {
-        const builtinNames = new Set(BUILTIN_SLASH_COMMANDS.map((command) => command.name));
         return extensionRunner
             .getRegisteredCommands()
-            .filter((command) => builtinNames.has(command.name))
+            .filter((command) => BUILTIN_SLASH_COMMAND_NAMES.has(command.name))
             .map((command) => ({
             type: "warning",
             message: command.invocationName === command.name
@@ -277,9 +279,20 @@ export class InteractiveMode {
             path: command.sourceInfo.path,
         }));
     }
+    getCodexFastModeCandidateModels() {
+        if (this.session.scopedModels.length > 0) {
+            return this.session.scopedModels
+                .map((scoped) => scoped.model)
+                .filter((model) => this.session.modelRegistry.hasConfiguredAuth(model));
+        }
+        return this.session.modelRegistry.getAvailable();
+    }
+    hasCodexFastModeSupportedModels() {
+        return hasSupportedCodexFastModeModel(this.getCodexFastModeCandidateModels());
+    }
     createBaseAutocompleteProvider() {
         // Define commands for autocomplete
-        const slashCommands = BUILTIN_SLASH_COMMANDS.map((command) => ({
+        const slashCommands = BUILTIN_SLASH_COMMANDS.filter((command) => command.name !== "fast" || this.hasCodexFastModeSupportedModels()).map((command) => ({
             name: command.name,
             description: command.description,
             getArgumentCompletions: command.getArgumentCompletions,
@@ -316,11 +329,13 @@ export class InteractiveMode {
             description: this.prefixAutocompleteDescription(cmd.description, cmd.sourceInfo),
             ...(cmd.argumentHint && { argumentHint: cmd.argumentHint }),
         }));
-        // Convert extension commands to SlashCommand format
-        const builtinCommandNames = new Set(slashCommands.map((c) => c.name));
+        // Convert extension commands to SlashCommand format. Built-in command names
+        // stay reserved even when a built-in is contextually hidden (for example,
+        // /fast without a supported OpenAI model) so extension visibility cannot
+        // change as auth/model state changes.
         const extensionCommands = this.session.extensionRunner
             .getRegisteredCommands()
-            .filter((cmd) => !builtinCommandNames.has(cmd.name))
+            .filter((cmd) => !BUILTIN_SLASH_COMMAND_NAMES.has(cmd.name))
             .map((cmd) => ({
             name: cmd.invocationName,
             description: this.prefixAutocompleteDescription(cmd.description, cmd.sourceInfo),
@@ -667,6 +682,18 @@ export class InteractiveMode {
         }
         return this.formatDisplayPath(absolutePath);
     }
+    getStartupModelLabel() {
+        const model = this.session.state.model;
+        let modelLabel = model?.id ?? "no-model";
+        if (model?.reasoning) {
+            modelLabel = `${modelLabel} ${this.session.thinkingLevel || "off"}`;
+        }
+        if (!model) {
+            return modelLabel;
+        }
+        const fastModeEnabled = shouldApplyCodexFastMode(model, this.session.settingsManager.getCodexFastModeSettings(), this.session.orchestrationContext);
+        return formatCodexFastModeModelLabel(modelLabel, fastModeEnabled);
+    }
     getStartupIdentityText() {
         const appLabel = APP_NAME.length > 0
             ? `${APP_NAME[0].toUpperCase()}${APP_NAME.slice(1)}`
@@ -674,8 +701,7 @@ export class InteractiveMode {
         const title = `${theme.bold(theme.fg("text", appLabel))} ${theme.fg("muted", `v${this.version}`)}`;
         const model = this.session.state.model;
         const provider = model ? theme.fg("dim", `(${model.provider})`) : theme.fg("dim", "(no-provider)");
-        const thinking = model?.reasoning ? ` ${this.session.thinkingLevel || "off"}` : "";
-        const modelLine = `${provider} ${theme.fg("muted", `${model?.id ?? "no-model"}${thinking}`)}`;
+        const modelLine = `${provider} ${theme.fg("muted", this.getStartupModelLabel())}`;
         const cwd = theme.fg("muted", this.formatDisplayPath(this.sessionManager.getCwd()));
         const metaLines = [title, modelLine, cwd];
         const markLines = this.getAtomicAnsiMarkLines();
@@ -1651,6 +1677,7 @@ export class InteractiveMode {
             confirm: (title, message, opts) => this.showExtensionConfirm(title, message, opts),
             input: (title, placeholder, opts) => this.showExtensionInput(title, placeholder, opts),
             notify: (message, type) => this.showExtensionNotify(message, type),
+            requestRender: () => this.ui.requestRender(),
             onTerminalInput: (handler) => this.addExtensionTerminalInputListener(handler),
             setStatus: (key, text) => this.setExtensionStatus(key, text),
             setWorkingMessage: (message) => {
@@ -1923,16 +1950,8 @@ export class InteractiveMode {
         return new Promise((resolve, reject) => {
             let component;
             let closed = false;
-            const close = (result) => {
-                if (closed)
-                    return;
-                closed = true;
-                if (isOverlay)
-                    this.ui.hideOverlay();
-                else
-                    restoreEditor();
-                // Note: both branches above already call requestRender
-                resolve(result);
+            let mounted = false;
+            const disposeComponent = () => {
                 try {
                     component?.dispose?.();
                 }
@@ -1940,10 +1959,54 @@ export class InteractiveMode {
                     /* ignore dispose errors */
                 }
             };
+            const cleanupAbortListener = () => {
+                options?.signal?.removeEventListener("abort", abortCustomUi);
+            };
+            const closeMountedUi = () => {
+                if (!mounted)
+                    return;
+                if (isOverlay)
+                    this.ui.hideOverlay();
+                else
+                    restoreEditor();
+            };
+            const close = (result) => {
+                if (closed)
+                    return;
+                closed = true;
+                cleanupAbortListener();
+                closeMountedUi();
+                resolve(result);
+                disposeComponent();
+            };
+            const rejectAndClose = (reason) => {
+                if (closed)
+                    return;
+                closed = true;
+                cleanupAbortListener();
+                closeMountedUi();
+                disposeComponent();
+                reject(reason);
+            };
+            function abortCustomUi() {
+                rejectAndClose(options?.signal?.reason ?? new Error("Extension custom UI aborted"));
+            }
+            if (options?.signal?.aborted) {
+                abortCustomUi();
+                return;
+            }
+            options?.signal?.addEventListener("abort", abortCustomUi, { once: true });
             Promise.resolve(factory(this.ui, theme, this.keybindings, close))
                 .then((c) => {
-                if (closed)
+                if (closed) {
+                    try {
+                        c.dispose?.();
+                    }
+                    catch {
+                        /* ignore dispose errors */
+                    }
                     return;
+                }
                 component = c;
                 if (isOverlay) {
                     // Resolve overlay options - can be static or dynamic function
@@ -1955,10 +2018,11 @@ export class InteractiveMode {
                             return opts;
                         }
                         // Fallback: use component's width property if available
-                        const w = component.width;
+                        const w = component?.width;
                         return w ? { width: w } : undefined;
                     };
                     const handle = this.ui.showOverlay(component, resolveOptions());
+                    mounted = true;
                     // Expose handle to caller for visibility control
                     options?.onHandle?.(handle);
                 }
@@ -1966,15 +2030,12 @@ export class InteractiveMode {
                     this.editorContainer.clear();
                     this.editorContainer.addChild(component);
                     this.ui.setFocus(component);
+                    mounted = true;
                     this.ui.requestRender();
                 }
             })
                 .catch((err) => {
-                if (closed)
-                    return;
-                if (!isOverlay)
-                    restoreEditor();
-                reject(err);
+                rejectAndClose(err);
             });
         });
     }
@@ -2083,6 +2144,11 @@ export class InteractiveMode {
                 this.editor.setText("");
                 return;
             }
+            if (text === "/fast") {
+                this.editor.setText("");
+                this.showFastModeSelector();
+                return;
+            }
             if (text === "/scoped-models") {
                 this.editor.setText("");
                 await this.showModelsSelector();
@@ -2379,11 +2445,13 @@ export class InteractiveMode {
                     this.streamingMessage = event.message;
                     let errorMessage;
                     if (this.streamingMessage.stopReason === "aborted") {
+                        const existingAbortMessage = this.streamingMessage.errorMessage && this.streamingMessage.errorMessage !== "Request was aborted"
+                            ? this.streamingMessage.errorMessage
+                            : undefined;
                         const retryAttempt = this.session.retryAttempt;
-                        errorMessage =
-                            retryAttempt > 0
-                                ? `Aborted after ${retryAttempt} retry attempt${retryAttempt > 1 ? "s" : ""}`
-                                : "Operation aborted";
+                        errorMessage = existingAbortMessage ?? (retryAttempt > 0
+                            ? `Aborted after ${retryAttempt} retry attempt${retryAttempt > 1 ? "s" : ""}`
+                            : "Operation aborted");
                         this.streamingMessage.errorMessage = errorMessage;
                     }
                     this.streamingComponent.updateContent(this.streamingMessage);
@@ -2695,6 +2763,7 @@ export class InteractiveMode {
             }
             default: {
                 const _exhaustive = message;
+                void _exhaustive;
             }
         }
     }
@@ -3259,6 +3328,36 @@ export class InteractiveMode {
         this.ui.setFocus(focus);
         this.ui.requestRender();
     }
+    showFastModeSelector() {
+        if (!this.hasCodexFastModeSupportedModels()) {
+            this.showWarning("Codex fast mode requires an available openai/* or openai-codex/* model.");
+            return;
+        }
+        this.showSelector((done) => {
+            let pendingStatusMessage;
+            const selector = new FastModeSelectorComponent(this.settingsManager.getCodexFastModeSettings(), {
+                onChange: (settings, changedRow) => {
+                    this.settingsManager.setCodexFastModeSettings({ [changedRow]: settings[changedRow] });
+                    const effectiveSettings = this.settingsManager.getCodexFastModeSettings();
+                    setCodexFastModeEnvironmentSettings(effectiveSettings);
+                    this.footer.invalidate();
+                    this.refreshBuiltInHeader();
+                    const changedLabel = changedRow === "chat" ? "Chat" : "Workflow";
+                    const changedState = effectiveSettings[changedRow] ? "on" : "off";
+                    pendingStatusMessage = `${changedLabel} fast mode ${changedState}`;
+                },
+                onCancel: async () => {
+                    await this.settingsManager.flush();
+                    done();
+                    if (pendingStatusMessage) {
+                        this.showStatus(pendingStatusMessage);
+                    }
+                    this.ui.requestRender();
+                },
+            });
+            return { component: selector, focus: selector };
+        });
+    }
     showSettingsSelector() {
         this.showSelector((done) => {
             const selector = new SettingsSelectorComponent({
@@ -3554,6 +3653,7 @@ export class InteractiveMode {
                 this.session.setScopedModels([]);
             }
             await this.updateAvailableProviderCount();
+            this.setupAutocompleteProvider();
             this.ui.requestRender();
         };
         this.showSelector((done) => {
@@ -3900,6 +4000,7 @@ export class InteractiveMode {
                     this.session.modelRegistry.authStorage.logout(providerOption.id);
                     this.session.modelRegistry.refresh();
                     await this.updateAvailableProviderCount();
+                    this.setupAutocompleteProvider();
                     const message = providerOption.authType === "oauth"
                         ? `Logged out of ${providerOption.name}`
                         : `Removed stored API key for ${providerOption.name}. Environment variables and models.json config are unchanged.`;
@@ -3950,6 +4051,7 @@ export class InteractiveMode {
             }
         }
         await this.updateAvailableProviderCount();
+        this.setupAutocompleteProvider();
         this.footer.invalidate();
         this.updateEditorBorderColor();
         if (selectedModel) {