@aiready/ast-mcp-server 0.1.6 → 0.2.0

package/dist/{chunk-KTDNJP4B.js → chunk-FQP7SHLM.js}

@@ -27,15 +27,9 @@ function validateWorkspacePath(inputPath) {
 
 // src/tools/search-code.ts
 var execFileAsync = promisify(execFile);
-async function searchCode(pattern, searchPath, filePattern, limit = 50, regex = true) {
+async function searchCode(pattern, searchPath, filePattern, limit = 50, regex = true, offset = 0) {
   const safePath = validateWorkspacePath(searchPath);
-  const args = [
-    "--json",
-    "--max-count",
-    limit.toString(),
-    "--max-columns",
-    "500"
-  ];
+  const args = ["--json", "--max-columns", "500"];
   if (!regex) {
     args.push("--fixed-strings");
   }
@@ -66,7 +60,7 @@ async function searchCode(pattern, searchPath, filePattern, limit = 50, regex =
         }
       }
     }
-    return results.slice(0, limit);
+    return results.slice(offset, offset + limit);
   } catch (error) {
     if (error.code === 1) {
       return [];
@@ -79,4 +73,4 @@ export {
   validateWorkspacePath,
   searchCode
 };
-//# sourceMappingURL=chunk-KTDNJP4B.js.map
+//# sourceMappingURL=chunk-FQP7SHLM.js.map

package/dist/chunk-FQP7SHLM.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/tools/search-code.ts","../src/security.ts"],"sourcesContent":["import { execFile } from 'child_process';\nimport { promisify } from 'util';\nimport { rgPath } from '@vscode/ripgrep';\nimport { validateWorkspacePath } from '../security.js';\n\nconst execFileAsync = promisify(execFile);\n\nexport interface SearchResult {\n  file: string;\n  line: number;\n  column: number;\n  text: string;\n}\n\nexport async function searchCode(\n  pattern: string,\n  searchPath: string,\n  filePattern?: string,\n  limit: number = 50,\n  regex: boolean = true,\n  offset: number = 0\n): Promise<SearchResult[]> {\n  const safePath = validateWorkspacePath(searchPath);\n\n  const args = ['--json', '--max-columns', '500'];\n\n  if (!regex) {\n    args.push('--fixed-strings');\n  }\n\n  args.push(pattern, safePath);\n\n  if (filePattern) {\n    args.push('--glob', filePattern);\n  }\n\n  // Common exclusions\n  args.push('--glob', '!**/node_modules/**');\n  args.push('--glob', '!**/dist/**');\n  args.push('--glob', '!**/.git/**');\n\n  try {\n    const { stdout } = await execFileAsync(rgPath, args);\n    const lines = stdout.split('\\n').filter(Boolean);\n    const results: SearchResult[] = [];\n\n    for (const line of lines) {\n      const data = JSON.parse(line);\n      if (data.type === 'match') {\n        const file = data.data.path.text;\n        const lineNumber = data.data.line_number;\n        const submatches = data.data.submatches;\n\n        for (const submatch of submatches) {\n          results.push({\n            file,\n            line: lineNumber,\n            column: submatch.start,\n            text: data.data.lines.text.trim(),\n          });\n        }\n      }\n    }\n\n    return results.slice(offset, offset + limit);\n  } catch (error: any) {\n    if (error.code === 1) {\n      // rg returns 1 if no matches found\n      return [];\n    }\n    throw error;\n  }\n}\n","import path from 'path';\nimport fs from 'fs';\n\nexport function resolveWorkspaceRoot(): string {\n  return process.env.AST_WORKSPACE_ROOT || process.cwd();\n}\n\nexport function validateWorkspacePath(inputPath: string): string {\n  const root = resolveWorkspaceRoot();\n  const resolved = path.resolve(root, inputPath);\n  const normalized = path.normalize(resolved);\n\n  // Reject path traversal\n  if (!normalized.startsWith(root)) {\n    throw new Error(\n      `Path traversal detected: ${inputPath} escapes workspace root`\n    );\n  }\n\n  // Reject null bytes\n  if (normalized.includes('\\0')) {\n    throw new Error('Path contains null bytes');\n  }\n\n  return normalized;\n}\n\nexport function validateFileExists(filePath: string): string {\n  const safe = validateWorkspacePath(filePath);\n  if (!fs.existsSync(safe)) {\n    throw new Error(`File not found: ${filePath}`);\n  }\n  if (!fs.statSync(safe).isFile()) {\n    throw new Error(`Not a file: ${filePath}`);\n  }\n  return safe;\n}\n"],"mappings":";;;AAAA,SAAS,gBAAgB;AACzB,SAAS,iBAAiB;AAC1B,SAAS,cAAc;;;ACFvB,OAAO,UAAU;AAGV,SAAS,uBAA+B;AAC7C,SAAO,QAAQ,IAAI,sBAAsB,QAAQ,IAAI;AACvD;AAEO,SAAS,sBAAsB,WAA2B;AAC/D,QAAM,OAAO,qBAAqB;AAClC,QAAM,WAAW,KAAK,QAAQ,MAAM,SAAS;AAC7C,QAAM,aAAa,KAAK,UAAU,QAAQ;AAG1C,MAAI,CAAC,WAAW,WAAW,IAAI,GAAG;AAChC,UAAM,IAAI;AAAA,MACR,4BAA4B,SAAS;AAAA,IACvC;AAAA,EACF;AAGA,MAAI,WAAW,SAAS,IAAI,GAAG;AAC7B,UAAM,IAAI,MAAM,0BAA0B;AAAA,EAC5C;AAEA,SAAO;AACT;;;ADpBA,IAAM,gBAAgB,UAAU,QAAQ;AASxC,eAAsB,WACpB,SACA,YACA,aACA,QAAgB,IAChB,QAAiB,MACjB,SAAiB,GACQ;AACzB,QAAM,WAAW,sBAAsB,UAAU;AAEjD,QAAM,OAAO,CAAC,UAAU,iBAAiB,KAAK;AAE9C,MAAI,CAAC,OAAO;AACV,SAAK,KAAK,iBAAiB;AAAA,EAC7B;AAEA,OAAK,KAAK,SAAS,QAAQ;AAE3B,MAAI,aAAa;AACf,SAAK,KAAK,UAAU,WAAW;AAAA,EACjC;AAGA,OAAK,KAAK,UAAU,qBAAqB;AACzC,OAAK,KAAK,UAAU,aAAa;AACjC,OAAK,KAAK,UAAU,aAAa;AAEjC,MAAI;AACF,UAAM,EAAE,OAAO,IAAI,MAAM,cAAc,QAAQ,IAAI;AACnD,UAAM,QAAQ,OAAO,MAAM,IAAI,EAAE,OAAO,OAAO;AAC/C,UAAM,UAA0B,CAAC;AAEjC,eAAW,QAAQ,OAAO;AACxB,YAAM,OAAO,KAAK,MAAM,IAAI;AAC5B,UAAI,KAAK,SAAS,SAAS;AACzB,cAAM,OAAO,KAAK,KAAK,KAAK;AAC5B,cAAM,aAAa,KAAK,KAAK;AAC7B,cAAM,aAAa,KAAK,KAAK;AAE7B,mBAAW,YAAY,YAAY;AACjC,kBAAQ,KAAK;AAAA,YACX;AAAA,YACA,MAAM;AAAA,YACN,QAAQ,SAAS;AAAA,YACjB,MAAM,KAAK,KAAK,MAAM,KAAK,KAAK;AAAA,UAClC,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF;AAEA,WAAO,QAAQ,MAAM,QAAQ,SAAS,KAAK;AAAA,EAC7C,SAAS,OAAY;AACnB,QAAI,MAAM,SAAS,GAAG;AAEpB,aAAO,CAAC;AAAA,IACV;AACA,UAAM;AAAA,EACR;AACF;","names":[]}

package/dist/index.cjs

@@ -73,15 +73,9 @@ var search_code_exports = {};
 __export(search_code_exports, {
   searchCode: () => searchCode
 });
-async function searchCode(pattern, searchPath, filePattern, limit = 50, regex = true) {
+async function searchCode(pattern, searchPath, filePattern, limit = 50, regex = true, offset = 0) {
   const safePath = validateWorkspacePath(searchPath);
-  const args = [
-    "--json",
-    "--max-count",
-    limit.toString(),
-    "--max-columns",
-    "500"
-  ];
+  const args = ["--json", "--max-columns", "500"];
   if (!regex) {
     args.push("--fixed-strings");
   }
@@ -112,7 +106,7 @@ async function searchCode(pattern, searchPath, filePattern, limit = 50, regex =
         }
       }
     }
-    return results.slice(0, limit);
+    return results.slice(offset, offset + limit);
   } catch (error) {
     if (error.code === 1) {
       return [];
@@ -172,7 +166,8 @@ var SearchCodeSchema = import_zod.z.object({
   pattern: import_zod.z.string().describe("Search pattern (regex by default)"),
   path: import_zod.z.string().describe("Directory to search in"),
   filePattern: import_zod.z.string().optional().describe('Glob filter (e.g., "*.ts")'),
-  limit: import_zod.z.number().optional().default(50).describe("Max matches to return"),
+  limit: import_zod.z.number().optional().default(50).describe("Max matches to return (default 50)"),
+  offset: import_zod.z.number().optional().default(0).describe("Pagination offset (default 0)"),
   regex: import_zod.z.boolean().optional().default(true).describe("Use regex mode (default true)")
 });
 var GetSymbolDocsSchema = import_zod.z.object({
@@ -190,6 +185,7 @@ init_cjs_shims();
 init_cjs_shims();
 var import_ts_morph3 = require("ts-morph");
 var import_fs3 = __toESM(require("fs"), 1);
+var import_path5 = __toESM(require("path"), 1);
 
 // src/project-manager.ts
 init_cjs_shims();
@@ -462,6 +458,9 @@ var SymbolIndex = class {
     const memoryUsage = process.memoryUsage().heapUsed / 1024 / 1024;
     return this.computeStats(cache, duration, Math.round(memoryUsage));
   }
+  isInitialized() {
+    return Object.keys(this.index).length > 0;
+  }
   lookup(name) {
     return this.index[name] || [];
   }
@@ -552,8 +551,23 @@ var TypeScriptAdapter = class {
     const poolSize = parseInt(process.env.AST_WORKER_POOL_SIZE || "2");
     this.pool = new WorkerPool(poolSize);
   }
-  async resolveDefinition(symbolName, path5) {
-    validateWorkspacePath(path5);
+  async ensureIndex(p) {
+    if (!symbolIndex.isInitialized()) {
+      const absolutePath = import_path5.default.resolve(p);
+      let searchPath = absolutePath;
+      if (!import_fs3.default.statSync(absolutePath).isDirectory()) {
+        const tsconfig = await projectManager.findNearestTsConfig(absolutePath);
+        searchPath = tsconfig ? import_path5.default.dirname(tsconfig) : import_path5.default.dirname(absolutePath);
+      }
+      console.error(
+        `[AST] Index not initialized. Building for ${searchPath}...`
+      );
+      await symbolIndex.buildIndex(searchPath);
+    }
+  }
+  async resolveDefinition(symbolName, path6) {
+    validateWorkspacePath(path6);
+    await this.ensureIndex(path6);
     const indexHits = symbolIndex.lookup(symbolName);
     if (indexHits.length > 0) {
       const results = [];
@@ -581,32 +595,33 @@ var TypeScriptAdapter = class {
       }
       return results;
     }
-    if (import_fs3.default.statSync(path5).isDirectory()) {
+    if (import_fs3.default.statSync(path6).isDirectory()) {
       return [];
     }
-    const tsconfig = await projectManager.findNearestTsConfig(path5);
+    const tsconfig = await projectManager.findNearestTsConfig(path6);
     if (!tsconfig) return [];
     try {
       const result = await this.pool.execute(
         "resolve_definition",
         {
           tsconfig,
-          file: path5,
+          file: path6,
           symbol: symbolName
         }
       );
       return result;
     } catch {
       const project = projectManager.ensureProject(tsconfig);
-      const sourceFile = project.addSourceFileAtPathIfExists(path5);
+      const sourceFile = project.addSourceFileAtPathIfExists(path6);
       if (!sourceFile) return [];
       const exported = sourceFile.getExportedDeclarations().get(symbolName);
       if (!exported) return [];
       return exported.map((decl) => this.mapToDefinitionLocation(decl));
     }
   }
-  async findReferences(symbolName, path5, limit = 50, offset = 0) {
-    validateWorkspacePath(path5);
+  async findReferences(symbolName, path6, limit = 50, offset = 0) {
+    validateWorkspacePath(path6);
+    await this.ensureIndex(path6);
     const hits = symbolIndex.lookup(symbolName);
     if (hits.length === 0) return { references: [], total_count: 0 };
     const hit = hits[0];
@@ -623,7 +638,7 @@ var TypeScriptAdapter = class {
       const { searchCode: searchCode2 } = await Promise.resolve().then(() => (init_search_code(), search_code_exports));
       const searchResults = await searchCode2(
         symbolName,
-        path5,
+        path6,
         "*.{ts,tsx,js,jsx}",
         1e3,
         false
@@ -655,8 +670,9 @@ var TypeScriptAdapter = class {
       total_count: unique.length
     };
   }
-  async findImplementations(symbolName, path5, limit = 50, offset = 0) {
-    validateWorkspacePath(path5);
+  async findImplementations(symbolName, path6, limit = 50, offset = 0) {
+    validateWorkspacePath(path6);
+    await this.ensureIndex(path6);
     const hits = symbolIndex.lookup(symbolName);
     if (hits.length === 0) return { implementations: [], total_count: 0 };
     const hit = hits[0];
@@ -687,7 +703,7 @@ var TypeScriptAdapter = class {
         const { searchCode: searchCode2 } = await Promise.resolve().then(() => (init_search_code(), search_code_exports));
         const searchResults = await searchCode2(
           symbolName,
-          path5,
+          path6,
           "*.{ts,tsx,js,jsx}",
           1e3,
           false
@@ -866,22 +882,22 @@ var TypeScriptAdapter = class {
 var typescriptAdapter = new TypeScriptAdapter();
 
 // src/tools/resolve-definition.ts
-async function resolveDefinition(symbol, path5) {
-  return await typescriptAdapter.resolveDefinition(symbol, path5);
+async function resolveDefinition(symbol, path6) {
+  return await typescriptAdapter.resolveDefinition(symbol, path6);
 }
 
 // src/tools/find-references.ts
 init_cjs_shims();
-async function findReferences(symbol, path5, limit = 50, offset = 0) {
-  return await typescriptAdapter.findReferences(symbol, path5, limit, offset);
+async function findReferences(symbol, path6, limit = 50, offset = 0) {
+  return await typescriptAdapter.findReferences(symbol, path6, limit, offset);
 }
 
 // src/tools/find-implementations.ts
 init_cjs_shims();
-async function findImplementations(symbol, path5, limit = 50, offset = 0) {
+async function findImplementations(symbol, path6, limit = 50, offset = 0) {
   return await typescriptAdapter.findImplementations(
     symbol,
-    path5,
+    path6,
     limit,
     offset
   );
@@ -928,8 +944,8 @@ async function getSymbolDocs(symbol, filePath) {
 
 // src/tools/build-symbol-index.ts
 init_cjs_shims();
-async function buildSymbolIndex(path5) {
-  return await symbolIndex.buildIndex(path5);
+async function buildSymbolIndex(path6) {
+  return await symbolIndex.buildIndex(path6);
 }
 
 // src/index.ts
@@ -1024,6 +1040,7 @@ var ASTExplorerServer = class {
                 path: { type: "string", description: "Directory to search" },
                 filePattern: { type: "string", description: "Glob filter" },
                 limit: { type: "number", default: 50 },
+                offset: { type: "number", default: 0 },
                 regex: { type: "boolean", default: true }
               },
               required: ["pattern", "path"]
@@ -1060,8 +1077,8 @@ var ASTExplorerServer = class {
       try {
         switch (name) {
           case "resolve_definition": {
-            const { symbol, path: path5 } = ResolveDefinitionSchema.parse(args);
-            const results = await resolveDefinition(symbol, path5);
+            const { symbol, path: path6 } = ResolveDefinitionSchema.parse(args);
+            const results = await resolveDefinition(symbol, path6);
             return {
               content: [
                 { type: "text", text: JSON.stringify(results, null, 2) }
@@ -1069,8 +1086,8 @@ var ASTExplorerServer = class {
             };
           }
           case "find_references": {
-            const { symbol, path: path5, limit, offset } = FindReferencesSchema.parse(args);
-            const results = await findReferences(symbol, path5, limit, offset);
+            const { symbol, path: path6, limit, offset } = FindReferencesSchema.parse(args);
+            const results = await findReferences(symbol, path6, limit, offset);
             return {
               content: [
                 { type: "text", text: JSON.stringify(results, null, 2) }
@@ -1078,10 +1095,10 @@ var ASTExplorerServer = class {
             };
           }
           case "find_implementations": {
-            const { symbol, path: path5, limit, offset } = FindImplementationsSchema.parse(args);
+            const { symbol, path: path6, limit, offset } = FindImplementationsSchema.parse(args);
             const results = await findImplementations(
               symbol,
-              path5,
+              path6,
               limit,
               offset
             );
@@ -1101,13 +1118,14 @@ var ASTExplorerServer = class {
             };
           }
           case "search_code": {
-            const { pattern, path: path5, filePattern, limit, regex } = SearchCodeSchema.parse(args);
+            const { pattern, path: path6, filePattern, limit, offset, regex } = SearchCodeSchema.parse(args);
             const results = await searchCode(
               pattern,
-              path5,
+              path6,
               filePattern,
               limit,
-              regex
+              regex,
+              offset
             );
             return {
               content: [
@@ -1116,15 +1134,15 @@ var ASTExplorerServer = class {
             };
           }
           case "get_symbol_docs": {
-            const { symbol, path: path5 } = GetSymbolDocsSchema.parse(args);
-            const docs = await getSymbolDocs(symbol, path5);
+            const { symbol, path: path6 } = GetSymbolDocsSchema.parse(args);
+            const docs = await getSymbolDocs(symbol, path6);
             return {
               content: [{ type: "text", text: JSON.stringify(docs, null, 2) }]
             };
           }
           case "build_symbol_index": {
-            const { path: path5 } = BuildSymbolIndexSchema.parse(args);
-            const stats = await buildSymbolIndex(path5);
+            const { path: path6 } = BuildSymbolIndexSchema.parse(args);
+            const stats = await buildSymbolIndex(path6);
             return {
               content: [{ type: "text", text: JSON.stringify(stats, null, 2) }]
             };