snow-ai 0.3.7 → 0.3.8

package/dist/ui/pages/ChatScreen.js

@@ -43,6 +43,7 @@ import '../../utils/commands/home.js';
 import '../../utils/commands/review.js';
 import '../../utils/commands/role.js';
 import '../../utils/commands/usage.js';
+import '../../utils/commands/export.js';
 export default function ChatScreen({ skipWelcome }) {
     const [messages, setMessages] = useState([]);
     const [isSaving] = useState(false);
@@ -260,6 +261,8 @@ export default function ChatScreen({ skipWelcome }) {
             streamingState.abortController) {
             // Abort the controller
             streamingState.abortController.abort();
+            // Clear retry status immediately when user cancels
+            streamingState.setRetryStatus(null);
             // Remove all pending tool call messages (those with toolPending: true)
             setMessages(prev => prev.filter(msg => !msg.toolPending));
             // Add discontinued message
@@ -279,10 +282,6 @@ export default function ChatScreen({ skipWelcome }) {
         }
     });
     const handleHistorySelect = async (selectedIndex, message) => {
-        // If rolling back to the first message (index 0), save the message content to restore in input
-        if (selectedIndex === 0) {
-            setRestoreInputContent(message);
-        }
         // Count total files that will be rolled back (from selectedIndex onwards)
         let totalFileCount = 0;
         for (const [index, count] of snapshotState.snapshotFileCount.entries()) {
@@ -301,39 +300,95 @@ export default function ChatScreen({ skipWelcome }) {
                 messageIndex: selectedIndex,
                 fileCount: filePaths.length, // Use actual unique file count
                 filePaths,
+                message, // Save message for restore after rollback
             });
         }
         else {
             // No files to rollback, just rollback conversation
-            performRollback(selectedIndex, false);
+            // Note: message is already in input buffer via useHistoryNavigation
+            await performRollback(selectedIndex, false);
         }
     };
     const performRollback = async (selectedIndex, rollbackFiles) => {
+        const currentSession = sessionManager.getCurrentSession();
         // Rollback workspace to checkpoint if requested
-        if (rollbackFiles) {
-            const currentSession = sessionManager.getCurrentSession();
-            if (currentSession) {
-                // Use rollbackToMessageIndex to rollback all snapshots >= selectedIndex
-                await incrementalSnapshotManager.rollbackToMessageIndex(currentSession.id, selectedIndex);
+        if (rollbackFiles && currentSession) {
+            // Use rollbackToMessageIndex to rollback all snapshots >= selectedIndex
+            await incrementalSnapshotManager.rollbackToMessageIndex(currentSession.id, selectedIndex);
+        }
+        // For session file: find the correct truncation point based on session messages
+        // We need to truncate to the same user message in the session file
+        if (currentSession) {
+            // Count how many user messages we're deleting (from selectedIndex onwards in UI)
+            const uiUserMessagesToDelete = messages
+                .slice(selectedIndex)
+                .filter(msg => msg.role === 'user').length;
+            // Find the corresponding user message in session to delete
+            // We start from the end and count backwards
+            let sessionUserMessageCount = 0;
+            let sessionTruncateIndex = currentSession.messages.length;
+            for (let i = currentSession.messages.length - 1; i >= 0; i--) {
+                const msg = currentSession.messages[i];
+                if (msg && msg.role === 'user') {
+                    sessionUserMessageCount++;
+                    if (sessionUserMessageCount === uiUserMessagesToDelete) {
+                        // We want to delete from this user message onwards
+                        sessionTruncateIndex = i;
+                        break;
+                    }
+                }
+            }
+            // Special case: rolling back to index 0 means deleting the entire session
+            if (sessionTruncateIndex === 0 && currentSession) {
+                // Delete all snapshots for this session
+                await incrementalSnapshotManager.clearAllSnapshots(currentSession.id);
+                // Delete the session file
+                await sessionManager.deleteSession(currentSession.id);
+                // Clear current session
+                sessionManager.clearCurrentSession();
+                // Clear all messages
+                setMessages([]);
+                // Clear saved messages
+                clearSavedMessages();
+                // Clear snapshot state
+                snapshotState.setSnapshotFileCount(new Map());
+                // Clear pending rollback dialog
+                snapshotState.setPendingRollback(null);
+                // Trigger remount
+                setRemountKey(prev => prev + 1);
+                return;
             }
+            // Delete snapshot files >= selectedIndex (regardless of whether files were rolled back)
+            await incrementalSnapshotManager.deleteSnapshotsFromIndex(currentSession.id, selectedIndex);
+            // Reload snapshot file counts from disk after deletion
+            const snapshots = await incrementalSnapshotManager.listSnapshots(currentSession.id);
+            const counts = new Map();
+            for (const snapshot of snapshots) {
+                counts.set(snapshot.messageIndex, snapshot.fileCount);
+            }
+            snapshotState.setSnapshotFileCount(counts);
+            // Truncate session messages
+            await sessionManager.truncateMessages(sessionTruncateIndex);
         }
-        // Truncate messages array to remove the selected user message and everything after it
+        // Truncate UI messages array to remove the selected user message and everything after it
         setMessages(prev => prev.slice(0, selectedIndex));
-        // Truncate session messages to match the UI state
-        await sessionManager.truncateMessages(selectedIndex);
         clearSavedMessages();
         setRemountKey(prev => prev + 1);
         // Clear pending rollback dialog
         snapshotState.setPendingRollback(null);
     };
-    const handleRollbackConfirm = (rollbackFiles) => {
+    const handleRollbackConfirm = async (rollbackFiles) => {
         if (rollbackFiles === null) {
             // User cancelled - just close the dialog without doing anything
             snapshotState.setPendingRollback(null);
             return;
         }
         if (snapshotState.pendingRollback) {
-            performRollback(snapshotState.pendingRollback.messageIndex, rollbackFiles);
+            // Restore message to input before rollback
+            if (snapshotState.pendingRollback.message) {
+                setRestoreInputContent(snapshotState.pendingRollback.message);
+            }
+            await performRollback(snapshotState.pendingRollback.messageIndex, rollbackFiles);
         }
     };
     const handleSessionPanelSelect = async (sessionId) => {
@@ -380,6 +435,8 @@ export default function ChatScreen({ skipWelcome }) {
         await processMessage(message, images);
     };
     const processMessage = async (message, images, useBasicModel, hideUserMessage) => {
+        // Clear any previous retry status when starting a new request
+        streamingState.setRetryStatus(null);
         // Parse and validate file references
         const { cleanContent, validFiles } = await parseAndValidateFileReferences(message);
         // Separate image files from regular files
@@ -472,6 +529,8 @@ export default function ChatScreen({ skipWelcome }) {
     const processPendingMessages = async () => {
         if (pendingMessages.length === 0)
             return;
+        // Clear any previous retry status when starting a new request
+        streamingState.setRetryStatus(null);
         // Get current pending messages and clear them immediately
         const messagesToProcess = [...pendingMessages];
         setPendingMessages([]);
@@ -588,10 +647,11 @@ export default function ChatScreen({ skipWelcome }) {
                                 workingDirectory)))),
                 ...messages
                     .filter(m => !m.streaming)
-                    .map((message, index) => {
+                    .map((message, index, filteredMessages) => {
                     // Determine tool message type and color
                     let toolStatusColor = 'cyan';
                     let isToolMessage = false;
+                    const isLastMessage = index === filteredMessages.length - 1;
                     if (message.role === 'assistant') {
                         if (message.content.startsWith('⚡')) {
                             isToolMessage = true;
@@ -609,7 +669,7 @@ export default function ChatScreen({ skipWelcome }) {
                             toolStatusColor = 'blue';
                         }
                     }
-                    return (React.createElement(Box, { key: `msg-${index}`, marginBottom: isToolMessage ? 0 : 1, paddingX: 1, flexDirection: "column", width: terminalWidth },
+                    return (React.createElement(Box, { key: `msg-${index}`, marginTop: index > 0 ? 1 : 0, marginBottom: isLastMessage ? 1 : 0, paddingX: 1, flexDirection: "column", width: terminalWidth },
                         React.createElement(Box, null,
                             React.createElement(Text, { color: message.role === 'user'
                                     ? 'green'
@@ -620,7 +680,7 @@ export default function ChatScreen({ skipWelcome }) {
                                 : message.role === 'command'
                                     ? '⌘'
                                     : '❆'),
-                            React.createElement(Box, { marginLeft: 1, marginBottom: 1, flexDirection: "column" }, message.role === 'command' ? (React.createElement(React.Fragment, null,
+                            React.createElement(Box, { marginLeft: 1, flexDirection: "column" }, message.role === 'command' ? (React.createElement(React.Fragment, null,
                                 React.createElement(Text, { color: "gray", dimColor: true },
                                     "\u2514\u2500 ",
                                     message.commandName),
@@ -641,8 +701,7 @@ export default function ChatScreen({ skipWelcome }) {
                                     ' ',
                                     arg.value))))),
                                 message.toolCall &&
-                                    (message.toolCall.name === 'filesystem-create' ||
-                                        message.toolCall.name === 'filesystem-write') &&
+                                    message.toolCall.name === 'filesystem-create' &&
                                     message.toolCall.arguments.content && (React.createElement(Box, { marginTop: 1 },
                                     React.createElement(DiffViewer, { newContent: message.toolCall.arguments.content, filename: message.toolCall.arguments.path }))),
                                 message.toolCall &&
@@ -661,47 +720,27 @@ export default function ChatScreen({ skipWelcome }) {
                                             .completeOldContent, completeNewContent: message.toolCall.arguments
                                             .completeNewContent, startLineNumber: message.toolCall.arguments.contextStartLine }))),
                                 message.toolCall &&
-                                    message.toolCall.name === 'terminal-execute' &&
-                                    message.toolCall.arguments.command && (React.createElement(Box, { marginTop: 1, flexDirection: "column" },
-                                    React.createElement(Text, { color: "gray", dimColor: true },
-                                        "\u2514\u2500 Command:",
-                                        ' ',
-                                        React.createElement(Text, { color: "white" }, message.toolCall.arguments.command)),
-                                    React.createElement(Text, { color: "gray", dimColor: true },
-                                        "\u2514\u2500 Exit Code:",
-                                        ' ',
-                                        React.createElement(Text, { color: message.toolCall.arguments.exitCode === 0
-                                                ? 'green'
-                                                : 'red' }, message.toolCall.arguments.exitCode)),
-                                    message.toolCall.arguments.stdout &&
-                                        message.toolCall.arguments.stdout.trim()
-                                            .length > 0 && (React.createElement(Box, { flexDirection: "column", marginTop: 1 },
-                                        React.createElement(Text, { color: "green", dimColor: true }, "\u2514\u2500 stdout:"),
-                                        React.createElement(Box, { paddingLeft: 2 },
-                                            React.createElement(Text, { color: "white" }, message.toolCall.arguments.stdout
-                                                .trim()
-                                                .split('\n')
-                                                .slice(0, 20)
-                                                .join('\n')),
-                                            message.toolCall.arguments.stdout
-                                                .trim()
-                                                .split('\n').length > 20 && (React.createElement(Text, { color: "gray", dimColor: true }, "... (output truncated)"))))),
-                                    message.toolCall.arguments.stderr &&
-                                        message.toolCall.arguments.stderr.trim()
-                                            .length > 0 && (React.createElement(Box, { flexDirection: "column", marginTop: 1 },
-                                        React.createElement(Text, { color: "red", dimColor: true }, "\u2514\u2500 stderr:"),
-                                        React.createElement(Box, { paddingLeft: 2 },
-                                            React.createElement(Text, { color: "red" }, message.toolCall.arguments.stderr
-                                                .trim()
-                                                .split('\n')
-                                                .slice(0, 10)
-                                                .join('\n')),
-                                            message.toolCall.arguments.stderr
-                                                .trim()
-                                                .split('\n').length > 10 && (React.createElement(Text, { color: "gray", dimColor: true }, "... (output truncated)"))))))),
+                                    (message.toolCall.name === 'filesystem-edit' ||
+                                        message.toolCall.name ===
+                                            'filesystem-edit_search') &&
+                                    message.toolCall.arguments.isBatch &&
+                                    message.toolCall.arguments.batchResults &&
+                                    Array.isArray(message.toolCall.arguments.batchResults) && (React.createElement(Box, { marginTop: 1, flexDirection: "column" }, message.toolCall.arguments.batchResults.map((fileResult, index) => {
+                                    if (fileResult.success &&
+                                        fileResult.oldContent &&
+                                        fileResult.newContent) {
+                                        return (React.createElement(Box, { key: index, flexDirection: "column", marginBottom: 1 },
+                                            React.createElement(Text, { bold: true, color: "cyan" }, `File ${index + 1}: ${fileResult.path}`),
+                                            React.createElement(DiffViewer, { oldContent: fileResult.oldContent, newContent: fileResult.newContent, filename: fileResult.path, completeOldContent: fileResult.completeOldContent, completeNewContent: fileResult.completeNewContent, startLineNumber: fileResult.contextStartLine })));
+                                    }
+                                    return null;
+                                }))),
                                 message.content.startsWith('✓') &&
                                     message.toolResult &&
-                                    !message.toolCall && (React.createElement(ToolResultPreview, { toolName: message.content
+                                    // 只在没有 diff 数据时显示预览（有 diff 的工具会用 DiffViewer 显示）
+                                    !(message.toolCall &&
+                                        (message.toolCall.arguments?.oldContent ||
+                                            message.toolCall.arguments?.batchResults)) && (React.createElement(ToolResultPreview, { toolName: message.content
                                         .replace('✓ ', '')
                                         .split('\n')[0] || '', result: message.toolResult, maxLines: 5 })),
                                 message.role === 'user' && message.systemInfo && (React.createElement(Box, { marginTop: 1, flexDirection: "column" },
@@ -793,7 +832,7 @@ export default function ChatScreen({ skipWelcome }) {
             !showMcpPanel &&
             !showUsagePanel &&
             !snapshotState.pendingRollback && (React.createElement(React.Fragment, null,
-            React.createElement(ChatInput, { onSubmit: handleMessageSubmit, onCommand: handleCommandExecution, placeholder: "Ask me anything about coding...", disabled: !!pendingToolConfirmation, chatHistory: messages, onHistorySelect: handleHistorySelect, yoloMode: yoloMode, contextUsage: streamingState.contextUsage
+            React.createElement(ChatInput, { onSubmit: handleMessageSubmit, onCommand: handleCommandExecution, placeholder: "Ask me anything about coding...", disabled: !!pendingToolConfirmation, isProcessing: streamingState.isStreaming || isSaving, chatHistory: messages, onHistorySelect: handleHistorySelect, yoloMode: yoloMode, contextUsage: streamingState.contextUsage
                     ? {
                         inputTokens: streamingState.contextUsage.prompt_tokens,
                         maxContextTokens: getOpenAiConfig().maxContextTokens || 4000,

package/dist/utils/chatExporter.d.ts

@@ -0,0 +1,9 @@
+import type { Message } from '../ui/components/MessageList.js';
+/**
+ * Format messages to plain text for export
+ */
+export declare function formatMessagesAsText(messages: Message[]): string;
+/**
+ * Export messages to a file
+ */
+export declare function exportMessagesToFile(messages: Message[], filePath: string): Promise<void>;

package/dist/utils/chatExporter.js

@@ -0,0 +1,126 @@
+import * as fs from 'fs/promises';
+/**
+ * Format messages to plain text for export
+ */
+export function formatMessagesAsText(messages) {
+    const lines = [];
+    const timestamp = new Date().toISOString();
+    // Add header
+    lines.push('=====================================');
+    lines.push('Snow AI - Chat Export');
+    lines.push(`Exported at: ${new Date(timestamp).toLocaleString()}`);
+    lines.push('=====================================');
+    lines.push('');
+    // Format each message
+    for (const message of messages) {
+        // Skip command messages
+        if (message.role === 'command') {
+            continue;
+        }
+        // Add role header
+        let roleLabel = '';
+        if (message.role === 'user') {
+            roleLabel = 'USER';
+        }
+        else if (message.role === 'assistant') {
+            roleLabel = 'ASSISTANT';
+        }
+        else if (message.role === 'subagent') {
+            roleLabel = 'SUBAGENT';
+        }
+        else {
+            roleLabel = 'UNKNOWN';
+        }
+        lines.push(`[${roleLabel}]`);
+        lines.push('-'.repeat(40));
+        // Add content (Message.content is always string based on the type definition)
+        lines.push(message.content);
+        // Add tool call information if present
+        if (message.toolCall) {
+            lines.push('');
+            lines.push(`[TOOL CALL: ${message.toolCall.name}]`);
+            try {
+                const argsStr = typeof message.toolCall.arguments === 'string'
+                    ? message.toolCall.arguments
+                    : JSON.stringify(message.toolCall.arguments, null, 2);
+                lines.push(argsStr);
+            }
+            catch {
+                lines.push(String(message.toolCall.arguments));
+            }
+        }
+        // Add tool display information if present
+        if (message.toolDisplay) {
+            lines.push('');
+            lines.push(`[TOOL: ${message.toolDisplay.toolName}]`);
+            for (const arg of message.toolDisplay.args) {
+                lines.push(`  ${arg.key}: ${arg.value}`);
+            }
+        }
+        // Add tool result if present
+        if (message.toolResult) {
+            lines.push('');
+            lines.push('[TOOL RESULT]');
+            try {
+                const result = JSON.parse(message.toolResult);
+                lines.push(JSON.stringify(result, null, 2));
+            }
+            catch {
+                lines.push(message.toolResult);
+            }
+        }
+        // Add terminal result if present
+        if (message.terminalResult) {
+            lines.push('');
+            if (message.terminalResult.command) {
+                lines.push(`[COMMAND: ${message.terminalResult.command}]`);
+            }
+            if (message.terminalResult.stdout) {
+                lines.push('[STDOUT]');
+                lines.push(message.terminalResult.stdout);
+            }
+            if (message.terminalResult.stderr) {
+                lines.push('[STDERR]');
+                lines.push(message.terminalResult.stderr);
+            }
+            if (message.terminalResult.exitCode !== undefined) {
+                lines.push(`[EXIT CODE: ${message.terminalResult.exitCode}]`);
+            }
+        }
+        // Add images information if present
+        if (message.images && message.images.length > 0) {
+            lines.push('');
+            lines.push(`[${message.images.length} image(s) attached]`);
+        }
+        // Add system info if present
+        if (message.systemInfo) {
+            lines.push('');
+            lines.push('[SYSTEM INFO]');
+            lines.push(`Platform: ${message.systemInfo.platform}`);
+            lines.push(`Shell: ${message.systemInfo.shell}`);
+            lines.push(`Working Directory: ${message.systemInfo.workingDirectory}`);
+        }
+        // Add files information if present
+        if (message.files && message.files.length > 0) {
+            lines.push('');
+            lines.push(`[${message.files.length} file(s) referenced]`);
+            for (const file of message.files) {
+                lines.push(`  - ${file.path}`);
+            }
+        }
+        lines.push('');
+        lines.push('');
+    }
+    // Add footer
+    lines.push('=====================================');
+    lines.push('End of Chat Export');
+    lines.push('=====================================');
+    return lines.join('\n');
+}
+/**
+ * Export messages to a file
+ */
+export async function exportMessagesToFile(messages, filePath) {
+    const textContent = formatMessagesAsText(messages);
+    await fs.writeFile(filePath, textContent, 'utf-8');
+}

package/dist/utils/commandExecutor.d.ts

@@ -1,7 +1,7 @@
 export interface CommandResult {
     success: boolean;
     message?: string;
-    action?: 'clear' | 'resume' | 'info' | 'showMcpInfo' | 'toggleYolo' | 'initProject' | 'compact' | 'showSessionPanel' | 'showMcpPanel' | 'showUsagePanel' | 'home' | 'review';
+    action?: 'clear' | 'resume' | 'info' | 'showMcpInfo' | 'toggleYolo' | 'initProject' | 'compact' | 'showSessionPanel' | 'showMcpPanel' | 'showUsagePanel' | 'home' | 'review' | 'exportChat';
     prompt?: string;
     alreadyConnected?: boolean;
 }

package/dist/utils/commands/export.d.ts

@@ -0,0 +1,2 @@
+declare const _default: {};
+export default _default;

package/dist/utils/commands/export.js

@@ -0,0 +1,12 @@
+import { registerCommand } from '../commandExecutor.js';
+// Export command handler - exports chat conversation to text file
+registerCommand('export', {
+    execute: () => {
+        return {
+            success: true,
+            action: 'exportChat',
+            message: 'Exporting conversation...'
+        };
+    }
+});
+export default {};

package/dist/utils/commands/init.js

@@ -83,11 +83,11 @@ License information (check package.json or LICENSE file)
 - Be thorough but concise - focus on essential information
 - If SNOW.md already exists, read it first and UPDATE it rather than replace
 - Format with proper Markdown syntax
-- After generating content, use filesystem-write or filesystem-create to save SNOW.md in the project root
+- After generating content, use filesystem-create to save SNOW.md in the project root
 - Confirm completion with a brief summary
 
-Begin your analysis now. Use every tool at your disposal to understand this project completely.`
+Begin your analysis now. Use every tool at your disposal to understand this project completely.`,
         };
-    }
+    },
 });
 export default {};

package/dist/utils/fileDialog.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Cross-platform file save dialog
+ * Opens a native file save dialog and returns the selected path
+ */
+export declare function showSaveDialog(defaultFilename?: string, title?: string): Promise<string | null>;
+/**
+ * Check if native file dialogs are available on this platform
+ */
+export declare function isFileDialogSupported(): boolean;

package/dist/utils/fileDialog.js

@@ -0,0 +1,74 @@
+import { exec } from 'child_process';
+import { promisify } from 'util';
+import * as path from 'path';
+import * as os from 'os';
+const execAsync = promisify(exec);
+/**
+ * Cross-platform file save dialog
+ * Opens a native file save dialog and returns the selected path
+ */
+export async function showSaveDialog(defaultFilename = 'export.txt', title = 'Save File') {
+    const platform = os.platform();
+    try {
+        if (platform === 'darwin') {
+            // macOS - use osascript (AppleScript)
+            const defaultPath = path.join(os.homedir(), 'Downloads', defaultFilename);
+            const script = `
+				set defaultPath to POSIX file "${defaultPath}"
+				set saveFile to choose file name with prompt "${title}" default location (POSIX file "${os.homedir()}/Downloads") default name "${defaultFilename}"
+				return POSIX path of saveFile
+			`;
+            const { stdout } = await execAsync(`osascript -e '${script.replace(/'/g, "'\\''")}'`);
+            return stdout.trim();
+        }
+        else if (platform === 'win32') {
+            // Windows - use PowerShell
+            const script = `
+				Add-Type -AssemblyName System.Windows.Forms
+				$dialog = New-Object System.Windows.Forms.SaveFileDialog
+				$dialog.Title = "${title}"
+				$dialog.Filter = "Text files (*.txt)|*.txt|Markdown files (*.md)|*.md|All files (*.*)|*.*"
+				$dialog.FileName = "${defaultFilename}"
+				$dialog.InitialDirectory = "${path.join(os.homedir(), 'Downloads').replace(/\\/g, '\\\\')}"
+				$result = $dialog.ShowDialog()
+				if ($result -eq 'OK') {
+					Write-Output $dialog.FileName
+				}
+			`;
+            const { stdout } = await execAsync(`powershell -NoProfile -Command "${script.replace(/"/g, '\\"')}"`);
+            const result = stdout.trim();
+            return result || null;
+        }
+        else {
+            // Linux - use zenity (most common) or kdialog as fallback
+            try {
+                const defaultPath = path.join(os.homedir(), 'Downloads', defaultFilename);
+                const { stdout } = await execAsync(`zenity --file-selection --save --title="${title}" --filename="${defaultPath}" --confirm-overwrite`);
+                return stdout.trim();
+            }
+            catch (error) {
+                // Try kdialog as fallback for KDE systems
+                try {
+                    const defaultPath = path.join(os.homedir(), 'Downloads', defaultFilename);
+                    const { stdout } = await execAsync(`kdialog --getsavefilename "${defaultPath}" "*.*|All Files" --title "${title}"`);
+                    return stdout.trim();
+                }
+                catch {
+                    // If both fail, return null
+                    return null;
+                }
+            }
+        }
+    }
+    catch (error) {
+        // User cancelled or error occurred
+        return null;
+    }
+}
+/**
+ * Check if native file dialogs are available on this platform
+ */
+export function isFileDialogSupported() {
+    const platform = os.platform();
+    return platform === 'darwin' || platform === 'win32' || platform === 'linux';
+}

package/dist/utils/fileUtils.js

@@ -177,9 +177,9 @@ export function createMessageWithFileInstructions(content, files, systemInfo, ed
         if (editorContext.cursorPosition) {
             editorLines.push(`└─ Cursor: Line ${editorContext.cursorPosition.line + 1}, Column ${editorContext.cursorPosition.character + 1}`);
         }
-        // if (editorContext.selectedText) {
-        // 	editorLines.push(`└─ Selected Code:\n\`\`\`\n${editorContext.selectedText}\n\`\`\``);
-        // }
+        if (editorContext.selectedText) {
+            editorLines.push(`└─ Selected Code:\n\`\`\`\n${editorContext.selectedText}\n\`\`\``);
+        }
         if (editorLines.length > 0) {
             parts.push(editorLines.join('\n'));
         }

package/dist/utils/incrementalSnapshot.d.ts

@@ -75,6 +75,13 @@ declare class IncrementalSnapshotManager {
      * @returns Number of files rolled back
      */
     rollbackToMessageIndex(sessionId: string, targetMessageIndex: number): Promise<number>;
+    /**
+     * Delete all snapshots >= targetMessageIndex
+     * This is used when user rolls back conversation to clean up snapshot files
+     * @param sessionId Session ID
+     * @param targetMessageIndex The message index to delete from (inclusive)
+     */
+    deleteSnapshotsFromIndex(sessionId: string, targetMessageIndex: number): Promise<number>;
     /**
      * Clear all snapshots for a session
      */

package/dist/utils/incrementalSnapshot.js

@@ -255,6 +255,41 @@ class IncrementalSnapshotManager {
             return 0;
         }
     }
+    /**
+     * Delete all snapshots >= targetMessageIndex
+     * This is used when user rolls back conversation to clean up snapshot files
+     * @param sessionId Session ID
+     * @param targetMessageIndex The message index to delete from (inclusive)
+     */
+    async deleteSnapshotsFromIndex(sessionId, targetMessageIndex) {
+        await this.ensureSnapshotsDir();
+        try {
+            const files = await fs.readdir(this.snapshotsDir);
+            let deletedCount = 0;
+            for (const file of files) {
+                if (file.startsWith(sessionId) && file.endsWith('.json')) {
+                    const snapshotPath = path.join(this.snapshotsDir, file);
+                    const content = await fs.readFile(snapshotPath, 'utf-8');
+                    const metadata = JSON.parse(content);
+                    // Delete snapshots with messageIndex >= targetMessageIndex
+                    if (metadata.messageIndex >= targetMessageIndex) {
+                        try {
+                            await fs.unlink(snapshotPath);
+                            deletedCount++;
+                        }
+                        catch (error) {
+                            console.error(`Failed to delete snapshot file ${snapshotPath}:`, error);
+                        }
+                    }
+                }
+            }
+            return deletedCount;
+        }
+        catch (error) {
+            console.error('Failed to delete snapshots from index:', error);
+            return 0;
+        }
+    }
     /**
      * Clear all snapshots for a session
      */