burnwatch 0.5.2 → 0.7.0

package/dist/interactive-init.js

@@ -60,7 +60,239 @@ async function fetchJson(url, options = {}) {
   }
 }
 
+// src/probes.ts
+function matchPlanByPrefix(detected, plans) {
+  const lower = detected.toLowerCase();
+  return plans.find((p) => {
+    if (p.type === "exclude") return false;
+    const firstWord = p.name.split(/[\s(]/)[0].toLowerCase();
+    return lower.includes(firstWord) || firstWord.includes(lower);
+  });
+}
+var probeScrapfly = async (apiKey, plans) => {
+  const result = await fetchJson(`https://api.scrapfly.io/account?key=${apiKey}`);
+  if (!result.ok || !result.data) return null;
+  const planName = result.data.subscription?.plan?.name;
+  let unitsUsed = 0;
+  let unitsTotal = 0;
+  if (result.data.subscription?.usage?.scrape) {
+    unitsUsed = result.data.subscription.usage.scrape.used ?? 0;
+    unitsTotal = result.data.subscription.usage.scrape.allowed ?? 0;
+  } else if (result.data.account) {
+    unitsUsed = result.data.account.credits_used ?? 0;
+    unitsTotal = result.data.account.credits_total ?? 0;
+  }
+  const matched = planName ? matchPlanByPrefix(planName, plans) : void 0;
+  return {
+    planName: planName ?? void 0,
+    matchedPlan: matched,
+    usage: {
+      unitsUsed,
+      unitsTotal,
+      unitName: "credits"
+    },
+    summary: matched ? `${matched.name} \u2014 ${formatK(unitsUsed)}/${formatK(unitsTotal)} credits used` : `${formatK(unitsUsed)}/${formatK(unitsTotal)} credits used`,
+    confidence: matched ? "high" : "medium"
+  };
+};
+var probeAnthropic = async (apiKey, _plans) => {
+  const now = /* @__PURE__ */ new Date();
+  const startOfMonth = new Date(now.getFullYear(), now.getMonth(), 1);
+  const params = new URLSearchParams({
+    start_date: startOfMonth.toISOString().split("T")[0],
+    end_date: now.toISOString().split("T")[0]
+  });
+  const result = await fetchJson(`https://api.anthropic.com/v1/organizations/cost_report?${params}`, {
+    headers: {
+      "x-api-key": apiKey,
+      "anthropic-version": "2023-06-01"
+    }
+  });
+  if (!result.ok || !result.data?.data) return null;
+  let totalCents = 0;
+  for (const entry of result.data.data) {
+    totalCents += parseFloat(entry.amount ?? "0");
+  }
+  const spend = totalCents / 100;
+  return {
+    usage: { spend, currency: "USD" },
+    summary: `$${spend.toFixed(2)} spent this billing period`,
+    confidence: "medium"
+  };
+};
+var probeOpenAI = async (apiKey, _plans) => {
+  const now = /* @__PURE__ */ new Date();
+  const startOfMonth = new Date(now.getFullYear(), now.getMonth(), 1);
+  const params = new URLSearchParams({
+    start_time: String(Math.floor(startOfMonth.getTime() / 1e3)),
+    end_time: String(Math.floor(now.getTime() / 1e3))
+  });
+  const result = await fetchJson(`https://api.openai.com/v1/organization/usage/completions?${params}`, {
+    headers: { Authorization: `Bearer ${apiKey}` }
+  });
+  if (!result.ok || !result.data?.data) return null;
+  let totalTokens = 0;
+  for (const bucket of result.data.data) {
+    for (const r of bucket.results ?? []) {
+      totalTokens += r.amount?.value ?? 0;
+    }
+  }
+  return {
+    usage: { unitsUsed: totalTokens, unitName: "tokens" },
+    summary: `${formatK(totalTokens)} tokens used this period`,
+    confidence: "medium"
+  };
+};
+var probeVercel = async (apiKey, plans) => {
+  const teamsResult = await fetchJson("https://api.vercel.com/v2/teams", {
+    headers: { Authorization: `Bearer ${apiKey}` }
+  });
+  if (teamsResult.ok && teamsResult.data?.teams?.[0]) {
+    const team = teamsResult.data.teams[0];
+    const planName = team.billing?.plan;
+    if (planName) {
+      const matched = matchPlanByPrefix(planName, plans);
+      return {
+        planName,
+        matchedPlan: matched,
+        summary: `Team "${team.name}" on ${planName} plan`,
+        confidence: matched ? "high" : "medium"
+      };
+    }
+  }
+  const userResult = await fetchJson("https://api.vercel.com/v2/user", {
+    headers: { Authorization: `Bearer ${apiKey}` }
+  });
+  if (userResult.ok && userResult.data?.user) {
+    const plan = userResult.data.user.billing?.plan ?? "hobby";
+    const matched = matchPlanByPrefix(plan, plans);
+    return {
+      planName: plan,
+      matchedPlan: matched,
+      summary: `Personal account on ${plan} plan`,
+      confidence: matched ? "high" : "low"
+    };
+  }
+  return null;
+};
+var probeSupabase = async (apiKey, plans) => {
+  const orgsResult = await fetchJson("https://api.supabase.com/v1/organizations", {
+    headers: { Authorization: `Bearer ${apiKey}` }
+  });
+  if (!orgsResult.ok || !orgsResult.data || !Array.isArray(orgsResult.data)) return null;
+  const org = orgsResult.data[0];
+  if (!org) return null;
+  const planName = org.billing?.plan;
+  if (planName) {
+    const matched = matchPlanByPrefix(planName, plans);
+    return {
+      planName,
+      matchedPlan: matched,
+      summary: `Org "${org.name}" on ${planName} plan`,
+      confidence: matched ? "high" : "medium"
+    };
+  }
+  return {
+    summary: `Org "${org.name}" found (plan not detected)`,
+    confidence: "low"
+  };
+};
+var probeStripe = async (apiKey, _plans) => {
+  const result = await fetchJson("https://api.stripe.com/v1/balance", {
+    headers: { Authorization: `Bearer ${apiKey}` }
+  });
+  if (!result.ok || !result.data) return null;
+  const available = result.data.available?.[0];
+  const pending = result.data.pending?.[0];
+  const totalCents = (available?.amount ?? 0) + (pending?.amount ?? 0);
+  const currency = (available?.currency ?? "usd").toUpperCase();
+  return {
+    usage: { spend: totalCents / 100, currency },
+    summary: `Balance: ${currency} ${(totalCents / 100).toFixed(2)} (${((available?.amount ?? 0) / 100).toFixed(2)} available)`,
+    confidence: "medium"
+  };
+};
+var probeBrowserbase = async (apiKey, _plans) => {
+  const projResult = await fetchJson("https://api.browserbase.com/v1/projects", {
+    headers: { "X-BB-API-Key": apiKey }
+  });
+  if (!projResult.ok || !projResult.data?.[0]?.id) return null;
+  const projectId = projResult.data[0].id;
+  const usageResult = await fetchJson(`https://api.browserbase.com/v1/projects/${projectId}/usage`, {
+    headers: { "X-BB-API-Key": apiKey }
+  });
+  if (!usageResult.ok || !usageResult.data) {
+    return {
+      summary: `Project "${projResult.data[0].name}" found`,
+      confidence: "low"
+    };
+  }
+  const sessions = usageResult.data.sessions_count ?? 0;
+  const hours = usageResult.data.browser_hours ?? 0;
+  return {
+    usage: { unitsUsed: sessions, unitName: "sessions" },
+    summary: `${sessions} sessions, ${hours.toFixed(1)} browser hours this period`,
+    confidence: "medium"
+  };
+};
+var probeUpstash = async (apiKey, _plans) => {
+  const result = await fetchJson("https://api.upstash.com/v2/redis/databases", {
+    headers: {
+      Authorization: `Basic ${Buffer.from(apiKey).toString("base64")}`
+    }
+  });
+  if (!result.ok) return null;
+  const dbCount = Array.isArray(result.data) ? result.data.length : 0;
+  return {
+    summary: `${dbCount} Redis database${dbCount !== 1 ? "s" : ""} found`,
+    confidence: "low"
+  };
+};
+var probePostHog = async (apiKey, _plans) => {
+  const result = await fetchJson("https://us.posthog.com/api/organizations/@current", {
+    headers: { Authorization: `Bearer ${apiKey}` }
+  });
+  if (!result.ok || !result.data) return null;
+  return {
+    summary: "Organization found",
+    confidence: "low"
+  };
+};
+var PROBES = /* @__PURE__ */ new Map([
+  ["scrapfly", probeScrapfly],
+  ["anthropic", probeAnthropic],
+  ["openai", probeOpenAI],
+  ["vercel", probeVercel],
+  ["supabase", probeSupabase],
+  ["stripe", probeStripe],
+  ["browserbase", probeBrowserbase],
+  ["upstash", probeUpstash],
+  ["posthog", probePostHog]
+]);
+async function probeService(serviceId, apiKey, plans) {
+  const probe = PROBES.get(serviceId);
+  if (!probe) return null;
+  try {
+    return await probe(apiKey, plans);
+  } catch {
+    return null;
+  }
+}
+function hasProbe(serviceId) {
+  return PROBES.has(serviceId);
+}
+function formatK(n) {
+  if (n >= 1e6) return `${(n / 1e6).toFixed(n % 1e6 === 0 ? 0 : 1)}M`;
+  if (n >= 1e3) return `${(n / 1e3).toFixed(n % 1e3 === 0 ? 0 : 1)}K`;
+  return String(n);
+}
+
 // src/interactive-init.ts
+function formatUnits(n) {
+  if (n >= 1e6) return `${(n / 1e6).toFixed(n % 1e6 === 0 ? 0 : 1)}M`;
+  if (n >= 1e3) return `${(n / 1e3).toFixed(n % 1e3 === 0 ? 0 : 1)}K`;
+  return String(n);
+}
 var RISK_ORDER = ["llm", "usage", "infra", "flat"];
 var RISK_LABELS = {
   llm: "LLM / AI Services (highest variable cost)",
@@ -101,16 +333,6 @@ function ask(rl, question) {
     });
   });
 }
