codebuff 1.0.334 → 1.0.335

package/dist/cli.js

@@ -1,812 +0,0 @@
-"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 (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.CLI = void 0;
-const fs_1 = __importStar(require("fs"));
-const os = __importStar(require("os"));
-const os_1 = require("os");
-const path_1 = __importStar(require("path"));
-const readline = __importStar(require("readline"));
-const analytics_events_1 = require("./common/constants/analytics-events");
-const file_1 = require("./common/util/file");
-const string_1 = require("./common/util/string");
-const picocolors_1 = require("picocolors");
-const background_process_manager_1 = require("./background-process-manager");
-const checkpoint_manager_1 = require("./checkpoints/checkpoint-manager");
-const api_key_1 = require("./cli-handlers/api-key");
-const checkpoint_1 = require("./cli-handlers/checkpoint");
-const diff_1 = require("./cli-handlers/diff");
-const easter_egg_1 = require("./cli-handlers/easter-egg");
-const inititalization_flow_1 = require("./cli-handlers/inititalization-flow");
-const client_1 = require("./client");
-const config_1 = require("./config");
-const credentials_1 = require("./credentials");
-const display_1 = require("./display");
-const parser_1 = require("./json-config/parser");
-const menu_1 = require("./menu");
-const project_files_1 = require("./project-files");
-const rage_detectors_1 = require("./rage-detectors");
-const startup_process_handler_1 = require("./startup-process-handler");
-const base_1 = require("./terminal/base");
-const analytics_1 = require("./utils/analytics");
-const logger_1 = require("./utils/logger");
-const spinner_1 = require("./utils/spinner");
-const with_hang_detection_1 = require("./utils/with-hang-detection");
-const PROMPT_HISTORY_PATH = path_1.default.join(credentials_1.CONFIG_DIR, 'prompt_history.json');
-class CLI {
-    static instance = null;
-    readyPromise;
-    git;
-    costMode;
-    isReceivingResponse = false;
-    stopResponse = null;
-    lastSigintTime = 0;
-    lastInputTime = 0;
-    consecutiveFastInputs = 0;
-    pastedContent = '';
-    isPasting = false;
-    shouldReconnectWhenIdle = false;
-    rl;
-    constructor(readyPromise, { git, costMode, model }) {
-        this.git = git;
-        this.costMode = costMode;
-        this.setupSignalHandlers();
-        this.initReadlineInterface();
-        client_1.Client.createInstance({
-            websocketUrl: config_1.websocketUrl,
-            onWebSocketError: this.onWebSocketError.bind(this),
-            onWebSocketReconnect: this.onWebSocketReconnect.bind(this),
-            freshPrompt: this.freshPrompt.bind(this),
-            reconnectWhenNextIdle: this.reconnectWhenNextIdle.bind(this),
-            costMode: this.costMode,
-            git: this.git,
-            model,
-        });
-        this.readyPromise = Promise.all([
-            readyPromise.then((results) => {
-                const [fileContext, ,] = results;
-                client_1.Client.getInstance().initAgentState(fileContext);
-                return client_1.Client.getInstance().warmContextCache();
-            }),
-            client_1.Client.getInstance().connect(),
-        ]);
-        this.setPrompt();
-        process.on('unhandledRejection', (reason, promise) => {
-            rage_detectors_1.rageDetectors.exitAfterErrorDetector.start();
-            console.error('\nUnhandled Rejection at:', promise, 'reason:', reason);
-            logger_1.logger.error({
-                errorMessage: reason instanceof Error ? reason.message : String(reason),
-                errorStack: reason instanceof Error ? reason.stack : undefined,
-            }, 'Unhandled Rejection');
-            this.freshPrompt();
-        });
-        process.on('uncaughtException', (err, origin) => {
-            rage_detectors_1.rageDetectors.exitAfterErrorDetector.start();
-            console.error(`\nCaught exception: ${err}\n` + `Exception origin: ${origin}`);
-            console.error(err.stack);
-            logger_1.logger.error({
-                errorMessage: err.message,
-                errorStack: err.stack,
-                origin,
-            }, 'Uncaught Exception');
-            this.freshPrompt();
-        });
-    }
-    static initialize(readyPromise, options) {
-        if (CLI.instance) {
-            throw new Error('CLI is already initialized');
-        }
-        CLI.instance = new CLI(readyPromise, options);
-    }
-    static getInstance() {
-        if (!CLI.instance) {
-            throw new Error('CLI must be initialized before getting an instance');
-        }
-        return CLI.instance;
-    }
-    setupSignalHandlers() {
-        process.on('exit', () => {
-            spinner_1.Spinner.get().restoreCursor();
-            // Kill the persistent PTY process first
-            if (base_1.persistentProcess?.type === 'pty') {
-                base_1.persistentProcess.pty.kill();
-            }
-            (0, background_process_manager_1.sendKillSignalToAllBackgroundProcesses)();
-            const isHomeDir = (0, project_files_1.getProjectRoot)() === os.homedir();
-            if (!isHomeDir) {
-                console.log((0, picocolors_1.green)('Codebuff out!'));
-            }
-            if (process.env.NEXT_PUBLIC_CB_ENVIRONMENT === 'dev') {
-                logger_1.logger.info('[dev] active handles on close', process._getActiveHandles());
-            }
-        });
-        for (const signal of ['SIGTERM', 'SIGHUP']) {
-            process.on(signal, async () => {
-                process.removeAllListeners('unhandledRejection');
-                process.removeAllListeners('uncaughtException');
-                spinner_1.Spinner.get().restoreCursor();
-                await (0, background_process_manager_1.killAllBackgroundProcesses)();
-                client_1.Client.getInstance().close();
-                await (0, analytics_1.flushAnalytics)();
-                process.exit(0);
-            });
-        }
-        process.on('SIGTSTP', async () => await this.handleExit());
-        // Doesn't catch SIGKILL (e.g. `kill -9`)
-    }
-    _loadHistory() {
-        try {
-            if (fs_1.default.existsSync(PROMPT_HISTORY_PATH)) {
-                const content = fs_1.default.readFileSync(PROMPT_HISTORY_PATH, 'utf8');
-                const history = JSON.parse(content);
-                // Filter out empty lines and reverse for readline
-                return history.filter((line) => line.trim()).reverse();
-            }
-        }
-        catch (error) {
-            console.error('Error loading prompt history:', error);
-            logger_1.logger.error({
-                errorMessage: error instanceof Error ? error.message : String(error),
-                errorStack: error instanceof Error ? error.stack : undefined,
-            }, 'Error loading prompt history');
-            // If file doesn't exist or is invalid JSON, create empty history file
-            fs_1.default.writeFileSync(PROMPT_HISTORY_PATH, '[]');
-        }
-        return [];
-    }
-    _appendToHistory(line) {
-        try {
-            let history = [];
-            if (fs_1.default.existsSync(PROMPT_HISTORY_PATH)) {
-                const content = fs_1.default.readFileSync(PROMPT_HISTORY_PATH, 'utf8');
-                history = JSON.parse(content);
-            }
-            const trimmedLine = line.trim();
-            if (trimmedLine) {
-                // Remove all previous occurrences of the line
-                history = history.filter((h) => h !== trimmedLine);
-                // Add the new line to the end
-                history.push(trimmedLine);
-                fs_1.default.writeFileSync(PROMPT_HISTORY_PATH, JSON.stringify(history, null, 2));
-            }
-        }
-        catch (error) {
-            console.error('Error appending to prompt history:', error);
-            logger_1.logger.error({
-                errorMessage: error instanceof Error ? error.message : String(error),
-                errorStack: error instanceof Error ? error.stack : undefined,
-            }, 'Error appending to prompt history');
-        }
-    }
-    initReadlineInterface() {
-        this.rl = readline.createInterface({
-            input: process.stdin,
-            output: process.stdout,
-            historySize: 1000,
-            terminal: true,
-            completer: this.inputCompleter.bind(this),
-        });
-        // Load and populate history
-        const history = this._loadHistory();
-        this.rl.history.push(...history);
-        this.rl.on('line', (line) => this.handleLine(line));
-        this.rl.on('SIGINT', async () => await this.handleSigint());
-        this.rl.on('close', async () => await this.handleExit());
-        process.stdin.on('keypress', (str, key) => this.handleKeyPress(str, key));
-    }
-    inputCompleter(line) {
-        const lastWord = line.split(' ').pop() || '';
-        if (line.startsWith('/')) {
-            const slashCommands = (0, menu_1.getSlashCommands)();
-            const currentInput = line.substring(1); // Text after '/'
-            const matches = slashCommands
-                .map((cmd) => cmd.baseCommand) // Get base command strings
-                .filter((cmdName) => cmdName && cmdName.startsWith(currentInput))
-                .map((cmdName) => `/${cmdName}`); // Add back the slash for display
-            if (matches.length > 0) {
-                return [matches, line]; // Return all matches and the full line typed so far
-            }
-            return [[], line]; // No slash command matches
-        }
-        // Handle @ prefix for token and file completion
-        if (lastWord.startsWith('@')) {
-            const client = client_1.Client.getInstance();
-            if (!client.fileContext?.fileTree)
-                return [[], line];
-            const searchTerm = lastWord.substring(1); // Remove @ prefix
-            const searchTermLower = searchTerm.toLowerCase();
-            // Get token names from fileTokenScores
-            const tokenNames = Object.values(client.fileContext.fileTokenScores).flatMap((o) => Object.keys(o));
-            // Get all file paths
-            const paths = this.getAllFilePaths(client.fileContext.fileTree);
-            // Combine tokens and paths for matching
-            const allCandidates = [...tokenNames, ...paths];
-            const matchingItems = allCandidates.filter((item) => item.toLowerCase().startsWith(searchTermLower) ||
-                item.toLowerCase().includes('/' + searchTermLower));
-            // Limit the number of results to keep completion manageable
-            const MAX_COMPLETION_RESULTS = 20;
-            const limitedMatches = matchingItems.slice(0, MAX_COMPLETION_RESULTS);
-            if (limitedMatches.length > 1) {
-                // Find common prefix among matches
-                const suffixes = limitedMatches.map((item) => {
-                    const index = item.toLowerCase().indexOf(searchTermLower);
-                    return item.slice(index + searchTerm.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) {
-                    // Return the completion with @ prefix preserved
-                    return [['@' + searchTerm + commonPrefix], lastWord];
-                }
-                // Multiple matches but no common prefix - show matches WITHOUT @ prefix but keep @ in input
-                return [limitedMatches, lastWord];
-            }
-            // Single match or no matches - remove @ prefix from completion
-            return [limitedMatches, lastWord];
-        }
-        // Original file path completion logic (unchanged)
-        const input = lastWord.startsWith('~')
-            ? (0, os_1.homedir)() + lastWord.slice(1)
-            : lastWord;
-        const directorySuffix = process.platform === 'win32' ? '\\' : '/';
-        const dir = input.endsWith(directorySuffix)
-            ? input.slice(0, input.length - 1)
-            : (0, path_1.dirname)(input);
-        const partial = input.endsWith(directorySuffix) ? '' : (0, path_1.basename)(input);
-        let baseDir = (0, path_1.isAbsolute)(dir) ? dir : path_1.default.join((0, project_files_1.getWorkingDirectory)(), dir);
-        try {
-            const files = (0, fs_1.readdirSync)(baseDir);
-            const fsMatches = files
-                .filter((file) => file.startsWith(partial))
-                .map((file) => file + ((0, file_1.isDir)(path_1.default.join(baseDir, file)) ? directorySuffix : ''));
-            return [fsMatches, partial];
-        }
-        catch {
-            return [[], line];
-        }
-    }
-    getAllFilePaths(nodes, basePath = '') {
-        return nodes.flatMap((node) => {
-            if (node.type === 'file') {
-                return [path_1.default.join(basePath, node.name)];
-            }
-            return this.getAllFilePaths(node.children || [], path_1.default.join(basePath, node.name));
-        });
-    }
-    getModeIndicator() {
-        const costModeIndicator = this.costMode !== 'normal' ? ` (${this.costMode})` : '';
-        return costModeIndicator;
-    }
-    setPrompt() {
-        const projectRoot = (0, project_files_1.getProjectRoot)();
-        const cwd = (0, project_files_1.getWorkingDirectory)();
-        const projectDirName = (0, path_1.parse)(projectRoot).base;
-        const ps1Dir = projectDirName +
-            (cwd === projectRoot
-                ? ''
-                : (os.platform() === 'win32' ? '\\' : '/') +
-                    path_1.default.relative(projectRoot, cwd));
-        const modeIndicator = this.getModeIndicator();
-        this.rl.setPrompt((0, picocolors_1.green)(`${ps1Dir}${modeIndicator} > `));
-    }
-    /**
-     * Prompts the user with a clean prompt state
-     */
-    freshPrompt(userInput = '') {
-        spinner_1.Spinner.get().stop();
-        this.isReceivingResponse = false;
-        if (this.shouldReconnectWhenIdle) {
-            client_1.Client.getInstance().reconnect();
-            this.shouldReconnectWhenIdle = false;
-        }
-        readline.cursorTo(process.stdout, 0);
-        const rlAny = this.rl;
-        // Check for pending auto-topup message before showing prompt
-        if (client_1.Client.getInstance().pendingTopUpMessageAmount > 0) {
-            console.log('\n\n' +
-                (0, picocolors_1.green)(`Auto top-up successful! ${client_1.Client.getInstance().pendingTopUpMessageAmount.toLocaleString()} credits added.`) +
-                '\n');
-            client_1.Client.getInstance().pendingTopUpMessageAmount = 0;
-        }
-        // clear line first
-        rlAny.line = '';
-        this.pastedContent = '';
-        this.setPrompt();
-        // then prompt
-        this.rl.prompt();
-        (0, display_1.disableSquashNewlines)();
-        if (!userInput) {
-            return;
-        }
-        // then rewrite new prompt
-        this.rl.write(' '.repeat(userInput.length)); // hacky way to move cursor
-        rlAny.line = userInput;
-        rlAny._refreshLine();
-    }
-    async printInitialPrompt({ initialInput, runInitFlow, }) {
-        const client = client_1.Client.getInstance();
-        if (client.user) {
-            (0, menu_1.displayGreeting)(this.costMode, client.user.name);
-        }
-        else {
-            console.log(`Welcome to Codebuff! Give us a sec to get your account set up...`);
-            await client_1.Client.getInstance().login();
-            return;
-        }
-        this.freshPrompt();
-        if (runInitFlow) {
-            process.stdout.write('init\n');
-            await this.handleUserInput('init');
-        }
-        if (initialInput) {
-            process.stdout.write(initialInput + '\n');
-            await this.handleUserInput(initialInput);
-        }
-    }
-    async printDiff() {
-        (0, diff_1.handleDiff)(client_1.Client.getInstance().lastChanges);
-        this.freshPrompt();
-    }
-    async handleLine(line) {
-        this.detectPasting();
-        if (this.isPasting) {
-            this.pastedContent += line + '\n';
-        }
-        else if (!this.isReceivingResponse) {
-            const input = (this.pastedContent + line).trim();
-            this.pastedContent = '';
-            await this.handleUserInput(input);
-            this._appendToHistory(input);
-        }
-    }
-    async handleUserInput(userInput) {
-        (0, display_1.enableSquashNewlines)();
-        this.rl.setPrompt('');
-        if (!userInput) {
-            this.freshPrompt();
-            return;
-        }
-        userInput = userInput.trim();
-        // Record input for frustration detection before processing
-        const cleanedInput = this.cleanCommandInput(userInput);
-        rage_detectors_1.rageDetectors.repeatInputDetector.recordEvent(cleanedInput.toLowerCase().trim());
-        const processedResult = await (0, with_hang_detection_1.withHangDetection)(userInput, () => this.processCommand(userInput));
-        if (processedResult === null) {
-            // Command was fully handled by processCommand
-            return;
-        }
-        // processedResult is the string to be forwarded as a prompt
-        await this.forwardUserInput(processedResult);
-    }
-    /**
-     * Cleans command input by removing leading slash while preserving special command syntax
-     * @param input The raw user input
-     * @returns The cleaned command string
-     */
-    cleanCommandInput(input) {
-        return input.startsWith('/') ? input.substring(1) : input;
-    }
-    /**
-     * Checks if a command is a known slash command
-     * @param command The command to check (without leading slash)
-     */
-    isKnownSlashCommand(command) {
-        return (0, menu_1.getSlashCommands)().some((cmd) => cmd.baseCommand === command);
-    }
-    /**
-     * Handles an unknown slash command by displaying an error message
-     * @param command The unknown command that was entered
-     */
-    handleUnknownCommand(command) {
-        console.log((0, picocolors_1.yellow)(`Unknown slash command: ${command}`) +
-            `\nType / to see available commands`);
-        this.freshPrompt();
-    }
-    async processCommand(userInput) {
-        // Handle cost mode commands with optional message: /lite, /lite message, /normal, /normal message, etc.
-        const costModeMatch = userInput.match(/^\/?(lite|normal|max|experimental|ask)(?:\s+(.*))?$/i);
-        if (costModeMatch) {
-            const mode = costModeMatch[1].toLowerCase();
-            const message = costModeMatch[2]?.trim() || '';
-            // Track the cost mode command usage
-            (0, analytics_1.trackEvent)(analytics_events_1.AnalyticsEvent.SLASH_COMMAND_USED, {
-                userId: client_1.Client.getInstance().user?.id || 'unknown',
-                command: mode,
-            });
-            this.costMode = mode;
-            client_1.Client.getInstance().setCostMode(mode);
-            if (mode === 'lite') {
-                console.log((0, picocolors_1.yellow)('✨ Switched to lite mode (faster, cheaper)'));
-            }
-            else if (mode === 'normal') {
-                console.log((0, picocolors_1.green)('⚖️ Switched to normal mode (balanced)'));
-            }
-            else if (mode === 'max') {
-                console.log((0, picocolors_1.blueBright)('⚡ Switched to max mode (slower, more thorough)'));
-            }
-            else if (mode === 'experimental') {
-                console.log((0, picocolors_1.magenta)('🧪 Switched to experimental mode (cutting-edge)'));
-            }
-            else if (mode === 'ask') {
-                console.log((0, picocolors_1.cyan)('💬 Switched to ask mode (questions & planning only, no code changes)'));
-                console.log((0, picocolors_1.gray)('Tip: Use /export to save conversation summary to a file after fleshing out a plan'));
-            }
-            if (!message) {
-                this.freshPrompt();
-                return null; // Fully handled, no message to forward
-            }
-            // Return the message part to be processed as user input
-            return message;
-        }
-        const cleanInput = this.cleanCommandInput(userInput);
-        // Handle empty slash command
-        if (userInput === '/') {
-            return userInput; // Let it be processed as a prompt
-        }
-        // Track slash command usage if it starts with '/'
-        if (userInput.startsWith('/') && !userInput.startsWith('/!')) {
-            const commandBase = cleanInput.split(' ')[0];
-            if (!this.isKnownSlashCommand(commandBase)) {
-                (0, analytics_1.trackEvent)(analytics_events_1.AnalyticsEvent.INVALID_COMMAND, {
-                    userId: client_1.Client.getInstance().user?.id || 'unknown',
-                    command: cleanInput,
-                });
-                this.handleUnknownCommand(userInput);
-                return null;
-            }
-            // Track successful slash command usage
-            (0, analytics_1.trackEvent)(analytics_events_1.AnalyticsEvent.SLASH_COMMAND_USED, {
-                userId: client_1.Client.getInstance().user?.id || 'unknown',
-                command: commandBase,
-            });
-        }
-        if (cleanInput === 'help' || cleanInput === 'h') {
-            (0, menu_1.displayMenu)();
-            this.freshPrompt();
-            return null;
-        }
-        if (cleanInput === 'login' || cleanInput === 'signin') {
-            await client_1.Client.getInstance().login();
-            checkpoint_manager_1.checkpointManager.clearCheckpoints();
-            return null;
-        }
-        if (cleanInput === 'logout' || cleanInput === 'signout') {
-            await client_1.Client.getInstance().logout();
-            this.freshPrompt();
-            return null;
-        }
-        if (cleanInput.startsWith('ref-')) {
-            // Referral codes can be entered with or without a leading slash.
-            // Pass the cleaned input (without slash) to the handler.
-            await client_1.Client.getInstance().handleReferralCode(cleanInput.trim());
-            return null;
-        }
-        // Detect potential API key input first
-        // API keys are not slash commands, so use userInput
-        const detectionResult = (0, api_key_1.detectApiKey)(userInput);
-        if (detectionResult.status !== 'not_found') {
-            await (0, api_key_1.handleApiKeyInput)(client_1.Client.getInstance(), detectionResult, this.readyPromise, this.freshPrompt.bind(this));
-            return null;
-        }
-        if (cleanInput === 'usage' || cleanInput === 'credits') {
-            await client_1.Client.getInstance().getUsage();
-            return null;
-        }
-        if (cleanInput === 'quit' || cleanInput === 'exit' || cleanInput === 'q') {
-            await this.handleExit();
-            return null;
-        }
-        if (cleanInput === 'reset') {
-            await this.readyPromise;
-            await client_1.Client.getInstance().resetContext();
-            const projectRoot = (0, project_files_1.getProjectRoot)();
-            (0, base_1.clearScreen)();
-            // from index.ts
-            const config = (0, parser_1.loadCodebuffConfig)();
-            await (0, background_process_manager_1.killAllBackgroundProcesses)();
-            const processStartPromise = (0, startup_process_handler_1.logAndHandleStartup)();
-            const initFileContextPromise = (0, project_files_1.initProjectFileContextWithWorker)(projectRoot, true);
-            this.readyPromise = Promise.all([
-                initFileContextPromise,
-                processStartPromise,
-            ]);
-            (0, menu_1.displayGreeting)(this.costMode, client_1.Client.getInstance().user?.name ?? null);
-            this.freshPrompt();
-            return null;
-        }
-        if (['diff', 'doff', 'dif', 'iff', 'd'].includes(cleanInput)) {
-            (0, diff_1.handleDiff)(client_1.Client.getInstance().lastChanges);
-            this.freshPrompt();
-            return null;
-        }
-        if (cleanInput === 'uuddlrlrba' ||
-            cleanInput === 'konami' ||
-            cleanInput === 'codebuffy') {
-            (0, easter_egg_1.showEasterEgg)(this.freshPrompt.bind(this));
-            return null;
-        }
-        // Checkpoint commands
-        if ((0, checkpoint_1.isCheckpointCommand)(cleanInput)) {
-            (0, analytics_1.trackEvent)(analytics_events_1.AnalyticsEvent.CHECKPOINT_COMMAND_USED, {
-                command: cleanInput, // Log the cleaned command
-            });
-            if ((0, checkpoint_1.isCheckpointCommand)(cleanInput, 'undo')) {
-                await (0, checkpoint_1.saveCheckpoint)(userInput, client_1.Client.getInstance(), this.readyPromise);
-                const toRestore = await (0, checkpoint_1.handleUndo)(client_1.Client.getInstance(), this.rl);
-                this.freshPrompt(toRestore);
-                return null;
-            }
-            if ((0, checkpoint_1.isCheckpointCommand)(cleanInput, 'redo')) {
-                await (0, checkpoint_1.saveCheckpoint)(userInput, client_1.Client.getInstance(), this.readyPromise);
-                const toRestore = await (0, checkpoint_1.handleRedo)(client_1.Client.getInstance(), this.rl);
-                this.freshPrompt(toRestore);
-                return null;
-            }
-            if ((0, checkpoint_1.isCheckpointCommand)(cleanInput, 'list')) {
-                await (0, checkpoint_1.saveCheckpoint)(userInput, client_1.Client.getInstance(), this.readyPromise);
-                await (0, checkpoint_1.listCheckpoints)();
-                this.freshPrompt();
-                return null;
-            }
-            const restoreMatch = (0, checkpoint_1.isCheckpointCommand)(cleanInput, 'restore');
-            if (restoreMatch) {
-                const id = parseInt(restoreMatch[1], 10);
-                await (0, checkpoint_1.saveCheckpoint)(userInput, client_1.Client.getInstance(), this.readyPromise);
-                const toRestore = await (0, checkpoint_1.handleRestoreCheckpoint)(id, client_1.Client.getInstance(), this.rl);
-                this.freshPrompt(toRestore);
-                return null;
-            }
-            if ((0, checkpoint_1.isCheckpointCommand)(cleanInput, 'clear')) {
-                (0, checkpoint_1.handleClearCheckpoints)();
-                this.freshPrompt();
-                return null;
-            }
-            if ((0, checkpoint_1.isCheckpointCommand)(cleanInput, 'save')) {
-                await (0, checkpoint_1.saveCheckpoint)(userInput, client_1.Client.getInstance(), this.readyPromise, true);
-                (0, checkpoint_1.displayCheckpointMenu)();
-                this.freshPrompt();
-                return null;
-            }
-            // Default checkpoint action (if just "checkpoint" or "/checkpoint" is typed)
-            (0, checkpoint_1.displayCheckpointMenu)();
-            this.freshPrompt();
-            return null;
-        }
-        if (cleanInput === 'init') {
-            (0, inititalization_flow_1.handleInitializationFlowLocally)();
-            // Also forward user input (original with / if present, or cleanInput) to the backend
-            // The original forwardUserInput takes the raw userInput.
-            return userInput; // Let it fall through to forwardUserInput
-        }
-        if (cleanInput === 'export') {
-            console.log((0, picocolors_1.yellow)('Exporting conversation to a file...'));
-            // Forward to backend like init command
-            return userInput; // Let it fall through to forwardUserInput
-        }
-        if (cleanInput === 'compact') {
-            console.log((0, picocolors_1.yellow)('Compacting conversation...'));
-            // Forward to backend
-            return userInput;
-        }
-        // If no command was matched, return the original userInput to be processed as a prompt
-        return userInput;
-    }
-    async forwardUserInput(promptContent) {
-        const cleanedInput = this.cleanCommandInput(promptContent);
-        await (0, checkpoint_1.saveCheckpoint)(cleanedInput, client_1.Client.getInstance(), this.readyPromise);
-        spinner_1.Spinner.get().start('Thinking...');
-        this.isReceivingResponse = true;
-        const { responsePromise, stopResponse } = await client_1.Client.getInstance().sendUserInput(cleanedInput);
-        this.stopResponse = stopResponse;
-        await responsePromise;
-        this.stopResponse = null;
-        this.isReceivingResponse = false;
-        spinner_1.Spinner.get().stop();
-        this.freshPrompt();
-    }
-    reconnectWhenNextIdle() {
-        if (!this.isReceivingResponse) {
-            client_1.Client.getInstance().reconnect();
-        }
-        else {
-            this.shouldReconnectWhenIdle = true;
-        }
-    }
-    onWebSocketError() {
-        rage_detectors_1.rageDetectors.exitAfterErrorDetector.start();
-        spinner_1.Spinner.get().stop();
-        this.isReceivingResponse = false;
-        if (this.stopResponse) {
-            this.stopResponse();
-            this.stopResponse = null;
-        }
-        console.error('\n' + (0, picocolors_1.yellow)('Could not connect. Retrying...'));
-        logger_1.logger.error({
-            errorMessage: 'Could not connect. Retrying...',
-        }, 'WebSocket connection error');
-        // Start hang detection for persistent connection issues
-        rage_detectors_1.rageDetectors.webSocketHangDetector.start({
-            connectionIssue: 'websocket_persistent_failure',
-            url: config_1.websocketUrl,
-        });
-    }
-    onWebSocketReconnect() {
-        // Stop hang detection on successful reconnection
-        rage_detectors_1.rageDetectors.webSocketHangDetector.stop();
-        console.log('\n' + (0, picocolors_1.green)('Reconnected!'));
-        this.freshPrompt();
-    }
-    handleKeyPress(str, key) {
-        // Track key mashing on repeated normal keys
-        const isModifier = key?.meta || key?.alt || key?.shift;
-        const isSpecialKey = key?.name === 'backspace' ||
-            key?.name === 'space' ||
-            key?.name === 'enter' ||
-            key?.name === 'tab';
-        // Control-C is a key mashing pattern we want to detect
-        const isControlC = key?.ctrl && key?.name === 'c';
-        if (isControlC) {
-            rage_detectors_1.rageDetectors.keyMashingDetector.recordEvent('ctrl-c');
-        }
-        else if (!isModifier && !isSpecialKey && key?.name) {
-            rage_detectors_1.rageDetectors.keyMashingDetector.recordEvent(key.name);
-        }
-        if (key.name === 'escape') {
-            this.handleEscKey();
-        }
-        if (str === '/') {
-            const currentLine = this.pastedContent + this.rl.line;
-            // Only track and show menu if '/' is the first character typed
-            if (currentLine === '/') {
-                (0, analytics_1.trackEvent)(analytics_events_1.AnalyticsEvent.SLASH_MENU_ACTIVATED, {
-                    userId: client_1.Client.getInstance().user?.id || 'unknown',
-                });
-                (0, menu_1.displaySlashCommandHelperMenu)();
-                // Call freshPrompt and pre-fill the line with the slash
-                // so the user can continue typing their command.
-                this.freshPrompt('/');
-            }
-        }
-        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();
-    }
-    async handleSigint() {
-        if ((0, base_1.isCommandRunning)()) {
-            (0, base_1.resetShell)((0, project_files_1.getProjectRoot)());
-        }
-        if (this.isReceivingResponse) {
-            this.handleStopResponse();
-        }
-        else {
-            const now = Date.now();
-            if (now - this.lastSigintTime < 5000 && !this.rl.line) {
-                await this.handleExit();
-            }
-            else {
-                this.lastSigintTime = now;
-                console.log('\nPress Ctrl-C again to exit');
-                this.freshPrompt();
-            }
-        }
-    }
-    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 handleExit() {
-        // Start exit time detector
-        rage_detectors_1.rageDetectors.exitTimeDetector.start();
-        // Call end() on the exit detector to check if user is exiting quickly after an error
-        rage_detectors_1.rageDetectors.exitAfterErrorDetector.end();
-        spinner_1.Spinner.get().restoreCursor();
-        process.removeAllListeners('unhandledRejection');
-        process.removeAllListeners('uncaughtException');
-        console.log('\n');
-        // Kill the persistent PTY process first
-        (0, base_1.killAndResetPersistentProcess)();
-        await (0, background_process_manager_1.killAllBackgroundProcesses)();
-        client_1.Client.getInstance().close(); // Close WebSocket
-        const client = client_1.Client.getInstance();
-        // Check for organization coverage first
-        const coverage = await client.checkRepositoryCoverage();
-        // Calculate session usage and total for display
-        const totalCreditsUsedThisSession = Object.values(client.creditsByPromptId)
-            .flat()
-            .reduce((sum, credits) => sum + credits, 0);
-        let exitUsageMessage = `${(0, string_1.pluralize)(totalCreditsUsedThisSession, 'credit')} used this session`;
-        if (client.usageData.remainingBalance !== null) {
-            exitUsageMessage += `, ${client.usageData.remainingBalance.toLocaleString()} credits left.`;
-        }
-        else {
-            exitUsageMessage += '.';
-        }
-        console.log(exitUsageMessage);
-        if (coverage.isCovered && coverage.organizationName) {
-            // When covered by an organization, show organization information
-            console.log((0, picocolors_1.green)(`Your usage in this repository was covered by the ${(0, picocolors_1.bold)(coverage.organizationName)} organization.`));
-        }
-        else {
-            // Only show personal credit renewal when not covered by an organization
-            if (client.usageData.next_quota_reset) {
-                const daysUntilReset = Math.ceil((new Date(client.usageData.next_quota_reset).getTime() - Date.now()) /
-                    (1000 * 60 * 60 * 24));
-                console.log(`Your free credits will reset in ${(0, string_1.pluralize)(daysUntilReset, 'day')}.`);
-            }
-        }
-        // End exit time detector right before process.exit
-        rage_detectors_1.rageDetectors.exitTimeDetector.end();
-        await (0, analytics_1.flushAnalytics)();
-        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;
-    }
-}
-exports.CLI = CLI;
-//# sourceMappingURL=cli.js.map