@visorcraft/idlehands 2.2.7 → 2.2.8

package/dist/agent.js

@@ -41,6 +41,7 @@ import { normalizeApprovalMode } from './shared/config-utils.js';
 import { collectSnapshot } from './sys/context.js';
 import { ToolError, ValidationError } from './tools/tool-error.js';
 import * as tools from './tools.js';
+import { EditTransaction } from './tools/transaction.js';
 import { stateDir, timestampedId } from './utils.js';
 import { VaultStore } from './vault.js';
 export { parseToolCallsFromContent };
@@ -489,6 +490,7 @@ export async function createSession(opts) {
     let inFlight = null;
     let initialConnectionProbeDone = false;
     let lastEditedPath;
+    let lastTurnTransaction;
     // Plan mode state (Phase 8)
     let planSteps = [];
     // Sub-agent queue state (Phase 18): enforce sequential execution on single-GPU setups.
@@ -3002,6 +3004,8 @@ export async function createSession(opts) {
                             const absPath = args.path.startsWith('/')
                                 ? args.path
                                 : path.resolve(projectDir, args.path);
+                            // Track in turn transaction for potential atomic rollback.
+                            turnTransaction.track(absPath);
                             // ── Pre-dispatch: block edits to files in a mutation spiral ──
                             if (fileMutationBlocked.has(absPath)) {
                                 const basename = path.basename(absPath);
@@ -3493,6 +3497,7 @@ export async function createSession(opts) {
                         return { id: callId, content: truncated.content };
                     };
                     const results = [];
+                    const turnTransaction = new EditTransaction();
                     let invalidArgsThisTurn = false;
                     // Helper: catch tool errors but re-throw AgentLoopBreak (those must break the outer loop)
                     const catchToolError = async (e, tc) => {
@@ -3647,6 +3652,11 @@ export async function createSession(opts) {
                             });
                         }
                     }
+                    // Store the turn transaction for potential post-turn rollback.
+                    if (turnTransaction.hasChanges) {
+                        turnTransaction.commit();
+                        lastTurnTransaction = turnTransaction;
+                    }
                     // Bail immediately if cancelled during tool execution
                     if (ac.signal.aborted)
                         break;
@@ -3918,6 +3928,16 @@ export async function createSession(opts) {
             return currentContextTokens > 0 ? currentContextTokens : estimateTokensFromMessages(messages);
         },
         ask,
+        rollbackLastTurnEdits: async () => {
+            if (!lastTurnTransaction || !lastTurnTransaction.hasChanges) {
+                return { ok: false, error: 'No file edits to roll back.' };
+            }
+            const tx = lastTurnTransaction;
+            lastTurnTransaction = undefined;
+            const callCtx = { cwd: projectDir, noConfirm: true, dryRun: false };
+            const results = await tx.rollback(callCtx);
+            return { ok: true, results };
+        },
         rollback: () => {
             const cp = conversationBranch.rollback();
             if (!cp)