@bastani/atomic 0.8.28-alpha.1 → 0.8.28-alpha.3

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

@@ -5,6 +5,7 @@
 import * as fs from "node:fs";
 import * as os from "node:os";
 import * as path from "node:path";
+import chalk from "chalk";
 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";
@@ -23,7 +24,7 @@ import { formatMissingSessionCwdPrompt, MissingSessionCwdError, } from "../../co
 import { SessionManager, } from "../../core/session-manager.js";
 import { BUILTIN_SLASH_COMMANDS } from "../../core/slash-commands.js";
 import { isInstallTelemetryEnabled } from "../../core/telemetry.js";
-import { getChangelogPath, getEntriesForVersion, getNewEntries, parseChangelog, } from "../../utils/changelog.js";
+import { getChangelogPath, getEntriesForVersion, getNewEntries, normalizeChangelogLinks, parseChangelog, } from "../../utils/changelog.js";
 import { copyToClipboard } from "../../utils/clipboard.js";
 import { combineQueuedMessagesForEditor, openExternalEditorForText, pasteClipboardImageToEditor, } from "./chat-input-actions.js";
 import { parseGitUrl } from "../../utils/git.js";
@@ -63,6 +64,8 @@ import { SettingsSelectorComponent } from "./components/settings-selector.js";
 import { SkillInvocationMessageComponent } from "./components/skill-invocation-message.js";
 import { ToolExecutionComponent } from "./components/tool-execution.js";
 import { TreeSelectorComponent } from "./components/tree-selector.js";
+import { TrustSelectorComponent } from "./components/trust-selector.js";
+import { hasProjectConfigDir, ProjectTrustStore } from "../../core/trust-manager.js";
 import { UserMessageComponent } from "./components/user-message.js";
 import { UserMessageSelectorComponent } from "./components/user-message-selector.js";
 import { getAvailableThemes, getAvailableThemesWithPaths, getEditorTheme, getMarkdownTheme, getThemeByName, initTheme, onThemeChange, setRegisteredThemes, setTheme, setThemeInstance, stopThemeWatcher, Theme, theme, } from "./theme/theme.js";
@@ -106,6 +109,27 @@ function isUnknownModel(model) {
         model.id === "unknown" &&
         model.api === "unknown");
 }
