codecane 1.0.156 → 1.0.171

package/dist/cli.js

@@ -24,35 +24,24 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.CLI = void 0;
-const lodash_1 = require("lodash");
-const project_file_tree_1 = require("./common/project-file-tree");
-const changes_1 = require("./common/util/changes");
-const readline = __importStar(require("readline"));
-const picocolors_1 = require("picocolors");
+const fs = __importStar(require("fs"));
 const path_1 = require("path");
-function rewriteLine(line) {
-    // Only do line rewriting if we have an interactive TTY
-    if (process.stdout.isTTY) {
-        readline.clearLine(process.stdout, 0);
-        readline.cursorTo(process.stdout, 0);
-        process.stdout.write(line);
-    }
-    else {
-        process.stdout.write(line + '\n');
-    }
-}
-const config_1 = require("./config");
+const picocolors_1 = require("picocolors");
+const readline = __importStar(require("readline"));
+const constants_1 = require("./common/constants");
+const project_file_tree_1 = require("./common/project-file-tree");
+const file_1 = require("./common/util/file");
+const string_1 = require("./common/util/string");
 const chat_storage_1 = require("./chat-storage");
+const checkpoints_1 = require("./checkpoints");
 const client_1 = require("./client");
+const config_1 = require("./config");
 const menu_1 = require("./menu");
 const project_files_1 = require("./project-files");
 const tool_handlers_1 = require("./tool-handlers");
+const spinner_1 = require("./utils/spinner");
 const terminal_1 = require("./utils/terminal");
-const constants_1 = require("./common/constants");
-const file_1 = require("./common/util/file");
 const web_scraper_1 = require("./web-scraper");
