@aiready/ast-mcp-server 0.1.2 → 0.1.3

package/dist/worker/ast-worker.js

@@ -0,0 +1,156 @@
+// src/worker/ast-worker.ts
+import { parentPort } from "worker_threads";
+import { Project } from "ts-morph";
+var projects = /* @__PURE__ */ new Map();
+function getProject(tsconfigPath) {
+  if (!projects.has(tsconfigPath)) {
+    projects.set(
+      tsconfigPath,
+      new Project({
+        tsConfigFilePath: tsconfigPath,
+        skipAddingFilesFromTsConfig: true
+      })
+    );
+  }
+  return projects.get(tsconfigPath);
+}
+parentPort?.on("message", async (msg) => {
+  try {
+    let result;
+    switch (msg.type) {
+      case "resolve_definition": {
+        const { tsconfig, file, symbol } = msg.payload;
+        const project = getProject(tsconfig);
+        const sf = project.addSourceFileAtPath(file);
+        if (!sf) {
+          result = [];
+          break;
+        }
+        const exported = sf.getExportedDeclarations().get(symbol);
+        if (!exported || exported.length === 0) {
+          result = [];
+          break;
+        }
+        result = exported.map((decl) => {
+          const sourceFile = decl.getSourceFile();
+          const lineAndColumn = sourceFile.getLineAndColumnAtPos(
+            decl.getStart()
+          );
+          return {
+            file: sourceFile.getFilePath(),
+            line: lineAndColumn.line,
+            column: lineAndColumn.column,
+            kind: decl.getKindName(),
+            snippet: decl.getText(),
+            documentation: getJsDoc(decl)
+          };
+        });
+        break;
+      }
+      case "find_references": {
+        const { tsconfig, file, symbol } = msg.payload;
+        const project = getProject(tsconfig);
+        const sf = project.addSourceFileAtPath(file);
+        if (!sf) {
+          result = { references: [], total_count: 0 };
+          break;
+        }
+        const exported = sf.getExportedDeclarations().get(symbol);
+        if (!exported || exported.length === 0) {
+          result = { references: [], total_count: 0 };
+          break;
+        }
+        const targetNode = exported[0];
+        const refSymbols = targetNode.findReferences?.();
+        if (!refSymbols) {
+          result = { references: [], total_count: 0 };
+          break;
+        }
+        const results = [];
+        for (const refSymbol of refSymbols) {
+          for (const ref of refSymbol.getReferences()) {
+            const refSf = ref.getSourceFile();
+            const lc = refSf.getLineAndColumnAtPos(
+              ref.getTextSpan().getStart()
+            );
+            results.push({
+              file: refSf.getFilePath(),
+              line: lc.line,
+              column: lc.column,
+              text: ref.getNode().getParent()?.getText() || ref.getNode().getText()
+            });
+          }
+        }
+        const unique = deduplicateLocations(results);
+        result = {
+          references: unique.slice(0, 50),
+          total_count: unique.length
+        };
+        break;
+      }
+      case "find_implementations": {
+        const { tsconfig, file, symbol } = msg.payload;
+        const project = getProject(tsconfig);
+        const sf = project.addSourceFileAtPath(file);
+        if (!sf) {
+          result = { implementations: [], total_count: 0 };
+          break;
+        }
+        const exported = sf.getExportedDeclarations().get(symbol);
+        if (!exported || exported.length === 0) {
+          result = { implementations: [], total_count: 0 };
+          break;
+        }
+        const targetNode = exported[0];
+        const implementations = targetNode.getImplementations?.();
+        if (!implementations) {
+          result = { implementations: [], total_count: 0 };
+          break;
+        }
+        const results = [];
+        for (const impl of implementations) {
+          const implSf = impl.getSourceFile();
+          const lc = implSf.getLineAndColumnAtPos(
+            impl.getTextSpan().getStart()
+          );
+          results.push({
+            file: implSf.getFilePath(),
+            line: lc.line,
+            column: lc.column,
+            text: impl.getNode().getParent()?.getText() || impl.getNode().getText()
+          });
+        }
+        const unique = deduplicateLocations(results);
+        result = {
+          implementations: unique.slice(0, 50),
+          total_count: unique.length
+        };
+        break;
+      }
+      default:
+        throw new Error(`Unknown task type: ${msg.type}`);
+    }
+    parentPort?.postMessage({ id: msg.id, result });
+  } catch (error) {
+    parentPort?.postMessage({ id: msg.id, error: error.message });
+  }
+});
+function getJsDoc(node) {
+  if (node.getJsDocs) {
+    const docs = node.getJsDocs();
+    if (docs.length > 0) {
+      return docs[0].getCommentText();
+    }
+  }
+  return void 0;
+}
+function deduplicateLocations(locations) {
+  const seen = /* @__PURE__ */ new Set();
+  return locations.filter((loc) => {
+    const key = `${loc.file}:${loc.line}:${loc.column}`;
+    if (seen.has(key)) return false;
+    seen.add(key);
+    return true;
+  });
+}
+//# sourceMappingURL=ast-worker.js.map

