@vibecheckai/cli 3.5.0 → 3.5.2

package/bin/runners/lib/engine/enforcement-extractor.js

@@ -0,0 +1,100 @@
+// bin/runners/lib/engine/enforcement-extractor.js
+// Optimized enforcement extraction using RepoIndex
+
+const path = require("path");
+const fs = require("fs");
+
+const ENFORCE_HINTS = [
+  "enforceFeature",
+  "enforceLimit",
+  "getEntitlements",
+  "requirePlan",
+  "requireTier",
+  "requireSubscription",
+  "checkAccess",
+  "entitlements",
+  "plan",
+  "tier",
+  "subscription",
+  "stripe",
+  "credits"
+];
+
+function looksPaidSurface(routePath) {
+  const p = String(routePath || "");
+  return (
+    p.includes("/api/ship") ||
+    p.includes("/api/verdict") ||
+    p.includes("/api/fix") ||
+    p.includes("/api/autopilot") ||
+    p.includes("/api/missions") ||
+    p.includes("/api/pr") ||
+    p.includes("/api/credits") ||
+    p.includes("/api/billing") ||
+    p.includes("/api/stripe") ||
+    p.includes("/api/entitlements")
+  );
+}
+
+/**
+ * Check if handler has enforcement signal using RepoIndex
+ * @param {import('./repo-index').RepoIndex} index
+ * @param {string} handlerRel
+ * @returns {{ ok: boolean, hits: string[] }}
+ */
+function handlerHasEnforcementSignalV2(index, handlerRel) {
+  const fileAbs = path.join(index.repoRoot, handlerRel);
+  
+  // Try to get content from index first (fast path)
+  let content = index.getContent(fileAbs);
+  
+  // Fall back to direct read if not in index
+  if (!content) {
+    try {
+      content = fs.readFileSync(fileAbs, "utf8");
+    } catch {
+      return { ok: false, hits: [] };
+    }
+  }
+
+  const hits = ENFORCE_HINTS.filter(k => content.includes(k));
+  return { ok: hits.length > 0, hits };
+}
+
+/**
+ * Build enforcement truth using RepoIndex (optimized)
+ * @param {import('./repo-index').RepoIndex} index
+ * @param {Array} serverRoutes
+ * @returns {Object}
+ */
+function buildEnforcementTruthV2(index, serverRoutes) {
+  const checked = [];
+  const routeList = serverRoutes || [];
+
+  for (const r of routeList) {
+    if (!r.handler) continue;
+    if (!looksPaidSurface(r.path)) continue;
+
+    const res = handlerHasEnforcementSignalV2(index, r.handler);
+    checked.push({
+      method: r.method,
+      path: r.path,
+      handler: r.handler,
+      enforced: res.ok,
+      hits: res.hits
+    });
+  }
+
+  return {
+    checkedCount: checked.length,
+    enforcedCount: checked.filter(x => x.enforced).length,
+    missingCount: checked.filter(x => !x.enforced).length,
+    checks: checked
+  };
+}
+
+module.exports = {
+  buildEnforcementTruthV2,
+  ENFORCE_HINTS,
+  looksPaidSurface,
+};

package/bin/runners/lib/engine/env-extractor.js

