context-compress 1.0.0 → 2026.3.13

package/dist/server.bundle.mjs

@@ -6800,85 +6800,8 @@ var require_dist = __commonJS({
 
 // src/config.ts
 import { readFileSync } from "node:fs";
+import { homedir } from "node:os";
 import { join } from "node:path";
-var DEFAULTS = {
-  passthroughEnvVars: [],
-  debug: false,
-  blockCurl: true,
-  blockWebFetch: true,
-  nudgeOnRead: true,
-  nudgeOnGrep: true,
-  intentSearchThreshold: 5e3,
-  maxOutputBytes: 102400,
-  hardCapBytes: 100 * 1024 * 1024,
-  searchMaxBytes: 40960,
-  batchMaxBytes: 81920,
-  searchLimit: 3,
-  searchWindowMs: 6e4,
-  searchReduceAfter: 3,
-  searchBlockAfter: 8
-};
-function loadFileConfig(projectDir) {
-  const paths = [
-    projectDir && join(projectDir, ".context-compress.json"),
-    join(process.env.HOME ?? "~", ".context-compress.json")
-  ].filter(Boolean);
-  for (const p of paths) {
-    try {
-      const raw = readFileSync(p, "utf-8");
-      return JSON.parse(raw);
-    } catch {
-    }
-  }
-  return {};
-}
-function loadEnvConfig() {
-  const partial2 = {};
-  if (process.env.CONTEXT_COMPRESS_DEBUG === "1") {
-    partial2.debug = true;
-  }
-  if (process.env.CONTEXT_COMPRESS_PASSTHROUGH_ENV) {
-    partial2.passthroughEnvVars = process.env.CONTEXT_COMPRESS_PASSTHROUGH_ENV.split(",").map((s) => s.trim()).filter(Boolean);
-  }
-  if (process.env.CONTEXT_COMPRESS_BLOCK_CURL !== void 0) {
-    partial2.blockCurl = process.env.CONTEXT_COMPRESS_BLOCK_CURL !== "0";
-  }
-  if (process.env.CONTEXT_COMPRESS_BLOCK_WEBFETCH !== void 0) {
-    partial2.blockWebFetch = process.env.CONTEXT_COMPRESS_BLOCK_WEBFETCH !== "0";
-  }
-  if (process.env.CONTEXT_COMPRESS_NUDGE_READ !== void 0) {
-    partial2.nudgeOnRead = process.env.CONTEXT_COMPRESS_NUDGE_READ !== "0";
-  }
-  if (process.env.CONTEXT_COMPRESS_NUDGE_GREP !== void 0) {
-    partial2.nudgeOnGrep = process.env.CONTEXT_COMPRESS_NUDGE_GREP !== "0";
-  }
-  return partial2;
-}
-var _config = null;
-function loadConfig(projectDir) {
-  if (_config) return _config;
-  const fileConfig = loadFileConfig(projectDir);
-  const envConfig = loadEnvConfig();
-  _config = { ...DEFAULTS, ...fileConfig, ...envConfig };
-  return _config;
-}
-function getConfig() {
-  if (!_config) return loadConfig();
-  return _config;
-}
-
-// src/logger.ts
-function debug(...args) {
-  if (getConfig().debug) {
-    process.stderr.write(`[context-compress] ${args.map(String).join(" ")}
-`);
-  }
-}
-
-// src/server.ts
-import { readFileSync as readFileSync2 } from "node:fs";
-import { dirname, join as join4, resolve } from "node:path";
-import { fileURLToPath } from "node:url";
 
 // node_modules/zod/v3/external.js
 var external_exports = {};
@@ -10921,6 +10844,163 @@ var coerce = {
 };
 var NEVER = INVALID;
 
+// src/config.ts
+var DEFAULTS = {
+  passthroughEnvVars: [],
+  debug: false,
+  blockCurl: true,
+  blockWebFetch: true,
+  nudgeOnRead: true,
+  nudgeOnGrep: true,
+  intentSearchThreshold: 5e3,
+  maxOutputBytes: 102400,
+  hardCapBytes: 100 * 1024 * 1024,
+  searchMaxBytes: 40960,
+  batchMaxBytes: 81920,
+  searchLimit: 3,
+  searchWindowMs: 6e4,
+  searchReduceAfter: 3,
+  searchBlockAfter: 8,
+  compressionLevel: "normal"
+};
+var LEVEL_OVERRIDES = {
+  normal: {},
+  compact: {
+    maxOutputBytes: 51200,
+    searchMaxBytes: 20480,
+    batchMaxBytes: 40960,
+    searchLimit: 2,
+    intentSearchThreshold: 3e3
+  },
+  ultra: {
+    maxOutputBytes: 25600,
+    searchMaxBytes: 10240,
+    batchMaxBytes: 20480,
+    searchLimit: 1,
+    intentSearchThreshold: 2e3
+  }
+};
+var ConfigSchema = external_exports.object({
+  passthroughEnvVars: external_exports.array(external_exports.string()).optional(),
+  debug: external_exports.boolean().optional(),
+  blockCurl: external_exports.boolean().optional(),
+  blockWebFetch: external_exports.boolean().optional(),
+  nudgeOnRead: external_exports.boolean().optional(),
+  nudgeOnGrep: external_exports.boolean().optional(),
+  intentSearchThreshold: external_exports.number().int().positive().optional(),
+  maxOutputBytes: external_exports.number().int().positive().optional(),
+  hardCapBytes: external_exports.number().int().positive().optional(),
+  searchMaxBytes: external_exports.number().int().positive().optional(),
+  batchMaxBytes: external_exports.number().int().positive().optional(),
+  searchLimit: external_exports.number().int().positive().optional(),
+  searchWindowMs: external_exports.number().int().positive().optional(),
+  searchReduceAfter: external_exports.number().int().nonnegative().optional(),
+  searchBlockAfter: external_exports.number().int().positive().optional(),
+  compressionLevel: external_exports.enum(["normal", "compact", "ultra"]).optional()
+});
+function parseIntEnv(key) {
+  const val = process.env[key];
+  if (val === void 0) return void 0;
+  const n = Number.parseInt(val, 10);
+  return Number.isNaN(n) ? void 0 : n;
+}
+function loadFileConfig(projectDir2) {
+  const paths = [
+    projectDir2 && join(projectDir2, ".context-compress.json"),
+    join(homedir(), ".context-compress.json")
+  ].filter(Boolean);
+  for (const p of paths) {
+    try {
+      const raw = readFileSync(p, "utf-8");
+      const parsed = JSON.parse(raw);
+      const result = ConfigSchema.safeParse(parsed);
+      if (result.success) {
+        return result.data;
+      }
+      return {};
+    } catch {
+    }
+  }
+  return {};
+}
+function loadEnvConfig() {
+  const partial2 = {};
+  if (process.env.CONTEXT_COMPRESS_DEBUG === "1") {
+    partial2.debug = true;
+  }
+  if (process.env.CONTEXT_COMPRESS_PASSTHROUGH_ENV) {
+    partial2.passthroughEnvVars = process.env.CONTEXT_COMPRESS_PASSTHROUGH_ENV.split(",").map((s) => s.trim()).filter(Boolean);
+  }
+  if (process.env.CONTEXT_COMPRESS_BLOCK_CURL !== void 0) {
+    partial2.blockCurl = process.env.CONTEXT_COMPRESS_BLOCK_CURL !== "0";
+  }
+  if (process.env.CONTEXT_COMPRESS_BLOCK_WEBFETCH !== void 0) {
+    partial2.blockWebFetch = process.env.CONTEXT_COMPRESS_BLOCK_WEBFETCH !== "0";
+  }
+  if (process.env.CONTEXT_COMPRESS_NUDGE_READ !== void 0) {
+    partial2.nudgeOnRead = process.env.CONTEXT_COMPRESS_NUDGE_READ !== "0";
+  }
+  if (process.env.CONTEXT_COMPRESS_NUDGE_GREP !== void 0) {
+    partial2.nudgeOnGrep = process.env.CONTEXT_COMPRESS_NUDGE_GREP !== "0";
+  }
+  const maxOutput = parseIntEnv("CONTEXT_COMPRESS_MAX_OUTPUT_BYTES");
+  if (maxOutput !== void 0) partial2.maxOutputBytes = maxOutput;
+  const hardCap = parseIntEnv("CONTEXT_COMPRESS_HARD_CAP_BYTES");
+  if (hardCap !== void 0) partial2.hardCapBytes = hardCap;
+  const searchMax = parseIntEnv("CONTEXT_COMPRESS_SEARCH_MAX_BYTES");
+  if (searchMax !== void 0) partial2.searchMaxBytes = searchMax;
+  const batchMax = parseIntEnv("CONTEXT_COMPRESS_BATCH_MAX_BYTES");
+  if (batchMax !== void 0) partial2.batchMaxBytes = batchMax;
+  const searchLimit = parseIntEnv("CONTEXT_COMPRESS_SEARCH_LIMIT");
+  if (searchLimit !== void 0) partial2.searchLimit = searchLimit;
+  const searchWindow = parseIntEnv("CONTEXT_COMPRESS_SEARCH_WINDOW_MS");
+  if (searchWindow !== void 0) partial2.searchWindowMs = searchWindow;
+  const searchReduce = parseIntEnv("CONTEXT_COMPRESS_SEARCH_REDUCE_AFTER");
+  if (searchReduce !== void 0) partial2.searchReduceAfter = searchReduce;
+  const searchBlock = parseIntEnv("CONTEXT_COMPRESS_SEARCH_BLOCK_AFTER");
+  if (searchBlock !== void 0) partial2.searchBlockAfter = searchBlock;
+  const intentThreshold = parseIntEnv("CONTEXT_COMPRESS_INTENT_SEARCH_THRESHOLD");
+  if (intentThreshold !== void 0) partial2.intentSearchThreshold = intentThreshold;
+  const level = process.env.CONTEXT_COMPRESS_LEVEL;
+  if (level === "normal" || level === "compact" || level === "ultra") {
+    partial2.compressionLevel = level;
+  }
+  return partial2;
+}
+var _config = null;
+function loadConfig(projectDir2) {
+  if (_config) return _config;
+  const fileConfig = loadFileConfig(projectDir2);
+  const envConfig = loadEnvConfig();
+  const merged = { ...DEFAULTS, ...fileConfig, ...envConfig };
+  const levelOverrides = LEVEL_OVERRIDES[merged.compressionLevel];
+  for (const [key, value] of Object.entries(levelOverrides)) {
+    const k = key;
+    if (!(k in fileConfig) && !(k in envConfig)) {
+      merged[k] = value;
+    }
+  }
+  _config = merged;
+  return _config;
+}
+function getConfig() {
+  if (!_config) return loadConfig();
+  return _config;
+}
+
+// src/logger.ts
+function debug(...args) {
+  if (getConfig().debug) {
+    process.stderr.write(`[context-compress] ${args.map(String).join(" ")}
+`);
+  }
+}
+
+// src/server.ts
+import { readFileSync as readFileSync2, statSync } from "node:fs";
+import { dirname, join as join4, resolve } from "node:path";
+import { fileURLToPath } from "node:url";
+
 // node_modules/zod/v4/core/core.js
 var NEVER2 = Object.freeze({
   status: "aborted"
@@ -21073,7 +21153,6 @@ import { mkdirSync, mkdtempSync, rmSync, writeFileSync } from "node:fs";
 import { tmpdir } from "node:os";
 import { join as join2 } from "node:path";
 var DEFAULT_TIMEOUT = 3e4;
-var DEFAULT_HARD_CAP = 100 * 1024 * 1024;
 var SAFE_ENV_KEYS = [
   "PATH",
   "HOME",
@@ -21126,6 +21205,86 @@ function killProcessTree(pid) {
     }
   }
 }
+function deduplicateLines(output) {
+  const lines = output.split("\n");
+  if (lines.length < 3) return output;
+  const result = [];
+  let prevLine = lines[0];
+  let count = 1;
+  for (let i = 1; i < lines.length; i++) {
+    if (lines[i] === prevLine && prevLine.trim().length > 0) {
+      count++;
+    } else {
+      if (count > 2) {
+        result.push(prevLine);
+        result.push(`  ... (\xD7${count} identical lines)`);
+      } else {
+        for (let j = 0; j < count; j++) result.push(prevLine);
+      }
+      prevLine = lines[i];
+      count = 1;
+    }
+  }
+  if (count > 2) {
+    result.push(prevLine);
+    result.push(`  ... (\xD7${count} identical lines)`);
+  } else {
+    for (let j = 0; j < count; j++) result.push(prevLine);
+  }
+  return result.join("\n");
+}
+function groupErrorLines(output) {
+  const lines = output.split("\n");
+  if (lines.length < 5) return output;
+  const ERROR_RE = /^(.*?(?:error|warning|Error|Warning|ERR|WARN)[:\s])\s*(.+?)(?:\s+(?:at|in|on)\s+(?:line\s+)?(\d+))?$/i;
+  const errorGroups = /* @__PURE__ */ new Map();
+  const resultLines = [];
+  let groupedCount = 0;
+  for (const line of lines) {
+    const match = line.match(ERROR_RE);
+    if (match) {
+      const prefix = match[1].trim();
+      const msg = match[2].trim();
+      const key = `${prefix}|${msg}`;
+      const existing = errorGroups.get(key);
+      if (existing) {
+        existing.count++;
+        if (match[3]) existing.locations.push(match[3]);
+        groupedCount++;
+        continue;
+      }
+      errorGroups.set(key, {
+        message: `${prefix} ${msg}`,
+        locations: match[3] ? [match[3]] : [],
+        count: 1
+      });
+      groupedCount++;
+      continue;
+    }
+    resultLines.push(line);
+  }
+  if (groupedCount < 4 || errorGroups.size === groupedCount) return output;
+  const grouped = [];
+  for (const [, group] of errorGroups) {
+    if (group.count === 1) {
+      grouped.push(
+        group.message + (group.locations.length ? ` at line ${group.locations[0]}` : "")
+      );
+    } else {
+      let line = `${group.message} (\xD7${group.count})`;
+      if (group.locations.length > 0) {
+        line += ` [lines: ${group.locations.join(", ")}]`;
+      }
+      grouped.push(line);
+    }
+  }
+  if (grouped.length > 0) {
+    resultLines.push("");
+    resultLines.push(`\u2500\u2500 Grouped errors/warnings (${groupedCount} \u2192 ${errorGroups.size}) \u2500\u2500`);
+    resultLines.push(...grouped);
+  }
+  return resultLines.join("\n");
+}
 function smartTruncate(output, maxBytes) {
   if (Buffer.byteLength(output) <= maxBytes) return output;
   const lines = output.split("\n");
@@ -21153,11 +21312,11 @@ function smartTruncate(output, maxBytes) {
   const truncatedLines = lines.length - headEnd - (lines.length - tailStart);
   const truncatedBytes = Buffer.byteLength(output) - headBytes - tailBytes;
   const separator = `
-... [${truncatedLines} lines / ${formatSize(truncatedBytes)} truncated \u2014 showing first ${headEnd} + last ${lines.length - tailStart} lines] ...
+... [${truncatedLines} lines / ${formatBytes(truncatedBytes)} truncated \u2014 showing first ${headEnd} + last ${lines.length - tailStart} lines] ...
 `;
   return headLines.join("\n") + separator + tailLines.join("\n");
 }
-function formatSize(bytes) {
+function formatBytes(bytes) {
   if (bytes < 1024) return `${bytes}B`;
   if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)}KB`;
   return `${(bytes / (1024 * 1024)).toFixed(1)}MB`;
@@ -21231,7 +21390,7 @@ var SubprocessExecutor = class {
         plugin.needsShell
       );
     } finally {
-      setTimeout(() => this.cleanupTempDir(tmpDir), 100);
+      setTimeout(() => this.cleanupTempDir(tmpDir), 100).unref();
     }
   }
   /**
@@ -21258,11 +21417,12 @@ var SubprocessExecutor = class {
   spawnAndCapture(cmd, args, cwd, timeout, maxOutput, useShell) {
     return new Promise((resolve2) => {
       const hardCap = this.config.hardCapBytes;
-      let stdoutBuf = Buffer.alloc(0);
-      let stderrBuf = Buffer.alloc(0);
+      const stdoutChunks = [];
+      const stderrChunks = [];
       let totalBytes = 0;
       let killed = false;
       let networkBytes;
+      let resolved = false;
       const proc = spawn(cmd, args, {
         cwd,
         env: { ...this.env, TMPDIR: cwd },
@@ -21278,7 +21438,7 @@ var SubprocessExecutor = class {
           if (proc.pid) killProcessTree(proc.pid);
           return;
         }
-        stdoutBuf = Buffer.concat([stdoutBuf, chunk]);
+        stdoutChunks.push(chunk);
       });
       proc.stderr?.on("data", (chunk) => {
         totalBytes += chunk.length;
@@ -21287,14 +21447,26 @@ var SubprocessExecutor = class {
           if (proc.pid) killProcessTree(proc.pid);
           return;
         }
-        stderrBuf = Buffer.concat([stderrBuf, chunk]);
+        stderrChunks.push(chunk);
       });
       proc.on("error", (err) => {
         debug("Process error:", err.message);
+        if (!resolved) {
+          resolved = true;
+          resolve2({
+            stdout: "",
+            stderr: err.message,
+            exitCode: 1,
+            truncated: false,
+            killed: false
+          });
+        }
       });
       proc.on("close", (code) => {
-        let stdout = stdoutBuf.toString("utf-8");
-        let stderr = stderrBuf.toString("utf-8");
+        if (resolved) return;
+        resolved = true;
+        let stdout = Buffer.concat(stdoutChunks).toString("utf-8");
+        let stderr = Buffer.concat(stderrChunks).toString("utf-8");
         const netMatch = stderr.match(/__CM_NET__:(\d+)/);
         if (netMatch) {
           networkBytes = Number.parseInt(netMatch[1], 10);
@@ -21302,8 +21474,10 @@ var SubprocessExecutor = class {
         }
         if (killed) {
           stdout += `
-[output capped at ${formatSize(hardCap)} \u2014 process killed]`;
+[output capped at ${formatBytes(hardCap)} \u2014 process killed]`;
         }
+        stdout = deduplicateLines(stdout);
+        stdout = groupErrorLines(stdout);
         const truncated = Buffer.byteLength(stdout) > maxOutput;
         if (truncated) {
           stdout = smartTruncate(stdout, maxOutput);
@@ -21333,7 +21507,7 @@ var SubprocessExecutor = class {
   }
 };
 function wrapWithNetworkTracking(code) {
-  const preamble = "let __cm_net=0;const __cm_f=globalThis.fetch;if(__cm_f){globalThis.fetch=async(...a)=>{const r=await __cm_f(...a);try{const cl=r.clone();const b=await cl.arrayBuffer();__cm_net+=b.byteLength}catch{}return r};}";
+  const preamble = "let __cm_net=0;const __cm_f=globalThis.fetch;if(__cm_f){globalThis.fetch=async(...a)=>{const r=await __cm_f(...a);try{const cl=r.headers.get('content-length');if(cl){__cm_net+=parseInt(cl,10)}else{const b=await r.clone().arrayBuffer();__cm_net+=b.byteLength}}catch{}return r};}";
   const epilogue = `
 process.stderr.write('__CM_NET__:'+__cm_net+'\\n');`;
   return `${preamble}
@@ -21341,6 +21515,25 @@ async function __cm_main(){${code}}
 __cm_main().then(()=>{${epilogue}}).catch(e=>{console.error(e);${epilogue}process.exit(1)});`;
 }
 
+// src/network.ts
+function isPrivateHost(hostname2) {
+  const h = hostname2.startsWith("[") && hostname2.endsWith("]") ? hostname2.slice(1, -1) : hostname2;
+  const lower = h.toLowerCase();
+  if (lower === "localhost" || lower === "0.0.0.0") return true;
+  if (/^127\./.test(h)) return true;
+  if (/^10\./.test(h)) return true;
+  if (/^172\.(1[6-9]|2\d|3[01])\./.test(h)) return true;
+  if (/^192\.168\./.test(h)) return true;
+  if (/^169\.254\./.test(h)) return true;
+  if (/^100\.(6[4-9]|[7-9]\d|1[01]\d|12[0-7])\./.test(h)) return true;
+  if (lower === "::1") return true;
+  const mappedMatch = lower.match(/^::ffff:(\d+\.\d+\.\d+\.\d+)$/);
+  if (mappedMatch) return isPrivateHost(mappedMatch[1]);
+  if (/^fe[89ab]/i.test(h)) return true;
+  if (/^f[cd]/i.test(h)) return true;
+  return false;
+}
+
 // src/runtime/index.ts
 import { exec } from "node:child_process";
 import { promisify } from "node:util";
@@ -21372,8 +21565,8 @@ var goPlugin = {
     return [runtime, "run", filePath];
   },
   preprocessCode(code) {
-    if (!code.includes("package ")) {
-      const hasImport = code.includes("import ");
+    if (!/^package\s/m.test(code)) {
+      const hasImport = /^import\s/m.test(code);
       if (hasImport) {
         return `package main
 
@@ -21392,7 +21585,7 @@ _ = fmt.Sprintf("")
   },
   wrapWithFileContent(code, filePath) {
     const escaped = JSON.stringify(filePath);
-    const hasPackage = code.includes("package ");
+    const hasPackage = /^package\s/m.test(code);
     if (hasPackage) {
       return code.replace(
         /(package\s+\w+\n)/,
@@ -21433,8 +21626,9 @@ var javascriptPlugin = {
   },
   wrapWithFileContent(code, filePath) {
     const escaped = JSON.stringify(filePath);
-    return `const FILE_CONTENT_PATH = ${escaped};
-const FILE_CONTENT = require("fs").readFileSync(FILE_CONTENT_PATH, "utf-8");
+    return `const {readFileSync: __cm_readFileSync} = await import("node:fs");
+const FILE_CONTENT_PATH = ${escaped};
+const FILE_CONTENT = __cm_readFileSync(FILE_CONTENT_PATH, "utf-8");
 ${code}`;
   }
 };
@@ -21506,7 +21700,7 @@ ${code}`;
 // src/runtime/languages/r.ts
 var rPlugin = {
   language: "r",
-  runtimeCandidates: ["Rscript", "r"],
+  runtimeCandidates: ["Rscript", "R"],
   fileExtension: ".R",
   buildCommand(runtime, filePath) {
     return [runtime, filePath];
@@ -21549,7 +21743,7 @@ var rustPlugin = {
     return [runtime, srcPath, "-o", binPath];
   },
   preprocessCode(code) {
-    if (!code.includes("fn main")) {
+    if (!/^fn\s+main\s*\(/m.test(code)) {
       return `fn main() {
 ${code}
 }`;
@@ -21562,7 +21756,7 @@ ${code}
 let file_content_path = ${escaped};
 let file_content = fs::read_to_string(file_content_path).unwrap();
 `;
-    if (code.includes("fn main")) {
+    if (/^fn\s+main\s*\(/m.test(code)) {
       return code.replace(/fn main\s*\(\s*\)\s*\{/, `fn main() {
 ${preamble}`);
     }
@@ -21598,8 +21792,9 @@ var typescriptPlugin = {
   },
   wrapWithFileContent(code, filePath) {
     const escaped = JSON.stringify(filePath);
-    return `const FILE_CONTENT_PATH = ${escaped};
-const FILE_CONTENT = require("fs").readFileSync(FILE_CONTENT_PATH, "utf-8");
+    return `const {readFileSync: __cm_readFileSync} = await import("node:fs");
+const FILE_CONTENT_PATH = ${escaped};
+const FILE_CONTENT = __cm_readFileSync(FILE_CONTENT_PATH, "utf-8");
 ${code}`;
   },
   // tsx and ts-node may be .cmd shims on Windows
@@ -21655,6 +21850,16 @@ function hasBun(runtimes) {
 }
 
 // src/stats.ts
+var BAR_WIDTH = 20;
+function asciiBar(ratio, width = BAR_WIDTH) {
+  const filled = Math.round(ratio * width);
+  const empty = width - filled;
+  return `[${"\u2588".repeat(filled)}${"\u2591".repeat(empty)}] ${(ratio * 100).toFixed(0)}%`;
+}
+function tokenCost(tokens) {
+  const cost = tokens / 1e6 * 3;
+  return cost >= 0.01 ? `~$${cost.toFixed(2)}` : "<$0.01";
+}
 var SessionTracker = class {
   stats = {
     calls: {},
@@ -21688,6 +21893,7 @@ var SessionTracker = class {
     const savingsRatio = totalReturned > 0 ? totalProcessed / totalReturned : 1;
     const reductionPct = totalProcessed > 0 ? ((1 - totalReturned / totalProcessed) * 100).toFixed(1) : "0.0";
     const estTokens = Math.round(totalReturned / 4);
+    const estTokensAvoided = Math.round(keptOut / 4);
     const lines = [];
     lines.push("## Session Statistics\n");
     lines.push("| Metric | Value |");
@@ -21697,18 +21903,31 @@ var SessionTracker = class {
     lines.push(`| Total data processed | ${formatBytes(totalProcessed)} |`);
     lines.push(`| Kept in sandbox | ${formatBytes(keptOut)} |`);
     lines.push(`| Context consumed | ${formatBytes(totalReturned)} |`);
-    lines.push(`| Est. tokens | ~${estTokens.toLocaleString()} |`);
+    lines.push(`| Est. tokens used | ~${estTokens.toLocaleString()} (${tokenCost(estTokens)}) |`);
+    lines.push(
+      `| Est. tokens saved | ~${estTokensAvoided.toLocaleString()} (${tokenCost(estTokensAvoided)}) |`
+    );
     lines.push(
       `| **Savings ratio** | **${savingsRatio.toFixed(1)}x** (${reductionPct}% reduction) |`
     );
+    if (totalProcessed > 0) {
+      const savingsBar = asciiBar(keptOut / totalProcessed);
+      lines.push(`
+**Context savings:** ${savingsBar}`);
+      lines.push(
+        `  Sandbox: ${formatBytes(keptOut)} kept out | Context: ${formatBytes(totalReturned)} entered`
+      );
+    }
     if (totalCalls > 0) {
       lines.push("\n## Per-Tool Breakdown\n");
-      lines.push("| Tool | Calls | Context bytes | Est. tokens |");
-      lines.push("|------|-------|--------------|-------------|");
+      const maxBytes = Math.max(...Object.values(snap.bytesReturned));
       for (const [name, calls] of Object.entries(snap.calls)) {
         const bytes = snap.bytesReturned[name] ?? 0;
+        const tokens = Math.round(bytes / 4);
+        const barRatio = maxBytes > 0 ? bytes / maxBytes : 0;
+        const bar = "\u2588".repeat(Math.max(1, Math.round(barRatio * 15)));
         lines.push(
-          `| ${name} | ${calls} | ${formatBytes(bytes)} | ~${Math.round(bytes / 4).toLocaleString()} |`
+          `  ${name.padEnd(16)} ${String(calls).padStart(3)} calls  ${bar} ${formatBytes(bytes)} (~${tokens.toLocaleString()} tok)`
         );
       }
     }
@@ -21719,11 +21938,6 @@ Context-compress kept ${formatBytes(keptOut)} out of context (${reductionPct}% s
     return lines.join("\n");
   }
 };
-function formatBytes(bytes) {
-  if (bytes < 1024) return `${bytes}B`;
-  if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)}KB`;
-  return `${(bytes / (1024 * 1024)).toFixed(1)}MB`;
-}
 
 // src/store.ts
 import { readdirSync, unlinkSync } from "node:fs";
@@ -21754,14 +21968,7 @@ function positionsFromHighlight(highlighted) {
   return positions;
 }
 function stripMarkers(text) {
-  let result = "";
-  for (let i = 0; i < text.length; i++) {
-    const ch = text[i];
-    if (ch !== STX && ch !== ETX) {
-      result += ch;
-    }
-  }
-  return result;
+  return text.replaceAll(STX, "").replaceAll(ETX, "");
 }
 function extractSnippet(highlighted, maxLen = DEFAULT_MAX_LEN) {
   const positions = positionsFromHighlight(highlighted);
@@ -21916,7 +22123,7 @@ var WORD_SPLIT_RE = /[^\p{L}\p{N}_-]+/u;
 function sanitizeQuery(raw) {
   const q = raw.replace(FTS_SPECIAL_RE, " ").replace(FTS_OPERATORS_RE, " ").trim();
   const words = q.split(/\s+/).filter((w) => w.length >= 2).map((w) => `"${w}"`);
-  return words.length > 0 ? words.join(" OR ") : '""';
+  return words.length > 0 ? words.join(" OR ") : "";
 }
 function levenshtein(a, b) {
   if (a.length === 0) return b.length;
@@ -21936,6 +22143,11 @@ function levenshtein(a, b) {
 var ContentStore = class {
   db;
   hasTrigramTable = false;
+  // Cached prepared statements (initialized in initSchema, always available after constructor)
+  insertSourceStmt;
+  insertChunkStmt;
+  vocabCountStmt;
+  vocabInsertStmt;
   constructor(dbPath) {
     const path = dbPath ?? join3(tmpdir2(), `context-compress-${process.pid}.db`);
     this.db = new Database(path);
@@ -21965,6 +22177,14 @@ var ContentStore = class {
 				word TEXT PRIMARY KEY
 			);
 		`);
+    this.insertSourceStmt = this.db.prepare(
+      "INSERT INTO sources (label, chunk_count, code_chunk_count) VALUES (?, ?, ?)"
+    );
+    this.insertChunkStmt = this.db.prepare(
+      "INSERT INTO chunks (title, content, source_id, content_type) VALUES (?, ?, ?, ?)"
+    );
+    this.vocabCountStmt = this.db.prepare("SELECT COUNT(*) as cnt FROM vocabulary");
+    this.vocabInsertStmt = this.db.prepare("INSERT OR IGNORE INTO vocabulary (word) VALUES (?)");
   }
   /** Lazily create trigram table only when porter search returns 0 results */
   ensureTrigramTable() {
@@ -21979,16 +22199,9 @@ var ContentStore = class {
 				tokenize='trigram'
 			);
 		`);
-    const rows = this.db.prepare("SELECT title, content, source_id, content_type FROM chunks").all();
-    const insert = this.db.prepare(
-      "INSERT INTO chunks_trigram (title, content, source_id, content_type) VALUES (?, ?, ?, ?)"
+    this.db.exec(
+      "INSERT INTO chunks_trigram (title, content, source_id, content_type) SELECT title, content, source_id, content_type FROM chunks"
     );
-    const tx = this.db.transaction(() => {
-      for (const row of rows) {
-        insert.run(row.title, row.content, row.source_id, row.content_type);
-      }
-    });
-    tx();
     this.hasTrigramTable = true;
   }
   /**
@@ -21997,12 +22210,8 @@ var ContentStore = class {
   index(content, label) {
     const isMarkdown = HEADING_RE.test(content) || content.includes("```") || content.includes("---");
     const chunks = isMarkdown ? chunkMarkdown(content) : chunkPlainText(content);
-    const insertSource = this.db.prepare(
-      "INSERT INTO sources (label, chunk_count, code_chunk_count) VALUES (?, ?, ?)"
-    );
-    const insertChunk = this.db.prepare(
-      "INSERT INTO chunks (title, content, source_id, content_type) VALUES (?, ?, ?, ?)"
-    );
+    const insertSource = this.insertSourceStmt;
+    const insertChunk = this.insertChunkStmt;
     const insertTrigram = this.hasTrigramTable ? this.db.prepare(
       "INSERT INTO chunks_trigram (title, content, source_id, content_type) VALUES (?, ?, ?, ?)"
     ) : null;
@@ -22034,6 +22243,9 @@ var ContentStore = class {
   search(query, options) {
     const limit = options?.limit ?? 3;
     const sanitized = sanitizeQuery(query);
+    if (!sanitized) {
+      return { query, results: [] };
+    }
     let hits = this.porterSearch(sanitized, options?.source, limit);
     if (hits.length > 0) {
       return { query, results: hits };
@@ -22046,9 +22258,11 @@ var ContentStore = class {
     const corrected = this.fuzzyCorrect(query);
     if (corrected && corrected !== query) {
       const correctedSanitized = sanitizeQuery(corrected);
-      hits = this.porterSearch(correctedSanitized, options?.source, limit);
-      if (hits.length > 0) {
-        return { query, results: hits, corrected };
+      if (correctedSanitized) {
+        hits = this.porterSearch(correctedSanitized, options?.source, limit);
+        if (hits.length > 0) {
+          return { query, results: hits, corrected };
+        }
       }
     }
     return { query, results: [] };
@@ -22129,7 +22343,7 @@ var ContentStore = class {
       const maxDist = word.length <= 4 ? 1 : word.length <= 12 ? 2 : 3;
       const minLen = word.length - maxDist;
       const maxLen = word.length + maxDist;
-      const candidates = this.db.prepare("SELECT word FROM vocabulary WHERE length(word) BETWEEN ? AND ?").all(minLen, maxLen);
+      const candidates = this.db.prepare("SELECT word FROM vocabulary WHERE length(word) BETWEEN ? AND ? LIMIT 500").all(minLen, maxLen);
       let bestWord = word;
       let bestDist = maxDist + 1;
       for (const { word: candidate } of candidates) {
@@ -22148,11 +22362,11 @@ var ContentStore = class {
    * Update vocabulary table from content (bounded to MAX_VOCABULARY).
    */
   updateVocabulary(content) {
-    const currentCount = this.db.prepare("SELECT COUNT(*) as cnt FROM vocabulary").get().cnt;
+    const currentCount = this.vocabCountStmt.get().cnt;
     if (currentCount >= MAX_VOCABULARY) return;
     const words = content.split(WORD_SPLIT_RE).filter((w) => w.length >= 3 && !STOPWORDS.has(w.toLowerCase()));
     const unique = new Set(words.map((w) => w.toLowerCase()));
-    const insert = this.db.prepare("INSERT OR IGNORE INTO vocabulary (word) VALUES (?)");
+    const insert = this.vocabInsertStmt;
     let added = 0;
     for (const word of unique) {
       if (currentCount + added >= MAX_VOCABULARY) break;
@@ -22169,7 +22383,7 @@ var ContentStore = class {
     ).get(...sourceId ? [sourceId] : []).cnt;
     if (totalChunks === 0) return [];
     const filter = sourceId ? " WHERE source_id = ?" : "";
-    const stmt = this.db.prepare(`SELECT content FROM chunks${filter}`);
+    const stmt = this.db.prepare(`SELECT content FROM chunks${filter} LIMIT 500`);
     const rows = sourceId ? stmt.all(sourceId) : stmt.all();
     const docFreq = /* @__PURE__ */ new Map();
     for (const row of rows) {
@@ -22194,6 +22408,21 @@ var ContentStore = class {
     scored.sort((a, b) => b.score - a.score);
     return scored.slice(0, 40).map((s) => s.word);
   }
+  /**
+   * List all indexed sources with metadata.
+   */
+  listSources() {
+    const rows = this.db.prepare(
+      "SELECT id, label, chunk_count, code_chunk_count, indexed_at FROM sources ORDER BY indexed_at DESC"
+    ).all();
+    return rows.map((row) => ({
+      id: row.id,
+      label: row.label,
+      chunkCount: row.chunk_count,
+      codeChunks: row.code_chunk_count,
+      indexedAt: row.indexed_at
+    }));
+  }
   /**
    * Get store statistics.
    */
@@ -22319,8 +22548,8 @@ function cleanupStaleDbs() {
   return cleaned;
 }
 
-// src/server.ts
-var LANGUAGES = [
+// src/types.ts
+var ALL_LANGUAGES = [
   "javascript",
   "typescript",
   "python",
@@ -22333,6 +22562,14 @@ var LANGUAGES = [
   "r",
   "elixir"
 ];
+
+// src/server.ts
+var LANGUAGE_ENUM = ALL_LANGUAGES;
+var projectDir = process.env.CLAUDE_PROJECT_DIR ?? process.cwd();
+function isWithinProject(absPath) {
+  const normalized = resolve(absPath);
+  return normalized === projectDir || normalized.startsWith(`${projectDir}/`);
+}
 function getVersion() {
   try {
     const __dirname = dirname(fileURLToPath(import.meta.url));
@@ -22343,6 +22580,18 @@ function getVersion() {
     return "1.0.0";
   }
 }
+function compactLabel(normal, level) {
+  if (level === "ultra") {
+    return normal.replace(/\*\*/g, "").replace(/Use search\(queries: \[\.\.\.]\) to retrieve.*$/gm, "\u2192 search() for more").replace(/Searchable terms: .+$/gm, "");
+  }
+  if (level === "compact") {
+    return normal.replace(
+      /Use search\(queries: \[\.\.\.]\) to retrieve full content of any section\./,
+      "\u2192 search() for details"
+    );
+  }
+  return normal;
+}
 async function createServer(config3) {
   const version2 = getVersion();
   debug("Version:", version2);
@@ -22353,6 +22602,15 @@ async function createServer(config3) {
   const executor = new SubprocessExecutor(runtimes, config3);
   const store = new ContentStore();
   const tracker = new SessionTracker();
+  const shutdown = () => {
+    try {
+      store.close();
+    } catch {
+    }
+  };
+  process.on("SIGINT", shutdown);
+  process.on("SIGTERM", shutdown);
+  process.on("beforeExit", shutdown);
   const searchCalls = [];
   const server2 = new McpServer({
     name: "context-compress",
@@ -22360,11 +22618,11 @@ async function createServer(config3) {
   });
   server2.tool(
     "execute",
-    `Execute code in a sandboxed subprocess. Only stdout enters context \u2014 raw data stays in the subprocess. Use instead of bash/cat when output would exceed 20 lines. ${bunDetected ? "(Bun detected \u2014 JS/TS runs 3-5x faster) " : ""}Available: ${LANGUAGES.join(", ")}.
+    `Execute code in a sandboxed subprocess. Only stdout enters context \u2014 raw data stays in the subprocess. Use instead of bash/cat when output would exceed 20 lines. ${bunDetected ? "(Bun detected \u2014 JS/TS runs 3-5x faster) " : ""}Available: ${ALL_LANGUAGES.join(", ")}.
 
 PREFER THIS OVER BASH for: API calls (gh, curl, aws), test runners (npm test, pytest), git queries (git log, git diff), data processing, and ANY CLI command that may produce large output. Bash should only be used for file mutations, git writes, and navigation.`,
     {
-      language: external_exports.enum(LANGUAGES).describe("Runtime language"),
+      language: external_exports.enum(LANGUAGE_ENUM).describe("Runtime language"),
       code: external_exports.string().describe(
         "Source code to execute. Use console.log (JS/TS), print (Python/Ruby/Perl/R), echo (Shell), echo (PHP), fmt.Println (Go), or IO.puts (Elixir) to output a summary to context."
       ),
@@ -22399,13 +22657,13 @@ ${result.stderr}`;
           filtered += `  - **${hit.title}**: ${hit.snippet.slice(0, 200)}
 `;
         }
-        if (terms.length > 0) {
+        if (terms.length > 0 && config3.compressionLevel !== "ultra") {
           filtered += `
 Searchable terms: ${terms.join(", ")}
 `;
         }
         filtered += "\nUse search(queries: [...]) to retrieve full content of any section.";
-        output = filtered;
+        output = compactLabel(filtered, config3.compressionLevel);
       }
       const responseBytes = Buffer.byteLength(output);
       tracker.trackCall("execute", responseBytes);
@@ -22417,7 +22675,7 @@ Searchable terms: ${terms.join(", ")}
     "Read a file and process it without loading contents into context. The file is read into a FILE_CONTENT variable inside the sandbox. Only your printed summary enters context.\n\nPREFER THIS OVER Read/cat for: log files, data files (CSV, JSON, XML), large source files for analysis, and any file where you need to extract specific information rather than read the entire content.",
     {
       path: external_exports.string().describe("Absolute file path or relative to project root"),
-      language: external_exports.enum(LANGUAGES).describe("Runtime language"),
+      language: external_exports.enum(LANGUAGE_ENUM).describe("Runtime language"),
       code: external_exports.string().describe(
         "Code to process FILE_CONTENT. Print summary via console.log/print/echo/IO.puts."
       ),
@@ -22425,7 +22683,17 @@ Searchable terms: ${terms.join(", ")}
       timeout: external_exports.number().default(3e4).describe("Max execution time in ms")
     },
     async ({ path: filePath, language, code, intent, timeout }) => {
-      const absPath = resolve(process.env.CLAUDE_PROJECT_DIR ?? process.cwd(), filePath);
+      const absPath = resolve(projectDir, filePath);
+      if (!isWithinProject(absPath)) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: `Error: path "${filePath}" is outside the project directory`
+            }
+          ]
+        };
+      }
       const result = await executor.executeFile({
         language,
         code,
@@ -22453,13 +22721,13 @@ ${result.stderr}`;
           filtered += `  - **${hit.title}**: ${hit.snippet.slice(0, 200)}
 `;
         }
-        if (terms.length > 0) {
+        if (terms.length > 0 && config3.compressionLevel !== "ultra") {
           filtered += `
 Searchable terms: ${terms.join(", ")}
 `;
         }
         filtered += "\nUse search(queries: [...]) to retrieve full content of any section.";
-        output = filtered;
+        output = compactLabel(filtered, config3.compressionLevel);
       }
       const responseBytes = Buffer.byteLength(output);
       tracker.trackCall("execute_file", responseBytes);
@@ -22478,7 +22746,28 @@ Searchable terms: ${terms.join(", ")}
       let text;
       let label = source ?? "indexed content";
       if (filePath) {
-        const absPath = resolve(process.env.CLAUDE_PROJECT_DIR ?? process.cwd(), filePath);
+        const absPath = resolve(projectDir, filePath);
+        if (!isWithinProject(absPath)) {
+          return {
+            content: [
+              {
+                type: "text",
+                text: `Error: path "${filePath}" is outside the project directory`
+              }
+            ]
+          };
+        }
+        const fileStat = statSync(absPath);
+        if (fileStat.size > 50 * 1024 * 1024) {
+          return {
+            content: [
+              {
+                type: "text",
+                text: `Error: file "${filePath}" is too large (${(fileStat.size / 1024 / 1024).toFixed(1)}MB). Max 50MB.`
+              }
+            ]
+          };
+        }
         text = readFileSync2(absPath, "utf-8");
         label = source ?? filePath;
       } else if (content) {
@@ -22561,6 +22850,25 @@ ${hit.snippet}
       source: external_exports.string().optional().describe("Label for the indexed content")
     },
     async ({ url, source }) => {
+      try {
+        const parsed = new URL(url);
+        if (parsed.protocol !== "http:" && parsed.protocol !== "https:") {
+          return {
+            content: [{ type: "text", text: "Error: only http/https URLs are allowed" }]
+          };
+        }
+        if (isPrivateHost(parsed.hostname)) {
+          return {
+            content: [
+              { type: "text", text: "Error: internal/private URLs are not allowed" }
+            ]
+          };
+        }
+      } catch {
+        return {
+          content: [{ type: "text", text: `Error: invalid URL "${url}"` }]
+        };
+      }
       const label = source ?? url;
       const fetchCode = buildFetchCode(url);
       const result = await executor.execute({
@@ -22690,7 +22998,7 @@ Searchable terms: ${terms.join(", ")}`;
   );
   server2.tool(
     "stats",
-    "Returns context consumption statistics for the current session. Shows total bytes returned to context, breakdown by tool, call counts, estimated token usage, and context savings ratio.",
+    "Returns context consumption statistics for the current session. Shows total bytes returned to context, breakdown by tool, call counts, estimated token usage, context savings ratio, and visual charts.",
     {},
     async () => {
       const report = tracker.formatReport();
@@ -22698,6 +23006,77 @@ Searchable terms: ${terms.join(", ")}`;
       return { content: [{ type: "text", text: report }] };
     }
   );
+  server2.tool(
+    "discover",
+    "Shows what's in the knowledge base and suggests optimization opportunities. Lists all indexed sources, chunk counts, searchable terms, and recommends next actions. Use this to understand what data is available for search.",
+    {},
+    async () => {
+      const storeStats = store.getStats();
+      const snap = tracker.getSnapshot();
+      const lines = [];
+      lines.push("## Knowledge Base Discovery\n");
+      if (storeStats.totalSources === 0) {
+        lines.push("No content indexed yet. Use these tools to build the knowledge base:\n");
+        lines.push("- `batch_execute` \u2014 run commands and auto-index output");
+        lines.push("- `execute` with `intent` \u2014 auto-indexes large output");
+        lines.push("- `index` \u2014 index documentation or files");
+        lines.push("- `fetch_and_index` \u2014 fetch and index web pages");
+      } else {
+        lines.push("| Metric | Value |");
+        lines.push("|--------|-------|");
+        lines.push(`| Indexed sources | ${storeStats.totalSources} |`);
+        lines.push(`| Total chunks | ${storeStats.totalChunks} |`);
+        lines.push(`| Vocabulary size | ${storeStats.vocabularySize} |`);
+        lines.push(
+          `| Trigram index | ${storeStats.hasTrigramTable ? "active" : "lazy (not yet needed)"} |`
+        );
+        const sources = store.listSources();
+        if (sources.length > 0) {
+          lines.push("\n### Indexed Sources\n");
+          for (const src of sources) {
+            lines.push(
+              `- **${src.label}** \u2014 ${src.chunkCount} chunks${src.codeChunks > 0 ? ` (${src.codeChunks} with code)` : ""}`
+            );
+          }
+        }
+        const terms = store.getDistinctiveTerms();
+        if (terms.length > 0) {
+          lines.push("\n### Top Searchable Terms\n");
+          lines.push(terms.slice(0, 20).join(", "));
+        }
+      }
+      lines.push("\n### Optimization Suggestions\n");
+      const totalCalls = Object.values(snap.calls).reduce((a, b) => a + b, 0);
+      if (totalCalls === 0) {
+        lines.push("- Start by using `batch_execute` to run multiple commands at once");
+      } else {
+        const searchCalls2 = snap.calls.search ?? 0;
+        const executeCalls = snap.calls.execute ?? 0;
+        const batchCalls = snap.calls.batch_execute ?? 0;
+        if (executeCalls > 3 && batchCalls === 0) {
+          lines.push(
+            "- **Use batch_execute** \u2014 you've made multiple execute calls that could be batched into one"
+          );
+        }
+        if (searchCalls2 > 5) {
+          lines.push("- **Batch your searches** \u2014 pass multiple queries in a single search() call");
+        }
+        if (storeStats.totalChunks > 50) {
+          lines.push(
+            "- **Use source filtering** \u2014 scope searches with `source` parameter for faster, targeted results"
+          );
+        }
+        if (storeStats.totalSources === 0 && totalCalls > 2) {
+          lines.push(
+            "- **Index more content** \u2014 use `intent` parameter in execute calls to auto-index large output"
+          );
+        }
+      }
+      const output = lines.join("\n");
+      tracker.trackCall("discover", Buffer.byteLength(output));
+      return { content: [{ type: "text", text: output }] };
+    }
+  );
   return {
     async start() {
       const transport = new StdioServerTransport();