-async function autoDetectScrapflyPlan(apiKey) {
-  try {
-    const result = await fetchJson(`https://api.scrapfly.io/account?key=${apiKey}`);
-    if (result.ok && result.data?.subscription?.plan?.name) {
-      return result.data.subscription.plan.name;
-    }
-  } catch {
-  }
-  return null;
-}
 function findEnvKey(service) {
   for (const pattern of service.envPatterns) {
     const val = process.env[pattern];
@@ -118,7 +340,7 @@ function findEnvKey(service) {
   }
   return void 0;
 }
-function autoConfigureServices(detected) {
+async function autoConfigureServices(detected) {
   const services = {};
   const groups = groupByRisk(detected);
   const globalConfig = readGlobalConfig();
@@ -151,25 +373,54 @@ function autoConfigureServices(detected) {
         } else if (defaultPlan.suggestedBudget !== void 0) {
           tracked.budget = defaultPlan.suggestedBudget;
         }
+        if (defaultPlan.includedUnits !== void 0 && defaultPlan.unitName) {
+          tracked.allowance = {
+            included: defaultPlan.includedUnits,
+            unitName: defaultPlan.unitName
+          };
+        }
       }
       const existingKey = globalConfig.services[service.id]?.apiKey;
       const envKey = findEnvKey(service);
       let keySource = "";
+      let apiKey;
       if (existingKey) {
         tracked.hasApiKey = true;
+        apiKey = existingKey;
         keySource = " (key: global config)";
       } else if (envKey) {
         tracked.hasApiKey = true;
+        apiKey = envKey;
         if (!globalConfig.services[service.id]) {
           globalConfig.services[service.id] = {};
         }
         globalConfig.services[service.id].apiKey = envKey;
         keySource = ` (key: ${service.envPatterns[0]})`;
       }
+      if (apiKey && hasProbe(service.id)) {
+        try {
+          const probe = await probeService(service.id, apiKey, plans);
+          if (probe?.matchedPlan && probe.confidence === "high") {
+            const mp = probe.matchedPlan;
+            tracked.planName = mp.name;
+            if (mp.type === "flat" && mp.monthlyBase !== void 0) {
+              tracked.planCost = mp.monthlyBase;
+              tracked.budget = mp.monthlyBase;
+            } else if (mp.suggestedBudget !== void 0) {
+              tracked.budget = mp.suggestedBudget;
+            }
+            if (mp.includedUnits !== void 0 && mp.unitName) {
+              tracked.allowance = { included: mp.includedUnits, unitName: mp.unitName };
+            }
+          }
+        } catch {
+        }
+      }
       const tierLabel = tracked.hasApiKey ? "LIVE" : tracked.planCost !== void 0 ? "CALC" : "BLIND";
       const planStr = tracked.planName ? ` ${tracked.planName}` : "";
+      const trackingStr = tracked.allowance ? `$${tracked.budget}/mo | ${formatUnits(tracked.allowance.included)} ${tracked.allowance.unitName}` : `$${tracked.budget}/mo`;
       console.log(
-        `    ${service.name}:${planStr} | ${tierLabel} | $${tracked.budget}/mo${keySource}`
+        `    ${service.name}:${planStr} | ${tierLabel} | ${trackingStr}${keySource}`
       );
       services[service.id] = tracked;
     }
@@ -221,21 +472,61 @@ async function runInteractiveInit(detected) {
         console.log("  -> Configured (no plan tiers in registry, budget: $0)");
         continue;
       }
-      const defaultIndex = plans.findIndex((p) => p.default);
-      console.log("");
-      for (let i = 0; i < plans.length; i++) {
-        const plan = plans[i];
-        const marker = i === defaultIndex ? " *" : "";
-        const costStr = plan.type === "exclude" ? "" : plan.monthlyBase !== void 0 ? ` - $${plan.monthlyBase}/mo` : " - variable";
-        console.log(`    ${i + 1}) ${plan.name}${costStr}${marker}`);
+      let apiKey;
+      const existingKey = globalConfig.services[service.id]?.apiKey;
+      const envKey = findEnvKey(service);
+      if (existingKey) {
+        apiKey = existingKey;
+        console.log(`  API key: found in global config`);
+      } else if (envKey) {
+        apiKey = envKey;
+        console.log(`  API key: found in environment (${service.envPatterns[0]})`);
+        if (!globalConfig.services[service.id]) {
+          globalConfig.services[service.id] = {};
+        }
+        globalConfig.services[service.id].apiKey = envKey;
+      }
+      let chosen;
+      if (apiKey && hasProbe(service.id)) {
+        console.log("  Probing API...");
+        const probe = await probeService(service.id, apiKey, plans);
+        if (probe) {
+          console.log(`  ${probe.summary}`);
+          if (probe.confidence === "high" && probe.matchedPlan) {
+            const plan = probe.matchedPlan;
+            const costStr = plan.monthlyBase !== void 0 ? `$${plan.monthlyBase}/mo` : "variable";
+            const unitsStr = plan.includedUnits && plan.unitName ? `, ${formatUnits(plan.includedUnits)} ${plan.unitName}` : "";
+            const confirm = await ask(
+              rl,
+              `  Detected: ${plan.name} (${costStr}${unitsStr}). Correct? [Y/n]: `
+            );
+            if (confirm === "" || confirm.toLowerCase().startsWith("y")) {
+              chosen = plan;
+            }
+          } else if (probe.confidence === "medium") {
+            if (probe.usage?.spend !== void 0) {
+              console.log(`  Current spend: $${probe.usage.spend.toFixed(2)}`);
+            }
+          }
+        }
+      }
+      if (!chosen) {
+        const defaultIndex = plans.findIndex((p) => p.default);
+        console.log("");
+        for (let i = 0; i < plans.length; i++) {
+          const plan = plans[i];
+          const marker = i === defaultIndex ? " *" : "";
+          const costStr = plan.type === "exclude" ? "" : plan.monthlyBase !== void 0 ? ` - $${plan.monthlyBase}/mo` : " - variable";
+          console.log(`    ${i + 1}) ${plan.name}${costStr}${marker}`);
+        }
+        const defaultChoice = defaultIndex >= 0 ? String(defaultIndex + 1) : "1";
+        const answer = await ask(
+          rl,
+          `  Which plan? [${defaultChoice}]: `
+        );
+        const choiceIndex = (answer === "" ? parseInt(defaultChoice) : parseInt(answer)) - 1;
+        chosen = plans[choiceIndex] ?? plans[defaultIndex >= 0 ? defaultIndex : 0];
       }
-      const defaultChoice = defaultIndex >= 0 ? String(defaultIndex + 1) : "1";
-      const answer = await ask(
-        rl,
-        `  Which plan? [${defaultChoice}]: `
-      );
-      const choiceIndex = (answer === "" ? parseInt(defaultChoice) : parseInt(answer)) - 1;
-      const chosen = plans[choiceIndex] ?? plans[defaultIndex >= 0 ? defaultIndex : 0];
       if (chosen.type === "exclude") {
         services[service.id] = {
           serviceId: service.id,
@@ -251,49 +542,37 @@ async function runInteractiveInit(detected) {
       const tracked2 = {
         serviceId: service.id,
         detectedVia: det.sources,
-        hasApiKey: false,
+        hasApiKey: !!apiKey,
         firstDetected: (/* @__PURE__ */ new Date()).toISOString(),
         planName: chosen.name
       };
       if (chosen.type === "flat" && chosen.monthlyBase !== void 0) {
         tracked2.planCost = chosen.monthlyBase;
       }
-      if (service.apiTier === "live") {
-        const existingKey = globalConfig.services[service.id]?.apiKey;
-        const envKey = findEnvKey(service);
-        if (existingKey) {
-          console.log(`  API key: found in global config`);
-          tracked2.hasApiKey = true;
-        } else if (envKey) {
-          console.log(`  API key: found in environment (${service.envPatterns[0]})`);
+      if (chosen.includedUnits !== void 0 && chosen.unitName) {
+        tracked2.allowance = {
+          included: chosen.includedUnits,
+          unitName: chosen.unitName
+        };
+      }
+      if (!apiKey && hasProbe(service.id)) {
+        const hint = API_KEY_HINTS[service.id];
+        if (hint) console.log(`  ${hint}`);
+        const keyAnswer = await ask(
+          rl,
+          `  API key for real-time tracking (Enter to skip): `
+        );
+        if (keyAnswer) {
           tracked2.hasApiKey = true;
+          apiKey = keyAnswer;
           if (!globalConfig.services[service.id]) {
             globalConfig.services[service.id] = {};
           }
-          globalConfig.services[service.id].apiKey = envKey;
-        } else {
-          const hint = API_KEY_HINTS[service.id];
-          if (hint) console.log(`  ${hint}`);
-          const keyAnswer = await ask(
-            rl,
-            `  API key for real-time tracking (Enter to skip): `
-          );
-          if (keyAnswer) {
-            tracked2.hasApiKey = true;
-            if (!globalConfig.services[service.id]) {
-              globalConfig.services[service.id] = {};
-            }
-            globalConfig.services[service.id].apiKey = keyAnswer;
-          }
-        }
-        if (service.autoDetectPlan && service.id === "scrapfly" && tracked2.hasApiKey) {
-          const key = globalConfig.services[service.id]?.apiKey;
-          if (key) {
-            console.log("  Detecting plan from API...");
-            const planName = await autoDetectScrapflyPlan(key);
-            if (planName) {
-              console.log(`  -> Detected plan: ${planName}`);
-              tracked2.planName = planName;
+          globalConfig.services[service.id].apiKey = keyAnswer;
+          if (hasProbe(service.id)) {
+            const probe = await probeService(service.id, keyAnswer, plans);
+            if (probe?.usage) {
+              console.log(`  ${probe.summary}`);
             }
           }
         }
@@ -311,8 +590,9 @@ async function runInteractiveInit(detected) {
       }
       services[service.id] = tracked2;
       const tierLabel = tracked2.hasApiKey ? "LIVE" : tracked2.planCost !== void 0 ? "CALC" : "BLIND";
+      const allowanceStr = tracked2.allowance ? ` | ${formatUnits(tracked2.allowance.included)} ${tracked2.allowance.unitName}` : "";
       console.log(
-        `  -> ${service.name}: ${tracked2.planName} | ${tierLabel} | $${tracked2.budget}/mo`
+        `  -> ${service.name}: ${tracked2.planName} | ${tierLabel} | $${tracked2.budget}/mo${allowanceStr}`
       );
     }
   }