@kill-switch/agent-guard 0.1.0 → 0.1.2

package/dist/report.js

@@ -1,12 +1,101 @@
 /**
  * Shared status report — the single computation behind `agent-guard status` and
  * `ks guard status`, so both emit an identical JSON shape and never drift.
+ *
+ * Two halves:
+ *   - the dollar budget (session + daily-rolling), always present; and
+ *   - the subscription rate-limit standing (5-hour + weekly pacing), present
+ *     once we've seen Anthropic's unified headers via the proxy, or estimated
+ *     when the user has pinned a plan tier. Alert-only — never blocks.
  */
 import { loadConfig } from "./config.js";
 import { isPaused, pauseExpiry } from "./config.js";
 import { loadLedger, rollingDailyCost } from "./ledger.js";
 import { evaluate } from "./budget.js";
+import { loadLimitsState } from "./limits.js";
+import { assessSnapshot, worstLevel } from "./pacing.js";
+import { estimateSnapshot } from "./estimate.js";
 const DAY_MS = 24 * 60 * 60 * 1000;
+function buildLimitsReport(cfg, ledger, now) {
+    const state = loadLimitsState();
+    const thresholds = cfg.limits;
+    const plan = cfg.limits.plan;
+    // Prefer real header data when we have it.
+    if (state.snapshot) {
+        const windows = assessSnapshot(state.snapshot, thresholds, now);
+        return {
+            source: "headers",
+            plan,
+            subscriptionDetected: state.subscriptionDetected,
+            observedAt: state.snapshot.observedAt,
+            windows,
+            level: worstLevel(windows),
+        };
+    }
+    // Otherwise estimate, but only when the user pinned a tier (opt-in, fuzzy).
+    if (plan === "pro" || plan === "max5" || plan === "max20") {
+        const snap = estimateSnapshot(ledger, plan, now);
+        const windows = assessSnapshot(snap, thresholds, now);
+        return {
+            source: "estimated",
+            plan,
+            subscriptionDetected: state.subscriptionDetected,
+            observedAt: snap.observedAt,
+            windows,
+            level: worstLevel(windows),
+        };
+    }
+    return {
+        source: "none",
+        plan,
+        subscriptionDetected: state.subscriptionDetected,
+        observedAt: null,
+        windows: [],
+        level: "ok",
+    };
+}
+function bar(frac) {
+    const pct = Math.max(0, Math.min(100, Math.round(frac * 100)));
+    const filled = Math.round(pct / 5);
+    return `[${"█".repeat(filled)}${"░".repeat(20 - filled)}]`;
+}
+function ageString(observedAt, now) {
+    const ms = now - observedAt;
+    if (ms < 60_000)
+        return "just now";
+    if (ms < 3_600_000)
+        return `${Math.round(ms / 60_000)}m ago`;
+    if (ms < 86_400_000)
+        return `${Math.round(ms / 3_600_000)}h ago`;
+    return `${Math.round(ms / 86_400_000)}d ago`;
+}
+/**
+ * Render the subscription rate-limit section as plain text lines (no color), so
+ * both the `agent-guard` and `ks guard` status views stay identical. Returns an
+ * empty array when there's nothing useful to show.
+ */
+export function formatLimitsLines(limits, now = Date.now()) {
+    if (limits.source === "none") {
+        // Only nudge if they haven't opted into either path.
+        if (!limits.subscriptionDetected) {
+            return [
+                "Claude Code plan limits: unknown.",
+                "  Run `ks guard proxy` and point Claude Code at it for exact 5-hour + weekly usage,",
+                "  or set your tier (`ks guard config --plan max5`) for an estimate.",
+            ];
+        }
+        return [];
+    }
+    const icon = limits.level === "danger" ? "🟥" : limits.level === "warn" ? "🟡" : "🟢";
+    const tag = limits.source === "estimated" ? " (estimated — run the proxy for exact)" : "";
+    const lines = [`${icon} Claude Code plan limits${tag}  ·  observed ${limits.observedAt ? ageString(limits.observedAt, now) : "—"}`];
+    for (const w of limits.windows) {
+        // w.message already leads with "<window> limit NN% used, …", so the bar
+        // carries the visual and the message carries the numbers + pacing.
+        lines.push(`  ${bar(w.utilization)}  ${w.message}`);
+    }
+    return lines;
+}
 /** Build the current status report from the on-disk config + ledger. */
 export function buildStatusReport(now = Date.now()) {
     const cfg = loadConfig();
@@ -26,5 +115,6 @@ export function buildStatusReport(now = Date.now()) {
         paused: isPaused(now),
         pauseUntil: pauseExpiry(),
         sessions,
+        limits: buildLimitsReport(cfg, ledger, now),
     };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kill-switch/agent-guard",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Kill Switch for coding agents — stop runaway Claude Code / Cursor / Aider sessions from racking up an LLM bill. Native hook + token-metering proxy with per-session and daily-rolling budgets.",
   "type": "module",
   "bin": {
@@ -12,7 +12,8 @@
   "scripts": {
     "build": "tsc",
     "dev": "tsx src/cli.ts",
-    "test": "vitest run"
+    "test": "vitest run",
+    "e2e": "tsc && node scripts/e2e-subscription.mjs"
   },
   "dependencies": {
     "commander": "^12.1.0"