@dyyz1993/pi-coding-agent 0.74.27 → 0.74.29

package/dist/core/agent-session.js

@@ -319,6 +319,35 @@ export class AgentSession {
             followUp: [...this._followUpMessages],
         });
     }
+    /** Emit entries_invalidated event to extensions when entries are removed from LLM context.
+     *  This is a notification-only event; it does not block or collect results. */
+    _emitEntriesInvalidated(invalidatedEntryIds, reason, operationEntryId) {
+        if (!this._extensionRunner || invalidatedEntryIds.length === 0)
+            return;
+        // Extract toolCallIds from invalidated tool result entries
+        const invalidatedToolCallIds = [];
+        for (const id of invalidatedEntryIds) {
+            const entry = this.sessionManager.getEntry(id);
+            if (entry && entry.type === "message" && entry.message.role === "toolResult") {
+                const toolCallId = entry.message.toolCallId;
+                if (toolCallId) {
+                    invalidatedToolCallIds.push(toolCallId);
+                }
+            }
+        }
+        // Fire-and-forget: don't await to avoid blocking SessionManager's synchronous _appendEntry
+        this._extensionRunner
+            .emit({
+            type: "entries_invalidated",
+            invalidatedEntryIds,
+            reason,
+            operationEntryId,
+            invalidatedToolCallIds,
+        })
+            .catch(() => {
+            // Silently swallow errors — this is a notification, not a critical path
+        });
+    }
     // Track last assistant message for auto-compaction check
     _lastAssistantMessage = undefined;
     /** Internal handler for agent events - shared by subscribe and reconnect */
@@ -2091,6 +2120,22 @@ export class AgentSession {
             }
         }
         this._extensionRunner = new ExtensionRunner(extensionsResult.extensions, extensionsResult.runtime, this._cwd, this.sessionManager, this._modelRegistry);
+        // Register SessionManager callback to detect entry lifecycle changes
+        // and emit entries_invalidated events to extensions.
+        this.sessionManager.setOnEntryAppended((entry) => {
+            if (entry.type === "deletion") {
+                const e = entry;
+                this._emitEntriesInvalidated(e.targetIds, "deletion", e.id);
+            }
+            else if (entry.type === "fold") {
+                const e = entry;
+                this._emitEntriesInvalidated([e.targetId], "fold", e.id);
+            }
+            else if (entry.type === "segment_summary") {
+                const e = entry;
+                this._emitEntriesInvalidated(e.targetIds, "segment_summary", e.id);
+            }
+        });
         const projectRoot = resolveProjectIdentity(this._cwd);
         this._extensionRunner.setContextDirFns({
             getProjectRoot: () => projectRoot,
@@ -2483,6 +2528,19 @@ export class AgentSession {
                 // Non-user message: leaf = selected node
                 newLeafId = targetId;
             }
+            // Safety guard: reject navigation that would eliminate ALL user messages.
+            // The new leaf path is root → ... → newLeafId (or null for root reset).
+            // If a summary is provided, it will be a child of newLeafId, so the effective
+            // leaf path includes newLeafId's ancestors. Check both cases.
+            const effectiveLeafId = summaryText ? newLeafId : newLeafId;
+            const userMsgCount = this.sessionManager.countUserMessagesOnPath(effectiveLeafId);
+            if (userMsgCount === 0) {
+                return {
+                    cancelled: true,
+                    editorText: undefined,
+                    reason: `Navigation to "${targetId}" would remove all user messages from the active path. This likely means the target is before the first user message. Use file-level rollback (snapshot.rollback) instead.`,
+                };
+            }
             // Switch leaf (with or without summary)
             // Summary is attached at the navigation target position (newLeafId), not the old branch
             let summaryEntry;