codebuff 1.0.148 → 1.0.150

package/dist/cli.js

@@ -40,23 +40,13 @@ const changes_1 = require("./common/util/changes");
 const readline = __importStar(require("readline"));
 const picocolors_1 = require("picocolors");
 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 chat_storage_1 = require("./chat-storage");
 const client_1 = require("./client");
 const menu_1 = require("./menu");
 const project_files_1 = require("./project-files");
 const tool_handlers_1 = require("./tool-handlers");
+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");
@@ -78,52 +68,13 @@ class CLI {
     consecutiveFastInputs = 0;
     pastedContent = '';
     isPasting = false;
+    // ==================== Initialization Methods ====================
     constructor(readyPromise, { git, costMode }) {
         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) => {
@@ -134,109 +85,63 @@ class CLI {
             this.client.connect(),
         ]);
         this.setPrompt();
-        this.rl.on('line', (line) => {
-            this.handleInput(line);
+    }
+    setupSignalHandlers() {
+        process.on('exit', () => this.restoreCursor());
+        process.on('SIGTERM', () => {
+            this.restoreCursor();
+            process.exit(0);
         });
-        this.rl.on('SIGINT', () => {
-            if (this.isReceivingResponse) {
-                this.handleStopResponse();
-            }
-            else {
-                const now = Date.now();
-                if (now - this.lastSigintTime < 3000) {
-                    // 3 second window
-                    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 (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} > `));
     }
+    // ==================== Public Methods ====================
     async printInitialPrompt(initialInput) {
         if (this.client.user) {
             (0, menu_1.displayGreeting)(this.costMode, this.client.user.name);
@@ -252,177 +157,83 @@ class CLI {
             this.handleUserInput(initialInput);
         }
     }
-    handleUndo() {
-        this.navigateFileVersion('undo');
-        this.rl.prompt();
-    }
-    handleRedo() {
-        this.navigateFileVersion('redo');
+    async printDiff() {
+        this.handleDiff();
         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();
-    }
-    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
+    // ==================== Input Handling Methods ====================
+    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);
+    }
+    /**
+     * Checks if the input matches a built-in command.
+     * Returns true if the command has been handled.
+     */
+    async processCommand(userInput) {
         if (userInput === 'help' || userInput === 'h') {
             (0, menu_1.displayMenu)();
             this.rl.prompt();
-            return;
+            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;
+            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;
+            return true;
         }
-        else if (userInput === 'redo' || userInput === 'r') {
+        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;
+            return true;
         }
         const runPrefix = '/run ';
         const hasRunPrefix = userInput.startsWith(runPrefix);
         if (hasRunPrefix ||
-            (!constants_1.SKIPPED_TERMINAL_COMMANDS.some((command) => userInput.toLowerCase().startsWith(command)) &&
+            (!constants_1.SKIPPED_TERMINAL_COMMANDS.some((cmd) => userInput.toLowerCase().startsWith(cmd)) &&
                 !userInput.includes('error ') &&
                 !userInput.includes("'") &&
                 userInput.split(' ').length <= 5)) {
@@ -431,15 +242,22 @@ class CLI {
             if (result !== 'command not found') {
                 this.setPrompt();
                 this.rl.prompt();
-                return;
+                return true;
             }
             else if (hasRunPrefix) {
                 process.stdout.write(stdout);
                 this.setPrompt();
                 this.rl.prompt();
-                return;
+                return true;
             }
         }
+        return false;
+    }
+    /**
+     * Processes regular user input (non-command) by gathering file changes and scraped content,
+     * sending the consolidated user message, and then applying assistant response changes.
+     */
+    async forwardUserInput(userInput) {
         this.startLoadingAnimation();
         await this.readyPromise;
         const currentChat = this.chatStorage.getCurrentChat();
@@ -493,7 +311,7 @@ class CLI {
             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.`);
             }
-            console.log('Complete! Type "diff" to see the changes made.');
+            console.log('Complete! Type "diff" to review changes or "undo" to revert.');
             this.client.showUsageWarning();
         }
         console.log();
@@ -507,6 +325,28 @@ class CLI {
         this.chatStorage.addNewFileState(updatedFiles);
         this.rl.prompt();
     }
+    // ==================== WebSocket and Response Handling ====================
+    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();
+    }
     async sendUserInputAndAwaitResponse() {
         const userInputId = `mc-input-` + Math.random().toString(36).substring(2, 15);
         const { responsePromise, stopResponse } = this.client.subscribeToResponse((chunk) => {
@@ -521,6 +361,197 @@ class CLI {
         this.stopResponse = null;
         return result;
     }
+    // ==================== File Version Navigation ====================
+    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`));
+        }
+    }
+    applyAndDisplayCurrentFileVersion() {
+        const currentVersion = this.chatStorage.getCurrentVersion();
+        if (currentVersion) {
+            (0, project_files_1.setFiles)(currentVersion.files);
+            return currentVersion.files;
+        }
+        return {};
+    }
+    // ==================== Terminal Animation and Cursor Management ====================
+    startLoadingAnimation() {
+        const chars = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
+        let i = 0;
+        // Hide cursor while spinner is active
+        process.stdout.write('\u001B[?25l');
+        this.loadingInterval = setInterval(() => {
+            this.rewriteLine((0, picocolors_1.green)(`${chars[i]} Thinking...`));
+            i = (i + 1) % chars.length;
+        }, 100);
+    }
+    stopLoadingAnimation() {
+        if (this.loadingInterval) {
+            clearInterval(this.loadingInterval);
+            this.loadingInterval = null;
+            this.rewriteLine(''); // Clear the spinner line
+            this.restoreCursor(); // Show cursor after spinner stops
+        }
+    }
+    rewriteLine(line) {
+        if (process.stdout.isTTY) {
+            readline.clearLine(process.stdout, 0);
+            readline.cursorTo(process.stdout, 0);
+            process.stdout.write(line);
+        }
+        else {
+            process.stdout.write(line + '\n');
+        }
+    }
+    restoreCursor() {
+        process.stdout.write('\u001B[?25h');
+    }
+    // ==================== Keyboard and SIGINT Handling ====================
+    handleKeyPress(str, key) {
+        if (key.name === 'escape') {
+            this.handleEscKey();
+        }
+        // Convert double spaces into newlines
+        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();
+            }
+        }
+        this.detectPasting();
+    }
+    handleSigint() {
+        if ((0, terminal_1.isCommandRunning)()) {
+            (0, terminal_1.resetShell)();
+        }
+        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();
+            }
+            else {
+                this.lastSigintTime = now;
+                console.log('\nPress Ctrl-C again to exit');
+                this.rl.prompt();
+            }
+        }
+    }
+    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();
+        }
+        this.stopLoadingAnimation();
+        this.restoreCursor();
+    }
+    // ==================== Exit Handling ====================
+    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);
+    }
+    // ==================== Pasting Detection ====================
+    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;
+    }
+    // ==================== Auto Commit Handling ====================
+    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;
+        }
+        return undefined;
+    }
+    // ==================== Diff Handling ====================
+    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);
+                }
+            });
+        });
+    }
 }
 exports.CLI = CLI;
 //# sourceMappingURL=cli.js.map