@@ -0,0 +1,207 @@
+// bin/runners/lib/engine/env-extractor.js
+// Optimized env var extraction using RepoIndex
+
+const path = require("path");
+const fs = require("fs");
+const crypto = require("crypto");
+const traverse = require("@babel/traverse").default;
+const t = require("@babel/types");
+const { globalASTCache } = require("./ast-cache");
+
+function sha256(text) {
+  return "sha256:" + crypto.createHash("sha256").update(text).digest("hex");
+}
+
+function isEnvName(s) {
+  return typeof s === "string" && /^[A-Z0-9_]+$/.test(s);
+}
+
+function extractEnvFromMemberExpr(node) {
+  if (!t.isMemberExpression(node)) return null;
+
+  // process.env.NAME
+  if (t.isMemberExpression(node.object) &&
+      t.isIdentifier(node.object.object, { name: "process" }) &&
+      t.isIdentifier(node.object.property, { name: "env" }) &&
+      t.isIdentifier(node.property)) {
+    return node.property.name;
+  }
+
+  // process.env["NAME"]
+  if (t.isMemberExpression(node.object) &&
+      t.isIdentifier(node.object.object, { name: "process" }) &&
+      t.isIdentifier(node.object.property, { name: "env" }) &&
+      t.isStringLiteral(node.property)) {
+    return node.property.value;
+  }
+
+  // import.meta.env.NAME
+  if (t.isMemberExpression(node.object) &&
+      t.isMemberExpression(node.object.object) &&
+      t.isIdentifier(node.object.object.object, { name: "import" }) &&
+      t.isIdentifier(node.object.object.property, { name: "meta" }) &&
+      t.isIdentifier(node.object.property, { name: "env" }) &&
+      t.isIdentifier(node.property)) {
+    return node.property.name;
+  }
+
+  return null;
+}
+
+function detectDefaultSignals(parentPath) {
+  const p = parentPath?.parentPath;
+  if (!p) return { hasDefault: false };
+
+  if (p.isLogicalExpression() && (p.node.operator === "||" || p.node.operator === "??")) {
+    return { hasDefault: true };
+  }
+  if (p.isConditionalExpression()) return { hasDefault: true };
+  if (p.isAssignmentExpression() && p.node.operator === "||=") return { hasDefault: true };
+  return { hasDefault: false };
+}
+
+function evidenceFromContent(content, fileRel, loc, reason) {
+  if (!loc || !content) return null;
+  const lines = content.split(/\r?\n/);
+  const start = Math.max(1, loc.start?.line || 1);
+  const end = Math.max(start, loc.end?.line || start);
+  const snippet = lines.slice(start - 1, end).join("\n");
+  return {
+    id: `ev_${crypto.randomBytes(4).toString("hex")}`,
+    file: fileRel,
+    lines: `${start}-${end}`,
+    snippetHash: sha256(snippet),
+    reason
+  };
+}
+
+function parseDotEnvLike(content) {
+  const out = new Set();
+  const lines = content.split(/\r?\n/);
+  for (const raw of lines) {
+    const line = raw.trim();
+    if (!line || line.startsWith("#")) continue;
+    const l = line.startsWith("export ") ? line.slice(7).trim() : line;
+    const eq = l.indexOf("=");
+    if (eq <= 0) continue;
+    const key = l.slice(0, eq).trim();
+    if (isEnvName(key)) out.add(key);
+  }
+  return out;
+}
+
+/**
+ * Extract env var usage using RepoIndex
+ * @param {import('./repo-index').RepoIndex} index
+ * @param {Object} stats
+ * @returns {Object} - Map of env var name to usage info
+ */
+function extractEnvUsage(index, stats) {
+  // Use token prefilter - only scan files that reference process.env or import.meta.env
+  const candidateAbs = index.getByAnyToken(["process.env", "import.meta.env"]);
+  
+  // Filter to JS/TS files
+  const jsExtensions = new Set([".ts", ".tsx", ".js", ".jsx", ".mjs", ".cjs"]);
+  const files = candidateAbs.filter(abs => {
+    const ext = path.extname(abs).toLowerCase();
+    return jsExtensions.has(ext);
+  });
+
+  const usageMap = {};
+
+  for (const fileAbs of files) {
+    const fileRel = index.relPath(fileAbs);
+    const content = index.getContent(fileAbs);
+    if (!content) continue;
+
+    const { ast, error } = globalASTCache.parse(content, fileAbs);
+    if (error || !ast) {
+      stats.parseErrors++;
+      continue;
+    }
+
+    try {
+      traverse(ast, {
+        MemberExpression(p) {
+          const name = extractEnvFromMemberExpr(p.node);
+          if (!name || !isEnvName(name)) return;
+
+          usageMap[name] = usageMap[name] || { name, references: [], signals: { hasDefault: false } };
+
+          const ev = evidenceFromContent(content, fileRel, p.node.loc, `Env usage: ${name}`);
+          if (ev) usageMap[name].references.push(ev);
+
+          const sig = detectDefaultSignals(p);
+          if (sig.hasDefault) usageMap[name].signals.hasDefault = true;
+        }
+      });
+    } catch {
+      stats.parseErrors++;
+    }
+  }
+
+  return usageMap;
+}
+
+/**
+ * Extract declared env vars from .env files
+ * @param {string} repoRoot
+ * @returns {{ declared: string[], sources: string[] }}
+ */
+function extractEnvDeclared(repoRoot) {
+  const candidates = [
+    ".env.example", ".env.template", ".env.sample",
+    ".env.local.example", ".env.development.example",
+    ".env"
+  ];
+
+  const declared = new Set();
+  const sources = [];
+
+  for (const rel of candidates) {
+    const abs = path.join(repoRoot, rel);
+    try {
+      if (!fs.existsSync(abs)) continue;
+      const content = fs.readFileSync(abs, "utf8");
+      const keys = parseDotEnvLike(content);
+      if (keys.size) {
+        for (const k of keys) declared.add(k);
+        sources.push(rel);
+      }
+    } catch {}
+  }
+
+  return { declared: Array.from(declared).sort(), sources };
+}
+
+/**
+ * Build env truth using RepoIndex (optimized)
+ * @param {import('./repo-index').RepoIndex} index
+ * @param {Object} stats
+ * @returns {Object}
+ */
+function buildEnvTruthV2(index, stats) {
+  const usageMap = extractEnvUsage(index, stats);
+  const declared = extractEnvDeclared(index.repoRoot);
+
+  const used = Object.values(usageMap).sort((a, b) => a.name.localeCompare(b.name));
+
+  const vars = used.map(u => ({
+    name: u.name,
+    required: !u.signals?.hasDefault,
+    references: u.references || [],
+    notes: u.signals?.hasDefault ? "Has default/fallback usage signal" : ""
+  }));
+
+  return {
+    vars,
+    declared: declared.declared,
+    declaredSources: declared.sources
+  };
+}
+
+module.exports = {
+  extractEnvUsage,
+  extractEnvDeclared,
+  buildEnvTruthV2,
+};

