@earendil-works/pi-coding-agent 0.78.0 → 0.79.0

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

@@ -24,6 +24,7 @@ import { formatMissingSessionCwdPrompt, MissingSessionCwdError } from "../../cor
 import { SessionManager } from "../../core/session-manager.js";
 import { BUILTIN_SLASH_COMMANDS } from "../../core/slash-commands.js";
 import { isInstallTelemetryEnabled } from "../../core/telemetry.js";
+import { hasProjectConfigDir, hasProjectTrustInputs, ProjectTrustStore } from "../../core/trust-manager.js";
 import { getChangelogPath, getNewEntries, parseChangelog } from "../../utils/changelog.js";
 import { copyToClipboard } from "../../utils/clipboard.js";
 import { extensionForImageMimeType, readClipboardImage } from "../../utils/clipboard-image.js";
@@ -59,6 +60,7 @@ 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 { 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";
@@ -213,6 +215,7 @@ export class InteractiveMode {
     // Custom header from extension (undefined = use built-in header)
     customHeader = undefined;
     options;
+    autoTrustOnReloadCwd;
     // Convenience accessors
     get session() {
         return this.runtimeHost.session;
@@ -229,6 +232,7 @@ export class InteractiveMode {
     constructor(runtimeHost, options = {}) {
         this.runtimeHost = runtimeHost;
         this.options = options;
+        this.autoTrustOnReloadCwd = options.autoTrustOnReloadCwd;
         this.runtimeHost.setBeforeSessionInvalidate(() => {
             this.resetExtensionUI();
         });
@@ -1129,6 +1133,7 @@ export class InteractiveMode {
         const uiContext = this.createExtensionUIContext();
         await this.session.bindExtensions({
             uiContext,
+            mode: "tui",
             abortHandler: () => {
                 this.restoreQueuedMessagesToEditor({ abort: true });
             },
@@ -1268,6 +1273,7 @@ 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,
@@ -1553,6 +1559,20 @@ export class InteractiveMode {
     /**
      * Create the ExtensionUIContext for extensions.
      */
+    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,
+            },
+        };
+    }
     createExtensionUIContext() {
         return {
             select: (title, options, opts) => this.showExtensionSelector(title, options, opts),
@@ -2057,6 +2077,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("");
@@ -2526,6 +2551,7 @@ export class InteractiveMode {
                         this.chatContainer.addChild(component);
                         // Render user message separately if present
                         if (skillBlock.userMessage) {
+                            this.chatContainer.addChild(new Spacer(1));
                             const userComponent = new UserMessageComponent(skillBlock.userMessage, this.getMarkdownThemeWithSettings());
                             this.chatContainer.addChild(userComponent);
                         }
@@ -2625,6 +2651,7 @@ export class InteractiveMode {
             updateFooter: true,
             populateHistory: true,
         });
+        this.renderProjectTrustWarningIfNeeded();
         // Show compaction info if session was compacted
         const allEntries = this.sessionManager.getEntries();
         const compactionCount = allEntries.filter((e) => e.type === "compaction").length;
@@ -2633,6 +2660,15 @@ export class InteractiveMode {
             this.showStatus(`Session compacted ${times}`);
         }
     }
+    renderProjectTrustWarningIfNeeded() {
+        if (this.settingsManager.isProjectTrusted() || !hasProjectTrustInputs(this.sessionManager.getCwd())) {
+            return;
+        }
+        if (this.chatContainer.children.length > 0) {
+            this.chatContainer.addChild(new Spacer(1));
+        }
+        this.chatContainer.addChild(new Text(theme.fg("warning", "This project is not trusted. Project instructions (AGENTS.md/CLAUDE.md), .pi resources, and project packages are ignored. Use /trust to save a trust decision, then restart pi."), 1, 0));
+    }
     async getUserInput() {
         const queuedInput = this.pendingUserInputs.shift();
         if (queuedInput !== undefined) {
@@ -3448,6 +3484,51 @@ export class InteractiveMode {
             // Ignore auth lookup failures for warning-only checks.
         }
     }
+    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.get(cwd);
+        this.showSelector((done) => {
+            const selector = new TrustSelectorComponent({
+                cwd,
+                savedDecision,
+                projectTrusted: this.settingsManager.isProjectTrusted(),
+                onSelect: (trusted) => {
+                    trustStore.set(cwd, trusted);
+                    done();
+                    this.showStatus(`Saved trust decision: ${trusted ? "trusted" : "untrusted"}. Restart pi for this to take effect.`);
+                },
+                onCancel: () => {
+                    done();
+                    this.ui.requestRender();
+                },
+            });
+            return { component: selector, focus: selector };
+        });
+    }
     showModelSelector(initialSearchInput) {
         this.showSelector((done) => {
             const selector = new ModelSelectorComponent(this.ui, this.session.model, this.settingsManager, this.session.modelRegistry, this.session.scopedModels, async (model) => {
@@ -3726,6 +3807,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;
@@ -3744,6 +3826,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;
@@ -4142,11 +4225,14 @@ export class InteractiveMode {
                 force: false,
                 showDiagnosticsWhenQuiet: true,
             });
+            const savedImplicitProjectTrust = this.maybeSaveImplicitProjectTrustAfterReload();
             const modelsJsonError = this.session.modelRegistry.getError();
             if (modelsJsonError) {
                 this.showError(`models.json error: ${modelsJsonError}`);
             }
-            this.showStatus("Reloaded keybindings, extensions, skills, prompts, themes");
+            this.showStatus(savedImplicitProjectTrust
+                ? "Reloaded keybindings, extensions, skills, prompts, themes; saved project trust"
+                : "Reloaded keybindings, extensions, skills, prompts, themes");
         }
         catch (error) {
             dismissReloadBox(previousEditor);