archtracker-mcp 0.2.0 → 0.2.1

package/dist/cli/index.js

@@ -1,10 +1,4 @@
 #!/usr/bin/env node
-var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
-  get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
-}) : x)(function(x) {
-  if (typeof require !== "undefined") return require.apply(this, arguments);
-  throw Error('Dynamic require of "' + x + '" is not supported');
-});
 
 // src/cli/index.ts
 import { Command } from "commander";
@@ -175,6 +169,306 @@ function detectCycles(edges) {
   return cycles;
 }
 
+// src/analyzer/engines/strip-comments.ts
+function stripComments(content, style) {
+  switch (style) {
+    case "c-style":
+      return stripCStyle(content);
+    case "hash":
+      return stripHash(content);
+    case "python":
+      return stripPython(content);
+    case "ruby":
+      return stripRuby(content);
+    case "php":
+      return stripPhp(content);
+  }
+}
+function stripCStyle(content) {
+  let result = "";
+  let i = 0;
+  while (i < content.length) {
+    if (content[i] === "/" && content[i + 1] === "/") {
+      while (i < content.length && content[i] !== "\n") {
+        result += " ";
+        i++;
+      }
+    } else if (content[i] === "/" && content[i + 1] === "*") {
+      result += " ";
+      i++;
+      result += " ";
+      i++;
+      while (i < content.length) {
+        if (content[i] === "*" && content[i + 1] === "/") {
+          result += " ";
+          i++;
+          result += " ";
+          i++;
+          break;
+        }
+        result += content[i] === "\n" ? "\n" : " ";
+        i++;
+      }
+    } else if (content[i] === '"') {
+      result += content[i];
+      i++;
+      while (i < content.length && content[i] !== '"') {
+        if (content[i] === "\\" && i + 1 < content.length) {
+          result += content[i];
+          i++;
+          result += content[i];
+          i++;
+        } else if (content[i] === "\n") {
+          result += "\n";
+          i++;
+        } else {
+          result += content[i];
+          i++;
+        }
+      }
+      if (i < content.length) {
+        result += content[i];
+        i++;
+      }
+    } else if (content[i] === "`") {
+      result += " ";
+      i++;
+      while (i < content.length && content[i] !== "`") {
+        result += content[i] === "\n" ? "\n" : " ";
+        i++;
+      }
+      if (i < content.length) {
+        result += " ";
+        i++;
+      }
+    } else {
+      result += content[i];
+      i++;
+    }
+  }
+  return result;
+}
+function stripHash(content) {
+  let result = "";
+  let i = 0;
+  while (i < content.length) {
+    if (content[i] === "#") {
+      while (i < content.length && content[i] !== "\n") {
+        result += " ";
+        i++;
+      }
+    } else if (content[i] === '"') {
+      result += content[i];
+      i++;
+      while (i < content.length && content[i] !== '"') {
+        if (content[i] === "\\" && i + 1 < content.length) {
+          result += content[i++];
+          result += content[i++];
+        } else {
+          result += content[i++];
+        }
+      }
+      if (i < content.length) {
+        result += content[i];
+        i++;
+      }
+    } else if (content[i] === "'") {
+      result += content[i];
+      i++;
+      while (i < content.length && content[i] !== "'") {
+        if (content[i] === "\\" && i + 1 < content.length) {
+          result += content[i++];
+          result += content[i++];
+        } else {
+          result += content[i++];
+        }
+      }
+      if (i < content.length) {
+        result += content[i];
+        i++;
+      }
+    } else {
+      result += content[i];
+      i++;
+    }
+  }
+  return result;
+}
+function stripPython(content) {
+  let result = "";
+  let i = 0;
+  while (i < content.length) {
+    if (content[i] === '"' && content[i + 1] === '"' && content[i + 2] === '"' || content[i] === "'" && content[i + 1] === "'" && content[i + 2] === "'") {
+      const quote = content[i];
+      const tripleQuote = quote + quote + quote;
+      result += "   ";
+      i += 3;
+      while (i < content.length) {
+        if (content[i] === quote && content[i + 1] === quote && content[i + 2] === quote) {
+          result += "   ";
+          i += 3;
+          break;
+        }
+        result += content[i] === "\n" ? "\n" : " ";
+        i++;
+      }
+    } else if (content[i] === "#") {
+      while (i < content.length && content[i] !== "\n") {
+        result += " ";
+        i++;
+      }
+    } else if (content[i] === '"' || content[i] === "'") {
+      const quote = content[i];
+      result += content[i];
+      i++;
+      while (i < content.length && content[i] !== quote) {
+        if (content[i] === "\\" && i + 1 < content.length) {
+          result += content[i++];
+          result += content[i++];
+        } else if (content[i] === "\n") {
+          result += "\n";
+          i++;
+          break;
+        } else {
+          result += content[i++];
+        }
+      }
+      if (i < content.length && content[i] === quote) {
+        result += content[i];
+        i++;
+      }
+    } else {
+      result += content[i];
+      i++;
+    }
+  }
+  return result;
+}
+function stripRuby(content) {
+  const lines = content.split("\n");
+  const result = [];
+  let inBlock = false;
+  for (const line of lines) {
+    if (!inBlock && line.startsWith("=begin")) {
+      inBlock = true;
+      result.push(" ".repeat(line.length));
+      continue;
+    }
+    if (inBlock) {
+      if (line.startsWith("=end")) {
+        inBlock = false;
+      }
+      result.push(" ".repeat(line.length));
+      continue;
+    }
+    let processed = "";
+    let i = 0;
+    while (i < line.length) {
+      if (line[i] === "#") {
+        processed += " ".repeat(line.length - i);
+        break;
+      } else if (line[i] === '"') {
+        processed += line[i];
+        i++;
+        while (i < line.length && line[i] !== '"') {
+          if (line[i] === "\\" && i + 1 < line.length) {
+            processed += line[i++];
+            processed += line[i++];
+          } else {
+            processed += line[i++];
+          }
+        }
+        if (i < line.length) {
+          processed += line[i];
+          i++;
+        }
+      } else if (line[i] === "'") {
+        processed += line[i];
+        i++;
+        while (i < line.length && line[i] !== "'") {
+          if (line[i] === "\\" && i + 1 < line.length) {
+            processed += line[i++];
+            processed += line[i++];
+          } else {
+            processed += line[i++];
+          }
+        }
+        if (i < line.length) {
+          processed += line[i];
+          i++;
+        }
+      } else {
+        processed += line[i];
+        i++;
+      }
+    }
+    result.push(processed);
+  }
+  return result.join("\n");
+}
+function stripPhp(content) {
+  let result = "";
+  let i = 0;
+  while (i < content.length) {
+    if (content[i] === "/" && content[i + 1] === "/" || content[i] === "#") {
+      while (i < content.length && content[i] !== "\n") {
+        result += " ";
+        i++;
+      }
+    } else if (content[i] === "/" && content[i + 1] === "*") {
+      result += " ";
+      i++;
+      result += " ";
+      i++;
+      while (i < content.length) {
+        if (content[i] === "*" && content[i + 1] === "/") {
+          result += " ";
+          i++;
+          result += " ";
+          i++;
+          break;
+        }
+        result += content[i] === "\n" ? "\n" : " ";
+        i++;
+      }
+    } else if (content[i] === '"') {
+      result += content[i];
+      i++;
+      while (i < content.length && content[i] !== '"') {
+        if (content[i] === "\\" && i + 1 < content.length) {
+          result += content[i++];
+          result += content[i++];
+        } else {
+          result += content[i++];
+        }
+      }
+      if (i < content.length) {
+        result += content[i];
+        i++;
+      }
+    } else if (content[i] === "'") {
+      result += content[i];
+      i++;
+      while (i < content.length && content[i] !== "'") {
+        if (content[i] === "\\" && i + 1 < content.length) {
+          result += content[i++];
+          result += content[i++];
+        } else {
+          result += content[i++];
+        }
+      }
+      if (i < content.length) {
+        result += content[i];
+        i++;
+      }
+    } else {
+      result += content[i];
+      i++;
+    }
+  }
+  return result;
+}
+
 // src/analyzer/engines/regex-engine.ts
 var RegexEngine = class {
   constructor(config) {
@@ -195,6 +489,7 @@ var RegexEngine = class {
     const projectFileSet = new Set(projectFiles);
     const files = {};
     const edges = [];
+    const edgeSet = /* @__PURE__ */ new Set();
     for (const filePath of projectFiles) {
       const relPath = relative(absRootDir, filePath);
       files[relPath] = {
@@ -212,7 +507,8 @@ var RegexEngine = class {
       } catch {
         continue;
       }
-      const imports = this.extractImports(content);
+      const stripped = stripComments(content, this.config.commentStyle);
+      const imports = this.extractImports(stripped);
       for (const importPath of imports) {
         const resolved = this.config.resolveImport(
           importPath,
@@ -223,6 +519,10 @@ var RegexEngine = class {
         if (!resolved) continue;
         const relTarget = relative(absRootDir, resolved);
         if (!files[relTarget]) continue;
+        if (relSource === relTarget) continue;
+        const edgeKey = `${relSource}\0${relTarget}`;
+        if (edgeSet.has(edgeKey)) continue;
+        edgeSet.add(edgeKey);
         edges.push({
           source: relSource,
           target: relTarget,
@@ -243,6 +543,9 @@ var RegexEngine = class {
     };
   }
   extractImports(content) {
+    if (this.config.extractImports) {
+      return this.config.extractImports(content);
+    }
     const imports = [];
     for (const pattern of this.config.importPatterns) {
       const regex = new RegExp(pattern.regex.source, pattern.regex.flags);
@@ -393,10 +696,12 @@ async function scanExtensions(dir, counts, maxDepth, currentDepth) {
 }
 
 // src/analyzer/engines/languages.ts
+import { readFileSync } from "fs";
 import { join as join3, dirname, resolve as resolve3 } from "path";
 var python = {
   id: "python",
   extensions: [".py"],
+  commentStyle: "python",
   importPatterns: [
     // from package.module import something
     { regex: /^from\s+(\.[\w.]*|\w[\w.]*)\s+import\b/gm },
@@ -424,12 +729,23 @@ function tryPythonResolve(base, projectFiles) {
 var rust = {
   id: "rust",
   extensions: [".rs"],
-  importPatterns: [
-    // use crate::module::item;
-    { regex: /\buse\s+crate::(\w[\w:]*)/gm },
-    // mod child;
-    { regex: /\bmod\s+(\w+)\s*;/gm }
-  ],
+  commentStyle: "c-style",
+  importPatterns: [],
+  // handled by extractImports
+  extractImports(content) {
+    const imports = [];
+    const modRegex = /\bmod\s+(\w+)\s*;/gm;
+    let match;
+    while ((match = modRegex.exec(content)) !== null) {
+      imports.push(match[1]);
+    }
+    const useRegex = /\buse\s+crate::([\s\S]*?);/gm;
+    while ((match = useRegex.exec(content)) !== null) {
+      const body = match[1].trim();
+      extractRustUsePaths(body, "", imports);
+    }
+    return imports;
+  },
   resolveImport(importPath, sourceFile, rootDir, projectFiles) {
     const srcDir = join3(rootDir, "src");
     if (!importPath.includes("::")) {
@@ -452,15 +768,84 @@ var rust = {
   },
   defaultExclude: ["target"]
 };
+function extractRustUsePaths(body, prefix, results) {
+  const trimmed = body.trim();
+  const braceStart = trimmed.indexOf("{");
+  if (braceStart === -1) {
+    const path = prefix ? `${prefix}::${trimmed}` : trimmed;
+    if (path && !path.includes("{")) {
+      results.push(path);
+    }
+    return;
+  }
+  let pathPrefix = trimmed.slice(0, braceStart).trim();
+  if (pathPrefix.endsWith("::")) {
+    pathPrefix = pathPrefix.slice(0, -2);
+  }
+  const fullPrefix = prefix ? `${prefix}::${pathPrefix}` : pathPrefix;
+  let depth = 0;
+  let braceEnd = -1;
+  for (let i = braceStart; i < trimmed.length; i++) {
+    if (trimmed[i] === "{") depth++;
+    else if (trimmed[i] === "}") {
+      depth--;
+      if (depth === 0) {
+        braceEnd = i;
+        break;
+      }
+    }
+  }
+  if (braceEnd === -1) return;
+  const inner = trimmed.slice(braceStart + 1, braceEnd).trim();
+  const items = splitByTopLevelComma(inner);
+  for (const item of items) {
+    const cleaned = item.trim();
+    if (cleaned === "self") {
+      results.push(fullPrefix);
+    } else if (cleaned) {
+      extractRustUsePaths(cleaned, fullPrefix, results);
+    }
+  }
+}
+function splitByTopLevelComma(s) {
+  const parts = [];
+  let depth = 0;
+  let start = 0;
+  for (let i = 0; i < s.length; i++) {
+    if (s[i] === "{") depth++;
+    else if (s[i] === "}") depth--;
+    else if (s[i] === "," && depth === 0) {
+      parts.push(s.slice(start, i));
+      start = i + 1;
+    }
+  }
+  parts.push(s.slice(start));
+  return parts;
+}
 var go = {
   id: "go",
   extensions: [".go"],
-  importPatterns: [
-    // import "path" or import ( "path" )
-    { regex: /\bimport\s+(?:\w+\s+)?"([^"]+)"/gm },
-    // import block entries
-    { regex: /^\s+(?:\w+\s+)?"([^"]+)"/gm }
-  ],
+  commentStyle: "c-style",
+  importPatterns: [],
+  // handled by extractImports
+  extractImports(content) {
+    const imports = [];
+    const singleRegex = /\bimport\s+(?:\w+\s+)?"([^"]+)"/gm;
+    let match;
+    while ((match = singleRegex.exec(content)) !== null) {
+      imports.push(match[1]);
+    }
+    const blockRegex = /\bimport\s*\(([^)]*)\)/gms;
+    while ((match = blockRegex.exec(content)) !== null) {
+      const block = match[1];
+      const entryRegex = /(?:\w+\s+)?"([^"]+)"/g;
+      let entry;
+      while ((entry = entryRegex.exec(block)) !== null) {
+        imports.push(entry[1]);
+      }
+    }
+    return imports;
+  },
   resolveImport(importPath, _sourceFile, rootDir, projectFiles) {
     const modPrefix = goModulePrefix(rootDir);
     if (!modPrefix || !importPath.startsWith(modPrefix)) return null;
@@ -477,8 +862,7 @@ var goModCache = /* @__PURE__ */ new Map();
 function goModulePrefix(rootDir) {
   if (goModCache.has(rootDir)) return goModCache.get(rootDir);
   try {
-    const fs = __require("fs");
-    const content = fs.readFileSync(join3(rootDir, "go.mod"), "utf-8");
+    const content = readFileSync(join3(rootDir, "go.mod"), "utf-8");
     const match = content.match(/^module\s+(.+)$/m);
     const prefix = match ? match[1].trim() : null;
     goModCache.set(rootDir, prefix);
@@ -491,6 +875,7 @@ function goModulePrefix(rootDir) {
 var java = {
   id: "java",
   extensions: [".java"],
+  commentStyle: "c-style",
   importPatterns: [
     // import com.example.ClassName;
     { regex: /^import\s+(?:static\s+)?([\w.]+);/gm }
@@ -508,6 +893,7 @@ var java = {
 var cCpp = {
   id: "c-cpp",
   extensions: [".c", ".cpp", ".cc", ".cxx", ".h", ".hpp"],
+  commentStyle: "c-style",
   importPatterns: [
     // #include "file.h" (skip <system> includes)
     { regex: /^#include\s+"([^"]+)"/gm }
@@ -528,6 +914,7 @@ var cCpp = {
 var ruby = {
   id: "ruby",
   extensions: [".rb"],
+  commentStyle: "ruby",
   importPatterns: [
     // require_relative 'path'
     { regex: /\brequire_relative\s+['"]([^'"]+)['"]/gm },
@@ -549,6 +936,7 @@ var ruby = {
 var php = {
   id: "php",
   extensions: [".php"],
+  commentStyle: "php",
   importPatterns: [
     // require/include/require_once/include_once 'path'
     { regex: /\b(?:require|include)(?:_once)?\s+['"]([^'"]+)['"]/gm },
@@ -576,11 +964,12 @@ var php = {
 var swift = {
   id: "swift",
   extensions: [".swift"],
+  commentStyle: "c-style",
   importPatterns: [
-    // import ModuleName
-    { regex: /^import\s+(?:class|struct|enum|protocol|func|var|let|typealias)?\s*(\w+)/gm }
+    // import ModuleName (for cross-module dependencies)
+    { regex: /^import\s+(?:class\s+|struct\s+|enum\s+|protocol\s+|func\s+|var\s+|let\s+|typealias\s+)?(\w+)/gm }
   ],
-  resolveImport(importPath, _sourceFile, rootDir, projectFiles) {
+  resolveImport(importPath, sourceFile, rootDir, projectFiles) {
     const spmDir = join3(rootDir, "Sources", importPath);
     for (const f of projectFiles) {
       if (f.startsWith(spmDir + "/") && f.endsWith(".swift")) return f;
@@ -592,6 +981,7 @@ var swift = {
 var kotlin = {
   id: "kotlin",
   extensions: [".kt", ".kts"],
+  commentStyle: "c-style",
   importPatterns: [
     // import com.example.ClassName
     { regex: /^import\s+([\w.]+)/gm }