burnwatch 0.2.0 → 0.3.0

package/dist/cli.js

@@ -122,7 +122,7 @@ function getAllServices(projectRoot) {
 function detectServices(projectRoot) {
   const registry = loadRegistry(projectRoot);
   const results = /* @__PURE__ */ new Map();
-  const pkgDeps = scanPackageJson(projectRoot);
+  const pkgDeps = scanAllPackageJsons(projectRoot);
   for (const [serviceId, service] of registry) {
     const matchedPkgs = service.packageNames.filter(
       (pkg) => pkgDeps.has(pkg)
@@ -134,7 +134,7 @@ function detectServices(projectRoot) {
       );
     }
   }
-  const envVars = new Set(Object.keys(process.env));
+  const envVars = collectEnvVars(projectRoot);
   for (const [serviceId, service] of registry) {
     const matchedEnvs = service.envPatterns.filter(
       (pattern) => envVars.has(pattern)
@@ -172,40 +172,118 @@ function getOrCreate(map, serviceId, service) {
   }
   return result;
 }
-function scanPackageJson(projectRoot) {
+function scanAllPackageJsons(projectRoot) {
   const deps = /* @__PURE__ */ new Set();
-  const pkgPath = path3.join(projectRoot, "package.json");
+  const pkgFiles = findFiles(projectRoot, "package.json", 4);
+  for (const pkgPath of pkgFiles) {
+    try {
+      const raw = fs3.readFileSync(pkgPath, "utf-8");
+      const pkg = JSON.parse(raw);
+      for (const name of Object.keys(pkg.dependencies ?? {})) deps.add(name);
+      for (const name of Object.keys(pkg.devDependencies ?? {})) deps.add(name);
+    } catch {
+    }
+  }
+  return deps;
+}
+function collectEnvVars(projectRoot) {
+  const envVars = new Set(Object.keys(process.env));
+  const envFiles = findEnvFiles(projectRoot, 3);
+  for (const envFile of envFiles) {
+    try {
+      const content = fs3.readFileSync(envFile, "utf-8");
+      const keys = content.split("\n").filter((line) => line.includes("=") && !line.startsWith("#")).map((line) => line.split("=")[0].trim()).filter(Boolean);
+      for (const key of keys) {
+        envVars.add(key);
+      }
+    } catch {
+    }
+  }
+  return envVars;
+}
+function findEnvFiles(dir, maxDepth) {
+  const results = [];
+  if (maxDepth <= 0) return results;
   try {
-    const raw = fs3.readFileSync(pkgPath, "utf-8");
-    const pkg = JSON.parse(raw);
-    for (const name of Object.keys(pkg.dependencies ?? {})) deps.add(name);
-    for (const name of Object.keys(pkg.devDependencies ?? {})) deps.add(name);
+    const entries = fs3.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (entry.name === "node_modules" || entry.name === ".git" || entry.name === "dist") continue;
+      const fullPath = path3.join(dir, entry.name);
+      if (entry.isDirectory()) {
+        results.push(...findEnvFiles(fullPath, maxDepth - 1));
+      } else if (entry.name.startsWith(".env")) {
+        results.push(fullPath);
+      }
+    }
   } catch {
   }
-  return deps;
+  return results;
+}
+function findFiles(dir, fileName, maxDepth) {
+  const results = [];
+  if (maxDepth <= 0) return results;
+  try {
+    const entries = fs3.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (entry.name === "node_modules" || entry.name === ".git" || entry.name === "dist") continue;
+      const fullPath = path3.join(dir, entry.name);
+      if (entry.isDirectory()) {
+        results.push(...findFiles(fullPath, fileName, maxDepth - 1));
+      } else if (entry.name === fileName) {
+        results.push(fullPath);
+      }
+    }
+  } catch {
+  }
+  return results;
 }
 function scanImports(projectRoot) {
   const imports = /* @__PURE__ */ new Set();
-  const srcDir = path3.join(projectRoot, "src");
-  if (!fs3.existsSync(srcDir)) return imports;
-  const files = walkDir(srcDir, /\.(ts|tsx|js|jsx|mjs|cjs)$/);
-  for (const file of files) {
-    try {
-      const content = fs3.readFileSync(file, "utf-8");
-      const importRegex = /(?:from\s+["']|require\s*\(\s*["'])([^./][^"']*?)(?:["'])/g;
-      let match;
-      while ((match = importRegex.exec(content)) !== null) {
-        const pkg = match[1];
-        if (pkg) {
-          const parts = pkg.split("/");
-          if (parts[0]?.startsWith("@") && parts.length >= 2) {
-            imports.add(`${parts[0]}/${parts[1]}`);
-          } else if (parts[0]) {
-            imports.add(parts[0]);
+  const codeDirs = ["src", "app", "lib", "pages", "components", "utils", "services", "hooks"];
+  const dirsToScan = [];
+  for (const dir of codeDirs) {
+    const fullPath = path3.join(projectRoot, dir);
+    if (fs3.existsSync(fullPath)) {
+      dirsToScan.push(fullPath);
+    }
+  }
+  try {
+    const entries = fs3.readdirSync(projectRoot, { withFileTypes: true });
+    for (const entry of entries) {
+      if (!entry.isDirectory()) continue;
+      if (entry.name === "node_modules" || entry.name === ".git" || entry.name === "dist" || entry.name.startsWith(".")) continue;
+      const subPkgPath = path3.join(projectRoot, entry.name, "package.json");
+      if (fs3.existsSync(subPkgPath)) {
+        for (const dir of codeDirs) {
+          const fullPath = path3.join(projectRoot, entry.name, dir);
+          if (fs3.existsSync(fullPath)) {
+            dirsToScan.push(fullPath);
           }
         }
       }
-    } catch {
+    }
+  } catch {
+  }
+  for (const dir of dirsToScan) {
+    const files = walkDir(dir, /\.(ts|tsx|js|jsx|mjs|cjs)$/);
+    for (const file of files) {
+      try {
+        const content = fs3.readFileSync(file, "utf-8");
+        const importRegex = /(?:from\s+["']|require\s*\(\s*["'])([^./][^"']*?)(?:["'])/g;
+        let match;
+        while ((match = importRegex.exec(content)) !== null) {
+          const pkg = match[1];
+          if (pkg) {
+            const parts = pkg.split("/");
+            if (parts[0]?.startsWith("@") && parts.length >= 2) {
+              imports.add(`${parts[0]}/${parts[1]}`);
+            } else if (parts[0]) {
+              imports.add(parts[0]);
+            }
+          }
+        }
+      } catch {
+      }
     }
   }
   return imports;
@@ -704,6 +782,7 @@ var command = args[0];
 async function main() {
   switch (command) {
     case "init":
+    case "setup":
       await cmdInit();
       break;
     case "add":
@@ -718,9 +797,6 @@ async function main() {
     case "reconcile":
       await cmdReconcile();
       break;
-    case "setup":
-      await cmdSetup();
-      break;
     case "help":
     case "--help":
     case "-h":
@@ -920,69 +996,6 @@ async function cmdStatus() {
     console.log("");
   }
 }
-async function cmdSetup() {
-  const projectRoot = process.cwd();
-  if (!isInitialized(projectRoot)) {
-    await cmdInit();
-  }
-  const config = readProjectConfig(projectRoot);
-  const detected = Object.values(config.services);
-  if (detected.length === 0) {
-    console.log("No paid services detected. You're all set!");
-    return;
-  }
-  console.log("\u{1F4CB} Auto-configuring detected services...\n");
-  const globalConfig = readGlobalConfig();
-  const liveServices = [];
-  const calcServices = [];
-  const estServices = [];
-  const blindServices = [];
-  for (const tracked of detected) {
-    const definition = getService(tracked.serviceId, projectRoot);
-    if (!definition) continue;
-    const hasKey = !!globalConfig.services[tracked.serviceId]?.apiKey;
-    if (hasKey && definition.apiTier === "live") {
-      tracked.hasApiKey = true;
-      liveServices.push(`${definition.name}`);
-    } else if (definition.apiTier === "calc") {
-      calcServices.push(`${definition.name}`);
-    } else if (definition.apiTier === "est") {
-      estServices.push(`${definition.name}`);
-    } else {
-      blindServices.push(`${definition.name}`);
-    }
-  }
-  writeProjectConfig(config, projectRoot);
-  if (liveServices.length > 0) {
-    console.log(`  \u2705 LIVE (real billing data): ${liveServices.join(", ")}`);
-  }
-  if (calcServices.length > 0) {
-    console.log(`  \u{1F7E1} CALC (flat-rate tracking): ${calcServices.join(", ")}`);
-  }
-  if (estServices.length > 0) {
-    console.log(`  \u{1F7E0} EST (estimated from usage): ${estServices.join(", ")}`);
-  }
-  if (blindServices.length > 0) {
-    console.log(`  \u{1F534} BLIND (detected, need API key): ${blindServices.join(", ")}`);
-  }
-  console.log("");
-  if (blindServices.length > 0) {
-    console.log("To upgrade BLIND services to LIVE, add API keys:");
-    for (const tracked of detected) {
-      const definition = getService(tracked.serviceId, projectRoot);
-      if (definition?.apiTier === "live" && !tracked.hasApiKey) {
-        const envHint = definition.envPatterns[0] ?? "YOUR_KEY";
-        console.log(`  burnwatch add ${tracked.serviceId} --key $${envHint} --budget <N>`);
-      }
-    }
-    console.log("");
-  }
-  console.log("To set budgets for any service:");
-  console.log("  burnwatch add <service> --budget <monthly_amount>");
-  console.log("");
-  console.log("Or use /setup-burnwatch in Claude Code for guided setup with budget suggestions.\n");
-  await cmdStatus();
-}
 function cmdServices() {
   const services = getAllServices();
   console.log(`
@@ -1066,24 +1079,47 @@ function cmdVersion() {
   }
 }
 function registerHooks(projectRoot) {
+  const sourceHooksDir = path5.resolve(
+    path5.dirname(new URL(import.meta.url).pathname),
+    "hooks"
+  );
+  const localHooksDir = path5.join(projectRoot, ".burnwatch", "hooks");
+  fs5.mkdirSync(localHooksDir, { recursive: true });
+  const hookFiles = [
+    "on-session-start.js",
+    "on-prompt.js",
+    "on-file-change.js",
+    "on-stop.js"
+  ];
+  for (const file of hookFiles) {
+    const src = path5.join(sourceHooksDir, file);
+    const dest = path5.join(localHooksDir, file);
+    try {
+      fs5.copyFileSync(src, dest);
+      const mapSrc = src + ".map";
+      if (fs5.existsSync(mapSrc)) {
+        fs5.copyFileSync(mapSrc, dest + ".map");
+      }
+    } catch (err) {
+      console.error(`   Warning: Could not copy hook ${file}: ${err instanceof Error ? err.message : err}`);
+    }
+  }
+  console.log(`   Hook scripts copied to ${localHooksDir}`);
   const claudeDir = path5.join(projectRoot, ".claude");
   const settingsPath = path5.join(claudeDir, "settings.json");
   fs5.mkdirSync(claudeDir, { recursive: true });
   let settings = {};
   try {
-    settings = JSON.parse(fs5.readFileSync(settingsPath, "utf-8"));
+    const existing = fs5.readFileSync(settingsPath, "utf-8");
+    settings = JSON.parse(existing);
+    console.log(`   Merging into existing ${settingsPath}`);
   } catch {
   }
-  const hookBase = `node "${path5.join(projectRoot, "node_modules", "burnwatch", "dist", "hooks")}"`;
-  const useNpx = !fs5.existsSync(
-    path5.join(projectRoot, "node_modules", "burnwatch")
-  );
-  const prefix = useNpx ? "npx --yes burnwatch-hook" : hookBase;
-  const hooksDir = path5.resolve(
-    path5.dirname(new URL(import.meta.url).pathname),
-    "hooks"
-  );
-  const hooks = settings["hooks"] ?? {};
+  if (!settings["hooks"] || typeof settings["hooks"] !== "object") {
+    settings["hooks"] = {};
+  }
+  const hooks = settings["hooks"];
+  const hooksDir = localHooksDir;
   if (!hooks["SessionStart"]) hooks["SessionStart"] = [];
   addHookIfMissing(hooks["SessionStart"], "SessionStart", {
     matcher: "startup|resume",