@scrylog/cli 0.3.0 → 0.3.2

package/src/usage/clients/anthropic-web-usage.ts

@@ -1,329 +0,0 @@
-import type { UsageClient } from "../client";
-import type { ClaudeUsageSnapshot, NamedRateWindowLegacy, ProviderCostSnapshotLegacy, RateWindowLegacy, ClaudeWebSession } from "../types";
-import { classifyUsageError, UsageError } from "../errors";
-import { usageWebSessionCache } from "../web-session-cache";
-import { CLAUDE_WEB_CACHE_KEY } from "../resolvers/claude-web";
-
-export class InvalidCredentialsError extends UsageError {
-  constructor(message: string) {
-    super("invalid-credentials", message);
-    this.name = "InvalidCredentialsError";
-  }
-}
-
-interface Org {
-  uuid: string;
-  name?: string;
-  capabilities?: string[];
-}
-
-interface UsageResponse {
-  five_hour?: { utilization?: number | null; resets_at?: string | null } | null;
-  seven_day?: { utilization?: number | null; resets_at?: string | null } | null;
-  seven_day_sonnet?: { utilization?: number | null; resets_at?: string | null } | null;
-  seven_day_opus?: { utilization?: number | null; resets_at?: string | null } | null;
-  seven_day_design?: { utilization?: number | null; resets_at?: string | null } | null;
-  seven_day_routines?: { utilization?: number | null; resets_at?: string | null } | null;
-  [key: string]: unknown;
-}
-
-interface OverageResponse {
-  monthly_credit_limit?: number;
-  currency?: string;
-  used_credits?: number;
-  is_enabled?: boolean;
-}
-
-interface AccountResponse {
-  email_address?: string;
-  memberships?: Array<{
-    organization?: {
-      uuid: string;
-      name?: string;
-      rate_limit_tier?: string;
-      billing_type?: string;
-    };
-  }>;
-}
-
-const DESIGN_ALTERNATES = [
-  "seven_day_design",
-  "seven_day_claude_design",
-  "claude_design",
-  "design",
-  "seven_day_omelette",
-  "omelette",
-  "omelette_promotional",
-];
-
-const ROUTINES_ALTERNATES = [
-  "seven_day_routines",
-  "seven_day_claude_routines",
-  "claude_routines",
-  "routines",
-  "routine",
-  "seven_day_cowork",
-  "cowork",
-];
-
-const OAUTH_APPS_ALTERNATES = ["seven_day_oauth_apps", "oauth_apps"];
-
-function pickWindowData(data: UsageResponse, keys: string[]): { utilization?: number | null; resets_at?: string | null } | null {
-  for (const key of keys) {
-    const val = data[key];
-    if (val && typeof val === "object") {
-      return val as { utilization?: number | null; resets_at?: string | null };
-    }
-  }
-  return null;
-}
-
-function toRateWindow(
-  utilization: number | null | undefined,
-  resetsAt: string | null | undefined,
-  windowMinutes: number,
-): RateWindowLegacy {
-  const usedPercent = typeof utilization === "number" ? Math.max(0, Math.min(100, utilization)) : 0;
-  const resetsAtDate = resetsAt ? new Date(resetsAt) : null;
-  const resetDescription = resetsAtDate && !Number.isNaN(resetsAtDate.getTime())
-    ? resetsAtDate.toLocaleString()
-    : null;
-  return {
-    usedPercent,
-    windowMinutes,
-    resetsAt: resetsAtDate && !Number.isNaN(resetsAtDate.getTime()) ? resetsAtDate : null,
-    resetDescription,
-  };
-}
-
-export class AnthropicWebUsageClient implements UsageClient<ClaudeWebSession, ClaudeUsageSnapshot> {
-  async fetch(creds: ClaudeWebSession, options?: { signal?: AbortSignal }): Promise<ClaudeUsageSnapshot> {
-    const candidates: string[] = [...new Set(creds.cookieCandidates ?? (creds.cookieHeader ? [creds.cookieHeader] : []))].slice(0, 2);
-    let expectedRevision = creds.cacheRevision ?? usageWebSessionCache.observe(CLAUDE_WEB_CACHE_KEY).revision;
-
-    for (let index = 0; index < candidates.length; index++) {
-      let rotated = false;
-      try {
-        const snapshot = await this.fetchWithCookie(candidates[index]!, options?.signal, (cookieHeader) => {
-          const stored = usageWebSessionCache.store(CLAUDE_WEB_CACHE_KEY, cookieHeader, expectedRevision);
-          if (stored) expectedRevision = stored.revision;
-          rotated = true;
-        });
-        if (!rotated) {
-          const stored = usageWebSessionCache.store(CLAUDE_WEB_CACHE_KEY, candidates[index]!, expectedRevision);
-          if (stored) expectedRevision = stored.revision;
-        }
-        return snapshot;
-      } catch (error) {
-        const failure = classifyUsageError(error);
-        if (failure.kind === "invalid-credentials") {
-          if (usageWebSessionCache.clear(CLAUDE_WEB_CACHE_KEY, expectedRevision)) expectedRevision = 0;
-          if (index + 1 < candidates.length) continue;
-          // Advance the demand-driven cursor at most once per attempt, keeping
-          // the two-candidate bound across imported and rotated candidates.
-          const imported = candidates.length < 2 ? (await creds.importNext?.()) ?? null : null;
-          if (imported && !candidates.includes(imported)) {
-            candidates.push(imported);
-            continue;
-          }
-        }
-        throw failure;
-      }
-    }
-    throw new InvalidCredentialsError("Session cookie invalid or expired");
-  }
-
-  private async fetchWithCookie(
-    initialCookieHeader: string,
-    signal: AbortSignal | undefined,
-    onCookieRotation: (cookieHeader: string) => void,
-  ): Promise<ClaudeUsageSnapshot> {
-    let cookieHeader = initialCookieHeader;
-    const request = async (url: string): Promise<Response> => {
-      let response: Response;
-      try {
-        response = await fetch(url, {
-          signal,
-          headers: { Cookie: cookieHeader, Accept: "application/json" },
-        });
-      } catch (error) {
-        throw classifyUsageError(error);
-      }
-      const rotated = applyClaudeSessionRotation(cookieHeader, response.headers);
-      if (rotated !== cookieHeader) {
-        cookieHeader = rotated;
-        onCookieRotation(cookieHeader);
-      }
-      return response;
-    };
-
-    // 1. Fetch organizations
-    const orgRes = await request("https://claude.ai/api/organizations");
-
-    if (orgRes.status === 401) {
-      throw new InvalidCredentialsError("Session cookie invalid or expired");
-    }
-    if (orgRes.status === 403) throw new UsageError("forbidden", "Organizations access forbidden");
-    if (orgRes.status === 429) throw new UsageError("rate-limited", "Organizations fetch rate limited");
-    if (orgRes.status >= 500) throw new UsageError("transient", `Organizations fetch failed: ${orgRes.status}`);
-    if (!orgRes.ok) {
-      throw new UsageError("invalid-response", `Organizations fetch failed: ${orgRes.status}`);
-    }
-
-    const orgs = (await orgRes.json()) as Org[];
-    if (!Array.isArray(orgs) || orgs.length === 0) {
-      throw new UsageError("invalid-response", "No organizations found");
-    }
-
-    // Select org
-    let org = orgs.find((o) => o.capabilities?.includes("chat"));
-    if (!org) {
-      org = orgs.find((o) => JSON.stringify(o.capabilities) !== '["api"]');
-    }
-    if (!org) {
-      org = orgs[0];
-    }
-
-    // 2. Fetch usage
-    const usageRes = await request(`https://claude.ai/api/organizations/${org.uuid}/usage`);
-
-    if (usageRes.status === 401) {
-      throw new InvalidCredentialsError("Session cookie invalid or expired");
-    }
-    if (usageRes.status === 403) throw new UsageError("forbidden", "Usage access forbidden");
-    if (usageRes.status === 429) throw new UsageError("rate-limited", "Usage fetch rate limited");
-    if (usageRes.status >= 500) throw new UsageError("transient", `Usage fetch failed: ${usageRes.status}`);
-    if (!usageRes.ok) {
-      throw new UsageError("invalid-response", `Usage fetch failed: ${usageRes.status}`);
-    }
-
-    const data = (await usageRes.json()) as UsageResponse;
-
-    // Null or missing windows mean "no usage recorded for the period" — a
-    // valid response, presented as an empty window at 0% (matches the
-    // reference implementation's "100% left" presentation).
-    const primaryWindow = pickWindowData(data, ["five_hour"]);
-    const secondaryWindow = pickWindowData(data, ["seven_day"]);
-
-    const primary = toRateWindow(
-      primaryWindow && typeof primaryWindow.utilization === "number" ? primaryWindow.utilization : 0,
-      primaryWindow?.resets_at ?? null,
-      300,
-    );
-    const secondary = secondaryWindow && typeof secondaryWindow.utilization === "number"
-      ? toRateWindow(secondaryWindow.utilization, secondaryWindow.resets_at, 10080)
-      : null;
-
-    const opusWindow = pickWindowData(data, ["seven_day_sonnet", "seven_day_opus"]);
-    const opus = opusWindow && typeof opusWindow.utilization === "number"
-      ? toRateWindow(opusWindow.utilization, opusWindow.resets_at, 10080)
-      : null;
-
-    const extraRateWindows: NamedRateWindowLegacy[] = [];
-    const designWindow = pickWindowData(data, DESIGN_ALTERNATES);
-    if (designWindow && typeof designWindow.utilization === "number") {
-      extraRateWindows.push({
-        id: "claude-design",
-        title: "Designs",
-        window: toRateWindow(designWindow.utilization, designWindow.resets_at, 10080),
-      });
-    }
-    const routinesWindow = pickWindowData(data, ROUTINES_ALTERNATES);
-    if (routinesWindow && typeof routinesWindow.utilization === "number") {
-      extraRateWindows.push({
-        id: "claude-routines",
-        title: "Daily Routines",
-        window: toRateWindow(routinesWindow.utilization, routinesWindow.resets_at, 10080),
-      });
-    }
-    const oauthAppsWindow = pickWindowData(data, OAUTH_APPS_ALTERNATES);
-    if (oauthAppsWindow && typeof oauthAppsWindow.utilization === "number") {
-      extraRateWindows.push({
-        id: "claude-oauth-apps",
-        title: "OAuth Apps",
-        window: toRateWindow(oauthAppsWindow.utilization, oauthAppsWindow.resets_at, 10080),
-      });
-    }
-
-    // 3. Best-effort overage
-    let providerCost: ProviderCostSnapshotLegacy | null = null;
-    try {
-      const overageRes = await request(`https://claude.ai/api/organizations/${org.uuid}/overage_spend_limit`);
-      if (overageRes.ok) {
-        const overage = (await overageRes.json()) as OverageResponse;
-        if (overage.is_enabled) {
-          providerCost = {
-            used: (overage.used_credits ?? 0) / 100,
-            limit: (overage.monthly_credit_limit ?? 0) / 100,
-            currencyCode: (overage.currency || "USD").trim() || "USD",
-            period: "Monthly",
-            resetsAt: null,
-            updatedAt: new Date(),
-          };
-        }
-      }
-    } catch (error) {
-      if (classifyUsageError(error).kind === "cancelled") throw error;
-      // Ignore best-effort failure
-    }
-
-    // 4. Best-effort account info
-    let accountEmail: string | null = null;
-    let accountOrganization: string | null = null;
-    let accountOrganizationId: string | null = null;
-    let loginMethod: string | null = null;
-    try {
-      const accountRes = await request("https://claude.ai/api/account");
-      if (accountRes.ok) {
-        const account = (await accountRes.json()) as AccountResponse;
-        if (account.email_address) {
-          accountEmail = account.email_address.trim();
-        }
-        const membership = account.memberships?.find((m) => m.organization?.uuid === org.uuid);
-        if (membership?.organization?.name) {
-          accountOrganization = membership.organization.name;
-          if (membership.organization.rate_limit_tier) {
-            loginMethod = membership.organization.rate_limit_tier;
-          }
-        }
-        accountOrganizationId = org.uuid;
-      }
-    } catch (error) {
-      if (classifyUsageError(error).kind === "cancelled") throw error;
-      // Ignore best-effort failure
-    }
-
-    return {
-      source: "web",
-      primary,
-      secondary,
-      opus,
-      extraRateWindows,
-      providerCost,
-      updatedAt: new Date(),
-      accountEmail,
-      accountOrganization,
-      accountOrganizationId,
-      loginMethod,
-      subscriptionType: null,
-      rateLimitTier: null,
-    };
-  }
-}
-
-function applyClaudeSessionRotation(current: string, headers: Headers): string {
-  const values = (headers as Headers & { getSetCookie?: () => string[] }).getSetCookie?.()
-    ?? (headers.get("set-cookie") ? [headers.get("set-cookie")!] : []);
-  let sessionKey = current.split(";").map((part) => part.trim()).find((part) => part.startsWith("sessionKey="));
-  for (const value of values) {
-    const [pair, ...attributes] = value.split(";");
-    if (!pair?.trim().startsWith("sessionKey=")) continue;
-    const expired = attributes.some((attribute) => {
-      const [name, raw = ""] = attribute.trim().split("=", 2);
-      if (name?.toLowerCase() === "max-age") return Number(raw) <= 0;
-      return name?.toLowerCase() === "expires" && Date.parse(raw) <= Date.now();
-    });
-    sessionKey = expired || pair.trim() === "sessionKey=" ? undefined : pair.trim();
-  }
-  return sessionKey ?? "";
-}