@mrxkun/mcfast-mcp 3.3.5 → 3.3.7

package/src/strategies/tree-sitter/languages.js

@@ -1,5 +1,9 @@
 /**
- * Tree-sitter Language Loader
+ * Tree-sitter Language Loader (v4.0+)
+ *
+ * Performance Optimization:
+ * - Cache Parser instances per language to avoid WASM re-initialization overhead
+ * - Reusing parsers significantly reduces latency for repeated operations
  */
 import path from 'path';
 import { createRequire } from 'module';
@@ -12,36 +16,31 @@ const require = createRequire(import.meta.url);
 const _Parser = require('web-tree-sitter');
 
 let Parser = _Parser;
-// If default export exists, use it
 if (Parser.default) {
     Parser = Parser.default;
 }
-// If Parser property exists and Parser itself isn't the constructor
 if (typeof Parser !== 'function' && Parser.Parser) {
     Parser = Parser.Parser;
 }
 
 let isInitialized = false;
 const languageCache = new Map();
+const parserCache = new Map();
 
-// Map internal language IDs to WASM filenames
 const WASM_MAP = {
     'go': 'tree-sitter-go.wasm',
     'rust': 'tree-sitter-rust.wasm',
     'java': 'tree-sitter-java.wasm',
     'javascript': 'tree-sitter-javascript.wasm',
-    'typescript': 'tree-sitter-javascript.wasm', // TS often uses JS or its own, using JS for now as fallback/compatible
+    'typescript': 'tree-sitter-javascript.wasm',
     'python': 'tree-sitter-python.wasm',
     'cpp': 'tree-sitter-cpp.wasm',
-    'c': 'tree-sitter-cpp.wasm', // C typically uses C++ parser or its own. Using CPP for now as it handles C.
+    'c': 'tree-sitter-cpp.wasm',
     'csharp': 'tree-sitter-c-sharp.wasm',
     'php': 'tree-sitter-php.wasm',
     'ruby': 'tree-sitter-ruby.wasm',
 };
 
-/**
- * Initialize web-tree-sitter
- */
 async function init() {
     if (isInitialized) return;
     try {
@@ -52,7 +51,6 @@ async function init() {
         isInitialized = true;
     } catch (e) {
         console.error('Failed to initialize web-tree-sitter:', e);
-        // Fallback: try without explicit path (may work if in same dir)
         try {
             await Parser.init();
             isInitialized = true;
@@ -62,10 +60,6 @@ async function init() {
     }
 }
 
-/**
- * Load language parser
- * @param {string} language - 'go', 'rust', 'java', etc.
- */
 export async function loadLanguage(language) {
     if (!isInitialized) await init();
 
@@ -78,12 +72,7 @@ export async function loadLanguage(language) {
         throw new Error(`Unsupported tree-sitter language: ${language}`);
     }
 
-    // Resolve path to wasm file
-    // Assuming this file is at packages/mcp-client/src/strategies/tree-sitter/languages.js
-    // and wasm files are at packages/mcp-client/src/strategies/tree-sitter/wasm/
     const wasmPath = path.resolve(__dirname, 'wasm', wasmFile);
-
-    // Resolve Language class
     let Language = Parser.Language;
     if (!Language && _Parser.Language) {
         Language = _Parser.Language;
@@ -94,20 +83,50 @@ export async function loadLanguage(language) {
         languageCache.set(language, lang);
         return lang;
     } catch (e) {
-        // Fallback for different environments where __dirname might behave differently
-        // Try relative path if absolute fails, or check common locations
         console.error(`Failed to load language ${language} from ${wasmPath}`, e);
         throw e;
     }
 }
 
+/**
+ * Get cached Parser instance for language (v4.0 optimization)
+ * Reuses Parser instances instead of creating new ones each time
+ */
 export async function getParser(language) {
+    if (!isInitialized) await init();
+
+    if (parserCache.has(language)) {
+        return parserCache.get(language);
+    }
+
     const lang = await loadLanguage(language);
     const parser = new Parser();
     parser.setLanguage(lang);
+    parserCache.set(language, parser);
     return parser;
 }
 
+/**
+ * Clear parser cache (useful for testing or memory management)
+ */
+export function clearParserCache() {
+    for (const parser of parserCache.values()) {
+        parser.delete?.();
+    }
+    parserCache.clear();
+}
+
+/**
+ * Get cache statistics
+ */
+export function getCacheStats() {
+    return {
+        languagesLoaded: languageCache.size,
+        parsersCached: parserCache.size,
+        supportedLanguages: Object.keys(WASM_MAP)
+    };
+}
+
 /**
  * Get compiled query for language
  */