snow-ai 0.2.12 → 0.2.13

package/dist/utils/commands/ide.js

@@ -3,6 +3,7 @@ import { vscodeConnection } from '../vscodeConnection.js';
 // IDE connection command handler
 registerCommand('ide', {
     execute: async () => {
+        // Check if already connected
         if (vscodeConnection.isConnected()) {
             return {
                 success: true,
@@ -10,15 +11,31 @@ registerCommand('ide', {
                 message: 'Already connected to VSCode editor'
             };
         }
+        // Check if server is already running (but not connected yet)
+        if (vscodeConnection.isServerRunning()) {
+            return {
+                success: true,
+                action: 'info',
+                message: `VSCode connection server is already running on port ${vscodeConnection.getPort()}\nWaiting for VSCode extension to connect...`
+            };
+        }
+        // Start the server
         try {
             await vscodeConnection.start();
             return {
                 success: true,
                 action: 'info',
-                message: `VSCode connection server started on port ${vscodeConnection.getPort()}\nPlease connect from the Snow CLI extension in VSCode`
+                message: `VSCode connection server started on port ${vscodeConnection.getPort()}\nWaiting for VSCode extension to connect...`
             };
         }
         catch (error) {
+            // Handle EADDRINUSE error specifically
+            if (error instanceof Error && 'code' in error && error.code === 'EADDRINUSE') {
+                return {
+                    success: false,
+                    message: `Port ${vscodeConnection.getPort()} is already in use. Please restart Snow CLI to reset the connection.`
+                };
+            }
             return {
                 success: false,
                 message: error instanceof Error ? error.message : 'Failed to start IDE connection'

package/dist/utils/mcpToolsManager.d.ts

@@ -1,6 +1,6 @@
 import { TodoService } from '../mcp/todo.js';
 export interface MCPTool {
-    type: "function";
+    type: 'function';
     function: {
         name: string;
         description: string;

package/dist/utils/mcpToolsManager.js

@@ -42,7 +42,7 @@ function isCacheValid() {
     if (!toolsCache)
         return false;
     const now = Date.now();
-    const isExpired = (now - toolsCache.lastUpdate) > CACHE_DURATION;
+    const isExpired = now - toolsCache.lastUpdate > CACHE_DURATION;
     const configChanged = toolsCache.configHash !== generateConfigHash();
     return !isExpired && !configChanged;
 }
@@ -66,44 +66,44 @@ async function refreshToolsCache() {
     const filesystemServiceTools = filesystemTools.map(tool => ({
         name: tool.name.replace('filesystem_', ''),
         description: tool.description,
-        inputSchema: tool.inputSchema
+        inputSchema: tool.inputSchema,
     }));
     servicesInfo.push({
         serviceName: 'filesystem',
         tools: filesystemServiceTools,
         isBuiltIn: true,
-        connected: true
+        connected: true,
     });
     for (const tool of filesystemTools) {
         allTools.push({
-            type: "function",
+            type: 'function',
             function: {
                 name: `filesystem-${tool.name.replace('filesystem_', '')}`,
                 description: tool.description,
-                parameters: tool.inputSchema
-            }
+                parameters: tool.inputSchema,
+            },
         });
     }
     // Add built-in terminal tools (always available)
     const terminalServiceTools = terminalTools.map(tool => ({
         name: tool.name.replace('terminal_', ''),
         description: tool.description,
-        inputSchema: tool.inputSchema
+        inputSchema: tool.inputSchema,
     }));
     servicesInfo.push({
         serviceName: 'terminal',
         tools: terminalServiceTools,
         isBuiltIn: true,
-        connected: true
+        connected: true,
     });
     for (const tool of terminalTools) {
         allTools.push({
-            type: "function",
+            type: 'function',
             function: {
                 name: `terminal-${tool.name.replace('terminal_', '')}`,
                 description: tool.description,
-                parameters: tool.inputSchema
-            }
+                parameters: tool.inputSchema,
+            },
         });
     }
     // Add built-in TODO tools (always available)
@@ -112,22 +112,22 @@ async function refreshToolsCache() {
     const todoServiceTools = todoTools.map(tool => ({
         name: tool.name.replace('todo-', ''),
         description: tool.description || '',
-        inputSchema: tool.inputSchema
+        inputSchema: tool.inputSchema,
     }));
     servicesInfo.push({
         serviceName: 'todo',
         tools: todoServiceTools,
         isBuiltIn: true,
-        connected: true
+        connected: true,
     });
     for (const tool of todoTools) {
         allTools.push({
-            type: "function",
+            type: 'function',
             function: {
                 name: tool.name,
                 description: tool.description || '',
-                parameters: tool.inputSchema
-            }
+                parameters: tool.inputSchema,
+            },
         });
     }
     // Add user-configured MCP server tools (probe for availability but don't maintain connections)
@@ -140,16 +140,16 @@ async function refreshToolsCache() {
                     serviceName,
                     tools: serviceTools,
                     isBuiltIn: false,
-                    connected: true
+                    connected: true,
                 });
                 for (const tool of serviceTools) {
                     allTools.push({
-                        type: "function",
+                        type: 'function',
                         function: {
                             name: `${serviceName}-${tool.name}`,
                             description: tool.description,
-                            parameters: tool.inputSchema
-                        }
+                            parameters: tool.inputSchema,
+                        },
                     });
                 }
             }
@@ -159,7 +159,7 @@ async function refreshToolsCache() {
                     tools: [],
                     isBuiltIn: false,
                     connected: false,
-                    error: error instanceof Error ? error.message : 'Unknown error'
+                    error: error instanceof Error ? error.message : 'Unknown error',
                 });
             }
         }
@@ -172,7 +172,7 @@ async function refreshToolsCache() {
         tools: allTools,
         servicesInfo,
         lastUpdate: Date.now(),
-        configHash: generateConfigHash()
+        configHash: generateConfigHash(),
     };
 }
 /**
@@ -226,7 +226,7 @@ async function connectAndGetTools(serviceName, server, timeoutMs = 10000) {
             name: `snow-cli-${serviceName}`,
             version: '1.0.0',
         }, {
-            capabilities: {}
+            capabilities: {},
         });
         // Create transport based on server configuration
         if (server.url) {
@@ -260,11 +260,11 @@ async function connectAndGetTools(serviceName, server, timeoutMs = 10000) {
                     }
                 }
                 transport = new StreamableHTTPClientTransport(url, {
-                    requestInit: { headers }
+                    requestInit: { headers },
                 });
                 await Promise.race([
                     client.connect(transport),
-                    new Promise((_, reject) => setTimeout(() => reject(new Error('HTTP connection timeout')), timeoutMs))
+                    new Promise((_, reject) => setTimeout(() => reject(new Error('HTTP connection timeout')), timeoutMs)),
                 ]);
             }
             catch (httpError) {
@@ -277,12 +277,12 @@ async function connectAndGetTools(serviceName, server, timeoutMs = 10000) {
                     name: `snow-cli-${serviceName}`,
                     version: '1.0.0',
                 }, {
-                    capabilities: {}
+                    capabilities: {},
                 });
                 transport = new SSEClientTransport(url);
                 await Promise.race([
                     client.connect(transport),
-                    new Promise((_, reject) => setTimeout(() => reject(new Error('SSE connection timeout')), timeoutMs))
+                    new Promise((_, reject) => setTimeout(() => reject(new Error('SSE connection timeout')), timeoutMs)),
                 ]);
             }
         }
@@ -309,13 +309,13 @@ async function connectAndGetTools(serviceName, server, timeoutMs = 10000) {
         // Get tools from the service
         const toolsResult = await Promise.race([
             client.listTools(),
-            new Promise((_, reject) => setTimeout(() => reject(new Error('ListTools timeout')), timeoutMs))
+            new Promise((_, reject) => setTimeout(() => reject(new Error('ListTools timeout')), timeoutMs)),
         ]);
-        return toolsResult.tools?.map(tool => ({
+        return (toolsResult.tools?.map(tool => ({
             name: tool.name,
             description: tool.description || '',
-            inputSchema: tool.inputSchema
-        })) || [];
+            inputSchema: tool.inputSchema,
+        })) || []);
     }
     finally {
         try {
@@ -391,8 +391,15 @@ export async function executeMCPTool(toolName, args) {
                 return await filesystemService.getFileInfo(args.filePath);
             case 'edit':
                 return await filesystemService.editFile(args.filePath, args.startLine, args.endLine, args.newContent, args.contextLines);
-            case 'search':
-                return await filesystemService.searchCode(args.query, args.dirPath, args.fileExtensions, args.caseSensitive, args.maxResults);
+            case 'search': {
+                // 兼容性处理：如果 searchMode 不存在或无效，默认使用 'text'
+                const validSearchModes = ['text', 'regex', 'ast'];
+                let searchMode = 'text';
+                if (args.searchMode && validSearchModes.includes(args.searchMode)) {
+                    searchMode = args.searchMode;
+                }
+                return await filesystemService.searchCode(args.query, args.dirPath, args.fileExtensions, args.caseSensitive, args.maxResults, searchMode);
+            }
             default:
                 throw new Error(`Unknown filesystem tool: ${actualToolName}`);
         }
@@ -430,7 +437,7 @@ async function executeOnExternalMCPService(serviceName, server, toolName, args)
             name: `snow-cli-${serviceName}`,
             version: '1.0.0',
         }, {
-            capabilities: {}
+            capabilities: {},
         });
         // Setup transport (similar to getServiceTools)
         let transport;
@@ -450,7 +457,9 @@ async function executeOnExternalMCPService(serviceName, server, toolName, args)
             transport = new StdioClientTransport({
                 command: server.command,
                 args: server.args || [],
-                env: server.env ? { ...process.env, ...server.env } : process.env,
+                env: server.env
+                    ? { ...process.env, ...server.env }
+                    : process.env,
             });
         }
         await client.connect(transport);
@@ -458,7 +467,7 @@ async function executeOnExternalMCPService(serviceName, server, toolName, args)
         // Execute the tool with the original tool name (not prefixed)
         const result = await client.callTool({
             name: toolName,
-            arguments: args
+            arguments: args,
         });
         logger.debug(`result from ${serviceName} tool ${toolName}:`, result);
         return result.content;

package/dist/utils/textBuffer.d.ts

@@ -28,11 +28,16 @@ export declare class TextBuffer {
     private pasteTimer;
     private pastePlaceholderPosition;
     private onUpdateCallback?;
+    private isDestroyed;
     private visualLines;
     private visualLineStarts;
     private visualCursorPos;
     private preferredVisualCol;
     constructor(viewport: Viewport, onUpdate?: () => void);
+    /**
+     * Cleanup method to be called when the buffer is no longer needed
+     */
+    destroy(): void;
     get text(): string;
     /**
      * 获取完整文本，包括替换占位符为原始内容

package/dist/utils/textBuffer.js

@@ -79,6 +79,12 @@ export class TextBuffer {
             writable: true,
             value: void 0
         }); // 更新回调函数
+        Object.defineProperty(this, "isDestroyed", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        }); // 标记是否已销毁
         Object.defineProperty(this, "visualLines", {
             enumerable: true,
             configurable: true,
@@ -107,6 +113,19 @@ export class TextBuffer {
         this.onUpdateCallback = onUpdate;
         this.recalculateVisualState();
     }
+    /**
+     * Cleanup method to be called when the buffer is no longer needed
+     */
+    destroy() {
+        this.isDestroyed = true;
+        if (this.pasteTimer) {
+            clearTimeout(this.pasteTimer);
+            this.pasteTimer = null;
+        }
+        this.pasteStorage.clear();
+        this.imageStorage.clear();
+        this.onUpdateCallback = undefined;
+    }
     get text() {
         return this.content;
     }
@@ -136,7 +155,7 @@ export class TextBuffer {
     }
     scheduleUpdate() {
         // Notify external components of updates
-        if (this.onUpdateCallback) {
+        if (!this.isDestroyed && this.onUpdateCallback) {
             this.onUpdateCallback();
         }
     }
@@ -191,7 +210,9 @@ export class TextBuffer {
             this.cursorIndex = this.pastePlaceholderPosition + cpLen(tempPlaceholder);
             // 设置150ms的定时器，如果150ms内没有新数据，则认为粘贴完成
             this.pasteTimer = setTimeout(() => {
-                this.finalizePaste();
+                if (!this.isDestroyed) {
+                    this.finalizePaste();
+                }
             }, 150);
             this.recalculateVisualState();
             this.scheduleUpdate();

package/dist/utils/vscodeConnection.js

@@ -41,6 +41,10 @@ class VSCodeConnectionManager {
             try {
                 this.server = new WebSocketServer({ port: this.port });
                 this.server.on('connection', (ws) => {
+                    // Close old client if exists
+                    if (this.client && this.client !== ws) {
+                        this.client.close();
+                    }
                     this.client = ws;
                     ws.on('message', (message) => {
                         try {
@@ -52,7 +56,12 @@ class VSCodeConnectionManager {
                         }
                     });
                     ws.on('close', () => {
-                        this.client = null;
+                        if (this.client === ws) {
+                            this.client = null;
+                        }
+                    });
+                    ws.on('error', () => {
+                        // Silently handle errors
                     });
                 });
                 this.server.on('listening', () => {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "snow-ai",
-	"version": "0.2.12",
+	"version": "0.2.13",
 	"description": "Intelligent Command Line Assistant powered by AI",
 	"license": "MIT",
 	"bin": {
@@ -59,6 +59,18 @@
 		"react": "^18.2.0",
 		"string-width": "^7.2.0",
 		"tiktoken": "^1.0.22",
+		"tree-sitter": "^0.25.0",
+		"tree-sitter-c": "^0.24.1",
+		"tree-sitter-c-sharp": "^0.23.1",
+		"tree-sitter-cpp": "^0.23.4",
+		"tree-sitter-go": "^0.25.0",
+		"tree-sitter-java": "^0.23.5",
+		"tree-sitter-javascript": "^0.25.0",
+		"tree-sitter-php": "^0.24.2",
+		"tree-sitter-python": "^0.25.0",
+		"tree-sitter-ruby": "^0.23.1",
+		"tree-sitter-rust": "^0.24.0",
+		"tree-sitter-typescript": "^0.23.2",
 		"ws": "^8.14.2"
 	},
 	"devDependencies": {

package/readme.md

@@ -1,6 +1,16 @@
+<div align="center">
+
+<img src="logo.png" alt="Snow AI CLI Logo" width="200"/>
+
 # snow-ai
 
-English | [中文](readme_zh.md)
+**English** | [中文](readme_zh.md)
+
+*An intelligent AI-powered CLI tool for developers*
+
+</div>
+
+---
 
 
 ## Install
@@ -66,7 +76,7 @@ $ npm uninstall --global snow-ai
 * **Commands**
 
 ![alt text](image-2.png)
-  - /clear —— Create a new session
+  - /clear - Create a new session
 
   - /resume - The recovery history has