vibe-code-explainer 0.3.4 → 0.3.6

package/dist/hooks/post-tool.js

@@ -1,12 +1,14 @@
 #!/usr/bin/env node
 import {
   buildClaudePrompt,
-  callOllama
-} from "../chunk-ABPTVWQ3.js";
+  callOllama,
+  parseResponse,
+  truncateText
+} from "../chunk-R5H62KGX.js";
 import {
   DEFAULT_CONFIG,
   loadConfig
-} from "../chunk-RK7ZFN4W.js";
+} from "../chunk-LWASVVBV.js";
 import {
   cleanStaleSessionFiles,
   formatDriftAlert,
@@ -15,10 +17,11 @@ import {
   formatSkipNotice,
   getCached,
   getRecentSummaries,
+  isSafeSessionId,
   readSession,
   recordEntry,
   setCached
-} from "../chunk-XW3S5GNV.js";
+} from "../chunk-VJN7Y4SI.js";
 import "../chunk-7OCVIDC7.js";
 
 // src/hooks/post-tool.ts
@@ -26,91 +29,6 @@ import { join } from "path";
 
 // src/engines/claude.ts
 import { execFile } from "child_process";
-function extractBalancedObject(text, startIdx) {
-  let depth = 0;
-  let inString = false;
-  let escape = false;
-  for (let i = startIdx; i < text.length; i++) {
-    const ch = text[i];
-    if (escape) {
-      escape = false;
-      continue;
-    }
-    if (ch === "\\") {
-      escape = true;
-      continue;
-    }
-    if (ch === '"') {
-      inString = !inString;
-      continue;
-    }
-    if (inString) continue;
-    if (ch === "{") depth++;
-    else if (ch === "}") {
-      depth--;
-      if (depth === 0) return text.slice(startIdx, i + 1);
-    }
-  }
-  return null;
-}
-function extractJson(text) {
-  const trimmed = text.trim();
-  if (trimmed.startsWith("{") && trimmed.endsWith("}")) return trimmed;
-  const fenceOpen = trimmed.match(/```(?:json)?\s*\n?/);
-  if (fenceOpen) {
-    const afterOpen = trimmed.slice(fenceOpen.index + fenceOpen[0].length);
-    const closingIdx = afterOpen.indexOf("```");
-    const inner = closingIdx !== -1 ? afterOpen.slice(0, closingIdx) : afterOpen;
-    const innerTrimmed = inner.trim();
-    if (innerTrimmed.startsWith("{")) {
-      const lastBrace = innerTrimmed.lastIndexOf("}");
-      if (lastBrace !== -1) return innerTrimmed.slice(0, lastBrace + 1);
-    }
-  }
-  const firstOpen = trimmed.indexOf("{");
-  if (firstOpen !== -1) {
-    const balanced = extractBalancedObject(trimmed, firstOpen);
-    if (balanced) return balanced;
-    const lastClose = trimmed.lastIndexOf("}");
-    if (lastClose > firstOpen) return trimmed.slice(firstOpen, lastClose + 1);
-  }
-  return null;
-}
-function coerceString(v) {
-  return typeof v === "string" ? v : "";
-}
-function coerceDeepDive(v) {
-  if (!Array.isArray(v)) return [];
-  return v.filter((it) => typeof it === "object" && it !== null).map((it) => ({
-    term: coerceString(it.term),
-    explanation: coerceString(it.explanation)
-  })).filter((it) => it.term.length > 0);
-}
-function parseResponse(rawText) {
-  const json = extractJson(rawText);
-  if (!json) return null;
-  try {
-    const parsed = JSON.parse(json);
-    const risk = coerceString(parsed.risk);
-    if (!["none", "low", "medium", "high"].includes(risk)) return null;
-    return {
-      impact: coerceString(parsed.impact),
-      howItWorks: coerceString(parsed.howItWorks),
-      why: coerceString(parsed.why),
-      deepDive: coerceDeepDive(parsed.deepDive),
-      isSamePattern: parsed.isSamePattern === true,
-      samePatternNote: coerceString(parsed.samePatternNote),
-      risk,
-      riskReason: coerceString(parsed.riskReason)
-    };
-  } catch {
-    return null;
-  }
-}
-function truncateText(text, max) {
-  if (text.length <= max) return text;
-  return text.slice(0, max) + "...";
-}
 function runClaude(prompt, timeoutMs) {
   return new Promise((resolve, reject) => {
     const child = execFile(
@@ -145,14 +63,23 @@ function runClaude(prompt, timeoutMs) {
   });
 }
 async function callClaude(inputs) {
-  const prompt = buildClaudePrompt(inputs.config.detailLevel, {
-    filePath: inputs.filePath,
-    diff: inputs.diff,
-    userPrompt: inputs.userPrompt,
-    language: inputs.config.language,
-    learnerLevel: inputs.config.learnerLevel,
-    recentSummaries: inputs.recentSummaries
-  });
+  let prompt;
+  try {
+    prompt = buildClaudePrompt(inputs.config.detailLevel, {
+      filePath: inputs.filePath,
+      diff: inputs.diff,
+      language: inputs.config.language,
+      learnerLevel: inputs.config.learnerLevel,
+      recentSummaries: inputs.recentSummaries
+    });
+  } catch (err) {
+    return {
+      kind: "error",
+      problem: "Failed to build Claude prompt",
+      cause: err.message || String(err),
+      fix: "Check detailLevel/learnerLevel/language values via 'npx vibe-code-explainer config'"
+    };
+  }
   try {
     const result = await runClaude(prompt, inputs.config.skipIfSlowMs);
     if (result.code !== 0) {
@@ -321,6 +248,20 @@ function extractNewFileDiff(filePath, cwd) {
   }
   return { kind: "empty" };
 }
+function looksBinary(buf) {
+  const sample = buf.length > 8192 ? buf.subarray(0, 8192) : buf;
+  if (sample.length === 0) return false;
+  if (sample.indexOf(0) !== -1) return true;
+  let nonPrint = 0;
+  for (let i = 0; i < sample.length; i++) {
+    const b = sample[i];
+    if (b === 9 || b === 10 || b === 13 || b >= 32 && b <= 126 || b >= 128) {
+      continue;
+    }
+    nonPrint++;
+  }
+  return nonPrint / sample.length > 0.3;
+}
 function readFileAsNewDiff(filePath) {
   if (!existsSync(filePath)) {
     return { kind: "skip", reason: `file not found: ${filePath}` };
@@ -330,13 +271,17 @@ function readFileAsNewDiff(filePath) {
     if (stat.size > 2 * 1024 * 1024) {
       return { kind: "skip", reason: `file too large (${Math.round(stat.size / 1024)}KB)` };
     }
-    const raw = readFileSync(filePath, "utf-8");
-    if (!raw.trim()) {
+    const buf = readFileSync(filePath);
+    if (buf.length === 0) {
       return { kind: "empty" };
     }
-    if (raw.includes("\0")) {
+    if (looksBinary(buf)) {
       return { kind: "binary", message: `Binary file created: ${filePath}` };
     }
+    const raw = buf.toString("utf-8");
+    if (!raw.trim()) {
+      return { kind: "empty" };
+    }
     const withMarkers = raw.split("\n").map((l) => `+ ${l}`).join("\n");
     const diff = `--- /dev/null
 +++ b/${filePath}
@@ -393,7 +338,32 @@ var MUTATING_COMMANDS = /* @__PURE__ */ new Set([
   "chown",
   "ln",
   "touch",
-  "dd"
+  "dd",
+  "tee",
+  "install",
+  "truncate",
+  "shred",
+  "rsync",
+  "scp",
+  "sftp",
+  "mount",
+  "umount",
+  "kill",
+  "killall",
+  "pkill",
+  "crontab",
+  "useradd",
+  "userdel",
+  "usermod",
+  "groupadd",
+  "groupdel",
+  "passwd",
+  "chpasswd",
+  "visudo",
+  "systemctl",
+  "service",
+  "launchctl"
+  // Note: brew is in CONTEXTUAL_COMMANDS (finer-grained control); do not add here.
 ]);
 var CONTEXTUAL_COMMANDS = {
   npm: /\b(install|add|remove|uninstall|update|ci|link|unlink|init|publish)\b/,
@@ -480,9 +450,12 @@ function subCommandShouldCapture(subCmd) {
     const rest = tokens.slice(idx + 1).join(" ");
     return contextPattern.test(rest);
   }
-  return false;
+  return true;
 }
-function shouldCaptureBash(command) {
+function shouldCaptureBash(command, capturePatterns = []) {
+  if (capturePatterns.length > 0 && capturePatterns.some((p) => command.includes(p))) {
+    return true;
+  }
   const parts = splitCommandChain(command);
   return parts.some((p) => subCommandShouldCapture(p));
 }
@@ -536,12 +509,11 @@ function shouldAlertDrift(entries) {
   const unrelatedFiles = Array.from(
     new Set(entries.filter((e) => e.unrelated).map((e) => e.file))
   );
-  const shouldAlert = unrelatedFiles.length > 0 && unrelatedFiles.length % DRIFT_ALERT_THRESHOLD === 0 && entries.filter((e) => e.unrelated).length === entries.filter((e) => e.unrelated).length;
   const lastEntry = entries[entries.length - 1];
   const lastWasUnrelated = lastEntry?.unrelated ?? false;
-  const crossedThreshold = lastWasUnrelated && unrelatedFiles.length % DRIFT_ALERT_THRESHOLD === 0;
+  const shouldAlert = lastWasUnrelated && unrelatedFiles.length === DRIFT_ALERT_THRESHOLD;
   return {
-    shouldAlert: crossedThreshold && shouldAlert,
+    shouldAlert,
     totalFiles: uniqueFiles.length,
     unrelatedFiles
   };
@@ -553,11 +525,14 @@ function addOutput(text) {
   output.push(text);
 }
 function safeExit() {
-  if (output.length > 0) {
-    const systemMessage = "\n" + output.join("\n");
-    process.stdout.write(JSON.stringify({ systemMessage }) + "\n");
+  if (output.length === 0) {
+    process.exit(0);
   }
-  process.exit(0);
+  const systemMessage = "\n" + output.join("\n");
+  const payload = JSON.stringify({ systemMessage }) + "\n";
+  process.stdout.write(payload, () => process.exit(0));
+  setTimeout(() => process.exit(0), 500);
+  throw new Error("unreachable");
 }
 async function readStdin() {
   return new Promise((resolve) => {
@@ -574,10 +549,11 @@ async function readStdin() {
 function parsePayload(raw) {
   try {
     const parsed = JSON.parse(raw);
-    if (typeof parsed.session_id === "string" && typeof parsed.tool_name === "string") {
-      return parsed;
-    }
-    return null;
+    if (typeof parsed !== "object" || parsed === null) return null;
+    if (!isSafeSessionId(parsed.session_id)) return null;
+    if (typeof parsed.tool_name !== "string") return null;
+    if (typeof parsed.tool_input !== "object" || parsed.tool_input === null) return null;
+    return parsed;
   } catch {
     return null;
   }
@@ -596,14 +572,48 @@ function isHookEnabled(toolName, config) {
   if (lower === "bash") return config.hooks.bash;
   return false;
 }
-async function runEngine(filePath, diff, config, userPrompt, recentSummaries, signal) {
+async function runEngine(filePath, diff, config, recentSummaries, signal) {
   if (signal.aborted) {
     return { kind: "skip", reason: "interrupted by user" };
   }
   if (config.engine === "ollama") {
     return callOllama({ filePath, diff, config, recentSummaries });
   }
-  return callClaude({ filePath, diff, config, userPrompt, recentSummaries });
+  return callClaude({ filePath, diff, config, recentSummaries });
+}
+function buildEditWriteDiff(payload, config, cwd) {
+  const lowerTool = payload.tool_name.toLowerCase();
+  const input = payload.tool_input;
+  const target = input.file_path ?? input.filePath;
+  if (!target) {
+    safeExit();
+  }
+  const filePath = target;
+  if (isExcluded(filePath, config.exclude)) {
+    safeExit();
+  }
+  let result;
+  if (lowerTool === "edit") {
+    const oldStr = input.old_string ?? input.oldString ?? "";
+    const newStr = input.new_string ?? input.newString ?? "";
+    result = oldStr || newStr ? buildDiffFromEdit(filePath, oldStr, newStr) : extractEditDiff(filePath, cwd);
+  } else if (lowerTool === "multiedit") {
+    result = input.edits && input.edits.length > 0 ? buildDiffFromMultiEdit(filePath, input.edits) : extractEditDiff(filePath, cwd);
+  } else {
+    result = extractNewFileDiff(filePath, cwd);
+  }
+  if (result.kind === "empty") {
+    safeExit();
+  }
+  if (result.kind === "skip") {
+    addOutput(formatSkipNotice(result.reason));
+    safeExit();
+  }
+  if (result.kind === "binary") {
+    addOutput(formatSkipNotice(result.message));
+    safeExit();
+  }
+  return { filePath, diff: result.content };
 }
 async function main() {
   const controller = new AbortController();
@@ -625,48 +635,20 @@ async function main() {
   let diff;
   const lowerTool = payload.tool_name.toLowerCase();
   if (lowerTool === "edit" || lowerTool === "multiedit" || lowerTool === "write") {
-    const input = payload.tool_input;
-    const target = input.file_path ?? input.filePath;
+    const target = buildEditWriteDiff(payload, config, cwd);
     if (!target) safeExit();
-    filePath = target;
-    if (isExcluded(filePath, config.exclude)) safeExit();
-    let result2;
-    if (lowerTool === "edit") {
-      const oldStr = input.old_string ?? input.oldString ?? "";
-      const newStr = input.new_string ?? input.newString ?? "";
-      if (oldStr || newStr) {
-        result2 = buildDiffFromEdit(filePath, oldStr, newStr);
-      } else {
-        result2 = extractEditDiff(filePath, cwd);
-      }
-    } else if (lowerTool === "multiedit") {
-      if (input.edits && input.edits.length > 0) {
-        result2 = buildDiffFromMultiEdit(filePath, input.edits);
-      } else {
-        result2 = extractEditDiff(filePath, cwd);
-      }
-    } else {
-      result2 = extractNewFileDiff(filePath, cwd);
-    }
-    if (result2.kind === "empty") safeExit();
-    if (result2.kind === "skip") {
-      addOutput(formatSkipNotice(result2.reason));
-      safeExit();
-    }
-    if (result2.kind === "binary") {
-      addOutput(formatSkipNotice(result2.message));
-      safeExit();
-    }
-    diff = result2.content;
+    ({ filePath, diff } = target);
   } else if (lowerTool === "bash") {
     const input = payload.tool_input;
     const command = input.command ?? "";
-    if (!command || !shouldCaptureBash(command)) safeExit();
+    if (!command || !shouldCaptureBash(command, config.bashFilter.capturePatterns)) safeExit();
     filePath = "<bash command>";
     diff = command;
   } else {
     safeExit();
   }
+  const isBash = filePath === "<bash command>";
+  const priorEntries = isBash ? [] : readSession(payload.session_id);
   const cacheKey = `${filePath}
 ${diff}`;
   const cached = getCached(payload.session_id, cacheKey);
@@ -674,15 +656,8 @@ ${diff}`;
   if (cached) {
     result = cached;
   } else {
-    const recentSummaries = getRecentSummaries(payload.session_id, 3);
-    const outcome = await runEngine(
-      filePath,
-      diff,
-      config,
-      void 0,
-      recentSummaries,
-      controller.signal
-    );
+    const recentSummaries = getRecentSummaries(payload.session_id, 3, priorEntries);
+    const outcome = await runEngine(filePath, diff, config, recentSummaries, controller.signal);
     if (outcome.kind === "skip") {
       addOutput(formatSkipNotice(outcome.reason));
       safeExit();
@@ -695,8 +670,7 @@ ${diff}`;
     setCached(payload.session_id, cacheKey, result);
   }
   let driftReason;
-  if (filePath !== "<bash command>") {
-    const priorEntries = readSession(payload.session_id);
+  if (!isBash) {
     const analysis = analyzeDrift(filePath, priorEntries);
     if (analysis.isUnrelated) {
       driftReason = analysis.reason;
@@ -718,8 +692,15 @@ ${diff}`;
     summary: summaryForTracking,
     unrelated: !!driftReason
   });
-  const updated = readSession(payload.session_id);
-  const driftCheck = shouldAlertDrift(updated);
+  const entryJustRecorded = {
+    file: filePath,
+    timestamp: Date.now(),
+    risk: result.risk,
+    summary: summaryForTracking,
+    unrelated: !!driftReason
+  };
+  const updatedEntries = [...priorEntries, entryJustRecorded];
+  const driftCheck = shouldAlertDrift(updatedEntries);
   if (driftCheck.shouldAlert) {
     addOutput(formatDriftAlert(driftCheck.totalFiles, driftCheck.unrelatedFiles, void 0, config.language));
   }