mason-context 0.2.2 → 0.3.0

package/dist/bin/mason-mcp.js

@@ -9,18 +9,115 @@ var __export = (target, all) => {
     __defProp(target, name, { get: all[name], enumerable: true });
 };
 
+// src/test-map.ts
+var test_map_exports = {};
+__export(test_map_exports, {
+  buildTestMap: () => buildTestMap
+});
+import path2 from "path";
+import fg3 from "fast-glob";
+async function buildTestMap(dir) {
+  const rootDir = path2.resolve(dir);
+  const testPatterns = [
+    "**/*.test.*",
+    "**/*.spec.*",
+    "**/*Test.kt",
+    "**/*Test.java",
+    "**/*Tests.kt",
+    "**/*Tests.java",
+    "**/test_*.py",
+    "**/*_test.py",
+    "**/*_test.go",
+    "**/*Tests.swift",
+    "**/*Test.swift",
+    "**/*_test.rs"
+  ];
+  const testFiles = await fg3(testPatterns, { cwd: rootDir, ignore: IGNORE });
+  const sourceFiles = await fg3(
+    "**/*.{ts,tsx,js,jsx,kt,kts,java,py,go,rs,swift,rb,cs,cpp,dart}",
+    { cwd: rootDir, ignore: IGNORE }
+  );
+  const sourceByBaseName = /* @__PURE__ */ new Map();
+  for (const file of sourceFiles) {
+    if (testFiles.includes(file)) continue;
+    const baseName = path2.basename(file).replace(/\.[^.]+$/, "");
+    const existing = sourceByBaseName.get(baseName) ?? [];
+    existing.push(file);
+    sourceByBaseName.set(baseName, existing);
+  }
+  const paired = [];
+  const unmatched = [];
+  for (const testFile of testFiles) {
+    const testBaseName = path2.basename(testFile).replace(/\.[^.]+$/, "");
+    const sourceName = testBaseName.replace(/Test$|Tests$|Spec$|\.test$|\.spec$/, "").replace(/^test_|_test$/, "");
+    if (!sourceName) {
+      unmatched.push(testFile);
+      continue;
+    }
+    const candidates = sourceByBaseName.get(sourceName);
+    if (candidates && candidates.length > 0) {
+      const testDir = path2.dirname(testFile);
+      const bestMatch = candidates.reduce((best, candidate) => {
+        const candidateDir = path2.dirname(candidate);
+        const bestDir = path2.dirname(best);
+        const candidateOverlap = commonSegments(testDir, candidateDir);
+        const bestOverlap = commonSegments(testDir, bestDir);
+        return candidateOverlap > bestOverlap ? candidate : best;
+      });
+      paired.push({
+        test: testFile,
+        source: bestMatch,
+        confidence: candidates.length === 1 ? "exact" : "best-guess"
+      });
+    } else {
+      unmatched.push(testFile);
+    }
+  }
+  return { totalTestFiles: testFiles.length, paired, unmatched };
+}
+function commonSegments(pathA, pathB) {
+  const segsA = pathA.split("/");
+  const segsB = pathB.split("/");
+  let count = 0;
+  for (let i = 0; i < Math.min(segsA.length, segsB.length); i++) {
+    if (segsA[i] === segsB[i]) count++;
+    else break;
+  }
+  return count;
+}
+var IGNORE;
+var init_test_map = __esm({
+  "src/test-map.ts"() {
+    "use strict";
+    IGNORE = [
+      "**/node_modules/**",
+      "**/dist/**",
+      "**/build/**",
+      "**/.gradle/**",
+      "**/target/**",
+      "**/.git/**",
+      "**/vendor/**",
+      "**/__pycache__/**",
+      "**/venv/**",
+      "**/.venv/**",
+      "**/*.min.*",
+      "**/*.map"
+    ];
+  }
+});
+
 // src/impact/impact.ts
 var impact_exports = {};
 __export(impact_exports, {
   analyzeImpact: () => analyzeImpact
 });
 import fs5 from "fs/promises";
-import path4 from "path";
+import path5 from "path";
 import { execFile as execFile7 } from "child_process";
 import { promisify as promisify7 } from "util";
