@mariozechner/pi-coding-agent 0.16.0 → 0.18.0

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

@@ -11,6 +11,7 @@ import { isBashExecutionMessage } from "../../core/messages.js";
 import { invalidateOAuthCache } from "../../core/model-config.js";
 import { listOAuthProviders, login, logout } from "../../core/oauth/index.js";
 import { getLatestCompactionEntry, SUMMARY_PREFIX, SUMMARY_SUFFIX } from "../../core/session-manager.js";
+import { loadProjectContextFiles } from "../../core/system-prompt.js";
 import { getChangelogPath, parseChangelog } from "../../utils/changelog.js";
 import { copyToClipboard } from "../../utils/clipboard.js";
 import { AssistantMessageComponent } from "./components/assistant-message.js";
@@ -19,6 +20,8 @@ import { CompactionComponent } from "./components/compaction.js";
 import { CustomEditor } from "./components/custom-editor.js";
 import { DynamicBorder } from "./components/dynamic-border.js";
 import { FooterComponent } from "./components/footer.js";
+import { HookInputComponent } from "./components/hook-input.js";
+import { HookSelectorComponent } from "./components/hook-selector.js";
 import { ModelSelectorComponent } from "./components/model-selector.js";
 import { OAuthSelectorComponent } from "./components/oauth-selector.js";
 import { QueueModeSelectorComponent } from "./components/queue-mode-selector.js";
