@nervekit/orchestrator 0.1.0 → 0.2.0

package/dist/usage/subscription-usage-service.js

@@ -1,131 +0,0 @@
-import { SUBSCRIPTION_USAGE_EVENT, } from "@nerve/shared";
-import { fetchAnthropicUsage as defaultFetchAnthropicUsage } from "./anthropic-client.js";
-import { fetchCodexUsage as defaultFetchCodexUsage, mergeCodexUsage, parseCodexUsageHeaders, writeCodexUsageCache, } from "./codex-client.js";
-const MIN_REFRESH_INTERVAL_MS = 30_000;
-const GLOBAL_PROVIDERS = [
-    "anthropic",
-    "openai-codex",
-];
-/**
- * Manages provider subscription usage (Anthropic 5h/7d, Codex
- * primary/secondary) for globally supported OAuth providers.
- *
- * Refreshing is request-driven: callers ask for fresh snapshots and the service
- * debounces upstream provider calls to at most one attempt per provider every
- * {@link MIN_REFRESH_INTERVAL_MS}. Codex also receives live header updates via
- * {@link applyCodexHeaders}.
- */
-export class SubscriptionUsageService {
-    #deps;
-    #snapshots = new Map();
-    #inFlight = new Map();
-    #lastAttemptAt = new Map();
-    constructor(deps) {
-        this.#deps = deps;
-    }
-    start() {
-        // Refreshes are triggered by GET /api/usage/subscription and by agent runs.
-    }
-    stop() {
-        // No background timer to stop.
-    }
-    /** Current snapshots, optionally refreshing all globally supported providers. */
-    async getSnapshots(options = {}) {
-        if (options.refresh) {
-            await Promise.all(GLOBAL_PROVIDERS.map((provider) => this.refreshProvider(provider)));
-        }
-        return this.configuredSnapshots();
-    }
-    /** Mark a model provider as used and request a debounced refresh. */
-    touchProvider(provider) {
-        if (!isSubscriptionUsageProvider(provider))
-            return;
-        void this.refreshProvider(provider);
-    }
-    /** Apply a Codex usage snapshot derived from provider response headers. */
-    applyCodexHeaders(headers) {
-        const parsed = parseCodexUsageHeaders(headers);
-        if (!parsed)
-            return;
-        const now = this.now();
-        this.#lastAttemptAt.set("openai-codex", now);
-        const merged = mergeCodexUsage(this.#snapshots.get("openai-codex") ?? null, parsed);
-        this.#snapshots.set("openai-codex", merged);
-        void writeCodexUsageCache(this.#deps.cacheDir, merged);
-        void this.#deps.events.publish(SUBSCRIPTION_USAGE_EVENT, merged, {
-            durability: "transient",
-        });
-    }
-    async configuredSnapshots() {
-        const snapshots = [];
-        for (const provider of GLOBAL_PROVIDERS) {
-            if ((await this.#deps.auth.credentialType(provider)) !== "oauth") {
-                this.#snapshots.delete(provider);
-                continue;
-            }
-            const snapshot = this.#snapshots.get(provider);
-            if (snapshot)
-                snapshots.push(snapshot);
-        }
-        return snapshots;
-    }
-    refreshProvider(provider) {
-        const existing = this.#inFlight.get(provider);
-        if (existing)
-            return existing;
-        const refresh = this.performRefresh(provider).finally(() => {
-            this.#inFlight.delete(provider);
-        });
-        this.#inFlight.set(provider, refresh);
-        return refresh;
-    }
-    async performRefresh(provider) {
-        try {
-            if ((await this.#deps.auth.credentialType(provider)) !== "oauth") {
-                this.#snapshots.delete(provider);
-                return;
-            }
-            const now = this.now();
-            const lastAttemptAt = this.#lastAttemptAt.get(provider);
-            if (lastAttemptAt !== undefined &&
-                now - lastAttemptAt < MIN_REFRESH_INTERVAL_MS) {
-                return;
-            }
-            const token = await this.#deps.auth.getApiKey(provider);
-            if (!token)
-                return;
-            this.#lastAttemptAt.set(provider, now);
-            const fetcher = provider === "anthropic"
-                ? (this.#deps.fetchAnthropicUsage ?? defaultFetchAnthropicUsage)
-                : (this.#deps.fetchCodexUsage ?? defaultFetchCodexUsage);
-            const data = await fetcher(token, this.#deps.cacheDir);
-            if (!data)
-                return;
-            const previous = this.#snapshots.get(provider);
-            this.#snapshots.set(provider, data);
-            if (!previous || hasChanged(previous, data)) {
-                await this.#deps.events.publish(SUBSCRIPTION_USAGE_EVENT, data, {
-                    durability: "transient",
-                });
-            }
-        }
-        catch {
-            // transient failure; keep last-known snapshot
-        }
-    }
-    now() {
-        return this.#deps.now?.() ?? Date.now();
-    }
-}
-function isSubscriptionUsageProvider(provider) {
-    return provider === "anthropic" || provider === "openai-codex";
-}
-function hasChanged(a, b) {
-    return (a.session?.usedPercent !== b.session?.usedPercent ||
-        a.weekly?.usedPercent !== b.weekly?.usedPercent ||
-        a.session?.resetsAt !== b.session?.resetsAt ||
-        a.weekly?.resetsAt !== b.weekly?.resetsAt ||
-        a.session?.resetAfterSeconds !== b.session?.resetAfterSeconds ||
-        a.weekly?.resetAfterSeconds !== b.weekly?.resetAfterSeconds);
-}
-//# sourceMappingURL=subscription-usage-service.js.map

package/dist/usage/subscription-usage-service.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"subscription-usage-service.js","sourceRoot":"","sources":["../../src/usage/subscription-usage-service.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,wBAAwB,GAGzB,MAAM,eAAe,CAAC;AAGvB,OAAO,EAAE,mBAAmB,IAAI,0BAA0B,EAAE,MAAM,uBAAuB,CAAC;AAC1F,OAAO,EACL,eAAe,IAAI,sBAAsB,EACzC,eAAe,EACf,sBAAsB,EACtB,oBAAoB,GACrB,MAAM,mBAAmB,CAAC;AAE3B,MAAM,uBAAuB,GAAG,MAAM,CAAC;AACvC,MAAM,gBAAgB,GAAyC;IAC7D,WAAW;IACX,cAAc;CACf,CAAC;AAiBF;;;;;;;;GAQG;AACH,MAAM,OAAO,wBAAwB;IAC1B,KAAK,CAA+B;IACpC,UAAU,GAAG,IAAI,GAAG,EAAgD,CAAC;IACrE,SAAS,GAAG,IAAI,GAAG,EAA4C,CAAC;IAChE,cAAc,GAAG,IAAI,GAAG,EAAqC,CAAC;IAEvE,YAAY,IAAkC;QAC5C,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;IACpB,CAAC;IAED,KAAK;QACH,4EAA4E;IAC9E,CAAC;IAED,IAAI;QACF,+BAA+B;IACjC,CAAC;IAED,iFAAiF;IACjF,KAAK,CAAC,YAAY,CAChB,UAAiC,EAAE;QAEnC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,OAAO,CAAC,GAAG,CACf,gBAAgB,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CACnE,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,mBAAmB,EAAE,CAAC;IACpC,CAAC;IAED,qEAAqE;IACrE,aAAa,CAAC,QAAgB;QAC5B,IAAI,CAAC,2BAA2B,CAAC,QAAQ,CAAC;YAAE,OAAO;QACnD,KAAK,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;IACtC,CAAC;IAED,2EAA2E;IAC3E,iBAAiB,CAAC,OAA+B;QAC/C,MAAM,MAAM,GAAG,sBAAsB,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM;YAAE,OAAO;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,eAAe,CAC5B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,IAAI,EAC3C,MAAM,CACP,CAAC;QACF,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;QAC5C,KAAK,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACvD,KAAK,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,wBAAwB,EAAE,MAAM,EAAE;YAC/D,UAAU,EAAE,WAAW;SACxB,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,mBAAmB;QAC/B,MAAM,SAAS,GAAwB,EAAE,CAAC;QAC1C,KAAK,MAAM,QAAQ,IAAI,gBAAgB,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,KAAK,OAAO,EAAE,CAAC;gBACjE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACjC,SAAS;YACX,CAAC;YACD,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC/C,IAAI,QAAQ;gBAAE,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,eAAe,CAAC,QAAmC;QACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC9C,IAAI,QAAQ;YAAE,OAAO,QAAQ,CAAC;QAE9B,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YACzD,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACtC,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,QAAmC;QAEnC,IAAI,CAAC;YACH,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,KAAK,OAAO,EAAE,CAAC;gBACjE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACjC,OAAO;YACT,CAAC;YACD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACxD,IACE,aAAa,KAAK,SAAS;gBAC3B,GAAG,GAAG,aAAa,GAAG,uBAAuB,EAC7C,CAAC;gBACD,OAAO;YACT,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACxD,IAAI,CAAC,KAAK;gBAAE,OAAO;YACnB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;YACvC,MAAM,OAAO,GACX,QAAQ,KAAK,WAAW;gBACtB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,mBAAmB,IAAI,0BAA0B,CAAC;gBAChE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,IAAI,sBAAsB,CAAC,CAAC;YAC7D,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YACvD,IAAI,CAAC,IAAI;gBAAE,OAAO;YAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC/C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YACpC,IAAI,CAAC,QAAQ,IAAI,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,CAAC;gBAC5C,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,wBAAwB,EAAE,IAAI,EAAE;oBAC9D,UAAU,EAAE,WAAW;iBACxB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,8CAA8C;QAChD,CAAC;IACH,CAAC;IAEO,GAAG;QACT,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;IAC1C,CAAC;CACF;AAED,SAAS,2BAA2B,CAClC,QAAgB;IAEhB,OAAO,QAAQ,KAAK,WAAW,IAAI,QAAQ,KAAK,cAAc,CAAC;AACjE,CAAC;AAED,SAAS,UAAU,CAAC,CAAoB,EAAE,CAAoB;IAC5D,OAAO,CACL,CAAC,CAAC,OAAO,EAAE,WAAW,KAAK,CAAC,CAAC,OAAO,EAAE,WAAW;QACjD,CAAC,CAAC,MAAM,EAAE,WAAW,KAAK,CAAC,CAAC,MAAM,EAAE,WAAW;QAC/C,CAAC,CAAC,OAAO,EAAE,QAAQ,KAAK,CAAC,CAAC,OAAO,EAAE,QAAQ;QAC3C,CAAC,CAAC,MAAM,EAAE,QAAQ,KAAK,CAAC,CAAC,MAAM,EAAE,QAAQ;QACzC,CAAC,CAAC,OAAO,EAAE,iBAAiB,KAAK,CAAC,CAAC,OAAO,EAAE,iBAAiB;QAC7D,CAAC,CAAC,MAAM,EAAE,iBAAiB,KAAK,CAAC,CAAC,MAAM,EAAE,iBAAiB,CAC5D,CAAC;AACJ,CAAC"}