package/dist/worker/ast-worker.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/worker/ast-worker.ts"],"sourcesContent":["import { parentPort } from 'worker_threads';\nimport { Project } from 'ts-morph';\nimport { validateWorkspacePath } from '../security.js';\n\nconst projects = new Map<string, Project>();\n\nfunction getProject(tsconfigPath: string): Project {\n  if (!projects.has(tsconfigPath)) {\n    projects.set(\n      tsconfigPath,\n      new Project({\n        tsConfigFilePath: tsconfigPath,\n        skipAddingFilesFromTsConfig: true,\n      })\n    );\n  }\n  return projects.get(tsconfigPath)!;\n}\n\ninterface WorkerMessage {\n  id: string;\n  type: string;\n  payload: any;\n}\n\ninterface DefinitionLocation {\n  file: string;\n  line: number;\n  column: number;\n  kind: string;\n  snippet: string;\n  documentation?: string;\n}\n\ninterface ReferenceLocation {\n  file: string;\n  line: number;\n  column: number;\n  text: string;\n}\n\nparentPort?.on('message', async (msg: WorkerMessage) => {\n  try {\n    let result: unknown;\n\n    switch (msg.type) {\n      case 'resolve_definition': {\n        const { tsconfig, file, symbol } = msg.payload;\n        const project = getProject(tsconfig);\n        const sf = project.addSourceFileAtPath(file);\n        if (!sf) {\n          result = [];\n          break;\n        }\n\n        const exported = sf.getExportedDeclarations().get(symbol);\n        if (!exported || exported.length === 0) {\n          result = [];\n          break;\n        }\n\n        result = exported.map((decl) => {\n          const sourceFile = decl.getSourceFile();\n          const lineAndColumn = sourceFile.getLineAndColumnAtPos(\n            decl.getStart()\n          );\n          return {\n            file: sourceFile.getFilePath(),\n            line: lineAndColumn.line,\n            column: lineAndColumn.column,\n            kind: decl.getKindName(),\n            snippet: decl.getText(),\n            documentation: getJsDoc(decl),\n          };\n        });\n        break;\n      }\n\n      case 'find_references': {\n        const { tsconfig, file, symbol } = msg.payload;\n        const project = getProject(tsconfig);\n        const sf = project.addSourceFileAtPath(file);\n        if (!sf) {\n          result = { references: [], total_count: 0 };\n          break;\n        }\n\n        const exported = sf.getExportedDeclarations().get(symbol);\n        if (!exported || exported.length === 0) {\n          result = { references: [], total_count: 0 };\n          break;\n        }\n\n        const targetNode = exported[0];\n        const refSymbols = (targetNode as any).findReferences?.();\n        if (!refSymbols) {\n          result = { references: [], total_count: 0 };\n          break;\n        }\n\n        const results: ReferenceLocation[] = [];\n        for (const refSymbol of refSymbols) {\n          for (const ref of refSymbol.getReferences()) {\n            const refSf = ref.getSourceFile();\n            const lc = refSf.getLineAndColumnAtPos(\n              ref.getTextSpan().getStart()\n            );\n            results.push({\n              file: refSf.getFilePath(),\n              line: lc.line,\n              column: lc.column,\n              text:\n                ref.getNode().getParent()?.getText() || ref.getNode().getText(),\n            });\n          }\n        }\n\n        const unique = deduplicateLocations(results);\n        result = {\n          references: unique.slice(0, 50),\n          total_count: unique.length,\n        };\n        break;\n      }\n\n      case 'find_implementations': {\n        const { tsconfig, file, symbol } = msg.payload;\n        const project = getProject(tsconfig);\n        const sf = project.addSourceFileAtPath(file);\n        if (!sf) {\n          result = { implementations: [], total_count: 0 };\n          break;\n        }\n\n        const exported = sf.getExportedDeclarations().get(symbol);\n        if (!exported || exported.length === 0) {\n          result = { implementations: [], total_count: 0 };\n          break;\n        }\n\n        const targetNode = exported[0];\n        const implementations = (targetNode as any).getImplementations?.();\n        if (!implementations) {\n          result = { implementations: [], total_count: 0 };\n          break;\n        }\n\n        const results: ReferenceLocation[] = [];\n        for (const impl of implementations) {\n          const implSf = impl.getSourceFile();\n          const lc = implSf.getLineAndColumnAtPos(\n            impl.getTextSpan().getStart()\n          );\n          results.push({\n            file: implSf.getFilePath(),\n            line: lc.line,\n            column: lc.column,\n            text:\n              impl.getNode().getParent()?.getText() || impl.getNode().getText(),\n          });\n        }\n\n        const unique = deduplicateLocations(results);\n        result = {\n          implementations: unique.slice(0, 50),\n          total_count: unique.length,\n        };\n        break;\n      }\n\n      default:\n        throw new Error(`Unknown task type: ${msg.type}`);\n    }\n\n    parentPort?.postMessage({ id: msg.id, result });\n  } catch (error: any) {\n    parentPort?.postMessage({ id: msg.id, error: error.message });\n  }\n});\n\nfunction getJsDoc(node: any): string | undefined {\n  if (node.getJsDocs) {\n    const docs = node.getJsDocs();\n    if (docs.length > 0) {\n      return docs[0].getCommentText();\n    }\n  }\n  return undefined;\n}\n\nfunction deduplicateLocations<\n  T extends { file: string; line: number; column: number },\n>(locations: T[]): T[] {\n  const seen = new Set<string>();\n  return locations.filter((loc) => {\n    const key = `${loc.file}:${loc.line}:${loc.column}`;\n    if (seen.has(key)) return false;\n    seen.add(key);\n    return true;\n  });\n}\n"],"mappings":";AAAA,SAAS,kBAAkB;AAC3B,SAAS,eAAe;AAGxB,IAAM,WAAW,oBAAI,IAAqB;AAE1C,SAAS,WAAW,cAA+B;AACjD,MAAI,CAAC,SAAS,IAAI,YAAY,GAAG;AAC/B,aAAS;AAAA,MACP;AAAA,MACA,IAAI,QAAQ;AAAA,QACV,kBAAkB;AAAA,QAClB,6BAA6B;AAAA,MAC/B,CAAC;AAAA,IACH;AAAA,EACF;AACA,SAAO,SAAS,IAAI,YAAY;AAClC;AAwBA,YAAY,GAAG,WAAW,OAAO,QAAuB;AACtD,MAAI;AACF,QAAI;AAEJ,YAAQ,IAAI,MAAM;AAAA,MAChB,KAAK,sBAAsB;AACzB,cAAM,EAAE,UAAU,MAAM,OAAO,IAAI,IAAI;AACvC,cAAM,UAAU,WAAW,QAAQ;AACnC,cAAM,KAAK,QAAQ,oBAAoB,IAAI;AAC3C,YAAI,CAAC,IAAI;AACP,mBAAS,CAAC;AACV;AAAA,QACF;AAEA,cAAM,WAAW,GAAG,wBAAwB,EAAE,IAAI,MAAM;AACxD,YAAI,CAAC,YAAY,SAAS,WAAW,GAAG;AACtC,mBAAS,CAAC;AACV;AAAA,QACF;AAEA,iBAAS,SAAS,IAAI,CAAC,SAAS;AAC9B,gBAAM,aAAa,KAAK,cAAc;AACtC,gBAAM,gBAAgB,WAAW;AAAA,YAC/B,KAAK,SAAS;AAAA,UAChB;AACA,iBAAO;AAAA,YACL,MAAM,WAAW,YAAY;AAAA,YAC7B,MAAM,cAAc;AAAA,YACpB,QAAQ,cAAc;AAAA,YACtB,MAAM,KAAK,YAAY;AAAA,YACvB,SAAS,KAAK,QAAQ;AAAA,YACtB,eAAe,SAAS,IAAI;AAAA,UAC9B;AAAA,QACF,CAAC;AACD;AAAA,MACF;AAAA,MAEA,KAAK,mBAAmB;AACtB,cAAM,EAAE,UAAU,MAAM,OAAO,IAAI,IAAI;AACvC,cAAM,UAAU,WAAW,QAAQ;AACnC,cAAM,KAAK,QAAQ,oBAAoB,IAAI;AAC3C,YAAI,CAAC,IAAI;AACP,mBAAS,EAAE,YAAY,CAAC,GAAG,aAAa,EAAE;AAC1C;AAAA,QACF;AAEA,cAAM,WAAW,GAAG,wBAAwB,EAAE,IAAI,MAAM;AACxD,YAAI,CAAC,YAAY,SAAS,WAAW,GAAG;AACtC,mBAAS,EAAE,YAAY,CAAC,GAAG,aAAa,EAAE;AAC1C;AAAA,QACF;AAEA,cAAM,aAAa,SAAS,CAAC;AAC7B,cAAM,aAAc,WAAmB,iBAAiB;AACxD,YAAI,CAAC,YAAY;AACf,mBAAS,EAAE,YAAY,CAAC,GAAG,aAAa,EAAE;AAC1C;AAAA,QACF;AAEA,cAAM,UAA+B,CAAC;AACtC,mBAAW,aAAa,YAAY;AAClC,qBAAW,OAAO,UAAU,cAAc,GAAG;AAC3C,kBAAM,QAAQ,IAAI,cAAc;AAChC,kBAAM,KAAK,MAAM;AAAA,cACf,IAAI,YAAY,EAAE,SAAS;AAAA,YAC7B;AACA,oBAAQ,KAAK;AAAA,cACX,MAAM,MAAM,YAAY;AAAA,cACxB,MAAM,GAAG;AAAA,cACT,QAAQ,GAAG;AAAA,cACX,MACE,IAAI,QAAQ,EAAE,UAAU,GAAG,QAAQ,KAAK,IAAI,QAAQ,EAAE,QAAQ;AAAA,YAClE,CAAC;AAAA,UACH;AAAA,QACF;AAEA,cAAM,SAAS,qBAAqB,OAAO;AAC3C,iBAAS;AAAA,UACP,YAAY,OAAO,MAAM,GAAG,EAAE;AAAA,UAC9B,aAAa,OAAO;AAAA,QACtB;AACA;AAAA,MACF;AAAA,MAEA,KAAK,wBAAwB;AAC3B,cAAM,EAAE,UAAU,MAAM,OAAO,IAAI,IAAI;AACvC,cAAM,UAAU,WAAW,QAAQ;AACnC,cAAM,KAAK,QAAQ,oBAAoB,IAAI;AAC3C,YAAI,CAAC,IAAI;AACP,mBAAS,EAAE,iBAAiB,CAAC,GAAG,aAAa,EAAE;AAC/C;AAAA,QACF;AAEA,cAAM,WAAW,GAAG,wBAAwB,EAAE,IAAI,MAAM;AACxD,YAAI,CAAC,YAAY,SAAS,WAAW,GAAG;AACtC,mBAAS,EAAE,iBAAiB,CAAC,GAAG,aAAa,EAAE;AAC/C;AAAA,QACF;AAEA,cAAM,aAAa,SAAS,CAAC;AAC7B,cAAM,kBAAmB,WAAmB,qBAAqB;AACjE,YAAI,CAAC,iBAAiB;AACpB,mBAAS,EAAE,iBAAiB,CAAC,GAAG,aAAa,EAAE;AAC/C;AAAA,QACF;AAEA,cAAM,UAA+B,CAAC;AACtC,mBAAW,QAAQ,iBAAiB;AAClC,gBAAM,SAAS,KAAK,cAAc;AAClC,gBAAM,KAAK,OAAO;AAAA,YAChB,KAAK,YAAY,EAAE,SAAS;AAAA,UAC9B;AACA,kBAAQ,KAAK;AAAA,YACX,MAAM,OAAO,YAAY;AAAA,YACzB,MAAM,GAAG;AAAA,YACT,QAAQ,GAAG;AAAA,YACX,MACE,KAAK,QAAQ,EAAE,UAAU,GAAG,QAAQ,KAAK,KAAK,QAAQ,EAAE,QAAQ;AAAA,UACpE,CAAC;AAAA,QACH;AAEA,cAAM,SAAS,qBAAqB,OAAO;AAC3C,iBAAS;AAAA,UACP,iBAAiB,OAAO,MAAM,GAAG,EAAE;AAAA,UACnC,aAAa,OAAO;AAAA,QACtB;AACA;AAAA,MACF;AAAA,MAEA;AACE,cAAM,IAAI,MAAM,sBAAsB,IAAI,IAAI,EAAE;AAAA,IACpD;AAEA,gBAAY,YAAY,EAAE,IAAI,IAAI,IAAI,OAAO,CAAC;AAAA,EAChD,SAAS,OAAY;AACnB,gBAAY,YAAY,EAAE,IAAI,IAAI,IAAI,OAAO,MAAM,QAAQ,CAAC;AAAA,EAC9D;AACF,CAAC;AAED,SAAS,SAAS,MAA+B;AAC/C,MAAI,KAAK,WAAW;AAClB,UAAM,OAAO,KAAK,UAAU;AAC5B,QAAI,KAAK,SAAS,GAAG;AACnB,aAAO,KAAK,CAAC,EAAE,eAAe;AAAA,IAChC;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,qBAEP,WAAqB;AACrB,QAAM,OAAO,oBAAI,IAAY;AAC7B,SAAO,UAAU,OAAO,CAAC,QAAQ;AAC/B,UAAM,MAAM,GAAG,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,MAAM;AACjD,QAAI,KAAK,IAAI,GAAG,EAAG,QAAO;AAC1B,SAAK,IAAI,GAAG;AACZ,WAAO;AAAA,EACT,CAAC;AACH;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aiready/ast-mcp-server",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "The AIReady AST-aware Model Context Protocol (MCP) Server. Explore and navigate multi-project TypeScript/JavaScript codebases with high precision using AST and type information.",
   "keywords": [
     "aiready",