-const git_1 = require("./common/util/git");
-const string_1 = require("./common/util/string");
 class CLI {
     client;
     chatStorage;
@@ -62,8 +51,6 @@ class CLI {
     rl;
     isReceivingResponse = false;
     stopResponse = null;
-    loadingInterval = null;
-    lastChanges = [];
     lastSigintTime = 0;
     lastInputTime = 0;
     consecutiveFastInputs = 0;
@@ -73,375 +60,223 @@ class CLI {
         this.git = git;
         this.costMode = costMode;
         this.chatStorage = new chat_storage_1.ChatStorage();
-        process.on('exit', () => this.restoreCursor());
-        process.on('SIGTERM', () => {
-            this.restoreCursor();
-            process.exit(0);
-        });
-        this.rl = readline.createInterface({
-            input: process.stdin,
-            output: process.stdout,
-            historySize: 1000,
-            terminal: true,
-            completer: (line) => {
-                if (!this.client.fileContext?.fileTree)
-                    return [[], line];
-                const tokenNames = Object.values(this.client.fileContext.fileTokenScores).flatMap((o) => Object.keys(o));
-                const paths = (0, project_file_tree_1.getAllFilePaths)(this.client.fileContext.fileTree);
-                const lastWord = line.split(' ').pop() || '';
-                const matchingTokens = [...tokenNames, ...paths].filter((token) => token.startsWith(lastWord) || token.includes('/' + lastWord));
-                if (matchingTokens.length > 1) {
-                    // Find common characters after lastWord
-                    const suffixes = matchingTokens.map((token) => {
-                        const index = token.indexOf(lastWord);
-                        return token.slice(index + lastWord.length);
-                    });
-                    let commonPrefix = '';
-                    const firstSuffix = suffixes[0];
-                    for (let i = 0; i < firstSuffix.length; i++) {
-                        const char = firstSuffix[i];
-                        if (suffixes.every((suffix) => suffix[i] === char)) {
-                            commonPrefix += char;
-                        }
-                        else {
-                            break;
-                        }
-                    }
-                    if (commonPrefix) {
-                        // Match the common prefix
-                        return [[lastWord + commonPrefix], lastWord];
-                    }
-                }
-                return [matchingTokens, lastWord];
-            },
-        });
+        this.setupSignalHandlers();
+        this.initReadlineInterface();
         this.client = new client_1.Client(config_1.websocketUrl, this.chatStorage, this.onWebSocketError.bind(this), this.onWebSocketReconnect.bind(this), this.returnControlToUser.bind(this), this.costMode, this.git, this.rl);
         this.readyPromise = Promise.all([
             readyPromise.then((results) => {
                 const [_, fileContext] = results;
-                this.client.initFileVersions(fileContext);
+                this.client.initAgentState(fileContext);
                 return this.client.warmContextCache();
             }),
             this.client.connect(),
         ]);
         this.setPrompt();
-        this.rl.on('line', (line) => {
-            this.handleInput(line);
+    }
+    setupSignalHandlers() {
+        process.on('exit', () => spinner_1.Spinner.get().restoreCursor());
+        process.on('SIGTERM', () => {
+            spinner_1.Spinner.get().restoreCursor();
+            process.exit(0);
         });
-        this.rl.on('SIGINT', () => {
-            // Kill any running terminal command
-            if ((0, terminal_1.isCommandRunning)()) {
-                (0, terminal_1.resetShell)();
-            }
-            // Clear the line buffer if it exists (readline internal API)
-            if ('line' in this.rl) {
-                ;
-                this.rl.line = '';
-            }
-            if (this.isReceivingResponse) {
-                this.handleStopResponse();
-            }
-            else {
-                const now = Date.now();
-                if (now - this.lastSigintTime < 5000) {
-                    this.handleExit();
+        process.on('SIGTSTP', () => this.handleExit());
+    }
+    initReadlineInterface() {
+        this.rl = readline.createInterface({
+            input: process.stdin,
+            output: process.stdout,
+            historySize: 1000,
+            terminal: true,
+            completer: this.completer.bind(this),
+        });
+        this.rl.on('line', (line) => this.handleLine(line));
+        this.rl.on('SIGINT', () => this.handleSigint());
+        this.rl.on('close', () => this.handleExit());
+        process.stdin.on('keypress', (str, key) => this.handleKeyPress(str, key));
+    }
+    completer(line) {
+        if (!this.client.fileContext?.fileTree)
+            return [[], line];
+        const tokenNames = Object.values(this.client.fileContext.fileTokenScores).flatMap((o) => Object.keys(o));
+        const paths = (0, project_file_tree_1.getAllFilePaths)(this.client.fileContext.fileTree);
+        const lastWord = line.split(' ').pop() || '';
+        const lastWordLower = lastWord.toLowerCase();
+        const matchingTokens = [...tokenNames, ...paths].filter((token) => token.toLowerCase().startsWith(lastWordLower) ||
+            token.toLowerCase().includes('/' + lastWordLower));
+        if (matchingTokens.length > 1) {
+            const suffixes = matchingTokens.map((token) => {
+                const index = token.toLowerCase().indexOf(lastWordLower);
+                return token.slice(index + lastWord.length);
+            });
+            let commonPrefix = '';
+            const firstSuffix = suffixes[0];
+            for (let i = 0; i < firstSuffix.length; i++) {
+                const char = firstSuffix[i];
+                if (suffixes.every((suffix) => suffix[i] === char)) {
+                    commonPrefix += char;
                 }
                 else {
-                    this.lastSigintTime = now;
-                    console.log('\nPress Ctrl-C again to exit');
-                    this.rl.prompt();
+                    break;
                 }
             }
-        });
-        this.rl.on('close', () => {
-            this.handleExit();
-        });
-        process.on('SIGTSTP', () => {
-            // Exit on Ctrl+Z
-            this.handleExit();
-        });
-        process.stdin.on('keypress', (str, key) => {
-            if (key.name === 'escape') {
-                this.handleEscKey();
-            }
-            // Make double spaces into newlines
-            if (!this.isPasting &&
-                str === ' ' &&
-                '_refreshLine' in this.rl &&
-                'line' in this.rl &&
-                'cursor' in this.rl) {
-                const rl = this.rl;
-                const { cursor, line } = rl;
-                const prevTwoChars = cursor > 1 ? line.slice(cursor - 2, cursor) : '';
-                if (prevTwoChars === '  ') {
-                    rl.line = line.slice(0, cursor - 2) + '\n\n' + line.slice(cursor);
-                    rl._refreshLine();
-                }
-            }
-            this.detectPasting();
-        });
-    }
-    returnControlToUser() {
-        this.rl.prompt();
-        this.isReceivingResponse = false;
-        if (this.stopResponse) {
-            this.stopResponse();
-        }
-        this.stopLoadingAnimation();
-    }
-    onWebSocketError() {
-        this.stopLoadingAnimation();
-        this.isReceivingResponse = false;
-        if (this.stopResponse) {
-            this.stopResponse();
-            this.stopResponse = null;
-        }
-        console.error((0, picocolors_1.yellow)('\nCould not connect. Retrying...'));
-    }
-    onWebSocketReconnect() {
-        console.log((0, picocolors_1.green)('\nReconnected!'));
-        this.returnControlToUser();
-    }
-    detectPasting() {
-        const currentTime = Date.now();
-        const timeDiff = currentTime - this.lastInputTime;
-        if (timeDiff < 10) {
-            this.consecutiveFastInputs++;
-            if (this.consecutiveFastInputs >= 2) {
-                this.isPasting = true;
-            }
-        }
-        else {
-            this.consecutiveFastInputs = 0;
-            if (this.isPasting) {
-                this.isPasting = false;
-            }
-        }
-        this.lastInputTime = currentTime;
-    }
-    handleInput(line) {
-        this.detectPasting();
-        if (this.isPasting) {
-            this.pastedContent += line + '\n';
-        }
-        else if (!this.isReceivingResponse) {
-            if (this.pastedContent) {
-                this.handleUserInput((this.pastedContent + line).trim());
-                this.pastedContent = '';
-            }
-            else {
-                this.handleUserInput(line.trim());
+            if (commonPrefix) {
+                return [[lastWord + commonPrefix], lastWord];
             }
         }
+        return [matchingTokens, lastWord];
     }
     setPrompt() {
-        this.rl.setPrompt((0, picocolors_1.green)(`${(0, path_1.parse)((0, project_files_1.getProjectRoot)()).base} > `));
+        this.rl.setPrompt((0, picocolors_1.green)(`${(0, path_1.parse)((0, project_files_1.getProjectRoot)()).base} ${checkpoints_1.checkpointManager.getNextId()} > `));
+    }
+    promptWithCheckpointNumber() {
+        this.setPrompt();
+        this.rl.prompt();
     }
     async printInitialPrompt(initialInput) {
         if (this.client.user) {
             (0, menu_1.displayGreeting)(this.costMode, this.client.user.name);
         }
         else {
-            console.log(`Welcome to Codebuff! Let's get your account set up.`);
+            console.log(`Welcome to Codebuff! Give us a sec to get your account set up...`);
             await this.client.login();
             return;
         }
-        this.rl.prompt();
+        this.promptWithCheckpointNumber();
         if (initialInput) {
             process.stdout.write(initialInput + '\n');
             this.handleUserInput(initialInput);
         }
     }
-    handleUndo() {
-        this.navigateFileVersion('undo');
-        this.rl.prompt();
-    }
-    handleRedo() {
-        this.navigateFileVersion('redo');
-        this.rl.prompt();
-    }
-    navigateFileVersion(direction) {
-        const currentVersion = this.chatStorage.getCurrentVersion();
-        const filePaths = Object.keys(currentVersion ? currentVersion.files : {});
-        const currentFiles = (0, project_files_1.getExistingFiles)(filePaths);
-        this.chatStorage.saveCurrentFileState(currentFiles);
-        const navigated = this.chatStorage.navigateVersion(direction);
-        if (navigated) {
-            console.log(direction === 'undo'
-                ? (0, picocolors_1.green)('Undo last change')
-                : (0, picocolors_1.green)('Redo last change'));
-            const files = this.applyAndDisplayCurrentFileVersion();
-            console.log((0, picocolors_1.green)('Loaded files:'), (0, picocolors_1.green)(Object.keys(files).join(', ')));
-        }
-        else {
-            console.log((0, picocolors_1.green)(`No more ${direction === 'undo' ? 'undo' : 'redo'}s`));
-        }
-    }
-    handleStopResponse() {
-        console.log((0, picocolors_1.yellow)('\n[Response stopped by user]'));
-        this.isReceivingResponse = false;
-        if (this.stopResponse) {
-            this.stopResponse();
-        }
-        this.stopLoadingAnimation();
-        this.restoreCursor();
+    async printDiff() {
+        this.handleDiff();
+        this.promptWithCheckpointNumber();
     }
-    restoreCursor() {
-        // Show cursor ANSI escape code
-        process.stdout.write('\u001B[?25h');
-    }
-    handleExit() {
-        this.restoreCursor();
-        console.log('\n\n');
-        console.log(`${(0, string_1.pluralize)(this.client.sessionCreditsUsed, 'credit')} used this session.`);
-        if (!!this.client.limit &&
-            !!this.client.usage &&
-            !!this.client.nextQuotaReset) {
-            const daysUntilReset = Math.max(0, Math.floor((this.client.nextQuotaReset.getTime() - Date.now()) /
-                (1000 * 60 * 60 * 24)));
-            console.log(`${Math.max(0, this.client.limit - this.client.usage)} / ${this.client.limit} credits remaining. Renews in ${(0, string_1.pluralize)(daysUntilReset, 'day')}.`);
-        }
-        console.log((0, picocolors_1.green)('Codebuff out!'));
-        process.exit(0);
-    }
-    handleEscKey() {
-        if (this.isReceivingResponse) {
-            this.handleStopResponse();
-        }
-    }
-    applyAndDisplayCurrentFileVersion() {
-        const currentVersion = this.chatStorage.getCurrentVersion();
-        if (currentVersion) {
-            (0, project_files_1.setFiles)(currentVersion.files);
-            return currentVersion.files;
-        }
-        return {};
-    }
-    startLoadingAnimation() {
-        const chars = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
-        let i = 0;
-        // Hide cursor while spinner is active
-        process.stdout.write('\u001B[?25l');
-        this.loadingInterval = setInterval(() => {
-            rewriteLine((0, picocolors_1.green)(`${chars[i]} Thinking...`));
-            i = (i + 1) % chars.length;
-        }, 100);
-    }
-    stopLoadingAnimation() {
-        if (this.loadingInterval) {
-            clearInterval(this.loadingInterval);
-            this.loadingInterval = null;
-            rewriteLine(''); // Clear the spinner line
-            this.restoreCursor(); // Show cursor after spinner stops
+    async handleLine(line) {
+        this.detectPasting();
+        if (this.isPasting) {
+            this.pastedContent += line + '\n';
         }
-    }
-    async autoCommitChanges() {
-        if ((0, git_1.hasStagedChanges)()) {
-            const stagedChanges = (0, git_1.getStagedChanges)();
-            if (!stagedChanges)
-                return;
-            const commitMessage = await this.client.generateCommitMessage(stagedChanges);
-            (0, git_1.commitChanges)(commitMessage);
-            return commitMessage;
+        else if (!this.isReceivingResponse) {
+            if (this.pastedContent) {
+                await this.handleUserInput((this.pastedContent + line).trim());
+                this.pastedContent = '';
+            }
+            else {
+                await this.handleUserInput(line.trim());
+            }
         }
-        return undefined;
-    }
-    handleDiff() {
-        if (this.lastChanges.length === 0) {
-            console.log((0, picocolors_1.yellow)('No changes found in the last assistant response.'));
-            return;
-        }
-        this.lastChanges.forEach((change) => {
-            console.log('-', change.filePath);
-            const lines = change.content
-                .split('\n')
-                .map((line) => (change.type === 'file' ? '+' + line : line));
-            lines.forEach((line) => {
-                if (line.startsWith('+')) {
-                    console.log((0, picocolors_1.green)(line));
-                }
-                else if (line.startsWith('-')) {
-                    console.log((0, picocolors_1.red)(line));
-                }
-                else {
-                    console.log(line);
-                }
-            });
-        });
     }
     async handleUserInput(userInput) {
         if (!userInput)
             return;
         userInput = userInput.trim();
-        // Handle commands
+        if (await this.processCommand(userInput)) {
+            return;
+        }
+        await this.forwardUserInput(userInput);
+    }
+    async beforeProcessCommand(userInput) {
+        await this.readyPromise;
+        // Save the current agent state
+        checkpoints_1.checkpointManager.addCheckpoint(this.client.agentState, userInput);
+    }
+    async processCommand(userInput) {
+        await this.beforeProcessCommand(userInput);
         if (userInput === 'help' || userInput === 'h') {
             (0, menu_1.displayMenu)();
-            this.rl.prompt();
-            return;
+            this.promptWithCheckpointNumber();
+            return true;
         }
         if (userInput === 'login' || userInput === 'signin') {
             await this.client.login();
-            return;
+            return true;
         }
-        else if (userInput === 'logout' || userInput === 'signout') {
+        if (userInput === 'logout' || userInput === 'signout') {
             await this.client.logout();
-            this.rl.prompt();
-            return;
+            this.promptWithCheckpointNumber();
+            return true;
         }
-        else if (userInput.startsWith('ref-')) {
+        if (userInput.startsWith('ref-')) {
             await this.client.handleReferralCode(userInput.trim());
-            return;
+            return true;
         }
-        else if (userInput === 'usage' || userInput === 'credits') {
+        if (userInput === 'usage' || userInput === 'credits') {
             this.client.getUsage();
-            return;
+            return true;
         }
-        else if (userInput === 'undo' || userInput === 'u') {
+        if (userInput === 'undo' || userInput === 'u') {
             this.handleUndo();
-            return;
-        }
-        else if (userInput === 'redo' || userInput === 'r') {
-            this.handleRedo();
-            return;
+            return true;
         }
-        else if (userInput === 'quit' ||
-            userInput === 'exit' ||
-            userInput === 'q') {
+        if (userInput === 'quit' || userInput === 'exit' || userInput === 'q') {
             this.handleExit();
-            return;
+            return true;
         }
-        else if (userInput === 'diff' ||
-            userInput === 'doff' ||
-            userInput === 'dif' ||
-            userInput === 'iff' ||
-            userInput === 'd') {
+        if (['diff', 'doff', 'dif', 'iff', 'd'].includes(userInput)) {
             this.handleDiff();
-            this.rl.prompt();
-            return;
+            this.promptWithCheckpointNumber();
+            return true;
+        }
+        if (userInput === 'uuddlrlrba' ||
+            userInput === 'konami' ||
+            userInput === 'codebuffy') {
+            this.showEasterEgg();
+            return true;
+        }
+        // Checkpoint commands
+        if (userInput === 'checkpoint list' || userInput === 'checkpoints') {
+            this.handleCheckpoints();
+            return true;
+        }
+        const checkpointDetailMatch = userInput.match(/^checkpoint\s+(\d+)$/);
+        if (checkpointDetailMatch) {
+            const id = parseInt(checkpointDetailMatch[1], 10);
+            this.handleCheckpointDetail(id);
+            return true;
+        }
+        const restoreMatch = userInput.match(/^restore\s+(\d+)$/);
+        if (restoreMatch) {
+            const id = parseInt(restoreMatch[1], 10);
+            this.handleRestoreCheckpoint(id);
+            return true;
+        }
+        if (userInput === 'checkpoint clear') {
+            this.handleClearCheckpoints();
+            return true;
         }
         const runPrefix = '/run ';
+        const bangPrefix = '!';
         const hasRunPrefix = userInput.startsWith(runPrefix);
+        const hasBangPrefix = userInput.startsWith(bangPrefix);
         if (hasRunPrefix ||
-            (!constants_1.SKIPPED_TERMINAL_COMMANDS.some((command) => userInput.toLowerCase().startsWith(command)) &&
+            hasBangPrefix ||
+            (!constants_1.SKIPPED_TERMINAL_COMMANDS.some((cmd) => userInput.toLowerCase().startsWith(cmd)) &&
                 !userInput.includes('error ') &&
                 !userInput.includes("'") &&
                 userInput.split(' ').length <= 5)) {
-            const withoutRunPrefix = userInput.replace(runPrefix, '');
-            const { result, stdout } = await (0, tool_handlers_1.handleRunTerminalCommand)({ command: withoutRunPrefix }, 'user', 'user');
+            let commandToRun = userInput;
+            if (hasRunPrefix) {
+                commandToRun = userInput.replace(runPrefix, '');
+            }
+            else if (hasBangPrefix) {
+                commandToRun = userInput.replace(bangPrefix, '');
+            }
+            const { result, stdout } = await (0, tool_handlers_1.handleRunTerminalCommand)({ command: commandToRun }, 'user', 'user', (0, project_files_1.getProjectRoot)());
             if (result !== 'command not found') {
-                this.setPrompt();
-                this.rl.prompt();
-                return;
+                this.promptWithCheckpointNumber();
+                return true;
             }
-            else if (hasRunPrefix) {
+            else if (hasRunPrefix || hasBangPrefix) {
                 process.stdout.write(stdout);
-                this.setPrompt();
-                this.rl.prompt();
-                return;
+                this.promptWithCheckpointNumber();
+                return true;
             }
         }
-        this.startLoadingAnimation();
-        await this.readyPromise;
+        return false;
+    }
+    async forwardUserInput(userInput) {
+        spinner_1.Spinner.get().start();
+        this.client.lastChanges = [];
         const currentChat = this.chatStorage.getCurrentChat();
         const { fileVersions } = currentChat;
         const currentFileVersion = fileVersions[fileVersions.length - 1]?.files ?? {};
@@ -466,60 +301,330 @@ class CLI {
         };
         this.chatStorage.addMessage(currentChat, newMessage);
         this.isReceivingResponse = true;
-        const { response, changes, changesAlreadyApplied } = await this.sendUserInputAndAwaitResponse();
+        const { responsePromise, stopResponse } = await this.client.sendUserInput(userInput);
+        this.stopResponse = stopResponse;
+        await responsePromise;
+        this.stopResponse = null;
         this.isReceivingResponse = false;
-        this.stopLoadingAnimation();
-        const allChanges = [...changesAlreadyApplied, ...changes];
-        const filesChanged = (0, lodash_1.uniq)(allChanges.map((change) => change.filePath));
-        const allFilesChanged = this.chatStorage.saveFilesChanged(filesChanged);
-        // Stage files about to be changed if flag was set
-        if (this.git === 'stage' && changes.length > 0) {
-            const didStage = (0, git_1.stagePatches)((0, project_files_1.getProjectRoot)(), changes);
-            if (didStage) {
-                console.log((0, picocolors_1.green)('\nStaged previous changes'));
+        spinner_1.Spinner.get().stop();
+        if (this.client.lastRequestCredits >= constants_1.REQUEST_CREDIT_SHOW_THRESHOLD) {
+            console.log(`\n${(0, string_1.pluralize)(this.client.lastRequestCredits, 'credit')} used for this request.`);
+        }
+        this.client.showUsageWarning();
+        console.log();
+        this.promptWithCheckpointNumber();
+    }
+    returnControlToUser() {
+        this.promptWithCheckpointNumber();
+        this.isReceivingResponse = false;
+        if (this.stopResponse) {
+            this.stopResponse();
+        }
+        spinner_1.Spinner.get().stop();
+    }
+    onWebSocketError() {
+        spinner_1.Spinner.get().stop();
+        this.isReceivingResponse = false;
+        if (this.stopResponse) {
+            this.stopResponse();
+            this.stopResponse = null;
+        }
+        console.error((0, picocolors_1.yellow)('\nCould not connect. Retrying...'));
+    }
+    onWebSocketReconnect() {
+        console.log((0, picocolors_1.green)('\nReconnected!'));
+        this.returnControlToUser();
+    }
+    handleUndo() {
+        // Get previous checkpoint number (not including undo command)
+        const checkpointId = checkpoints_1.checkpointManager.getLatestCheckpoint().id - 1;
+        if (checkpointId < 1) {
+            console.log((0, picocolors_1.red)('Nothing to undo.'));
+            return;
+        }
+        this.restoreAgentStateAndFiles(checkpoints_1.checkpointManager.getCheckpoint(checkpointId));
+        console.log((0, picocolors_1.green)(`Restored to checkpoint #${checkpointId}.`));
+        this.promptWithCheckpointNumber();
+    }
+    handleKeyPress(str, key) {
+        if (key.name === 'escape') {
+            this.handleEscKey();
+        }
+        if (!this.isPasting &&
+            str === ' ' &&
+            '_refreshLine' in this.rl &&
+            'line' in this.rl &&
+            'cursor' in this.rl) {
+            const rlAny = this.rl;
+            const { cursor, line } = rlAny;
+            const prevTwoChars = cursor > 1 ? line.slice(cursor - 2, cursor) : '';
+            if (prevTwoChars === '  ') {
+                rlAny.line = line.slice(0, cursor - 2) + '\n\n' + line.slice(cursor);
+                rlAny._refreshLine();
             }
         }
-        const { created, modified } = (0, changes_1.applyChanges)((0, project_files_1.getProjectRoot)(), changes);
-        if (created.length > 0 || modified.length > 0) {
-            console.log();
+        this.detectPasting();
+    }
+    handleSigint() {
+        if ((0, terminal_1.isCommandRunning)()) {
+            (0, terminal_1.resetShell)((0, project_files_1.getProjectRoot)());
         }
-        for (const file of created) {
-            console.log((0, picocolors_1.green)(`- Created ${file}`));
+        if ('line' in this.rl) {
+            ;
+            this.rl.line = '';
         }
-        for (const file of modified) {
-            console.log((0, picocolors_1.green)(`- Updated ${file}`));
+        if (this.isReceivingResponse) {
+            this.handleStopResponse();
         }
-        if (created.length > 0 || modified.length > 0) {
-            if (this.client.lastRequestCredits > constants_1.REQUEST_CREDIT_SHOW_THRESHOLD) {
-                console.log(`\n${(0, string_1.pluralize)(this.client.lastRequestCredits, 'credit')} used for this request.`);
+        else {
+            const now = Date.now();
+            if (now - this.lastSigintTime < 5000) {
+                this.handleExit();
+            }
+            else {
+                this.lastSigintTime = now;
+                console.log('\nPress Ctrl-C again to exit');
+                this.promptWithCheckpointNumber();
             }
-            console.log('Complete! Type "diff" to see the changes made.');
-            this.client.showUsageWarning();
         }
-        console.log();
-        this.lastChanges = allChanges;
-        const assistantMessage = {
-            role: 'assistant',
-            content: response,
-        };
-        this.chatStorage.addMessage(this.chatStorage.getCurrentChat(), assistantMessage);
-        const updatedFiles = (0, project_files_1.getExistingFiles)(allFilesChanged);
-        this.chatStorage.addNewFileState(updatedFiles);
-        this.rl.prompt();
     }
-    async sendUserInputAndAwaitResponse() {
-        const userInputId = `mc-input-` + Math.random().toString(36).substring(2, 15);
-        const { responsePromise, stopResponse } = this.client.subscribeToResponse((chunk) => {
-            process.stdout.write(chunk);
-        }, userInputId, () => {
-            this.stopLoadingAnimation();
-            process.stdout.write((0, picocolors_1.green)((0, picocolors_1.underline)('\nCodebuff') + ':') + ' ');
+    handleEscKey() {
+        if (this.isReceivingResponse) {
+            this.handleStopResponse();
+        }
+    }
+    handleStopResponse() {
+        console.log((0, picocolors_1.yellow)('\n[Response stopped by user]'));
+        this.isReceivingResponse = false;
+        if (this.stopResponse) {
+            this.stopResponse();
+        }
+        spinner_1.Spinner.get().stop();
+    }
+    async showEasterEgg() {
+        const text = 'codebuffy';
+        // Utility: clear the terminal screen
+        function clearScreen() {
+            process.stdout.write('\u001b[2J\u001b[0;0H');
+        }
+        const termWidth = process.stdout.columns;
+        const termHeight = process.stdout.rows;
+        const baselineWidth = 80;
+        const baselineHeight = 24;
+        const scaleFactor = Math.min(termWidth / baselineWidth, termHeight / baselineHeight);
+        // Utility: Generate a set of points tracing a "C" shape using an arc.
+        function generateCPath(cx, cy, r, steps) {
+            const points = [];
+            // A typical "C" opens to the right: from 45° to 315° (in radians)
+            const startAngle = Math.PI / 4;
+            const endAngle = (7 * Math.PI) / 4;
+            const angleStep = (endAngle - startAngle) / steps;
+            for (let i = 0; i <= steps; i++) {
+                const angle = startAngle + i * angleStep;
+                const x = Math.floor(cx + r * Math.cos(angle));
+                const y = Math.floor(cy + r * Math.sin(angle));
+                points.push({ x, y });
+            }
+            return points;
+        }
+        // Utility: Generate points along a quadratic Bézier curve.
+        function quadraticBezier(P0, P1, P2, steps) {
+            const points = [];
+            for (let i = 0; i <= steps; i++) {
+                const t = i / steps;
+                const x = Math.round((1 - t) ** 2 * P0.x + 2 * (1 - t) * t * P1.x + t ** 2 * P2.x);
+                const y = Math.round((1 - t) ** 2 * P0.y + 2 * (1 - t) * t * P1.y + t ** 2 * P2.y);
+                points.push({ x, y });
+            }
+            return points;
+        }
+        // Generate a vertical line from startY to endY at a given x.
+        function generateVerticalLine(x, startY, endY) {
+            const points = [];
+            const step = startY < endY ? 1 : -1;
+            for (let y = startY; y !== endY; y += step) {
+                points.push({ x, y });
+            }
+            points.push({ x, y: endY });
+            return points;
+        }
+        // Generate a path approximating a B shape using two quadratic Bézier curves
+        // for the rounded bubbles, and then closing the shape with a vertical spine.
+        function generateBPath(bX, bYTop, bYBottom, bWidth, bGap, stepsPerCurve) {
+            let points = [];
+            const middle = Math.floor((bYTop + bYBottom) / 2);
+            // Upper bubble: from top-left (spine) out then back to the spine at the middle.
+            const upperStart = { x: bX, y: bYTop };
+            const upperControl = {
+                x: bX + bWidth + bGap - 10,
+                y: Math.floor((bYTop + middle) / 2),
+            };
+            const upperEnd = { x: bX, y: middle };
+            const upperCurve = quadraticBezier(upperStart, upperControl, upperEnd, stepsPerCurve);
+            // Lower bubble: from the middle to the bottom.
+            const lowerStart = { x: bX, y: middle };
+            const lowerControl = {
+                x: bX + bWidth + bGap,
+                y: Math.floor((middle + bYBottom) / 2),
+            };
+            const lowerEnd = { x: bX, y: bYBottom };
+            const lowerCurve = quadraticBezier(lowerStart, lowerControl, lowerEnd, stepsPerCurve);
+            // Combine the curves.
+            points = points.concat(upperCurve, lowerCurve);
+            // Add a vertical line from the bottom of the B back up to the top.
+            const closingLine = generateVerticalLine(bX, bYBottom, bYTop);
+            points = points.concat(closingLine);
+            return points;
+        }
+        // Dynamically scale parameters for the shapes.
+        // Use Math.max to ensure values don't get too small.
+        const cCenterX = Math.floor(termWidth * 0.3);
+        const cCenterY = Math.floor(termHeight / 2);
+        const cRadius = Math.max(2, Math.floor(8 * scaleFactor));
+        const cSteps = Math.max(10, Math.floor(30 * scaleFactor));
+        const bX = Math.floor(termWidth * 0.55);
+        const bYTop = Math.floor(termHeight / 2 - 7 * scaleFactor);
+        const bYBottom = Math.floor(termHeight / 2 + 7 * scaleFactor);
+        const bWidth = Math.max(2, Math.floor(8 * scaleFactor));
+        const bGap = Math.max(1, Math.floor(35 * scaleFactor));
+        const bStepsPerCurve = Math.max(10, Math.floor(20 * scaleFactor));
+        // Generate the paths.
+        const fullPath = [
+            ...generateCPath(cCenterX, cCenterY, cRadius, cSteps),
+            ...generateBPath(bX, bYTop, bYBottom, bWidth, bGap, bStepsPerCurve),
+        ];
+        // Array of picocolors functions for random colors.
+        const colors = [picocolors_1.red, picocolors_1.green, picocolors_1.yellow, picocolors_1.blue, picocolors_1.magenta, picocolors_1.cyan];
+        function getRandomColor() {
+            return colors[Math.floor(Math.random() * colors.length)];
+        }
+        // Animation state: index into the fullPath.
+        let index = 0;
+        let completedCycle = false;
+        // Main animation function
+        function animate() {
+            if (index >= fullPath.length) {
+                completedCycle = true;
+                return;
+            }
+            const { x, y } = fullPath[index];
+            const cursorPosition = `\u001b[${y + 1};${x + 1}H`;
+            process.stdout.write(cursorPosition + getRandomColor()(text));
+            index++;
+        }
+        clearScreen();
+        const interval = setInterval(() => {
+            animate();
+            if (completedCycle) {
+                clearInterval(interval);
+                clearScreen();
+                this.returnControlToUser();
+            }
+        }, 100);
+    }
+    handleExit() {
+        spinner_1.Spinner.get().restoreCursor();
+        console.log('\n\n');
+        console.log(`${(0, string_1.pluralize)(this.client.sessionCreditsUsed, 'credit')} used this session.`);
+        if (this.client.limit && this.client.usage && this.client.nextQuotaReset) {
+            const daysUntilReset = Math.max(0, Math.floor((this.client.nextQuotaReset.getTime() - Date.now()) /
+                (1000 * 60 * 60 * 24)));
+            console.log(`${Math.max(0, this.client.limit - this.client.usage)} / ${this.client.limit} credits remaining. Renews in ${(0, string_1.pluralize)(daysUntilReset, 'day')}.`);
+        }
+        console.log((0, picocolors_1.green)('Codebuff out!'));
+        process.exit(0);
+    }
+    detectPasting() {
+        const currentTime = Date.now();
+        const timeDiff = currentTime - this.lastInputTime;
+        if (timeDiff < 10) {
+            this.consecutiveFastInputs++;
+            if (this.consecutiveFastInputs >= 2) {
+                this.isPasting = true;
+            }
+        }
+        else {
+            this.consecutiveFastInputs = 0;
+            if (this.isPasting) {
+                this.isPasting = false;
+            }
+        }
+        this.lastInputTime = currentTime;
+    }
+    handleDiff() {
+        if (this.client.lastChanges.length === 0) {
+            console.log((0, picocolors_1.yellow)('No changes found in the last assistant response.'));
+            return;
+        }
+        this.client.lastChanges.forEach((change) => {
+            console.log((0, picocolors_1.bold)(`___${change.path}___`));
+            const lines = change.content
+                .split('\n')
+                .map((line) => (change.type === 'file' ? '+' + line : line));
+            lines.forEach((line) => {
+                if (line.startsWith('+')) {
+                    console.log((0, picocolors_1.green)(line));
+                }
+                else if (line.startsWith('-')) {
+                    console.log((0, picocolors_1.red)(line));
+                }
+                else if (line.startsWith('@@')) {
+                    console.log((0, picocolors_1.cyan)(line));
+                }
+                else {
+                    console.log(line);
+                }
+            });
         });
-        this.stopResponse = stopResponse;
-        this.client.sendUserInput([], userInputId);
-        const result = await responsePromise;
-        this.stopResponse = null;
-        return result;
+    }
+    // Checkpoint command handlers
+    handleCheckpoints() {
+        console.log(checkpoints_1.checkpointManager.getCheckpointsAsString());
+        this.promptWithCheckpointNumber();
+    }
+    handleCheckpointDetail(id) {
+        const checkpoint = checkpoints_1.checkpointManager.getCheckpoint(id);
+        if (!checkpoint) {
+            console.log((0, picocolors_1.red)(`Checkpoint #${id} not found.`));
+        }
+        else {
+            console.log(checkpoints_1.checkpointManager.getCheckpointDetails(id));
+        }
+        this.promptWithCheckpointNumber();
+    }
+    async handleRestoreCheckpoint(id) {
+        const checkpoint = checkpoints_1.checkpointManager.getCheckpoint(id);
+        if (!checkpoint) {
+            console.log((0, picocolors_1.red)(`Checkpoint #${id} not found.`));
+            return;
+        }
+        this.restoreAgentStateAndFiles(checkpoint);
+        console.log((0, picocolors_1.green)(`Restored to checkpoint #${id}.`));
+        // Insert the original user input that created this checkpoint
+        this.promptWithCheckpointNumber();
+        this.rl.write(checkpoint.userInput);
+    }
+    restoreAgentStateAndFiles(checkpoint) {
+        // Restore the agentState
+        this.client.agentState = JSON.parse(checkpoint.agentStateString);
+        // Restore file state
+        const toChange = {
+            ...this.client.originalFileVersions,
+            ...checkpoint.fileVersions,
+        };
+        for (const [filePath, fileContents] of Object.entries(toChange)) {
+            if (fileContents === null) {
+                // delete file
+                fs.unlinkSync(filePath);
+            }
+            else {
+                fs.writeFileSync(filePath, fileContents);
+            }
+        }
+    }
+    async handleClearCheckpoints() {
+        checkpoints_1.checkpointManager.clearCheckpoints();
+        this.promptWithCheckpointNumber();
     }
 }
 exports.CLI = CLI;