@@ -66,6 +69,9 @@ export class InteractiveMode {
     // Auto-compaction state
     autoCompactionLoader = null;
     autoCompactionEscapeHandler;
+    // Hook UI state
+    hookSelector = null;
+    hookInput = null;
     // Convenience accessors
     get agent() {
         return this.session.agent;
@@ -189,6 +195,8 @@ export class InteractiveMode {
         // Start the UI
         this.ui.start();
         this.isInitialized = true;
+        // Initialize hooks with TUI-based UI context
+        await this.initHooks();
         // Subscribe to agent events
         this.subscribeToAgent();
         // Set up theme file watcher
@@ -202,6 +210,161 @@ export class InteractiveMode {
             this.ui.requestRender();
         });
     }
+    // =========================================================================
+    // Hook System
+    // =========================================================================
+    /**
+     * Initialize the hook system with TUI-based UI context.
+     */
+    async initHooks() {
+        // Show loaded project context files
+        const contextFiles = loadProjectContextFiles();
+        if (contextFiles.length > 0) {
+            const contextList = contextFiles.map((f) => theme.fg("dim", `  ${f.path}`)).join("\n");
+            this.chatContainer.addChild(new Text(theme.fg("muted", "Loaded context:\n") + contextList, 0, 0));
+            this.chatContainer.addChild(new Spacer(1));
+        }
+        const hookRunner = this.session.hookRunner;
+        if (!hookRunner) {
+            return; // No hooks loaded
+        }
+        // Set TUI-based UI context on the hook runner
+        hookRunner.setUIContext(this.createHookUIContext(), true);
+        hookRunner.setSessionFile(this.session.sessionFile);
+        // Subscribe to hook errors
+        hookRunner.onError((error) => {
+            this.showHookError(error.hookPath, error.error);
+        });
+        // Set up send handler for pi.send()
+        hookRunner.setSendHandler((text, attachments) => {
+            this.handleHookSend(text, attachments);
+        });
+        // Show loaded hooks
+        const hookPaths = hookRunner.getHookPaths();
+        if (hookPaths.length > 0) {
+            const hookList = hookPaths.map((p) => theme.fg("dim", `  ${p}`)).join("\n");
+            this.chatContainer.addChild(new Text(theme.fg("muted", "Loaded hooks:\n") + hookList, 0, 0));
+            this.chatContainer.addChild(new Spacer(1));
+        }
+        // Emit session_start event
+        await hookRunner.emit({ type: "session_start" });
+    }
+    /**
+     * Create the UI context for hooks.
+     */
+    createHookUIContext() {
+        return {
+            select: (title, options) => this.showHookSelector(title, options),
+            confirm: (title, message) => this.showHookConfirm(title, message),
+            input: (title, placeholder) => this.showHookInput(title, placeholder),
+            notify: (message, type) => this.showHookNotify(message, type),
+        };
+    }
+    /**
+     * Show a selector for hooks.
+     */
+    showHookSelector(title, options) {
+        return new Promise((resolve) => {
+            this.hookSelector = new HookSelectorComponent(title, options, (option) => {
+                this.hideHookSelector();
+                resolve(option);
+            }, () => {
+                this.hideHookSelector();
+                resolve(null);
+            });
+            this.editorContainer.clear();
+            this.editorContainer.addChild(this.hookSelector);
+            this.ui.setFocus(this.hookSelector);
+            this.ui.requestRender();
+        });
+    }
+    /**
+     * Hide the hook selector.
+     */
+    hideHookSelector() {
+        this.editorContainer.clear();
+        this.editorContainer.addChild(this.editor);
+        this.hookSelector = null;
+        this.ui.setFocus(this.editor);
+        this.ui.requestRender();
+    }
+    /**
+     * Show a confirmation dialog for hooks.
+     */
+    async showHookConfirm(title, message) {
+        const result = await this.showHookSelector(`${title}\n${message}`, ["Yes", "No"]);
+        return result === "Yes";
+    }
+    /**
+     * Show a text input for hooks.
+     */
+    showHookInput(title, placeholder) {
+        return new Promise((resolve) => {
+            this.hookInput = new HookInputComponent(title, placeholder, (value) => {
+                this.hideHookInput();
+                resolve(value);
+            }, () => {
+                this.hideHookInput();
+                resolve(null);
+            });
+            this.editorContainer.clear();
+            this.editorContainer.addChild(this.hookInput);
+            this.ui.setFocus(this.hookInput);
+            this.ui.requestRender();
+        });
+    }
+    /**
+     * Hide the hook input.
+     */
+    hideHookInput() {
+        this.editorContainer.clear();
+        this.editorContainer.addChild(this.editor);
+        this.hookInput = null;
+        this.ui.setFocus(this.editor);
+        this.ui.requestRender();
+    }
+    /**
+     * Show a notification for hooks.
+     */
+    showHookNotify(message, type) {
+        if (type === "error") {
+            this.showError(message);
+        }
+        else if (type === "warning") {
+            this.showWarning(message);
+        }
+        else {
+            this.showStatus(message);
+        }
+    }
+    /**
+     * Show a hook error in the UI.
+     */
+    showHookError(hookPath, error) {
+        const errorText = new Text(theme.fg("error", `Hook "${hookPath}" error: ${error}`), 1, 0);
+        this.chatContainer.addChild(errorText);
+        this.ui.requestRender();
+    }
+    /**
+     * Handle pi.send() from hooks.
+     * If streaming, queue the message. Otherwise, start a new agent loop.
+     */
+    handleHookSend(text, attachments) {
+        if (this.session.isStreaming) {
+            // Queue the message for later (note: attachments are lost when queuing)
+            this.session.queueMessage(text);
+            this.updatePendingMessagesDisplay();
+        }
+        else {
+            // Start a new agent loop immediately
+            this.session.prompt(text, { attachments }).catch((err) => {
+                this.showError(err instanceof Error ? err.message : String(err));
+            });
+        }
+    }
+    // =========================================================================
+    // Key Handlers
+    // =========================================================================
     setupKeyHandlers() {
         this.editor.onEscape = () => {
             if (this.loadingAnimation) {
@@ -350,6 +513,10 @@ export class InteractiveMode {
                     return;
                 }
             }
+            // Block input during compaction (will retry automatically)
+            if (this.session.isCompacting) {
+                return;
+            }
             // Queue message if agent is streaming
             if (this.session.isStreaming) {
                 await this.session.queueMessage(text);
@@ -484,19 +651,21 @@ export class InteractiveMode {
                 this.pendingTools.clear();
                 this.ui.requestRender();
                 break;
-            case "auto_compaction_start":
+            case "auto_compaction_start": {
                 // Set up escape to abort auto-compaction
                 this.autoCompactionEscapeHandler = this.editor.onEscape;
                 this.editor.onEscape = () => {
                     this.session.abortCompaction();
                 };
-                // Show compacting indicator
+                // Show compacting indicator with reason
                 this.statusContainer.clear();
-                this.autoCompactionLoader = new Loader(this.ui, (spinner) => theme.fg("accent", spinner), (text) => theme.fg("muted", text), "Auto-compacting... (esc to cancel)");
+                const reasonText = event.reason === "overflow" ? "Context overflow detected, " : "";
+                this.autoCompactionLoader = new Loader(this.ui, (spinner) => theme.fg("accent", spinner), (text) => theme.fg("muted", text), `${reasonText}Auto-compacting... (esc to cancel)`);
                 this.statusContainer.addChild(this.autoCompactionLoader);
                 this.ui.requestRender();
                 break;
-            case "auto_compaction_end":
+            }
+            case "auto_compaction_end": {
                 // Restore escape handler
                 if (this.autoCompactionEscapeHandler) {
                     this.editor.onEscape = this.autoCompactionEscapeHandler;
@@ -524,6 +693,7 @@ export class InteractiveMode {
                 }
                 this.ui.requestRender();
                 break;
+            }
         }
     }
     /** Extract text content from a user message */
@@ -639,6 +809,13 @@ export class InteractiveMode {
     }
     renderInitialMessages(state) {
         this.renderMessages(state.messages, { updateFooter: true, populateHistory: true });
+        // Show compaction info if session was compacted
+        const entries = this.sessionManager.loadEntries();
+        const compactionCount = entries.filter((e) => e.type === "compaction").length;
+        if (compactionCount > 0) {
+            const times = compactionCount === 1 ? "1 time" : `${compactionCount} times`;
+            this.showStatus(`Session compacted ${times}`);
+        }
     }
     async getUserInput() {
         return new Promise((resolve) => {
@@ -869,12 +1046,18 @@ export class InteractiveMode {
             return;
         }
         this.showSelector((done) => {
-            const selector = new UserMessageSelectorComponent(userMessages.map((m) => ({ index: m.entryIndex, text: m.text })), (entryIndex) => {
-                const selectedText = this.session.branch(entryIndex);
+            const selector = new UserMessageSelectorComponent(userMessages.map((m) => ({ index: m.entryIndex, text: m.text })), async (entryIndex) => {
+                const result = await this.session.branch(entryIndex);
+                if (result.skipped) {
+                    // Hook requested to skip conversation restore
+                    done();
+                    this.ui.requestRender();
+                    return;
+                }
                 this.chatContainer.clear();
                 this.isFirstUserMessage = true;
                 this.renderInitialMessages(this.session.state);
-                this.editor.setText(selectedText);
+                this.editor.setText(result.selectedText);
                 done();
                 this.showStatus("Branched to new session");
             }, () => {