codebuff 1.0.168 → 1.0.170

package/dist/checkpoints.js

@@ -0,0 +1,171 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.checkpointManager = exports.CheckpointManager = void 0;
+const picocolors_1 = require("picocolors");
+const checkpointFileManager = __importStar(require("./checkpoint-file-manager"));
+/**
+ * Simple in-memory checkpoint manager for agent states
+ */
+class CheckpointManager {
+    checkpoints = new Map();
+    nextId = 1;
+    /**
+     * Add a new checkpoint
+     * @param agentState - The agent state to checkpoint
+     * @param userInput - The user input associated with this checkpoint
+     * @returns The ID of the created checkpoint
+     */
+    async addCheckpoint(agentState, userInput) {
+        // Use incremental ID starting at 1
+        const id = this.nextId++;
+        const fileStateIdPromise = checkpointFileManager.storeFileState(`Checkpoint ${id}`);
+        const checkpoint = {
+            agentStateString: JSON.stringify(agentState), // Deep clone to prevent reference issues
+            fileStateIdPromise,
+            historyLength: agentState.messageHistory.length,
+            id,
+            timestamp: Date.now(),
+            userInput,
+        };
+        // Add to map
+        this.checkpoints.set(id, checkpoint);
+        return id;
+    }
+    /**
+     * Get a checkpoint by ID
+     * @param id The checkpoint ID
+     * @returns The checkpoint or null if not found
+     */
+    getCheckpoint(id) {
+        if (id === null) {
+            return null;
+        }
+        const checkpoint = this.checkpoints.get(id);
+        return checkpoint || null;
+    }
+    /**
+     * Get all checkpoints
+     * @returns Array of all checkpoints
+     */
+    getAllCheckpoints() {
+        return Array.from(this.checkpoints.values());
+    }
+    /**
+     * Get the most recent checkpoint
+     * @returns The most recent checkpoint or null if none exist
+     */
+    getLatestCheckpoint() {
+        return this.checkpoints.get(this.nextId - 1) || null;
+    }
+    async restoreFileState(id) {
+        const checkpoint = this.getCheckpoint(id);
+        if (!checkpoint) {
+            return false;
+        }
+        checkpointFileManager.checkoutFileState(await checkpoint.fileStateIdPromise);
+        return true;
+    }
+    /**
+     * Clear all checkpoints
+     */
+    clearCheckpoints() {
+        this.checkpoints.clear();
+        this.nextId = 1; // Reset the ID counter when clearing
+    }
+    /**
+     * Get a formatted string representation of all checkpoints
+     * @param detailed Whether to include detailed information about each checkpoint
+     * @returns A formatted string representation of all checkpoints
+     */
+    getCheckpointsAsString(detailed = false) {
+        const checkpoints = this.getAllCheckpoints().sort((a, b) => a.id - b.id);
+        if (checkpoints.length === 0) {
+            return (0, picocolors_1.yellow)('No checkpoints available.');
+        }
+        const lines = [(0, picocolors_1.bold)((0, picocolors_1.underline)('Agent State Checkpoints:')), ''];
+        checkpoints.forEach((checkpoint) => {
+            const date = new Date(checkpoint.timestamp);
+            const formattedDate = date.toLocaleString();
+            const userInputOneLine = checkpoint.userInput.replaceAll('\n', ' ');
+            const userInput = userInputOneLine.length > 50
+                ? userInputOneLine.substring(0, 47) + '...'
+                : userInputOneLine;
+            lines.push(`${(0, picocolors_1.cyan)((0, picocolors_1.bold)(`#${checkpoint.id}`))} ${(0, picocolors_1.gray)(`[${formattedDate}]`)}:`);
+            lines.push(`  ${(0, picocolors_1.blue)('Input')}: ${userInput}`);
+            if (detailed) {
+                // Add more details about the agent state if needed
+                const messageCount = checkpoint.historyLength;
+                lines.push(`  ${(0, picocolors_1.blue)('Messages')}: ${messageCount}`);
+                // You can add more detailed information here as needed
+                // For example, file context information, etc.
+            }
+            lines.push(''); // Empty line between checkpoints
+        });
+        return lines.join('\n');
+    }
+    /**
+     * Get detailed information about a specific checkpoint
+     * @param id The checkpoint ID
+     * @returns A formatted string with detailed information about the checkpoint, or an error message if not found
+     */
+    getCheckpointDetails(id) {
+        const checkpoint = this.getCheckpoint(id);
+        if (!checkpoint) {
+            return (0, picocolors_1.cyan)(`\nCheckpoint #${id} not found.`);
+        }
+        const lines = [
+            (0, picocolors_1.cyan)(`Detailed information for checkpoint #${id}:`),
+        ];
+        const date = new Date(checkpoint.timestamp);
+        const formattedDate = date.toLocaleString();
+        lines.push(`${(0, picocolors_1.blue)('Created at')}: ${formattedDate}`);
+        if (checkpoint.userInput) {
+            lines.push(`${(0, picocolors_1.blue)('User input')}: ${checkpoint.userInput}`);
+        }
+        // Display more detailed information about the agent state
+        const messageCount = checkpoint.historyLength;
+        lines.push(`${(0, picocolors_1.blue)('Message history')}: ${messageCount} messages`);
+        // You could add more detailed information here as needed
+        return lines.join('\n');
+    }
+    getNextId() {
+        return this.nextId;
+    }
+}
+exports.CheckpointManager = CheckpointManager;
+// Export a singleton instance for use throughout the application
+exports.checkpointManager = new CheckpointManager();
+//# sourceMappingURL=checkpoints.js.map

package/dist/checkpoints.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"checkpoints.js","sourceRoot":"","sources":["../src/checkpoints.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2CASmB;AAGnB,iFAAkE;AAclE;;GAEG;AACH,MAAa,iBAAiB;IACpB,WAAW,GAA4B,IAAI,GAAG,EAAE,CAAA;IAChD,MAAM,GAAW,CAAC,CAAA;IAE1B;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CACjB,UAAsB,EACtB,SAAiB;QAEjB,mCAAmC;QACnC,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAA;QAExB,MAAM,kBAAkB,GAAG,qBAAqB,CAAC,cAAc,CAC7D,cAAc,EAAE,EAAE,CACnB,CAAA;QAED,MAAM,UAAU,GAAe;YAC7B,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,yCAAyC;YACvF,kBAAkB;YAClB,aAAa,EAAE,UAAU,CAAC,cAAc,CAAC,MAAM;YAC/C,EAAE;YACF,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS;SACV,CAAA;QAED,aAAa;QACb,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,CAAA;QAEpC,OAAO,EAAE,CAAA;IACX,CAAC;IAED;;;;OAIG;IACH,aAAa,CAAC,EAAiB;QAC7B,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;YAChB,OAAO,IAAI,CAAA;QACb,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QAC3C,OAAO,UAAU,IAAI,IAAI,CAAA;IAC3B,CAAC;IAED;;;OAGG;IACH,iBAAiB;QACf,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAA;IAC9C,CAAC;IAED;;;OAGG;IACH,mBAAmB;QACjB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,IAAI,CAAA;IACtD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,EAAU;QAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAA;QACzC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,KAAK,CAAA;QACd,CAAC;QAED,qBAAqB,CAAC,iBAAiB,CAAC,MAAM,UAAU,CAAC,kBAAkB,CAAC,CAAA;QAC5E,OAAO,IAAI,CAAA;IACb,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAA;QACxB,IAAI,CAAC,MAAM,GAAG,CAAC,CAAA,CAAC,qCAAqC;IACvD,CAAC;IAED;;;;OAIG;IACH,sBAAsB,CAAC,WAAoB,KAAK;QAC9C,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAA;QAExE,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO,IAAA,mBAAM,EAAC,2BAA2B,CAAC,CAAA;QAC5C,CAAC;QAED,MAAM,KAAK,GAAa,CAAC,IAAA,iBAAI,EAAC,IAAA,sBAAS,EAAC,0BAA0B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QAEzE,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE;YACjC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;YAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,EAAE,CAAA;YAE3C,MAAM,gBAAgB,GAAG,UAAU,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;YACnE,MAAM,SAAS,GACb,gBAAgB,CAAC,MAAM,GAAG,EAAE;gBAC1B,CAAC,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK;gBAC3C,CAAC,CAAC,gBAAgB,CAAA;YAEtB,KAAK,CAAC,IAAI,CACR,GAAG,IAAA,iBAAI,EAAC,IAAA,iBAAI,EAAC,IAAI,UAAU,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,IAAA,iBAAI,EAAC,IAAI,aAAa,GAAG,CAAC,GAAG,CACpE,CAAA;YAED,KAAK,CAAC,IAAI,CAAC,KAAK,IAAA,iBAAI,EAAC,OAAO,CAAC,KAAK,SAAS,EAAE,CAAC,CAAA;YAE9C,IAAI,QAAQ,EAAE,CAAC;gBACb,mDAAmD;gBACnD,MAAM,YAAY,GAAG,UAAU,CAAC,aAAa,CAAA;gBAC7C,KAAK,CAAC,IAAI,CAAC,KAAK,IAAA,iBAAI,EAAC,UAAU,CAAC,KAAK,YAAY,EAAE,CAAC,CAAA;gBAEpD,uDAAuD;gBACvD,8CAA8C;YAChD,CAAC;YAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA,CAAC,iCAAiC;QAClD,CAAC,CAAC,CAAA;QAEF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IAED;;;;OAIG;IACH,oBAAoB,CAAC,EAAU;QAC7B,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAA;QACzC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,IAAA,iBAAI,EAAC,iBAAiB,EAAE,aAAa,CAAC,CAAA;QAC/C,CAAC;QAED,MAAM,KAAK,GAAa;YACtB,IAAA,iBAAI,EAAC,wCAAwC,EAAE,GAAG,CAAC;SACpD,CAAA;QAED,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAA;QAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,EAAE,CAAA;QAC3C,KAAK,CAAC,IAAI,CAAC,GAAG,IAAA,iBAAI,EAAC,YAAY,CAAC,KAAK,aAAa,EAAE,CAAC,CAAA;QAErD,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;YACzB,KAAK,CAAC,IAAI,CAAC,GAAG,IAAA,iBAAI,EAAC,YAAY,CAAC,KAAK,UAAU,CAAC,SAAS,EAAE,CAAC,CAAA;QAC9D,CAAC;QAED,0DAA0D;QAC1D,MAAM,YAAY,GAAG,UAAU,CAAC,aAAa,CAAA;QAC7C,KAAK,CAAC,IAAI,CAAC,GAAG,IAAA,iBAAI,EAAC,iBAAiB,CAAC,KAAK,YAAY,WAAW,CAAC,CAAA;QAElE,yDAAyD;QAEzD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAA;IACpB,CAAC;CACF;AAnKD,8CAmKC;AAED,iEAAiE;AACpD,QAAA,iBAAiB,GAAG,IAAI,iBAAiB,EAAE,CAAA"}

package/dist/cli.d.ts

@@ -2,38 +2,34 @@ import { ProjectFileContext } from './common/util/file';
 import { CliOptions } from './types';
 export declare class CLI {
     private client;
-    private chatStorage;
     private readyPromise;
     private git;
     private costMode;
     private rl;
     private isReceivingResponse;
     private stopResponse;
-    private lastChanges;
     private lastSigintTime;
     private lastInputTime;
     private consecutiveFastInputs;
     private pastedContent;
     private isPasting;
-    constructor(readyPromise: Promise<[void, ProjectFileContext]>, { git, costMode }: CliOptions);
+    constructor(readyPromise: Promise<[void, ProjectFileContext, void]>, { git, costMode }: CliOptions);
     private setupSignalHandlers;
     private initReadlineInterface;
     private completer;
     private setPrompt;
+    private promptWithCheckpointNumber;
     printInitialPrompt(initialInput?: string): Promise<void>;
     printDiff(): Promise<void>;
     private handleLine;
     private handleUserInput;
+    private beforeProcessCommand;
     private processCommand;
     private forwardUserInput;
     private returnControlToUser;
     private onWebSocketError;
     private onWebSocketReconnect;
-    private sendUserInputAndAwaitResponse;
     private handleUndo;
-    private handleRedo;
-    private navigateFileVersion;
-    private applyAndDisplayCurrentFileVersion;
     private handleKeyPress;
     private handleSigint;
     private handleEscKey;
@@ -41,6 +37,9 @@ export declare class CLI {
     private showEasterEgg;
     private handleExit;
     private detectPasting;
-    private autoCommitChanges;
     private handleDiff;
+    private handleCheckpoints;
+    private handleRestoreCheckpoint;
+    private restoreAgentStateAndFiles;
+    private handleClearCheckpoints;
 }