mason-context 0.3.2 → 0.3.4

package/dist/src/cli.js

@@ -405,13 +405,12 @@ function spawnWithStdin(command, args, input) {
   });
 }
 async function callClaudeCLI(system, userMessage) {
-  const prompt = `${system}
-
-${userMessage}`;
-  return spawnWithStdin("claude", ["-p"], prompt);
+  return spawnWithStdin("claude", ["-p", "--system-prompt", system], userMessage);
 }
 async function callGeminiCLI(system, userMessage) {
-  const prompt = `${system}
+  const prompt = `<system>
+${system}
+</system>
 
 ${userMessage}`;
   return spawnWithStdin("gemini", ["-p", ""], prompt);
@@ -519,10 +518,22 @@ async function loadProjectConfig(rootDir) {
     return {};
   }
 }
+async function getTrackedFiles(rootDir) {
+  try {
+    const { stdout } = await exec5("git", ["ls-files", "--cached", "--others", "--exclude-standard"], {
+      cwd: rootDir,
+      maxBuffer: 1e7
+    });
+    return new Set(stdout.trim().split("\n").filter(Boolean));
+  } catch {
+    return null;
+  }
+}
 async function sampleFiles(rootDir, maxFiles = 25) {
   const selected = /* @__PURE__ */ new Map();
   const projectConfig = await loadProjectConfig(rootDir);
   const ignorePatterns = [...IGNORE_PATTERNS, ...projectConfig.ignore ?? []];
+  const trackedFiles = await getTrackedFiles(rootDir);
   for (const filePath of projectConfig.alwaysInclude ?? []) {
     if (selected.size >= maxFiles) break;
     const resolvedPath = path2.resolve(rootDir, filePath);
@@ -701,6 +712,7 @@ async function sampleFiles(rootDir, maxFiles = 25) {
       const fullPath = path2.resolve(rootDir, filePath);
       if (!fullPath.startsWith(path2.resolve(rootDir))) continue;
       if (isSensitiveFile(filePath)) continue;
+      if (trackedFiles && !trackedFiles.has(filePath)) continue;
       const stat = await fs3.stat(fullPath);
       if (stat.size > 1e5) continue;
       const content = await fs3.readFile(fullPath, "utf-8");
@@ -1094,6 +1106,7 @@ import fs4 from "fs/promises";
 import path4 from "path";
 import { execFile as execFile6 } from "child_process";
 import { promisify as promisify6 } from "util";
+import fg4 from "fast-glob";
 function snapshotDir(rootDir) {
   return path4.join(rootDir, ".mason");
 }
@@ -1157,7 +1170,31 @@ function parseSnapshotResponse(raw) {
 }
 async function createSnapshot(rootDir, config) {
   const resolvedRoot = path4.resolve(rootDir);
-  const sampled = await sampleFiles(resolvedRoot, 25);
+  const allFiles = await fg4(
+    "**/*.{ts,tsx,js,jsx,kt,kts,java,py,go,rs,swift,rb,cs,cpp,c,dart}",
+    {
+      cwd: resolvedRoot,
+      ignore: [
+        "**/node_modules/**",
+        "**/dist/**",
+        "**/build/**",
+        "**/.gradle/**",
+        "**/target/**",
+        "**/.git/**",
+        "**/vendor/**",
+        "**/__pycache__/**",
+        "**/venv/**",
+        "**/.venv/**",
+        "**/*.min.*",
+        "**/*.map",
+        "**/generated/**",
+        "**/R.java",
+        "**/BuildConfig.java"
+      ]
+    }
+  );
+  const sampleCount = Math.min(80, Math.max(20, Math.round(allFiles.length * 0.15)));
+  const sampled = await sampleFiles(resolvedRoot, sampleCount);
   const filesWithContent = [];
   for (const sample of sampled) {
     const full = await readFullFile(resolvedRoot, sample.path);
@@ -1324,7 +1361,7 @@ import fs5 from "fs/promises";
 import path5 from "path";
 import { execFile as execFile7 } from "child_process";
 import { promisify as promisify7 } from "util";
-import fg4 from "fast-glob";
+import fg5 from "fast-glob";
 async function analyzeImpact(rootDir, targetFiles) {
   const resolvedRoot = path5.resolve(rootDir);
   const resolvedTargets = await resolveTargetFiles(resolvedRoot, targetFiles);
@@ -1347,7 +1384,7 @@ async function resolveTargetFiles(rootDir, targets) {
       resolved.push(target);
       continue;
     }
-    const matches = await fg4(`**/${target}`, {
+    const matches = await fg5(`**/${target}`, {
       cwd: rootDir,
       ignore: IGNORE2
     });
@@ -1355,7 +1392,7 @@ async function resolveTargetFiles(rootDir, targets) {
       resolved.push(matches[0]);
     } else {
       const noExt = target.replace(/\.[^.]+$/, "");
-      const extMatches = await fg4(`**/${noExt}.*`, {
+      const extMatches = await fg5(`**/${noExt}.*`, {
         cwd: rootDir,
         ignore: IGNORE2
       });
@@ -1412,7 +1449,7 @@ async function getReferences(rootDir, targetFiles) {
     const basename = path5.basename(target).replace(/\.[^.]+$/, "");
     searchNames.add(basename);
   }
-  const allSourceFiles = await fg4(`**/${SOURCE_EXTENSIONS2}`, {
+  const allSourceFiles = await fg5(`**/${SOURCE_EXTENSIONS2}`, {
     cwd: rootDir,
     ignore: IGNORE2
   });
@@ -1461,7 +1498,7 @@ async function getRelatedTests(rootDir, targetFiles) {
     "**/*Test.swift",
     "**/*_test.rs"
   ];
-  const testFiles = await fg4(testPatterns, { cwd: rootDir, ignore: IGNORE2 });
+  const testFiles = await fg5(testPatterns, { cwd: rootDir, ignore: IGNORE2 });
   const results = [];
   for (const target of targetFiles) {
     const targetBaseName = path5.basename(target).replace(/\.[^.]+$/, "");
@@ -1508,7 +1545,7 @@ import fs6 from "fs/promises";
 import path6 from "path";
 import { execFile as execFile8 } from "child_process";
 import { promisify as promisify8 } from "util";
-import fg5 from "fast-glob";
+import fg6 from "fast-glob";
 async function buildContext(dir) {
   return {
     rootDir: dir,
@@ -1588,7 +1625,7 @@ async function detectProjectSnapshot(rootDir) {
   ];
   const testInfo = {};
   for (const pattern of testDirs) {
-    const files = await fg5(`${pattern}/**/*`, {
+    const files = await fg6(`${pattern}/**/*`, {
       cwd: rootDir,
       ignore: IGNORE3,
       onlyFiles: true
@@ -1608,12 +1645,12 @@ async function detectProjectSnapshot(rootDir) {
     { pattern: "**/*_test.rs", label: "*_test.rs" }
   ];
   for (const { pattern, label } of testFilePatterns) {
-    const files = await fg5(pattern, { cwd: rootDir, ignore: IGNORE3 });
+    const files = await fg6(pattern, { cwd: rootDir, ignore: IGNORE3 });
     if (files.length > 0) {
       testInfo[label] = files.length;
     }
   }
-  const sourceFiles = await fg5("**/*.{ts,tsx,js,jsx,kt,kts,java,py,go,rs,swift,rb,cs,cpp,c,dart}", {
+  const sourceFiles = await fg6("**/*.{ts,tsx,js,jsx,kt,kts,java,py,go,rs,swift,rb,cs,cpp,c,dart}", {
     cwd: rootDir,
     ignore: IGNORE3
   });
@@ -1646,7 +1683,7 @@ async function getCodeSamples(dir, count = 15) {
 }
 async function getProjectStructure(dir) {
   const rootDir = path6.resolve(dir);
-  const allFiles = await fg5("**/*", {
+  const allFiles = await fg6("**/*", {
     cwd: rootDir,
     ignore: IGNORE3,
     onlyFiles: true
@@ -1829,7 +1866,7 @@ function createMcpServer() {
   const server = new McpServer(
     {
       name: "mason",
-      version: "0.3.2"
+      version: "0.3.4"
     },
     {
       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."
@@ -2014,7 +2051,7 @@ function createCLI() {
   const program = new Command();
   program.name("mason").description(
     "Context engineering CLI & MCP server \u2014 generates intelligent CLAUDE.md files"
-  ).version("0.3.2");
+  ).version("0.3.4");
   program.command("setup").description("Register Mason as an MCP server with Claude Code").option("--scope <scope>", "Config scope: user or project", "user").action(async (opts) => {
     const { execFile: execFile9 } = await import("child_process");
     const { promisify: promisify9 } = await import("util");