+function quoteIfNeeded(value) {
+    if (value.length > 0 && !/[^a-zA-Z0-9_\-./~:@]/.test(value)) {
+        return value;
+    }
+    return `'${value.replace(/'/g, `'\\''`)}'`;
+}
+export function formatResumeCommand(sessionManager) {
+    if (!process.stdout.isTTY)
+        return undefined;
+    if (!sessionManager.isPersisted())
+        return undefined;
+    const sessionFile = sessionManager.getSessionFile();
+    if (!sessionFile || !fs.existsSync(sessionFile))
+        return undefined;
+    const args = [APP_NAME];
+    if (!sessionManager.usesDefaultSessionDir()) {
+        args.push("--session-dir", quoteIfNeeded(sessionManager.getSessionDir()));
+    }
+    args.push("--session", sessionManager.getSessionId());
+    return args.join(" ");
+}
 function hasDefaultModelProvider(providerId) {
     return providerId in defaultModelPerProvider;
 }
@@ -137,6 +161,7 @@ export class InteractiveMode {
     constructor(runtimeHost, options = {}) {
         this.autocompleteProviderWrappers = [];
         this.isInitialized = false;
+        this.pendingUserInputs = [];
         this.loadingAnimation = undefined;
         this.workingMessage = undefined;
         this.workingVisible = true;
@@ -200,6 +225,7 @@ export class InteractiveMode {
          */
         this.isShuttingDown = false;
         this.options = options;
+        this.autoTrustOnReloadCwd = options.autoTrustOnReloadCwd;
         this.runtimeHost = runtimeHost;
         this.runtimeHost.setBeforeSessionInvalidate(() => {
             this.resetExtensionUI();
@@ -639,7 +665,7 @@ export class InteractiveMode {
         if (currentEntries.length > 0) {
             this.settingsManager.setLastChangelogVersion(VERSION);
             this.reportInstallTelemetry(VERSION);
-            return currentEntries.map((e) => e.content).join("\n\n");
+            return currentEntries.map((e) => normalizeChangelogLinks(e.content, e)).join("\n\n");
         }
         return undefined;
     }
@@ -1236,6 +1262,7 @@ export class InteractiveMode {
         const uiContext = this.createExtensionUIContext();
         await this.session.bindExtensions({
             uiContext,
+            mode: "tui",
             commandContextActions: {
                 waitForIdle: () => this.session.agent.waitForIdle(),
                 newSession: async (options) => {
@@ -1372,12 +1399,14 @@ export class InteractiveMode {
         // Create a context for shortcut handlers
         const createContext = () => ({
             ui: this.createExtensionUIContext(),
+            mode: "tui",
             hasUI: true,
             cwd: this.sessionManager.getCwd(),
             sessionManager: this.sessionManager,
             modelRegistry: this.session.modelRegistry,
             model: this.session.model,
             isIdle: () => !this.session.isStreaming,
+            isProjectTrusted: () => this.session.settingsManager.isProjectTrusted(),
             signal: this.session.agent.signal,
             abort: () => this.session.abort(),
             hasPendingMessages: () => this.session.pendingMessageCount > 0,
@@ -1679,6 +1708,20 @@ export class InteractiveMode {
         }
         this.extensionTerminalInputUnsubscribers.clear();
     }
+    createProjectTrustContext(cwd) {
+        const ui = this.createExtensionUIContext();
+        return {
+            cwd,
+            mode: "tui",
+            hasUI: true,
+            ui: {
+                select: ui.select,
+                confirm: ui.confirm,
+                input: ui.input,
+                notify: ui.notify,
+            },
+        };
+    }
     /**
      * Create the ExtensionUIContext for extensions.
      */
@@ -2233,6 +2276,11 @@ export class InteractiveMode {
                 this.editor.setText("");
                 return;
             }
+            if (text === "/trust") {
+                this.showTrustSelector();
+                this.editor.setText("");
+                return;
+            }
             if (text === "/login") {
                 this.showOAuthSelector("login");
                 this.editor.setText("");
@@ -2334,6 +2382,9 @@ export class InteractiveMode {
             if (this.onInputCallback) {
                 this.onInputCallback(text);
             }
+            else {
+                this.pendingUserInputs.push(text);
+            }
             this.editor.addToHistory?.(text);
         };
     }
@@ -2863,6 +2914,10 @@ export class InteractiveMode {
         }
     }
     async getUserInput() {
+        const queuedInput = this.pendingUserInputs.shift();
+        if (queuedInput !== undefined) {
+            return queuedInput;
+        }
         return new Promise((resolve) => {
             this.onInputCallback = (text) => {
                 this.onInputCallback = undefined;
@@ -2892,16 +2947,28 @@ export class InteractiveMode {
         // Only called when editor is empty (enforced by CustomEditor)
         void this.shutdown();
     }
-    async shutdown() {
+    async shutdown(options) {
         if (this.isShuttingDown)
             return;
         this.isShuttingDown = true;
         this.unregisterSignalHandlers();
+        if (options?.fromSignal) {
+            await this.runtimeHost.dispose();
+            // Drain any in-flight Kitty key release events before stopping.
+            // This prevents escape sequences from leaking to the parent shell over slow SSH.
+            await this.ui.terminal.drainInput(1000);
+            this.stop();
+            process.exit(0);
+        }
         // Drain any in-flight Kitty key release events before stopping.
         // This prevents escape sequences from leaking to the parent shell over slow SSH.
         await this.ui.terminal.drainInput(1000);
         this.stop();
         await this.runtimeHost.dispose();
+        const resumeCommand = formatResumeCommand(this.sessionManager);
+        if (resumeCommand) {
+            process.stdout.write(`${chalk.dim("To resume this session:")} ${resumeCommand}\n`);
+        }
         process.exit(0);
     }
     emergencyTerminalExit() {
@@ -2960,11 +3027,8 @@ export class InteractiveMode {
         }
         for (const signal of signals) {
             const handler = () => {
-                if (signal === "SIGHUP") {
-                    this.emergencyTerminalExit();
-                }
                 killTrackedDetachedChildren();
-                void this.shutdown();
+                void this.shutdown({ fromSignal: true });
             };
             process.prependListener(signal, handler);
             this.signalCleanupHandlers.push(() => process.off(signal, handler));
@@ -3438,6 +3502,7 @@ export class InteractiveMode {
                 editorPaddingX: this.settingsManager.getEditorPaddingX(),
                 autocompleteMaxVisible: this.settingsManager.getAutocompleteMaxVisible(),
                 quietStartup: this.settingsManager.getQuietStartup(),
+                defaultProjectTrust: this.settingsManager.getDefaultProjectTrust(),
                 clearOnShrink: this.settingsManager.getClearOnShrink(),
                 showTerminalProgress: this.settingsManager.getShowTerminalProgress(),
                 warnings: this.settingsManager.getWarnings(),
@@ -3526,6 +3591,9 @@ export class InteractiveMode {
                 onQuietStartupChange: (enabled) => {
                     this.settingsManager.setQuietStartup(enabled);
                 },
+                onDefaultProjectTrustChange: (defaultProjectTrust) => {
+                    this.settingsManager.setDefaultProjectTrust(defaultProjectTrust);
+                },
                 onDoubleEscapeActionChange: (action) => {
                     this.settingsManager.setDoubleEscapeAction(action);
                 },
@@ -3786,6 +3854,51 @@ export class InteractiveMode {
             this.showError(error instanceof Error ? error.message : String(error));
         }
     }
+    maybeSaveImplicitProjectTrustAfterReload() {
+        const cwd = this.sessionManager.getCwd();
+        if (this.autoTrustOnReloadCwd !== cwd) {
+            return false;
+        }
+        if (!this.settingsManager.isProjectTrusted() || !hasProjectConfigDir(cwd)) {
+            return false;
+        }
+        const trustStore = new ProjectTrustStore(this.runtimeHost.services.agentDir);
+        try {
+            if (trustStore.get(cwd) !== null) {
+                this.autoTrustOnReloadCwd = undefined;
+                return false;
+            }
+            trustStore.set(cwd, true);
+            this.autoTrustOnReloadCwd = undefined;
+            return true;
+        }
+        catch (error) {
+            this.showWarning(`Could not save project trust after reload: ${error instanceof Error ? error.message : String(error)}`);
+            return false;
+        }
+    }
+    showTrustSelector() {
+        const cwd = this.sessionManager.getCwd();
+        const trustStore = new ProjectTrustStore(this.runtimeHost.services.agentDir);
+        const savedDecision = trustStore.getEntry(cwd);
+        this.showSelector((done) => {
+            const selector = new TrustSelectorComponent({
+                cwd,
+                savedDecision,
+                projectTrusted: this.settingsManager.isProjectTrusted(),
+                onSelect: (selection) => {
+                    trustStore.setMany(selection.updates);
+                    done();
+                    this.showStatus(`Saved trust decision: ${selection.trusted ? "trusted" : "untrusted"}. Restart ${APP_NAME} for this to take effect.`);
+                },
+                onCancel: () => {
+                    done();
+                    this.ui.requestRender();
+                },
+            });
+            return { component: selector, focus: selector };
+        });
+    }
     showTreeSelector(initialSelectedId) {
         const tree = this.sessionManager.getTree();
         const realLeafId = this.sessionManager.getLeafId();
@@ -3886,7 +3999,9 @@ export class InteractiveMode {
     }
     showSessionSelector() {
         this.showSelector((done) => {
-            const selector = new SessionSelectorComponent((onProgress) => SessionManager.list(this.sessionManager.getCwd(), this.sessionManager.getSessionDir(), onProgress), SessionManager.listAll, async (sessionPath) => {
+            const selector = new SessionSelectorComponent((onProgress) => SessionManager.list(this.sessionManager.getCwd(), this.sessionManager.getSessionDir(), onProgress), (onProgress) => this.sessionManager.usesDefaultSessionDir()
+                ? SessionManager.listAll(onProgress)
+                : SessionManager.listAll(this.sessionManager.getSessionDir(), onProgress), async (sessionPath) => {
                 done();
                 await this.handleResumeSession(sessionPath);
             }, () => {
@@ -3917,6 +4032,7 @@ export class InteractiveMode {
         try {
             const result = await this.runtimeHost.switchSession(sessionPath, {
                 withSession: options?.withSession,
+                projectTrustContextFactory: (cwd) => this.createProjectTrustContext(cwd),
             });
             if (result.cancelled) {
                 return result;
@@ -3935,6 +4051,7 @@ export class InteractiveMode {
                 const result = await this.runtimeHost.switchSession(sessionPath, {
                     cwdOverride: selectedCwd,
                     withSession: options?.withSession,
+                    projectTrustContextFactory: (cwd) => this.createProjectTrustContext(cwd),
                 });
                 if (result.cancelled) {
                     return result;
@@ -4341,10 +4458,14 @@ export class InteractiveMode {
             this.setupExtensionShortcuts(runner);
             this.rebuildChatFromMessages();
             dismissReloadBox(this.editor);
+            const savedImplicitProjectTrust = this.maybeSaveImplicitProjectTrustAfterReload();
             this.showLoadedResources({
                 force: false,
                 showDiagnosticsWhenQuiet: true,
             });
+            if (savedImplicitProjectTrust) {
+                this.showStatus("Saved project trust for future sessions");
+            }
             const modelsJsonError = this.session.modelRegistry.getError();
             if (modelsJsonError) {
                 this.showError(`models.json error: ${modelsJsonError}`);
@@ -4606,7 +4727,7 @@ export class InteractiveMode {
         const changelogMarkdown = allEntries.length > 0
             ? allEntries
                 .reverse()
-                .map((e) => e.content)
+                .map((e) => normalizeChangelogLinks(e.content, e))
                 .join("\n\n")
             : "No changelog entries found.";
         this.chatContainer.addChild(new Spacer(1));