@aiready/ast-mcp-server 0.1.2 → 0.1.3

package/dist/index.cjs

@@ -31,9 +31,12 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // ../../node_modules/.pnpm/tsup@8.5.1_jiti@2.6.1_postcss@8.5.8_tsx@4.21.0_typescript@5.9.3_yaml@2.8.2/node_modules/tsup/assets/cjs_shims.js
+var getImportMetaUrl, importMetaUrl;
 var init_cjs_shims = __esm({
   "../../node_modules/.pnpm/tsup@8.5.1_jiti@2.6.1_postcss@8.5.8_tsx@4.21.0_typescript@5.9.3_yaml@2.8.2/node_modules/tsup/assets/cjs_shims.js"() {
     "use strict";
+    getImportMetaUrl = () => typeof document === "undefined" ? new URL(`file:${__filename}`).href : document.currentScript && document.currentScript.tagName.toUpperCase() === "SCRIPT" ? document.currentScript.src : new URL("main.js", document.baseURI).href;
+    importMetaUrl = /* @__PURE__ */ getImportMetaUrl();
   }
 });
 
@@ -469,21 +472,103 @@ var symbolIndex = new SymbolIndex();
 
 // src/adapters/typescript-adapter.ts
 init_security();
+
+// src/worker/pool.ts
+init_cjs_shims();
+var import_worker_threads = require("worker_threads");
+var import_path4 = __toESM(require("path"), 1);
+var import_url = require("url");
+var WorkerPool = class {
+  constructor(poolSize) {
+    this.poolSize = poolSize;
+  }
+  workers = [];
+  available = [];
+  queue = [];
+  activeJobs = /* @__PURE__ */ new Map();
+  taskId = 0;
+  async init() {
+    const __dirname = import_path4.default.dirname((0, import_url.fileURLToPath)(importMetaUrl));
+    const workerPath = import_path4.default.join(__dirname, "ast-worker.js");
+    for (let i = 0; i < this.poolSize; i++) {
+      const worker = new import_worker_threads.Worker(workerPath);
+      worker.on("message", (msg) => this.handleResult(msg));
+      worker.on("error", (err) => this.handleWorkerError(worker, err));
+      this.workers.push(worker);
+      this.available.push(worker);
+    }
+  }
+  async execute(type, payload) {
+    return new Promise((resolve, reject) => {
+      const id = String(++this.taskId);
+      const task = { id, type, payload, resolve, reject };
+      const worker = this.available.pop();
+      if (worker) {
+        this.dispatch(worker, task);
+      } else {
+        this.queue.push(task);
+      }
+    });
+  }
+  dispatch(worker, task) {
+    this.activeJobs.set(task.id, task);
+    worker.postMessage({ id: task.id, type: task.type, payload: task.payload });
+  }
+  handleResult(msg) {
+    const task = this.activeJobs.get(msg.id);
+    if (!task) return;
+    this.activeJobs.delete(msg.id);
+    if (msg.error) {
+      task.reject(new Error(msg.error));
+    } else {
+      task.resolve(msg.result);
+    }
+    const worker = this.workers.find(
+      (w) => !this.available.includes(w) && ![...this.activeJobs.values()].some((t) => t.id === msg.id)
+      // simplistic
+    );
+  }
+  handleWorkerError(worker, err) {
+    const idx = this.workers.indexOf(worker);
+    if (idx !== -1) {
+      worker.terminate();
+      const __dirname = import_path4.default.dirname((0, import_url.fileURLToPath)(importMetaUrl));
+      const workerPath = import_path4.default.join(__dirname, "ast-worker.js");
+      const newWorker = new import_worker_threads.Worker(workerPath);
+      newWorker.on("message", (msg) => this.handleResult(msg));
+      newWorker.on("error", (e) => this.handleWorkerError(newWorker, e));
+      this.workers[idx] = newWorker;
+      this.available.push(newWorker);
+    }
+  }
+  async terminate() {
+    await Promise.all(this.workers.map((w) => w.terminate()));
+    this.workers = [];
+    this.available = [];
+  }
+};
+
+// src/adapters/typescript-adapter.ts
 var TypeScriptAdapter = class {
-  async resolveDefinition(symbolName, path4) {
-    validateWorkspacePath(path4);
+  pool;
+  constructor() {
+    const poolSize = parseInt(process.env.AST_WORKER_POOL_SIZE || "2");
+    this.pool = new WorkerPool(poolSize);
+  }
+  async resolveDefinition(symbolName, path5) {
+    validateWorkspacePath(path5);
     const indexHits = symbolIndex.lookup(symbolName);
     if (indexHits.length > 0) {
       const results = [];
       for (const hit of indexHits) {
         const tsconfig2 = await projectManager.findNearestTsConfig(hit.file);
         if (tsconfig2) {
-          const project2 = projectManager.ensureProject(tsconfig2);
-          const sourceFile2 = project2.addSourceFileAtPathIfExists(hit.file);
-          if (sourceFile2) {
-            const exported2 = sourceFile2.getExportedDeclarations().get(symbolName);
-            if (exported2 && exported2.length > 0) {
-              results.push(this.mapToDefinitionLocation(exported2[0]));
+          const project = projectManager.ensureProject(tsconfig2);
+          const sourceFile = project.addSourceFileAtPathIfExists(hit.file);
+          if (sourceFile) {
+            const exported = sourceFile.getExportedDeclarations().get(symbolName);
+            if (exported && exported.length > 0) {
+              results.push(this.mapToDefinitionLocation(exported[0]));
               continue;
             }
           }
@@ -499,20 +584,32 @@ var TypeScriptAdapter = class {
       }
       return results;
     }
-    if (import_fs3.default.statSync(path4).isDirectory()) {
+    if (import_fs3.default.statSync(path5).isDirectory()) {
       return [];
     }
-    const tsconfig = await projectManager.findNearestTsConfig(path4);
+    const tsconfig = await projectManager.findNearestTsConfig(path5);
     if (!tsconfig) return [];
-    const project = projectManager.ensureProject(tsconfig);
-    const sourceFile = project.addSourceFileAtPathIfExists(path4);
-    if (!sourceFile) return [];
-    const exported = sourceFile.getExportedDeclarations().get(symbolName);
-    if (!exported) return [];
-    return exported.map((decl) => this.mapToDefinitionLocation(decl));
+    try {
+      const result = await this.pool.execute(
+        "resolve_definition",
+        {
+          tsconfig,
+          file: path5,
+          symbol: symbolName
+        }
+      );
+      return result;
+    } catch {
+      const project = projectManager.ensureProject(tsconfig);
+      const sourceFile = project.addSourceFileAtPathIfExists(path5);
+      if (!sourceFile) return [];
+      const exported = sourceFile.getExportedDeclarations().get(symbolName);
+      if (!exported) return [];
+      return exported.map((decl) => this.mapToDefinitionLocation(decl));
+    }
   }
-  async findReferences(symbolName, path4, limit = 50, offset = 0) {
-    validateWorkspacePath(path4);
+  async findReferences(symbolName, path5, limit = 50, offset = 0) {
+    validateWorkspacePath(path5);
     const hits = symbolIndex.lookup(symbolName);
     if (hits.length === 0) return { references: [], total_count: 0 };
     const hit = hits[0];
@@ -529,13 +626,12 @@ var TypeScriptAdapter = class {
       const { searchCode: searchCode2 } = await Promise.resolve().then(() => (init_search_code(), search_code_exports));
       const searchResults = await searchCode2(
         symbolName,
-        path4,
+        path5,
         "*.{ts,tsx,js,jsx}",
         1e3,
         false
       );
       const filesToLoad = [...new Set(searchResults.map((r) => r.file))];
-      console.log("Search code files to load:", filesToLoad);
       for (const file of filesToLoad) {
         project.addSourceFileAtPathIfExists(file);
       }
@@ -562,57 +658,69 @@ var TypeScriptAdapter = class {
       total_count: unique.length
     };
   }
-  async findImplementations(symbolName, path4, limit = 50, offset = 0) {
-    validateWorkspacePath(path4);
+  async findImplementations(symbolName, path5, limit = 50, offset = 0) {
+    validateWorkspacePath(path5);
     const hits = symbolIndex.lookup(symbolName);
     if (hits.length === 0) return { implementations: [], total_count: 0 };
     const hit = hits[0];
     const tsconfig = await projectManager.findNearestTsConfig(hit.file);
     if (!tsconfig) return { implementations: [], total_count: 0 };
-    const project = projectManager.ensureProject(tsconfig);
-    const sourceFile = project.addSourceFileAtPathIfExists(hit.file);
-    if (!sourceFile) return { implementations: [], total_count: 0 };
-    const exported = sourceFile.getExportedDeclarations().get(symbolName);
-    if (!exported || exported.length === 0)
-      return { implementations: [], total_count: 0 };
-    const targetNode = exported[0];
-    if (!import_ts_morph3.Node.isClassDeclaration(targetNode) && !import_ts_morph3.Node.isInterfaceDeclaration(targetNode)) {
-      return { implementations: [], total_count: 0 };
-    }
     try {
-      const { searchCode: searchCode2 } = await Promise.resolve().then(() => (init_search_code(), search_code_exports));
-      const searchResults = await searchCode2(
-        symbolName,
-        path4,
-        "*.{ts,tsx,js,jsx}",
-        1e3,
-        false
-      );
-      const filesToLoad = [...new Set(searchResults.map((r) => r.file))];
-      for (const file of filesToLoad) {
-        project.addSourceFileAtPathIfExists(file);
+      const result = await this.pool.execute("find_implementations", {
+        tsconfig,
+        file: hit.file,
+        symbol: symbolName
+      });
+      return {
+        implementations: result.implementations.slice(offset, offset + limit),
+        total_count: result.total_count
+      };
+    } catch {
+      const project = projectManager.ensureProject(tsconfig);
+      const sourceFile = project.addSourceFileAtPathIfExists(hit.file);
+      if (!sourceFile) return { implementations: [], total_count: 0 };
+      const exported = sourceFile.getExportedDeclarations().get(symbolName);
+      if (!exported || exported.length === 0)
+        return { implementations: [], total_count: 0 };
+      const targetNode = exported[0];
+      if (!import_ts_morph3.Node.isClassDeclaration(targetNode) && !import_ts_morph3.Node.isInterfaceDeclaration(targetNode)) {
+        return { implementations: [], total_count: 0 };
       }
-    } catch (_e) {
-    }
-    const results = [];
-    const implementations = targetNode.getImplementations?.();
-    if (implementations) {
-      for (const impl of implementations) {
-        const sf = impl.getSourceFile();
-        const lc = sf.getLineAndColumnAtPos(impl.getTextSpan().getStart());
-        results.push({
-          file: sf.getFilePath(),
-          line: lc.line,
-          column: lc.column,
-          text: impl.getNode().getParent()?.getText() || impl.getNode().getText()
-        });
+      try {
+        const { searchCode: searchCode2 } = await Promise.resolve().then(() => (init_search_code(), search_code_exports));
+        const searchResults = await searchCode2(
+          symbolName,
+          path5,
+          "*.{ts,tsx,js,jsx}",
+          1e3,
+          false
+        );
+        const filesToLoad = [...new Set(searchResults.map((r) => r.file))];
+        for (const file of filesToLoad) {
+          project.addSourceFileAtPathIfExists(file);
+        }
+      } catch (_e) {
+      }
+      const results = [];
+      const implementations = targetNode.getImplementations?.();
+      if (implementations) {
+        for (const impl of implementations) {
+          const sf = impl.getSourceFile();
+          const lc = sf.getLineAndColumnAtPos(impl.getTextSpan().getStart());
+          results.push({
+            file: sf.getFilePath(),
+            line: lc.line,
+            column: lc.column,
+            text: impl.getNode().getParent()?.getText() || impl.getNode().getText()
+          });
+        }
       }
+      const unique = this.deduplicateLocations(results);
+      return {
+        implementations: unique.slice(offset, offset + limit),
+        total_count: unique.length
+      };
     }
-    const unique = this.deduplicateLocations(results);
-    return {
-      implementations: unique.slice(offset, offset + limit),
-      total_count: unique.length
-    };
   }
   async getFileStructure(filePath) {
     const safePath = validateWorkspacePath(filePath);
@@ -639,6 +747,9 @@ var TypeScriptAdapter = class {
     };
     return structure;
   }
+  async shutdown() {
+    await this.pool.terminate();
+  }
   mapToDefinitionLocation(node) {
     const sourceFile = node.getSourceFile();
     const lineAndColumn = sourceFile.getLineAndColumnAtPos(node.getStart());
@@ -754,22 +865,22 @@ var TypeScriptAdapter = class {
 var typescriptAdapter = new TypeScriptAdapter();
 
 // src/tools/resolve-definition.ts
-async function resolveDefinition(symbol, path4) {
-  return await typescriptAdapter.resolveDefinition(symbol, path4);
+async function resolveDefinition(symbol, path5) {
+  return await typescriptAdapter.resolveDefinition(symbol, path5);
 }
 
 // src/tools/find-references.ts
 init_cjs_shims();
-async function findReferences(symbol, path4, limit = 50, offset = 0) {
-  return await typescriptAdapter.findReferences(symbol, path4, limit, offset);
+async function findReferences(symbol, path5, limit = 50, offset = 0) {
+  return await typescriptAdapter.findReferences(symbol, path5, limit, offset);
 }
 
 // src/tools/find-implementations.ts
 init_cjs_shims();
-async function findImplementations(symbol, path4, limit = 50, offset = 0) {
+async function findImplementations(symbol, path5, limit = 50, offset = 0) {
   return await typescriptAdapter.findImplementations(
     symbol,
-    path4,
+    path5,
     limit,
     offset
   );
@@ -816,8 +927,8 @@ async function getSymbolDocs(symbol, filePath) {
 
 // src/tools/build-symbol-index.ts
 init_cjs_shims();
-async function buildSymbolIndex(path4) {
-  return await symbolIndex.buildIndex(path4);
+async function buildSymbolIndex(path5) {
+  return await symbolIndex.buildIndex(path5);
 }
 
 // src/index.ts
@@ -948,8 +1059,8 @@ var ASTExplorerServer = class {
       try {
         switch (name) {
           case "resolve_definition": {
-            const { symbol, path: path4 } = ResolveDefinitionSchema.parse(args);
-            const results = await resolveDefinition(symbol, path4);
+            const { symbol, path: path5 } = ResolveDefinitionSchema.parse(args);
+            const results = await resolveDefinition(symbol, path5);
             return {
               content: [
                 { type: "text", text: JSON.stringify(results, null, 2) }
@@ -957,8 +1068,8 @@ var ASTExplorerServer = class {
             };
           }
           case "find_references": {
-            const { symbol, path: path4, limit, offset } = FindReferencesSchema.parse(args);
-            const results = await findReferences(symbol, path4, limit, offset);
+            const { symbol, path: path5, limit, offset } = FindReferencesSchema.parse(args);
+            const results = await findReferences(symbol, path5, limit, offset);
             return {
               content: [
                 { type: "text", text: JSON.stringify(results, null, 2) }
@@ -966,10 +1077,10 @@ var ASTExplorerServer = class {
             };
           }
           case "find_implementations": {
-            const { symbol, path: path4, limit, offset } = FindImplementationsSchema.parse(args);
+            const { symbol, path: path5, limit, offset } = FindImplementationsSchema.parse(args);
             const results = await findImplementations(
               symbol,
-              path4,
+              path5,
               limit,
               offset
             );
@@ -989,10 +1100,10 @@ var ASTExplorerServer = class {
             };
           }
           case "search_code": {
-            const { pattern, path: path4, filePattern, limit, regex } = SearchCodeSchema.parse(args);
+            const { pattern, path: path5, filePattern, limit, regex } = SearchCodeSchema.parse(args);
             const results = await searchCode(
               pattern,
-              path4,
+              path5,
               filePattern,
               limit,
               regex
@@ -1004,15 +1115,15 @@ var ASTExplorerServer = class {
             };
           }
           case "get_symbol_docs": {
-            const { symbol, path: path4 } = GetSymbolDocsSchema.parse(args);
-            const docs = await getSymbolDocs(symbol, path4);
+            const { symbol, path: path5 } = GetSymbolDocsSchema.parse(args);
+            const docs = await getSymbolDocs(symbol, path5);
             return {
               content: [{ type: "text", text: JSON.stringify(docs, null, 2) }]
             };
           }
           case "build_symbol_index": {
-            const { path: path4 } = BuildSymbolIndexSchema.parse(args);
-            const stats = await buildSymbolIndex(path4);
+            const { path: path5 } = BuildSymbolIndexSchema.parse(args);
+            const stats = await buildSymbolIndex(path5);
             return {
               content: [{ type: "text", text: JSON.stringify(stats, null, 2) }]
             };