claude-cup 0.4.1 → 0.7.0

package/src/usage-api.js

@@ -1,250 +1,250 @@
-// Polls Anthropic's OAuth usage endpoint (the same data /usage shows in Claude Code).
-// Strictly read-only on credentials. Cached aggressively: the endpoint rate-limits hard.
-import { EventEmitter } from 'node:events';
-import { readFileSync, writeFileSync, mkdirSync } from 'node:fs';
-import { join, dirname } from 'node:path';
-
-const USAGE_URL = 'https://api.anthropic.com/api/oauth/usage';
-const POLL_MS = 5 * 60 * 1000;
-const BACKOFF_429_MS = 10 * 60 * 1000;
-
-// utilization is a percentage (verified against the live endpoint and the
-// Claude UI side by side: raw value 1 === "1%"). Never reinterpret it.
-function normPct(v) {
-  if (typeof v !== 'number' || !Number.isFinite(v)) return null;
-  return Math.max(0, Math.min(100, Math.round(v * 10) / 10));
-}
-
-function normBucket(b) {
-  if (!b || typeof b !== 'object') return null;
-  const pct = normPct(b.utilization);
-  if (pct === null) return null;
-  return { pct, resetsAt: b.resets_at || null };
-}
-
-export class UsagePoller extends EventEmitter {
-  constructor({ configDir, cachePath, intervalMs = POLL_MS, fetchImpl = fetch, now = () => Date.now() } = {}) {
-    super();
-    this.configDir = configDir;
-    this.cachePath = cachePath || null;
-    this.intervalMs = intervalMs;
-    this.fetch = fetchImpl;
-    this.now = now;
-    this.state = { status: 'unavailable', reason: 'not polled yet', fetchedAt: 0 };
-    this.samples = []; // {t, pct} history of the 5h window, for pace estimation
-    this.backoffUntil = 0;
-    this.timer = null;
-    this._loadDiskCache();
-  }
-
-  /** Keep only samples after the most recent window reset, sorted, max 90 min old. */
-  _normalizeSamples(arr) {
-    const cutoff = this.now() - 90 * 60000;
-    const clean = arr
-      .filter((s) => s && typeof s.t === 'number' && typeof s.pct === 'number' && s.t >= cutoff)
-      .sort((a, b) => a.t - b.t);
-    let cut = 0;
-    for (let i = 1; i < clean.length; i++) {
-      if (clean[i].pct < clean[i - 1].pct - 0.5) cut = i; // window reset happened here
-    }
-    return clean.slice(cut);
-  }
-
-  _loadDiskCache() {
-    if (!this.cachePath) return;
-    try {
-      const c = JSON.parse(readFileSync(this.cachePath, 'utf8'));
-      if (c && c.fiveHour !== undefined) {
-        const { samples, ...cachedState } = c;
-        this.state = { ...cachedState, status: 'stale', reason: 'cached from previous run' };
-        // restore pace history persisted by any previous claude-cup process
-        if (Array.isArray(samples)) {
-          this.samples = this._normalizeSamples(samples);
-        } else if (c.fiveHour && typeof c.fetchedAt === 'number' && this.now() - c.fetchedAt < 45 * 60000) {
-          this.samples = [{ t: c.fetchedAt, pct: c.fiveHour.pct }];
-        }
-        if (this.state.fiveHour) this.state.timeLeft = this._timeLeft();
-      }
-    } catch {
-      /* no cache yet */
-    }
-  }
-
-  /**
-   * Instance cooperation: if another claude-cup process (TUI, web, statusline)
-   * fetched fresh data recently, adopt its result and pace samples from the
-   * shared cache file instead of hitting the rate-limited endpoint again.
-   */
-  _adoptSharedCache() {
-    if (!this.cachePath) return false;
-    try {
-      const c = JSON.parse(readFileSync(this.cachePath, 'utf8'));
-      if (!c || typeof c.fetchedAt !== 'number' || !c.fiveHour) return false;
-      if (this.now() - c.fetchedAt >= Math.min(this.intervalMs, POLL_MS)) return false;
-      const newer = c.fetchedAt > (this.state.fetchedAt || 0);
-      const refreshable = this.state.status !== 'ok' && c.fetchedAt >= (this.state.fetchedAt || 0);
-      if (!newer && !refreshable) return false;
-      const { samples, ...freshState } = c;
-      const merged = [...this.samples];
-      const seen = new Set(merged.map((s) => s.t));
-      for (const s of Array.isArray(samples) ? samples : []) {
-        if (s && !seen.has(s.t)) merged.push(s);
-      }
-      this.samples = this._normalizeSamples(merged);
-      this.state = { ...freshState, status: 'ok', reason: null };
-      this.state.timeLeft = this._timeLeft();
-      this.emit('usage', this.state);
-      return true;
-    } catch {
-      return false;
-    }
-  }
-
-  _recordSample() {
-    const pct = this.state.fiveHour?.pct;
-    if (pct == null) return;
-    const t = this.now();
-    const last = this.samples[this.samples.length - 1];
-    // a noticeable drop means the 5h window reset - old pace is meaningless
-    if (last && pct < last.pct - 0.5) this.samples = [];
-    this.samples.push({ t, pct });
-    const cutoff = t - 90 * 60000;
-    this.samples = this.samples.filter((s) => s.t >= cutoff);
-  }
-
-  /**
-   * Projects time until the 5h limit is hit, from the measured pace of the
-   * official used-% (covers ALL the user's Claude surfaces - desktop, CLI, web).
-   * @returns {null | {outlasts: true} | {minutes: number, outlasts: false}}
-   *   null = not enough data yet ("measuring")
-   *   outlasts = at this pace the window resets before the limit is reached
-   */
-  _timeLeft() {
-    const f = this.state.fiveHour;
-    if (!f || f.pct == null) return null;
-    const t = this.now();
-    const win = this.samples.filter((s) => s.t >= t - 60 * 60000);
-    if (win.length < 2) return null;
-    const first = win[0];
-    const last = win[win.length - 1];
-    const spanMin = (last.t - first.t) / 60000;
-    if (spanMin < 4) return null;
-    const slope = (last.pct - first.pct) / spanMin; // % per minute
-    const resetsInMin = f.resetsAt ? Math.max(0, (Date.parse(f.resetsAt) - t) / 60000) : null;
-    if (slope <= 0.02) return { outlasts: true }; // idle or near-idle burn
-    const minutes = (100 - last.pct) / slope;
-    if (resetsInMin !== null && minutes >= resetsInMin) return { outlasts: true };
-    return { minutes: Math.round(minutes), outlasts: false };
-  }
-
-  _saveDiskCache() {
-    if (!this.cachePath) return;
-    try {
-      mkdirSync(dirname(this.cachePath), { recursive: true });
-      writeFileSync(this.cachePath, JSON.stringify({ ...this.state, samples: this.samples }));
-    } catch {
-      /* non-fatal */
-    }
-  }
-
-  readToken() {
-    const file = join(this.configDir, '.credentials.json');
-    try {
-      const creds = JSON.parse(readFileSync(file, 'utf8'));
-      const oauth = creds.claudeAiOauth;
-      if (!oauth || !oauth.accessToken) return { error: 'no token in credentials file' };
-      return {
-        token: oauth.accessToken,
-        expired: typeof oauth.expiresAt === 'number' && oauth.expiresAt < Date.now(),
-        tier: oauth.rateLimitTier || oauth.subscriptionType || null,
-      };
-    } catch {
-      return { error: 'credentials file not found (is Claude Code logged in?)' };
-    }
-  }
-
-  start() {
-    this.pollNow();
-    this.timer = setInterval(() => this.pollNow(), this.intervalMs);
-    this.timer.unref?.();
-  }
-
-  stop() {
-    if (this.timer) clearInterval(this.timer);
-  }
-
-  _degrade(reason) {
-    if (this.state.fiveHour) {
-      this.state = { ...this.state, status: 'stale', reason };
-      // pace projection still works from the samples we already have
-      this.state.timeLeft = this._timeLeft();
-    } else {
-      this.state = { status: 'unavailable', reason, fetchedAt: 0 };
-    }
-    this.emit('usage', this.state);
-  }
-
-  async pollNow() {
-    if (Date.now() < this.backoffUntil) return this.state;
-    // another claude-cup process may have fetched moments ago - reuse its data
-    if (this._adoptSharedCache()) return this.state;
-    const cred = this.readToken();
-    if (cred.error) {
-      this._degrade(cred.error);
-      return this.state;
-    }
-
-    try {
-      const ctrl = new AbortController();
-      const t = setTimeout(() => ctrl.abort(), 10_000);
-      const res = await this.fetch(USAGE_URL, {
-        headers: {
-          Authorization: `Bearer ${cred.token}`,
-          'anthropic-beta': 'oauth-2025-04-20',
-        },
-        signal: ctrl.signal,
-      });
-      clearTimeout(t);
-
-      if (res.status === 429) {
-        this.backoffUntil = Date.now() + BACKOFF_429_MS;
-        this._degrade('rate limited by usage endpoint, backing off');
-        return this.state;
-      }
-      if (res.status === 401 || res.status === 403) {
-        this._degrade(cred.expired ? 'token expired (open Claude Code to refresh it)' : 'token rejected');
-        return this.state;
-      }
-      if (!res.ok) {
-        this._degrade(`usage endpoint returned ${res.status}`);
-        return this.state;
-      }
-
-      const json = await res.json();
-      this.state = {
-        status: 'ok',
-        reason: null,
-        fetchedAt: this.now(),
-        tier: cred.tier,
-        fiveHour: normBucket(json.five_hour),
-        sevenDay: normBucket(json.seven_day),
-        sevenDaySonnet: normBucket(json.seven_day_sonnet),
-        sevenDayOpus: normBucket(json.seven_day_opus),
-        extraUsage: json.extra_usage && json.extra_usage.is_enabled
-          ? {
-              monthlyLimit: json.extra_usage.monthly_limit ?? null,
-              usedCredits: json.extra_usage.used_credits ?? null,
-            }
-          : null,
-      };
-      this._recordSample();
-      this.state.timeLeft = this._timeLeft();
-      this._saveDiskCache();
-      this.emit('usage', this.state);
-      return this.state;
-    } catch (err) {
-      this._degrade(err.name === 'AbortError' ? 'usage request timed out' : 'network error reaching usage endpoint');
-      return this.state;
-    }
-  }
-}
+// Polls Anthropic's OAuth usage endpoint (the same data /usage shows in Claude Code).
+// Strictly read-only on credentials. Cached aggressively: the endpoint rate-limits hard.
+import { EventEmitter } from 'node:events';
+import { readFileSync, writeFileSync, mkdirSync } from 'node:fs';
+import { join, dirname } from 'node:path';
+
+const USAGE_URL = 'https://api.anthropic.com/api/oauth/usage';
+const POLL_MS = 5 * 60 * 1000;
+const BACKOFF_429_MS = 10 * 60 * 1000;
+
+// utilization is a percentage (verified against the live endpoint and the
+// Claude UI side by side: raw value 1 === "1%"). Never reinterpret it.
+function normPct(v) {
+  if (typeof v !== 'number' || !Number.isFinite(v)) return null;
+  return Math.max(0, Math.min(100, Math.round(v * 10) / 10));
+}
+
+function normBucket(b) {
+  if (!b || typeof b !== 'object') return null;
+  const pct = normPct(b.utilization);
+  if (pct === null) return null;
+  return { pct, resetsAt: b.resets_at || null };
+}
+
+export class UsagePoller extends EventEmitter {
+  constructor({ configDir, cachePath, intervalMs = POLL_MS, fetchImpl = fetch, now = () => Date.now() } = {}) {
+    super();
+    this.configDir = configDir;
+    this.cachePath = cachePath || null;
+    this.intervalMs = intervalMs;
+    this.fetch = fetchImpl;
+    this.now = now;
+    this.state = { status: 'unavailable', reason: 'not polled yet', fetchedAt: 0 };
+    this.samples = []; // {t, pct} history of the 5h window, for pace estimation
+    this.backoffUntil = 0;
+    this.timer = null;
+    this._loadDiskCache();
+  }
+
+  /** Keep only samples after the most recent window reset, sorted, max 90 min old. */
+  _normalizeSamples(arr) {
+    const cutoff = this.now() - 90 * 60000;
+    const clean = arr
+      .filter((s) => s && typeof s.t === 'number' && typeof s.pct === 'number' && s.t >= cutoff)
+      .sort((a, b) => a.t - b.t);
+    let cut = 0;
+    for (let i = 1; i < clean.length; i++) {
+      if (clean[i].pct < clean[i - 1].pct - 0.5) cut = i; // window reset happened here
+    }
+    return clean.slice(cut);
+  }
+
+  _loadDiskCache() {
+    if (!this.cachePath) return;
+    try {
+      const c = JSON.parse(readFileSync(this.cachePath, 'utf8'));
+      if (c && c.fiveHour !== undefined) {
+        const { samples, ...cachedState } = c;
+        this.state = { ...cachedState, status: 'stale', reason: 'cached from previous run' };
+        // restore pace history persisted by any previous claude-cup process
+        if (Array.isArray(samples)) {
+          this.samples = this._normalizeSamples(samples);
+        } else if (c.fiveHour && typeof c.fetchedAt === 'number' && this.now() - c.fetchedAt < 45 * 60000) {
+          this.samples = [{ t: c.fetchedAt, pct: c.fiveHour.pct }];
+        }
+        if (this.state.fiveHour) this.state.timeLeft = this._timeLeft();
+      }
+    } catch {
+      /* no cache yet */
+    }
+  }
+
+  /**
+   * Instance cooperation: if another claude-cup process (TUI, web, statusline)
+   * fetched fresh data recently, adopt its result and pace samples from the
+   * shared cache file instead of hitting the rate-limited endpoint again.
+   */
+  _adoptSharedCache() {
+    if (!this.cachePath) return false;
+    try {
+      const c = JSON.parse(readFileSync(this.cachePath, 'utf8'));
+      if (!c || typeof c.fetchedAt !== 'number' || !c.fiveHour) return false;
+      if (this.now() - c.fetchedAt >= Math.min(this.intervalMs, POLL_MS)) return false;
+      const newer = c.fetchedAt > (this.state.fetchedAt || 0);
+      const refreshable = this.state.status !== 'ok' && c.fetchedAt >= (this.state.fetchedAt || 0);
+      if (!newer && !refreshable) return false;
+      const { samples, ...freshState } = c;
+      const merged = [...this.samples];
+      const seen = new Set(merged.map((s) => s.t));
+      for (const s of Array.isArray(samples) ? samples : []) {
+        if (s && !seen.has(s.t)) merged.push(s);
+      }
+      this.samples = this._normalizeSamples(merged);
+      this.state = { ...freshState, status: 'ok', reason: null };
+      this.state.timeLeft = this._timeLeft();
+      this.emit('usage', this.state);
+      return true;
+    } catch {
+      return false;
+    }
+  }
+
+  _recordSample() {
+    const pct = this.state.fiveHour?.pct;
+    if (pct == null) return;
+    const t = this.now();
+    const last = this.samples[this.samples.length - 1];
+    // a noticeable drop means the 5h window reset - old pace is meaningless
+    if (last && pct < last.pct - 0.5) this.samples = [];
+    this.samples.push({ t, pct });
+    const cutoff = t - 90 * 60000;
+    this.samples = this.samples.filter((s) => s.t >= cutoff);
+  }
+
+  /**
+   * Projects time until the 5h limit is hit, from the measured pace of the
+   * official used-% (covers ALL the user's Claude surfaces - desktop, CLI, web).
+   * @returns {null | {outlasts: true} | {minutes: number, outlasts: false}}
+   *   null = not enough data yet ("measuring")
+   *   outlasts = at this pace the window resets before the limit is reached
+   */
+  _timeLeft() {
+    const f = this.state.fiveHour;
+    if (!f || f.pct == null) return null;
+    const t = this.now();
+    const win = this.samples.filter((s) => s.t >= t - 60 * 60000);
+    if (win.length < 2) return null;
+    const first = win[0];
+    const last = win[win.length - 1];
+    const spanMin = (last.t - first.t) / 60000;
+    if (spanMin < 4) return null;
+    const slope = (last.pct - first.pct) / spanMin; // % per minute
+    const resetsInMin = f.resetsAt ? Math.max(0, (Date.parse(f.resetsAt) - t) / 60000) : null;
+    if (slope <= 0.02) return { outlasts: true }; // idle or near-idle burn
+    const minutes = (100 - last.pct) / slope;
+    if (resetsInMin !== null && minutes >= resetsInMin) return { outlasts: true };
+    return { minutes: Math.round(minutes), outlasts: false };
+  }
+
+  _saveDiskCache() {
+    if (!this.cachePath) return;
+    try {
+      mkdirSync(dirname(this.cachePath), { recursive: true });
+      writeFileSync(this.cachePath, JSON.stringify({ ...this.state, samples: this.samples }));
+    } catch {
+      /* non-fatal */
+    }
+  }
+
+  readToken() {
+    const file = join(this.configDir, '.credentials.json');
+    try {
+      const creds = JSON.parse(readFileSync(file, 'utf8'));
+      const oauth = creds.claudeAiOauth;
+      if (!oauth || !oauth.accessToken) return { error: 'no token in credentials file' };
+      return {
+        token: oauth.accessToken,
+        expired: typeof oauth.expiresAt === 'number' && oauth.expiresAt < Date.now(),
+        tier: oauth.rateLimitTier || oauth.subscriptionType || null,
+      };
+    } catch {
+      return { error: 'credentials file not found (is Claude Code logged in?)' };
+    }
+  }
+
+  start() {
+    this.pollNow();
+    this.timer = setInterval(() => this.pollNow(), this.intervalMs);
+    this.timer.unref?.();
+  }
+
+  stop() {
+    if (this.timer) clearInterval(this.timer);
+  }
+
+  _degrade(reason) {
+    if (this.state.fiveHour) {
+      this.state = { ...this.state, status: 'stale', reason };
+      // pace projection still works from the samples we already have
+      this.state.timeLeft = this._timeLeft();
+    } else {
+      this.state = { status: 'unavailable', reason, fetchedAt: 0 };
+    }
+    this.emit('usage', this.state);
+  }
+
+  async pollNow() {
+    if (Date.now() < this.backoffUntil) return this.state;
+    // another claude-cup process may have fetched moments ago - reuse its data
+    if (this._adoptSharedCache()) return this.state;
+    const cred = this.readToken();
+    if (cred.error) {
+      this._degrade(cred.error);
+      return this.state;
+    }
+
+    try {
+      const ctrl = new AbortController();
+      const t = setTimeout(() => ctrl.abort(), 10_000);
+      const res = await this.fetch(USAGE_URL, {
+        headers: {
+          Authorization: `Bearer ${cred.token}`,
+          'anthropic-beta': 'oauth-2025-04-20',
+        },
+        signal: ctrl.signal,
+      });
+      clearTimeout(t);
+
+      if (res.status === 429) {
+        this.backoffUntil = Date.now() + BACKOFF_429_MS;
+        this._degrade('rate limited by usage endpoint, backing off');
+        return this.state;
+      }
+      if (res.status === 401 || res.status === 403) {
+        this._degrade(cred.expired ? 'token expired (open Claude Code to refresh it)' : 'token rejected');
+        return this.state;
+      }
+      if (!res.ok) {
+        this._degrade(`usage endpoint returned ${res.status}`);
+        return this.state;
+      }
+
+      const json = await res.json();
+      this.state = {
+        status: 'ok',
+        reason: null,
+        fetchedAt: this.now(),
+        tier: cred.tier,
+        fiveHour: normBucket(json.five_hour),
+        sevenDay: normBucket(json.seven_day),
+        sevenDaySonnet: normBucket(json.seven_day_sonnet),
+        sevenDayOpus: normBucket(json.seven_day_opus),
+        extraUsage: json.extra_usage && json.extra_usage.is_enabled
+          ? {
+              monthlyLimit: json.extra_usage.monthly_limit ?? null,
+              usedCredits: json.extra_usage.used_credits ?? null,
+            }
+          : null,
+      };
+      this._recordSample();
+      this.state.timeLeft = this._timeLeft();
+      this._saveDiskCache();
+      this.emit('usage', this.state);
+      return this.state;
+    } catch (err) {
+      this._degrade(err.name === 'AbortError' ? 'usage request timed out' : 'network error reaching usage endpoint');
+      return this.state;
+    }
+  }
+}