package/bin/runners/lib/engine/express-extractor.js

@@ -0,0 +1,208 @@
+// bin/runners/lib/engine/express-extractor.js
+// Optimized Express route extraction using RepoIndex
+
+const path = require("path");
+const crypto = require("crypto");
+const traverse = require("@babel/traverse").default;
+const t = require("@babel/types");
+const { globalASTCache } = require("./ast-cache");
+
+function sha256(text) {
+  return "sha256:" + crypto.createHash("sha256").update(text).digest("hex");
+}
+
+function canonicalizeMethod(m) {
+  const u = String(m || "").toUpperCase();
+  if (u === "ALL" || u === "ANY" || u === "*") return "*";
+  return u;
+}
+
+function canonicalizePath(p) {
+  let s = String(p || "").trim();
+  if (!s.startsWith("/")) s = "/" + s;
+  s = s.replace(/\/+/g, "/");
+  s = s.replace(/\[\[\.{3}([^\]]+)\]\]/g, "*$1?");
+  s = s.replace(/\[\.{3}([^\]]+)\]/g, "*$1");
+  s = s.replace(/\[([^\]]+)\]/g, ":$1");
+  if (s.length > 1) s = s.replace(/\/$/, "");
+  return s;
+}
+
+function joinPaths(prefix, p) {
+  const a = canonicalizePath(prefix || "/");
+  const b = canonicalizePath(p || "/");
+  if (a === "/") return b;
+  if (b === "/") return a;
+  return canonicalizePath(a + "/" + b);
+}
+
+function evidenceFromContent(content, fileRel, loc, reason) {
+  if (!loc || !content) return null;
+  const lines = content.split(/\r?\n/);
+  const start = Math.max(1, loc.start?.line || 1);
+  const end = Math.max(start, loc.end?.line || start);
+  const snippet = lines.slice(start - 1, end).join("\n");
+  return {
+    id: `ev_${crypto.randomBytes(4).toString("hex")}`,
+    file: fileRel,
+    lines: `${start}-${end}`,
+    snippetHash: sha256(snippet),
+    reason
+  };
+}
+
+const EXPRESS_METHODS = new Set(["get", "post", "put", "patch", "delete", "options", "head", "all", "use"]);
+const EXPRESS_NAMES = new Set(["app", "router", "express", "server"]);
+
+function evalStaticString(node, constMap = new Map(), depth = 0) {
+  if (!node || depth > 5) return null;
+  if (t.isStringLiteral(node)) return node.value;
+  if (t.isTemplateLiteral(node) && node.expressions.length === 0) {
+    return node.quasis.map(q => q.value.cooked || "").join("");
+  }
+  if (t.isIdentifier(node) && constMap.has(node.name)) {
+    return evalStaticString(constMap.get(node.name), constMap, depth + 1);
+  }
+  if (t.isBinaryExpression(node, { operator: "+" })) {
+    const l = evalStaticString(node.left, constMap, depth + 1);
+    const r = evalStaticString(node.right, constMap, depth + 1);
+    if (typeof l === "string" && typeof r === "string") return l + r;
+  }
+  return null;
+}
+
+/**
+ * Extract Express routes using RepoIndex
+ * @param {import('./repo-index').RepoIndex} index
+ * @param {Object} stats
+ * @returns {Array}
+ */
+function extractExpressRoutes(index, stats) {
+  if (!index.hasFramework("express")) return [];
+
+  // Use token prefilter - only scan files that mention express/app/router
+  const candidateAbs = index.getByAnyToken(["express", "app.get", "app.post", "router.get", "router.post"]);
+  
+  // Filter to JS/TS files, exclude tests
+  const jsExtensions = new Set([".ts", ".tsx", ".js", ".jsx", ".mjs", ".cjs"]);
+  const files = candidateAbs.filter(abs => {
+    const ext = path.extname(abs).toLowerCase();
+    if (!jsExtensions.has(ext)) return false;
+    const rel = index.relPath(abs);
+    if (rel.includes("/test/") || rel.includes("/tests/") || rel.includes("/__tests__/")) return false;
+    if (rel.includes(".test.") || rel.includes(".spec.")) return false;
+    if (rel.endsWith(".d.ts")) return false;
+    return true;
+  });
+
+  const routes = [];
+
+  for (const fileAbs of files) {
+    const fileRel = index.relPath(fileAbs);
+    const content = index.getContent(fileAbs);
+    if (!content) continue;
+
+    // Quick content check
+    if (!content.includes("express") && !content.includes("app.") && !content.includes("router.")) continue;
+
+    const { ast, error } = globalASTCache.parse(content, fileAbs);
+    if (error || !ast) {
+      stats.parseErrors++;
+      continue;
+    }
+
+    const constMap = new Map();
+    const routerPrefixes = new Map();
+
+    try {
+      // First pass: collect constants and router.use prefixes
+      traverse(ast, {
+        VariableDeclarator(p) {
+          if (t.isIdentifier(p.node.id) && p.node.init) {
+            const v = evalStaticString(p.node.init, constMap);
+            if (typeof v === "string") constMap.set(p.node.id.name, p.node.init);
+
+            // Detect express/router assignment
+            if (t.isCallExpression(p.node.init)) {
+              const callee = p.node.init.callee;
+              if (t.isIdentifier(callee, { name: "express" }) ||
+                  (t.isMemberExpression(callee) && t.isIdentifier(callee.property, { name: "Router" }))) {
+                EXPRESS_NAMES.add(p.node.id.name);
+              }
+            }
+          }
+        },
+        CallExpression(p) {
+          const callee = p.node.callee;
+          if (!t.isMemberExpression(callee)) return;
+          if (!t.isIdentifier(callee.object) || !t.isIdentifier(callee.property)) return;
+          
+          const obj = callee.object.name;
+          const method = callee.property.name;
+
+          // app.use("/api", router) - track prefix
+          if (method === "use" && EXPRESS_NAMES.has(obj)) {
+            const args = p.node.arguments;
+            if (args.length >= 2) {
+              const prefixArg = args[0];
+              const routerArg = args[1];
+              const prefix = evalStaticString(prefixArg, constMap);
+              if (typeof prefix === "string" && t.isIdentifier(routerArg)) {
+                routerPrefixes.set(routerArg.name, prefix);
+              }
+            }
+          }
+        }
+      });
+
+      // Second pass: extract routes
+      traverse(ast, {
+        CallExpression(p) {
+          const callee = p.node.callee;
+          if (!t.isMemberExpression(callee)) return;
+          if (!t.isIdentifier(callee.object) || !t.isIdentifier(callee.property)) return;
+
+          const obj = callee.object.name;
+          const method = callee.property.name.toLowerCase();
+
+          if (!EXPRESS_NAMES.has(obj) && !routerPrefixes.has(obj)) return;
+          if (!EXPRESS_METHODS.has(method)) return;
+
+          // Get the path argument
+          const pathArg = p.node.arguments[0];
+          const routePath = evalStaticString(pathArg, constMap);
+          if (!routePath) return;
+
+          // Skip middleware-only patterns
+          if (method === "use" && !routePath.startsWith("/")) return;
+
+          // Get prefix for this router
+          const prefix = routerPrefixes.get(obj) || "/";
+          const fullPath = joinPaths(prefix, routePath);
+
+          const ev = evidenceFromContent(content, fileRel, p.node.loc, `Express ${method.toUpperCase()}("${routePath}")`);
+
+          routes.push({
+            method: method === "use" ? "*" : canonicalizeMethod(method),
+            path: fullPath,
+            handler: fileRel,
+            confidence: "med",
+            framework: "express",
+            evidence: ev ? [ev] : []
+          });
+        }
+      });
+    } catch {
+      stats.parseErrors++;
+    }
+  }
+
+  return routes;
+}
+
+module.exports = {
+  extractExpressRoutes,
+  canonicalizeMethod,
+  canonicalizePath,
+  joinPaths,
+};