-import fg3 from "fast-glob";
+import fg4 from "fast-glob";
 async function analyzeImpact(rootDir, targetFiles) {
-  const resolvedRoot = path4.resolve(rootDir);
+  const resolvedRoot = path5.resolve(rootDir);
   const resolvedTargets = await resolveTargetFiles(resolvedRoot, targetFiles);
   const [cochange, references, tests] = await Promise.all([
     getCochangeFiles(resolvedRoot, resolvedTargets),
@@ -41,17 +138,17 @@ async function resolveTargetFiles(rootDir, targets) {
       resolved.push(target);
       continue;
     }
-    const matches = await fg3(`**/${target}`, {
+    const matches = await fg4(`**/${target}`, {
       cwd: rootDir,
-      ignore: IGNORE
+      ignore: IGNORE2
     });
     if (matches.length > 0) {
       resolved.push(matches[0]);
     } else {
       const noExt = target.replace(/\.[^.]+$/, "");
-      const extMatches = await fg3(`**/${noExt}.*`, {
+      const extMatches = await fg4(`**/${noExt}.*`, {
         cwd: rootDir,
-        ignore: IGNORE
+        ignore: IGNORE2
       });
       if (extMatches.length > 0) {
         resolved.push(extMatches[0]);
@@ -103,12 +200,12 @@ async function getCochangeFiles(rootDir, targetFiles) {
 async function getReferences(rootDir, targetFiles) {
   const searchNames = /* @__PURE__ */ new Set();
   for (const target of targetFiles) {
-    const basename = path4.basename(target).replace(/\.[^.]+$/, "");
+    const basename = path5.basename(target).replace(/\.[^.]+$/, "");
     searchNames.add(basename);
   }
-  const allSourceFiles = await fg3(`**/${SOURCE_EXTENSIONS2}`, {
+  const allSourceFiles = await fg4(`**/${SOURCE_EXTENSIONS2}`, {
     cwd: rootDir,
-    ignore: IGNORE
+    ignore: IGNORE2
   });
   const targetSet = new Set(targetFiles);
   const filesToSearch = allSourceFiles.filter((f) => !targetSet.has(f));
@@ -120,7 +217,7 @@ async function getReferences(rootDir, targetFiles) {
       batch.map(async (file) => {
         try {
           const content = await fs5.readFile(
-            path4.join(rootDir, file),
+            path5.join(rootDir, file),
             "utf-8"
           );
           for (const name of searchNames) {
@@ -155,12 +252,12 @@ async function getRelatedTests(rootDir, targetFiles) {
     "**/*Test.swift",
     "**/*_test.rs"
   ];
-  const testFiles = await fg3(testPatterns, { cwd: rootDir, ignore: IGNORE });
+  const testFiles = await fg4(testPatterns, { cwd: rootDir, ignore: IGNORE2 });
   const results = [];
   for (const target of targetFiles) {
-    const targetBaseName = path4.basename(target).replace(/\.[^.]+$/, "");
+    const targetBaseName = path5.basename(target).replace(/\.[^.]+$/, "");
     for (const testFile of testFiles) {
-      const testBaseName = path4.basename(testFile).replace(/\.[^.]+$/, "");
+      const testBaseName = path5.basename(testFile).replace(/\.[^.]+$/, "");
       const sourceName = testBaseName.replace(/Test$|Tests$|Spec$|\.test$|\.spec$/, "").replace(/^test_|_test$/, "");
       if (sourceName === targetBaseName) {
         results.push({
@@ -175,12 +272,12 @@ async function getRelatedTests(rootDir, targetFiles) {
 function escapeRegex(str) {
   return str.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
 }
-var exec7, IGNORE, SOURCE_EXTENSIONS2;
+var exec7, IGNORE2, SOURCE_EXTENSIONS2;
 var init_impact = __esm({
   "src/impact/impact.ts"() {
     "use strict";
     exec7 = promisify7(execFile7);
-    IGNORE = [
+    IGNORE2 = [
       "**/node_modules/**",
       "**/dist/**",
       "**/build/**",
@@ -204,10 +301,10 @@ import { z } from "zod";
 
 // src/mcp/tools.ts
 import fs6 from "fs/promises";
-import path5 from "path";
+import path6 from "path";
 import { execFile as execFile8 } from "child_process";
 import { promisify as promisify8 } from "util";
-import fg4 from "fast-glob";
+import fg5 from "fast-glob";
 
 // src/analyzers/git-history.ts
 import { execFile } from "child_process";
@@ -810,23 +907,24 @@ function isSensitiveFile(filePath) {
 
 // src/snapshot/snapshot.ts
 import fs4 from "fs/promises";
-import path3 from "path";
+import path4 from "path";
 import { execFile as execFile6 } from "child_process";
 import { promisify as promisify6 } from "util";
+init_test_map();
 
 // src/llm/providers.ts
-import { execFile as execFile5 } from "child_process";
+import { execFile as execFile5, spawn } from "child_process";
 import { promisify as promisify5 } from "util";
 
 // src/llm/config.ts
 import fs3 from "fs/promises";
-import path2 from "path";
+import path3 from "path";
 import os from "os";
 import { execFile as execFile4 } from "child_process";
 import { promisify as promisify4 } from "util";
 var exec4 = promisify4(execFile4);
-var CONFIG_DIR = path2.join(os.homedir(), ".mason");
-var CONFIG_FILE = path2.join(CONFIG_DIR, "config.json");
+var CONFIG_DIR = path3.join(os.homedir(), ".mason");
+var CONFIG_FILE = path3.join(CONFIG_DIR, "config.json");
 
 // src/llm/providers.ts
 var exec5 = promisify5(execFile5);
@@ -834,10 +932,10 @@ var exec5 = promisify5(execFile5);
 // src/snapshot/snapshot.ts
 var exec6 = promisify6(execFile6);
 function snapshotDir(rootDir) {
-  return path3.join(rootDir, ".mason");
+  return path4.join(rootDir, ".mason");
 }
 function snapshotPath(rootDir) {
-  return path3.join(snapshotDir(rootDir), "snapshot.json");
+  return path4.join(snapshotDir(rootDir), "snapshot.json");
 }
 async function loadSnapshot(rootDir) {
   try {
@@ -870,7 +968,7 @@ async function getCurrentGitHash(rootDir) {
 
 // src/mcp/tools.ts
 var exec8 = promisify8(execFile8);
-var IGNORE2 = [
+var IGNORE3 = [
   "**/node_modules/**",
   "**/dist/**",
   "**/build/**",
@@ -891,7 +989,7 @@ async function buildContext(dir) {
   };
 }
 async function analyzeProject(dir) {
-  const rootDir = path5.resolve(dir);
+  const rootDir = path6.resolve(dir);
   const context = await buildContext(rootDir);
   const results = await runAll(context);
   const projectSnapshot = await detectProjectSnapshot(rootDir);
@@ -945,7 +1043,7 @@ async function detectProjectSnapshot(rootDir) {
   const present = [];
   for (const file of buildFiles) {
     try {
-      await fs6.access(path5.join(rootDir, file));
+      await fs6.access(path6.join(rootDir, file));
       present.push(file);
     } catch {
     }
@@ -963,9 +1061,9 @@ async function detectProjectSnapshot(rootDir) {
   ];
   const testInfo = {};
   for (const pattern of testDirs) {
-    const files = await fg4(`${pattern}/**/*`, {
+    const files = await fg5(`${pattern}/**/*`, {
       cwd: rootDir,
-      ignore: IGNORE2,
+      ignore: IGNORE3,
       onlyFiles: true
     });
     if (files.length > 0) {
@@ -983,18 +1081,18 @@ async function detectProjectSnapshot(rootDir) {
     { pattern: "**/*_test.rs", label: "*_test.rs" }
   ];
   for (const { pattern, label } of testFilePatterns) {
-    const files = await fg4(pattern, { cwd: rootDir, ignore: IGNORE2 });
+    const files = await fg5(pattern, { cwd: rootDir, ignore: IGNORE3 });
     if (files.length > 0) {
       testInfo[label] = files.length;
     }
   }
-  const sourceFiles = await fg4("**/*.{ts,tsx,js,jsx,kt,kts,java,py,go,rs,swift,rb,cs,cpp,c,dart}", {
+  const sourceFiles = await fg5("**/*.{ts,tsx,js,jsx,kt,kts,java,py,go,rs,swift,rb,cs,cpp,c,dart}", {
     cwd: rootDir,
-    ignore: IGNORE2
+    ignore: IGNORE3
   });
   const fileCounts = {};
   for (const file of sourceFiles) {
-    const ext = path5.extname(file).slice(1);
+    const ext = path6.extname(file).slice(1);
     fileCounts[ext] = (fileCounts[ext] ?? 0) + 1;
   }
   return {
@@ -1005,7 +1103,7 @@ async function detectProjectSnapshot(rootDir) {
   };
 }
 async function getCodeSamples(dir, count = 15) {
-  const rootDir = path5.resolve(dir);
+  const rootDir = path6.resolve(dir);
   const samples = await sampleFiles(rootDir, count);
   const output = {
     note: "These are previews (first ~60 lines). Use get_file_content to read the full file if needed.",
@@ -1020,10 +1118,10 @@ async function getCodeSamples(dir, count = 15) {
   return JSON.stringify(output, null, 2);
 }
 async function getProjectStructure(dir) {
-  const rootDir = path5.resolve(dir);
-  const allFiles = await fg4("**/*", {
+  const rootDir = path6.resolve(dir);
+  const allFiles = await fg5("**/*", {
     cwd: rootDir,
-    ignore: IGNORE2,
+    ignore: IGNORE3,
     onlyFiles: true
   });
   const dirInfo = /* @__PURE__ */ new Map();
@@ -1036,7 +1134,7 @@ async function getProjectStructure(dir) {
       }
       const info = dirInfo.get(dirPath);
       info.fileCount++;
-      const ext = path5.extname(file).slice(1);
+      const ext = path6.extname(file).slice(1);
       if (ext) {
         info.extensions.set(ext, (info.extensions.get(ext) ?? 0) + 1);
       }
@@ -1058,81 +1156,12 @@ async function getProjectStructure(dir) {
   return JSON.stringify(output, null, 2);
 }
 async function getTestMap(dir) {
-  const rootDir = path5.resolve(dir);
-  const testPatterns = [
-    "**/*.test.*",
-    "**/*.spec.*",
-    "**/*Test.kt",
-    "**/*Test.java",
-    "**/*Tests.kt",
-    "**/*Tests.java",
-    "**/test_*.py",
-    "**/*_test.py",
-    "**/*_test.go",
-    "**/*Tests.swift",
-    "**/*Test.swift",
-    "**/*_test.rs"
-  ];
-  const testFiles = await fg4(testPatterns, { cwd: rootDir, ignore: IGNORE2 });
-  const sourceFiles = await fg4(
-    "**/*.{ts,tsx,js,jsx,kt,kts,java,py,go,rs,swift,rb,cs,cpp,dart}",
-    { cwd: rootDir, ignore: IGNORE2 }
-  );
-  const sourceByBaseName = /* @__PURE__ */ new Map();
-  for (const file of sourceFiles) {
-    if (testFiles.includes(file)) continue;
-    const baseName = path5.basename(file).replace(/\.[^.]+$/, "");
-    const existing = sourceByBaseName.get(baseName) ?? [];
-    existing.push(file);
-    sourceByBaseName.set(baseName, existing);
-  }
-  const pairs = [];
-  const unmatched = [];
-  for (const testFile of testFiles) {
-    const testBaseName = path5.basename(testFile).replace(/\.[^.]+$/, "");
-    const sourceName = testBaseName.replace(/Test$|Tests$|Spec$|\.test$|\.spec$/, "").replace(/^test_|_test$/, "");
-    if (!sourceName) {
-      unmatched.push(testFile);
-      continue;
-    }
-    const candidates = sourceByBaseName.get(sourceName);
-    if (candidates && candidates.length > 0) {
-      const testDir = path5.dirname(testFile);
-      const bestMatch = candidates.reduce((best, candidate) => {
-        const candidateDir = path5.dirname(candidate);
-        const bestDir = path5.dirname(best);
-        const candidateOverlap = commonSegments(testDir, candidateDir);
-        const bestOverlap = commonSegments(testDir, bestDir);
-        return candidateOverlap > bestOverlap ? candidate : best;
-      });
-      pairs.push({
-        test: testFile,
-        source: bestMatch,
-        confidence: candidates.length === 1 ? "exact" : "best-guess"
-      });
-    } else {
-      unmatched.push(testFile);
-    }
-  }
-  const output = {
-    totalTestFiles: testFiles.length,
-    paired: pairs,
-    unmatched
-  };
-  return JSON.stringify(output, null, 2);
-}
-function commonSegments(pathA, pathB) {
-  const segsA = pathA.split("/");
-  const segsB = pathB.split("/");
-  let count = 0;
-  for (let i = 0; i < Math.min(segsA.length, segsB.length); i++) {
-    if (segsA[i] === segsB[i]) count++;
-    else break;
-  }
-  return count;
+  const { buildTestMap: buildTestMap2 } = await Promise.resolve().then(() => (init_test_map(), test_map_exports));
+  const result = await buildTestMap2(dir);
+  return JSON.stringify(result, null, 2);
 }
 async function getSnapshot(dir) {
-  const rootDir = path5.resolve(dir);
+  const rootDir = path6.resolve(dir);
   const snapshot = await loadSnapshot(rootDir);
   if (!snapshot) {
     return JSON.stringify({
@@ -1148,7 +1177,11 @@ async function getSnapshot(dir) {
     const unique = feat.files.filter((f) => !seenFiles.has(f));
     if (unique.length === 0) continue;
     for (const f of unique) seenFiles.add(f);
-    compactFeatures[name] = unique;
+    const entry = { files: unique };
+    if (feat.tests && feat.tests.length > 0) {
+      entry.tests = feat.tests;
+    }
+    compactFeatures[name] = entry;
   }
   const compactFlows = {};
   for (const [name, flow] of Object.entries(snapshot.flows)) {
@@ -1156,6 +1189,7 @@ async function getSnapshot(dir) {
   }
   const output = {
     exists: true,
+    updatedAt: snapshot.updatedAt,
     features: compactFeatures,
     flows: compactFlows,
     stale: isStale
@@ -1166,7 +1200,7 @@ async function getSnapshot(dir) {
   return JSON.stringify(output);
 }
 async function fullAnalysis(dir) {
-  const rootDir = path5.resolve(dir);
+  const rootDir = path6.resolve(dir);
   const [analysis, structure, samples, testMap, snapshot] = await Promise.all([
     analyzeProject(dir),
     getProjectStructure(dir),
@@ -1192,7 +1226,7 @@ async function fullAnalysis(dir) {
   return JSON.stringify(output, null, 2);
 }
 async function saveSnapshotData(dir, features, flows) {
-  const rootDir = path5.resolve(dir);
+  const rootDir = path6.resolve(dir);
   const gitHash = await getCurrentGitHash(rootDir);
   const now = (/* @__PURE__ */ new Date()).toISOString();
   const existing = await loadSnapshot(rootDir);
@@ -1225,7 +1259,7 @@ async function saveSnapshotData(dir, features, flows) {
 }
 async function getImpact(dir, files) {
   const { analyzeImpact: analyzeImpact2 } = await Promise.resolve().then(() => (init_impact(), impact_exports));
-  const rootDir = path5.resolve(dir);
+  const rootDir = path6.resolve(dir);
   const result = await analyzeImpact2(rootDir, files);
   return JSON.stringify(result, null, 2);
 }
@@ -1235,7 +1269,7 @@ function createMcpServer() {
   const server = new McpServer(
     {
       name: "mason",
-      version: "0.2.2"
+      version: "0.3.0"
     },
     {
       instructions: "Mason is a context engineering tool. Always call get_snapshot before using Explore agents, Glob, or Grep to understand the codebase. The snapshot is a concept map that maps features and flows to their implementing files \u2014 it eliminates the need to search. This applies to ANY question about architecture, features, flows, how things work, cross-feature interactions, or bug investigation. Workflow: 1) Call get_snapshot first. 2) If no snapshot, call full_analysis and then save_snapshot to create one. 3) If the snapshot is stale, tell the user and offer to update it. 4) Use your native file reading tool to read files the snapshot points to. 5) Before modifying a file, call get_impact to check what else might be affected. 6) After making significant changes (new features, refactors, architecture changes), call save_snapshot to update the concept map."