@ashx-j/lunr-dev 0.2.13-dev.1.1

package/dist/core/usage-limit.js

@@ -0,0 +1,79 @@
+/**
+ * Provider error classification for subscription-key rotation.
+ *
+ * lunr: these classify provider free-text error bodies — regex-on-English,
+ * so wording drift across providers (or upstream changes to error text) can
+ * silently reclassify an error. Treat matches as heuristics, not contracts.
+ */
+const USAGE_LIMIT_ERROR_PATTERNS = [
+    // Provider-declared subscription/free-tier limit error types and wording
+    // (pi-ai NON_RETRYABLE_PROVIDER_LIMIT_ERROR_PATTERN spirit).
+    "GoUsageLimitError",
+    "FreeUsageLimitError",
+    "Monthly usage limit reached",
+    "available balance",
+    "insufficient_quota",
+    "out of budget",
+    "quota exceeded",
+    "billing",
+];
+const USAGE_LIMIT_ERROR_REGEXES = [
+    // Goal runtime USAGE_LIMIT_GOAL_ERROR_PATTERNS: usage caps, quota exhaustion,
+    // credit/balance exhaustion, payment required.
+    /usage[_\s-]*(?:limit|cap)|chatgpt.{0,32}usage/i,
+    /quota.{0,32}(?:reached|exceeded|exhausted|depleted)|(?:reached|exceeded|exhausted|depleted).{0,32}quota/i,
+    /insufficient[_\s-]*(?:quota|credits?)|out of credits|out of budget|available balance|payment required/i,
+    /(?:credit|balance).{0,32}(?:low|exhausted|depleted)|billing/i,
+];
+const AUTH_INVALID_ERROR_PATTERN = /\b401\b|unauthori[sz]ed|unauthenticated|invalid.{0,16}(?:api.?key|token|credentials?)|authentication.{0,16}failed/i;
+export function isUsageLimitError(errorMessage) {
+    if (!errorMessage)
+        return false;
+    if (USAGE_LIMIT_ERROR_PATTERNS.some((pattern) => errorMessage.toLowerCase().includes(pattern.toLowerCase()))) {
+        return true;
+    }
+    return USAGE_LIMIT_ERROR_REGEXES.some((pattern) => pattern.test(errorMessage));
+}
+export function isAuthInvalidError(errorMessage) {
+    if (!errorMessage)
+        return false;
+    return AUTH_INVALID_ERROR_PATTERN.test(errorMessage);
+}
+const RESET_TIME_PATTERNS = [
+    /reset(?:s|ting)?\s*(?:in|after|at)?\s*(\d+)\s*s(ec)?/i,
+    /reset(?:s|ting)?\s*(?:in|after|at)?\s*(\d+)\s*m(in)?/i,
+    /reset(?:s|ting)?\s*(?:in|after|at)?\s*(\d+)\s*h(our|r)?/i,
+    /(?:retry|try(?:\s+again)?|available)\s+(?:in|after|at)\s*(\d+)\s*s(ec)?/i,
+    /(?:retry|try(?:\s+again)?|available)\s+(?:in|after|at)\s*(\d+)\s*m(in)?/i,
+    /(?:retry|try(?:\s+again)?|available)\s+(?:in|after|at)\s*(\d+)\s*h(our|r)?/i,
+];
+/**
+ * Parse a quota-reset time out of an error message: relative forms like
+ * "reset in 2h" / "retry in 30m" / "available in 3600s", or an ISO-8601
+ * timestamp. Returns epoch ms in the future, or undefined when unparseable.
+ * Port of the goal runtime's parseUsageResetTimeMs.
+ */
+export function parseResetTimeMs(errorMessage) {
+    if (!errorMessage)
+        return undefined;
+    const now = Date.now();
+    for (const pattern of RESET_TIME_PATTERNS) {
+        const match = errorMessage.match(pattern);
+        if (!match)
+            continue;
+        const value = Number.parseInt(match[1] ?? "", 10);
+        if (!Number.isFinite(value) || value <= 0)
+            continue;
+        const unit = pattern.source.includes("\\s*h") ? 3600_000 : pattern.source.includes("\\s*m") ? 60_000 : 1_000;
+        return now + value * unit;
+    }
+    // ISO-8601 timestamp fallback (e.g. "reset at 2026-08-01T09:00:00Z").
+    const isoMatch = errorMessage.match(/(?:reset|available).{0,12}(\d{4}-\d{2}-\d{2}T\d{2}:\d{2}(:\d{2})?(?:\.\d+)?(?:Z|[+-]\d{2}:?\d{2})?)/i);
+    if (isoMatch?.[1]) {
+        const parsed = Date.parse(isoMatch[1]);
+        if (Number.isFinite(parsed) && parsed > now)
+            return parsed;
+    }
+    return undefined;
+}
+//# sourceMappingURL=usage-limit.js.map

package/dist/core/usage-limit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-limit.js","sourceRoot":"","sources":["../../src/core/usage-limit.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,MAAM,0BAA0B,GAAG;IAClC,yEAAyE;IACzE,6DAA6D;IAC7D,mBAAmB;IACnB,qBAAqB;IACrB,6BAA6B;IAC7B,mBAAmB;IACnB,oBAAoB;IACpB,eAAe;IACf,gBAAgB;IAChB,SAAS;CACA,CAAC;AAEX,MAAM,yBAAyB,GAAG;IACjC,8EAA8E;IAC9E,+CAA+C;IAC/C,gDAAgD;IAChD,0GAA0G;IAC1G,wGAAwG;IACxG,8DAA8D;CACrD,CAAC;AAEX,MAAM,0BAA0B,GAC/B,oHAAoH,CAAC;AAEtH,MAAM,UAAU,iBAAiB,CAAC,YAAoB,EAAW;IAChE,IAAI,CAAC,YAAY;QAAE,OAAO,KAAK,CAAC;IAChC,IAAI,0BAA0B,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,EAAE,CAAC;QAC9G,OAAO,IAAI,CAAC;IACb,CAAC;IACD,OAAO,yBAAyB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;AAAA,CAC/E;AAED,MAAM,UAAU,kBAAkB,CAAC,YAAoB,EAAW;IACjE,IAAI,CAAC,YAAY;QAAE,OAAO,KAAK,CAAC;IAChC,OAAO,0BAA0B,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AAAA,CACrD;AAED,MAAM,mBAAmB,GAAG;IAC3B,uDAAuD;IACvD,uDAAuD;IACvD,0DAA0D;IAC1D,0EAA0E;IAC1E,0EAA0E;IAC1E,6EAA6E;CACpE,CAAC;AAEX;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAAC,YAAoB,EAAsB;IAC1E,IAAI,CAAC,YAAY;QAAE,OAAO,SAAS,CAAC;IACpC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,KAAK,MAAM,OAAO,IAAI,mBAAmB,EAAE,CAAC;QAC3C,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC1C,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAClD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC;YAAE,SAAS;QACpD,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC;QAC7G,OAAO,GAAG,GAAG,KAAK,GAAG,IAAI,CAAC;IAC3B,CAAC;IACD,sEAAsE;IACtE,MAAM,QAAQ,GAAG,YAAY,CAAC,KAAK,CAClC,sGAAsG,CACtG,CAAC;IACF,IAAI,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QACvC,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,GAAG;YAAE,OAAO,MAAM,CAAC;IAC5D,CAAC;IACD,OAAO,SAAS,CAAC;AAAA,CACjB","sourcesContent":["/**\n * Provider error classification for subscription-key rotation.\n *\n * lunr: these classify provider free-text error bodies — regex-on-English,\n * so wording drift across providers (or upstream changes to error text) can\n * silently reclassify an error. Treat matches as heuristics, not contracts.\n */\n\nconst USAGE_LIMIT_ERROR_PATTERNS = [\n\t// Provider-declared subscription/free-tier limit error types and wording\n\t// (pi-ai NON_RETRYABLE_PROVIDER_LIMIT_ERROR_PATTERN spirit).\n\t\"GoUsageLimitError\",\n\t\"FreeUsageLimitError\",\n\t\"Monthly usage limit reached\",\n\t\"available balance\",\n\t\"insufficient_quota\",\n\t\"out of budget\",\n\t\"quota exceeded\",\n\t\"billing\",\n] as const;\n\nconst USAGE_LIMIT_ERROR_REGEXES = [\n\t// Goal runtime USAGE_LIMIT_GOAL_ERROR_PATTERNS: usage caps, quota exhaustion,\n\t// credit/balance exhaustion, payment required.\n\t/usage[_\\s-]*(?:limit|cap)|chatgpt.{0,32}usage/i,\n\t/quota.{0,32}(?:reached|exceeded|exhausted|depleted)|(?:reached|exceeded|exhausted|depleted).{0,32}quota/i,\n\t/insufficient[_\\s-]*(?:quota|credits?)|out of credits|out of budget|available balance|payment required/i,\n\t/(?:credit|balance).{0,32}(?:low|exhausted|depleted)|billing/i,\n] as const;\n\nconst AUTH_INVALID_ERROR_PATTERN =\n\t/\\b401\\b|unauthori[sz]ed|unauthenticated|invalid.{0,16}(?:api.?key|token|credentials?)|authentication.{0,16}failed/i;\n\nexport function isUsageLimitError(errorMessage: string): boolean {\n\tif (!errorMessage) return false;\n\tif (USAGE_LIMIT_ERROR_PATTERNS.some((pattern) => errorMessage.toLowerCase().includes(pattern.toLowerCase()))) {\n\t\treturn true;\n\t}\n\treturn USAGE_LIMIT_ERROR_REGEXES.some((pattern) => pattern.test(errorMessage));\n}\n\nexport function isAuthInvalidError(errorMessage: string): boolean {\n\tif (!errorMessage) return false;\n\treturn AUTH_INVALID_ERROR_PATTERN.test(errorMessage);\n}\n\nconst RESET_TIME_PATTERNS = [\n\t/reset(?:s|ting)?\\s*(?:in|after|at)?\\s*(\\d+)\\s*s(ec)?/i,\n\t/reset(?:s|ting)?\\s*(?:in|after|at)?\\s*(\\d+)\\s*m(in)?/i,\n\t/reset(?:s|ting)?\\s*(?:in|after|at)?\\s*(\\d+)\\s*h(our|r)?/i,\n\t/(?:retry|try(?:\\s+again)?|available)\\s+(?:in|after|at)\\s*(\\d+)\\s*s(ec)?/i,\n\t/(?:retry|try(?:\\s+again)?|available)\\s+(?:in|after|at)\\s*(\\d+)\\s*m(in)?/i,\n\t/(?:retry|try(?:\\s+again)?|available)\\s+(?:in|after|at)\\s*(\\d+)\\s*h(our|r)?/i,\n] as const;\n\n/**\n * Parse a quota-reset time out of an error message: relative forms like\n * \"reset in 2h\" / \"retry in 30m\" / \"available in 3600s\", or an ISO-8601\n * timestamp. Returns epoch ms in the future, or undefined when unparseable.\n * Port of the goal runtime's parseUsageResetTimeMs.\n */\nexport function parseResetTimeMs(errorMessage: string): number | undefined {\n\tif (!errorMessage) return undefined;\n\tconst now = Date.now();\n\tfor (const pattern of RESET_TIME_PATTERNS) {\n\t\tconst match = errorMessage.match(pattern);\n\t\tif (!match) continue;\n\t\tconst value = Number.parseInt(match[1] ?? \"\", 10);\n\t\tif (!Number.isFinite(value) || value <= 0) continue;\n\t\tconst unit = pattern.source.includes(\"\\\\s*h\") ? 3600_000 : pattern.source.includes(\"\\\\s*m\") ? 60_000 : 1_000;\n\t\treturn now + value * unit;\n\t}\n\t// ISO-8601 timestamp fallback (e.g. \"reset at 2026-08-01T09:00:00Z\").\n\tconst isoMatch = errorMessage.match(\n\t\t/(?:reset|available).{0,12}(\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}(:\\d{2})?(?:\\.\\d+)?(?:Z|[+-]\\d{2}:?\\d{2})?)/i,\n\t);\n\tif (isoMatch?.[1]) {\n\t\tconst parsed = Date.parse(isoMatch[1]);\n\t\tif (Number.isFinite(parsed) && parsed > now) return parsed;\n\t}\n\treturn undefined;\n}\n"]}

package/dist/core/usage-service.d.ts

@@ -0,0 +1,75 @@
+/**
+ * Plan/subscription usage service. Per-provider adapters query provider quota
+ * endpoints (60-second cache each) so /usage and the footer can show subscription windows
+ * alongside session/context totals.
+ *
+ * Deliberately absent adapters:
+ * - anthropic: Anthropic policy prohibits subscription usage tracking in
+ *   third-party coding agents (2026-07-18 decision). OAuth Anthropic sessions
+ *   fall back to hiding the cost meter.
+ * - ollama-cloud: no usage API exists.
+ *
+ * Adapter/network errors resolve to `undefined` (callers omit the plan UI).
+ * Auth failures are different: they return `error` and are not cached as empty.
+ */
+import type { ModelRuntime } from "./model-runtime.ts";
+import type { SettingsManager } from "./settings-manager.ts";
+export interface PlanUsageWindow {
+    label: string;
+    usedPercent: number;
+    /** Epoch ms when the window resets. */
+    resetsAt?: number;
+}
+export interface PlanUsage {
+    provider: string;
+    planLabel?: string;
+    windows: PlanUsageWindow[];
+}
+export interface PlanUsageResult {
+    /** First plan, kept for footer and compatibility callers. */
+    usage?: PlanUsage;
+    /** Every plan returned by the provider. */
+    usages?: PlanUsage[];
+    /** Actionable auth failure. Not cached as an empty success. */
+    error?: string;
+}
+export interface AllPlanUsageResult {
+    usages: PlanUsage[];
+    errors: string[];
+}
+/** Classify adapter/getAuth failures that mean the SuperGrok session is dead. */
+export declare function planUsageAuthError(error: unknown): string | undefined;
+export declare const PLAN_USAGE_CACHE_TTL_MS: number;
+export type PlanUsageWindowPreference = "5h" | "weekly";
+export declare function isWeeklyPlanWindow(window: PlanUsageWindow): boolean;
+export declare function isFiveHourPlanWindow(window: PlanUsageWindow): boolean;
+/** Preferred window, falling back to the other, skipping Extra/on-demand. */
+export declare function pickPlanWindow(usage: PlanUsage | undefined, preferred: PlanUsageWindowPreference): PlanUsageWindow | undefined;
+export declare function footerPlanLabel(window: PlanUsageWindow): string;
+/** Whether a plan-usage adapter exists for this provider. */
+export declare function hasPlanUsageAdapter(providerId: string): boolean;
+/** Plan usage for a provider, or undefined when unsupported/unavailable. Never throws. */
+export declare function getPlanUsage(providerId: string, runtime: ModelRuntime): Promise<PlanUsage | undefined>;
+/** Like getPlanUsage, but surfaces auth failures instead of hiding them. */
+export declare function getPlanUsageResult(providerId: string, runtime: ModelRuntime): Promise<PlanUsageResult>;
+/** Fetch every stored-credential plan plus the current provider, in parallel. */
+export declare function getAllPlanUsageResults(currentProvider: string | undefined, runtime: ModelRuntime): Promise<AllPlanUsageResult>;
+/** Last fetched usage, including expired cache (footer can show stale while refetching). */
+export declare function peekPlanUsage(providerId: string): PlanUsage | undefined;
+/** Clear the adapter cache (tests, /login changes). */
+export declare function clearPlanUsageCache(): void;
+export declare const USAGE_SERVICE_BRIDGE_SYMBOL: unique symbol;
+export interface FooterPlanSegment {
+    label: string;
+    usedPercent: number;
+}
+export interface UsageServiceBridge {
+    peek(providerId: string): PlanUsage | undefined;
+    prefetch(providerId: string): void;
+    getPreferredWindow(): PlanUsageWindowPreference;
+    pickForFooter(providerId: string): FooterPlanSegment | undefined;
+    setOnUpdate(fn: (() => void) | undefined): void;
+}
+export declare function registerUsageServiceBridge(runtime: ModelRuntime, settingsManager: SettingsManager): void;
+export declare function getUsageServiceBridge(): UsageServiceBridge | undefined;
+//# sourceMappingURL=usage-service.d.ts.map

package/dist/core/usage-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-service.d.ts","sourceRoot":"","sources":["../../src/core/usage-service.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAM7D,MAAM,WAAW,eAAe;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,uCAAuC;IACvC,QAAQ,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,SAAS;IACzB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,eAAe,EAAE,CAAC;CAC3B;AAED,MAAM,WAAW,eAAe;IAC/B,6DAA6D;IAC7D,KAAK,CAAC,EAAE,SAAS,CAAC;IAClB,2CAA2C;IAC3C,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC;IACrB,+DAA+D;IAC/D,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,kBAAkB;IAClC,MAAM,EAAE,SAAS,EAAE,CAAC;IACpB,MAAM,EAAE,MAAM,EAAE,CAAC;CACjB;AAoBD,iFAAiF;AACjF,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAOrE;AAWD,eAAO,MAAM,uBAAuB,QAAY,CAAC;AAEjD,MAAM,MAAM,yBAAyB,GAAG,IAAI,GAAG,QAAQ,CAAC;AAExD,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,eAAe,GAAG,OAAO,CAEnE;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,eAAe,GAAG,OAAO,CAErE;AAED,6EAA6E;AAC7E,wBAAgB,cAAc,CAC7B,KAAK,EAAE,SAAS,GAAG,SAAS,EAC5B,SAAS,EAAE,yBAAyB,GAClC,eAAe,GAAG,SAAS,CAO7B;AAED,wBAAgB,eAAe,CAAC,MAAM,EAAE,eAAe,GAAG,MAAM,CAI/D;AAKD,6DAA6D;AAC7D,wBAAgB,mBAAmB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAE/D;AAED,0FAA0F;AAC1F,wBAAsB,YAAY,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC,CAE5G;AAED,4EAA4E;AAC5E,wBAAsB,kBAAkB,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,eAAe,CAAC,CAyB5G;AAED,iFAAiF;AACjF,wBAAsB,sBAAsB,CAC3C,eAAe,EAAE,MAAM,GAAG,SAAS,EACnC,OAAO,EAAE,YAAY,GACnB,OAAO,CAAC,kBAAkB,CAAC,CAU7B;AAED,4FAA4F;AAC5F,wBAAgB,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,CAEvE;AAED,uDAAuD;AACvD,wBAAgB,mBAAmB,IAAI,IAAI,CAG1C;AAMD,eAAO,MAAM,2BAA2B,eAAoC,CAAC;AAE7E,MAAM,WAAW,iBAAiB;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IAClC,IAAI,CAAC,UAAU,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,CAAC;IAChD,QAAQ,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IACnC,kBAAkB,IAAI,yBAAyB,CAAC;IAChD,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,iBAAiB,GAAG,SAAS,CAAC;IACjE,WAAW,CAAC,EAAE,EAAE,CAAC,MAAM,IAAI,CAAC,GAAG,SAAS,GAAG,IAAI,CAAC;CAChD;AA+BD,wBAAgB,0BAA0B,CAAC,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,eAAe,GAAG,IAAI,CAIxG;AAED,wBAAgB,qBAAqB,IAAI,kBAAkB,GAAG,SAAS,CAEtE","sourcesContent":["/**\n * Plan/subscription usage service. Per-provider adapters query provider quota\n * endpoints (60-second cache each) so /usage and the footer can show subscription windows\n * alongside session/context totals.\n *\n * Deliberately absent adapters:\n * - anthropic: Anthropic policy prohibits subscription usage tracking in\n *   third-party coding agents (2026-07-18 decision). OAuth Anthropic sessions\n *   fall back to hiding the cost meter.\n * - ollama-cloud: no usage API exists.\n *\n * Adapter/network errors resolve to `undefined` (callers omit the plan UI).\n * Auth failures are different: they return `error` and are not cached as empty.\n */\n\nimport type { ModelRuntime } from \"./model-runtime.ts\";\nimport type { SettingsManager } from \"./settings-manager.ts\";\nimport { fetchKimiPlanUsage } from \"./usage-adapters/kimi-coding.ts\";\nimport { fetchCodexPlanUsage } from \"./usage-adapters/openai-codex.ts\";\nimport { fetchXaiPlanUsage } from \"./usage-adapters/xai.ts\";\nimport { fetchZaiPlanUsage } from \"./usage-adapters/zai.ts\";\n\nexport interface PlanUsageWindow {\n\tlabel: string;\n\tusedPercent: number;\n\t/** Epoch ms when the window resets. */\n\tresetsAt?: number;\n}\n\nexport interface PlanUsage {\n\tprovider: string;\n\tplanLabel?: string;\n\twindows: PlanUsageWindow[];\n}\n\nexport interface PlanUsageResult {\n\t/** First plan, kept for footer and compatibility callers. */\n\tusage?: PlanUsage;\n\t/** Every plan returned by the provider. */\n\tusages?: PlanUsage[];\n\t/** Actionable auth failure. Not cached as an empty success. */\n\terror?: string;\n}\n\nexport interface AllPlanUsageResult {\n\tusages: PlanUsage[];\n\terrors: string[];\n}\n\nconst XAI_RELOGIN = \"xAI login expired. Run /login xai.\";\n\nfunction errorChain(error: unknown): string {\n\tconst parts: string[] = [];\n\tlet current: unknown = error;\n\tconst seen = new Set<unknown>();\n\twhile (current && typeof current === \"object\" && !seen.has(current)) {\n\t\tseen.add(current);\n\t\tif (current instanceof Error) {\n\t\t\tparts.push(current.message);\n\t\t\tcurrent = current.cause;\n\t\t\tcontinue;\n\t\t}\n\t\tbreak;\n\t}\n\treturn parts.join(\" \");\n}\n\n/** Classify adapter/getAuth failures that mean the SuperGrok session is dead. */\nexport function planUsageAuthError(error: unknown): string | undefined {\n\tconst chain = errorChain(error);\n\tif (/timed out|aborted|cancelled|ECONN|ENOTFOUND|network/i.test(chain)) return undefined;\n\t// 403 on billing is often entitlement, not a dead refresh token.\n\tif (/xai billing rejected the session \\(HTTP 401\\)/i.test(chain)) return XAI_RELOGIN;\n\tif (/invalid_grant|refresh token revoked/i.test(chain)) return XAI_RELOGIN;\n\treturn undefined;\n}\n\ntype UsageAdapter = (runtime: ModelRuntime) => Promise<PlanUsage | PlanUsage[] | undefined>;\n\nconst ADAPTERS: Readonly<Record<string, UsageAdapter>> = {\n\t\"openai-codex\": fetchCodexPlanUsage,\n\t\"kimi-coding\": fetchKimiPlanUsage,\n\tzai: fetchZaiPlanUsage,\n\txai: fetchXaiPlanUsage,\n};\n\nexport const PLAN_USAGE_CACHE_TTL_MS = 60 * 1000;\n\nexport type PlanUsageWindowPreference = \"5h\" | \"weekly\";\n\nexport function isWeeklyPlanWindow(window: PlanUsageWindow): boolean {\n\treturn /week/i.test(window.label);\n}\n\nexport function isFiveHourPlanWindow(window: PlanUsageWindow): boolean {\n\treturn /5h/i.test(window.label) || window.label === \"5h\";\n}\n\n/** Preferred window, falling back to the other, skipping Extra/on-demand. */\nexport function pickPlanWindow(\n\tusage: PlanUsage | undefined,\n\tpreferred: PlanUsageWindowPreference,\n): PlanUsageWindow | undefined {\n\tif (!usage?.windows.length) return undefined;\n\tconst usable = usage.windows.filter((w) => !/extra/i.test(w.label));\n\tconst weekly = usable.find(isWeeklyPlanWindow);\n\tconst fiveH = usable.find(isFiveHourPlanWindow);\n\tif (preferred === \"5h\") return fiveH ?? weekly ?? usable[0] ?? usage.windows[0];\n\treturn weekly ?? fiveH ?? usable[0] ?? usage.windows[0];\n}\n\nexport function footerPlanLabel(window: PlanUsageWindow): string {\n\tif (isFiveHourPlanWindow(window)) return \"5h\";\n\tif (isWeeklyPlanWindow(window)) return \"wk\";\n\treturn window.label.length <= 4 ? window.label : window.label.slice(0, 4);\n}\n\nconst cache = new Map<string, { expiresAt: number; value: PlanUsage[] }>();\nconst inflight = new Map<string, Promise<PlanUsageResult>>();\n\n/** Whether a plan-usage adapter exists for this provider. */\nexport function hasPlanUsageAdapter(providerId: string): boolean {\n\treturn providerId in ADAPTERS;\n}\n\n/** Plan usage for a provider, or undefined when unsupported/unavailable. Never throws. */\nexport async function getPlanUsage(providerId: string, runtime: ModelRuntime): Promise<PlanUsage | undefined> {\n\treturn (await getPlanUsageResult(providerId, runtime)).usage;\n}\n\n/** Like getPlanUsage, but surfaces auth failures instead of hiding them. */\nexport async function getPlanUsageResult(providerId: string, runtime: ModelRuntime): Promise<PlanUsageResult> {\n\tconst adapter = ADAPTERS[providerId];\n\tif (!adapter) return {};\n\tconst now = Date.now();\n\tconst cached = cache.get(providerId);\n\tif (cached && cached.expiresAt > now) return { usage: cached.value[0], usages: cached.value };\n\tconst pending = inflight.get(providerId);\n\tif (pending) return pending;\n\tconst task = (async (): Promise<PlanUsageResult> => {\n\t\ttry {\n\t\t\tconst value = await adapter(runtime);\n\t\t\tconst usages = value === undefined ? [] : Array.isArray(value) ? value : [value];\n\t\t\tcache.set(providerId, { expiresAt: Date.now() + PLAN_USAGE_CACHE_TTL_MS, value: usages });\n\t\t\treturn { usage: usages[0], usages };\n\t\t} catch (error) {\n\t\t\tconst authError = planUsageAuthError(error);\n\t\t\tif (authError) return { error: authError };\n\t\t\tcache.set(providerId, { expiresAt: Date.now() + PLAN_USAGE_CACHE_TTL_MS, value: [] });\n\t\t\treturn {};\n\t\t} finally {\n\t\t\tinflight.delete(providerId);\n\t\t}\n\t})();\n\tinflight.set(providerId, task);\n\treturn task;\n}\n\n/** Fetch every stored-credential plan plus the current provider, in parallel. */\nexport async function getAllPlanUsageResults(\n\tcurrentProvider: string | undefined,\n\truntime: ModelRuntime,\n): Promise<AllPlanUsageResult> {\n\tconst stored = new Set((await runtime.listCredentials()).map((credential) => credential.providerId));\n\tconst providers = Object.keys(ADAPTERS).filter(\n\t\t(providerId) => stored.has(providerId) || providerId === currentProvider,\n\t);\n\tconst results = await Promise.all(providers.map((providerId) => getPlanUsageResult(providerId, runtime)));\n\treturn {\n\t\tusages: results.flatMap((result) => result.usages ?? []),\n\t\terrors: results.flatMap((result) => (result.error ? [`${result.error}`] : [])),\n\t};\n}\n\n/** Last fetched usage, including expired cache (footer can show stale while refetching). */\nexport function peekPlanUsage(providerId: string): PlanUsage | undefined {\n\treturn cache.get(providerId)?.value[0];\n}\n\n/** Clear the adapter cache (tests, /login changes). */\nexport function clearPlanUsageCache(): void {\n\tcache.clear();\n\tinflight.clear();\n}\n\n// ---------------------------------------------------------------------------\n// Extension bridge\n// ---------------------------------------------------------------------------\n\nexport const USAGE_SERVICE_BRIDGE_SYMBOL = Symbol.for(\"@lunr/usage-service\");\n\nexport interface FooterPlanSegment {\n\tlabel: string;\n\tusedPercent: number;\n}\n\nexport interface UsageServiceBridge {\n\tpeek(providerId: string): PlanUsage | undefined;\n\tprefetch(providerId: string): void;\n\tgetPreferredWindow(): PlanUsageWindowPreference;\n\tpickForFooter(providerId: string): FooterPlanSegment | undefined;\n\tsetOnUpdate(fn: (() => void) | undefined): void;\n}\n\nlet activeRuntime: ModelRuntime | undefined;\nlet activeUsageSettings: SettingsManager | undefined;\nlet onUsageUpdate: (() => void) | undefined;\n\nfunction preferredWindow(): PlanUsageWindowPreference {\n\treturn activeUsageSettings?.getPlanUsageWindow() ?? \"weekly\";\n}\n\nconst usageBridge: UsageServiceBridge = {\n\tpeek(providerId: string): PlanUsage | undefined {\n\t\treturn peekPlanUsage(providerId);\n\t},\n\tprefetch(providerId: string): void {\n\t\tif (!activeRuntime || !hasPlanUsageAdapter(providerId)) return;\n\t\tvoid getPlanUsage(providerId, activeRuntime).then(() => onUsageUpdate?.());\n\t},\n\tsetOnUpdate(fn: (() => void) | undefined): void {\n\t\tonUsageUpdate = fn;\n\t},\n\tgetPreferredWindow(): PlanUsageWindowPreference {\n\t\treturn preferredWindow();\n\t},\n\tpickForFooter(providerId: string): FooterPlanSegment | undefined {\n\t\tconst window = pickPlanWindow(peekPlanUsage(providerId), preferredWindow());\n\t\tif (!window) return undefined;\n\t\treturn { label: footerPlanLabel(window), usedPercent: window.usedPercent };\n\t},\n};\n\nexport function registerUsageServiceBridge(runtime: ModelRuntime, settingsManager: SettingsManager): void {\n\tactiveRuntime = runtime;\n\tactiveUsageSettings = settingsManager;\n\t(globalThis as Record<symbol, unknown>)[USAGE_SERVICE_BRIDGE_SYMBOL] = usageBridge;\n}\n\nexport function getUsageServiceBridge(): UsageServiceBridge | undefined {\n\treturn (globalThis as Record<symbol, unknown>)[USAGE_SERVICE_BRIDGE_SYMBOL] as UsageServiceBridge | undefined;\n}\n"]}

package/dist/core/usage-service.js

@@ -0,0 +1,180 @@
+/**
+ * Plan/subscription usage service. Per-provider adapters query provider quota
+ * endpoints (60-second cache each) so /usage and the footer can show subscription windows
+ * alongside session/context totals.
+ *
+ * Deliberately absent adapters:
+ * - anthropic: Anthropic policy prohibits subscription usage tracking in
+ *   third-party coding agents (2026-07-18 decision). OAuth Anthropic sessions
+ *   fall back to hiding the cost meter.
+ * - ollama-cloud: no usage API exists.
+ *
+ * Adapter/network errors resolve to `undefined` (callers omit the plan UI).
+ * Auth failures are different: they return `error` and are not cached as empty.
+ */
+import { fetchKimiPlanUsage } from "./usage-adapters/kimi-coding.js";
+import { fetchCodexPlanUsage } from "./usage-adapters/openai-codex.js";
+import { fetchXaiPlanUsage } from "./usage-adapters/xai.js";
+import { fetchZaiPlanUsage } from "./usage-adapters/zai.js";
+const XAI_RELOGIN = "xAI login expired. Run /login xai.";
+function errorChain(error) {
+    const parts = [];
+    let current = error;
+    const seen = new Set();
+    while (current && typeof current === "object" && !seen.has(current)) {
+        seen.add(current);
+        if (current instanceof Error) {
+            parts.push(current.message);
+            current = current.cause;
+            continue;
+        }
+        break;
+    }
+    return parts.join(" ");
+}
+/** Classify adapter/getAuth failures that mean the SuperGrok session is dead. */
+export function planUsageAuthError(error) {
+    const chain = errorChain(error);
+    if (/timed out|aborted|cancelled|ECONN|ENOTFOUND|network/i.test(chain))
+        return undefined;
+    // 403 on billing is often entitlement, not a dead refresh token.
+    if (/xai billing rejected the session \(HTTP 401\)/i.test(chain))
+        return XAI_RELOGIN;
+    if (/invalid_grant|refresh token revoked/i.test(chain))
+        return XAI_RELOGIN;
+    return undefined;
+}
+const ADAPTERS = {
+    "openai-codex": fetchCodexPlanUsage,
+    "kimi-coding": fetchKimiPlanUsage,
+    zai: fetchZaiPlanUsage,
+    xai: fetchXaiPlanUsage,
+};
+export const PLAN_USAGE_CACHE_TTL_MS = 60 * 1000;
+export function isWeeklyPlanWindow(window) {
+    return /week/i.test(window.label);
+}
+export function isFiveHourPlanWindow(window) {
+    return /5h/i.test(window.label) || window.label === "5h";
+}
+/** Preferred window, falling back to the other, skipping Extra/on-demand. */
+export function pickPlanWindow(usage, preferred) {
+    if (!usage?.windows.length)
+        return undefined;
+    const usable = usage.windows.filter((w) => !/extra/i.test(w.label));
+    const weekly = usable.find(isWeeklyPlanWindow);
+    const fiveH = usable.find(isFiveHourPlanWindow);
+    if (preferred === "5h")
+        return fiveH ?? weekly ?? usable[0] ?? usage.windows[0];
+    return weekly ?? fiveH ?? usable[0] ?? usage.windows[0];
+}
+export function footerPlanLabel(window) {
+    if (isFiveHourPlanWindow(window))
+        return "5h";
+    if (isWeeklyPlanWindow(window))
+        return "wk";
+    return window.label.length <= 4 ? window.label : window.label.slice(0, 4);
+}
+const cache = new Map();
+const inflight = new Map();
+/** Whether a plan-usage adapter exists for this provider. */
+export function hasPlanUsageAdapter(providerId) {
+    return providerId in ADAPTERS;
+}
+/** Plan usage for a provider, or undefined when unsupported/unavailable. Never throws. */
+export async function getPlanUsage(providerId, runtime) {
+    return (await getPlanUsageResult(providerId, runtime)).usage;
+}
+/** Like getPlanUsage, but surfaces auth failures instead of hiding them. */
+export async function getPlanUsageResult(providerId, runtime) {
+    const adapter = ADAPTERS[providerId];
+    if (!adapter)
+        return {};
+    const now = Date.now();
+    const cached = cache.get(providerId);
+    if (cached && cached.expiresAt > now)
+        return { usage: cached.value[0], usages: cached.value };
+    const pending = inflight.get(providerId);
+    if (pending)
+        return pending;
+    const task = (async () => {
+        try {
+            const value = await adapter(runtime);
+            const usages = value === undefined ? [] : Array.isArray(value) ? value : [value];
+            cache.set(providerId, { expiresAt: Date.now() + PLAN_USAGE_CACHE_TTL_MS, value: usages });
+            return { usage: usages[0], usages };
+        }
+        catch (error) {
+            const authError = planUsageAuthError(error);
+            if (authError)
+                return { error: authError };
+            cache.set(providerId, { expiresAt: Date.now() + PLAN_USAGE_CACHE_TTL_MS, value: [] });
+            return {};
+        }
+        finally {
+            inflight.delete(providerId);
+        }
+    })();
+    inflight.set(providerId, task);
+    return task;
+}
+/** Fetch every stored-credential plan plus the current provider, in parallel. */
+export async function getAllPlanUsageResults(currentProvider, runtime) {
+    const stored = new Set((await runtime.listCredentials()).map((credential) => credential.providerId));
+    const providers = Object.keys(ADAPTERS).filter((providerId) => stored.has(providerId) || providerId === currentProvider);
+    const results = await Promise.all(providers.map((providerId) => getPlanUsageResult(providerId, runtime)));
+    return {
+        usages: results.flatMap((result) => result.usages ?? []),
+        errors: results.flatMap((result) => (result.error ? [`${result.error}`] : [])),
+    };
+}
+/** Last fetched usage, including expired cache (footer can show stale while refetching). */
+export function peekPlanUsage(providerId) {
+    return cache.get(providerId)?.value[0];
+}
+/** Clear the adapter cache (tests, /login changes). */
+export function clearPlanUsageCache() {
+    cache.clear();
+    inflight.clear();
+}
+// ---------------------------------------------------------------------------
+// Extension bridge
+// ---------------------------------------------------------------------------
+export const USAGE_SERVICE_BRIDGE_SYMBOL = Symbol.for("@lunr/usage-service");
+let activeRuntime;
+let activeUsageSettings;
+let onUsageUpdate;
+function preferredWindow() {
+    return activeUsageSettings?.getPlanUsageWindow() ?? "weekly";
+}
+const usageBridge = {
+    peek(providerId) {
+        return peekPlanUsage(providerId);
+    },
+    prefetch(providerId) {
+        if (!activeRuntime || !hasPlanUsageAdapter(providerId))
+            return;
+        void getPlanUsage(providerId, activeRuntime).then(() => onUsageUpdate?.());
+    },
+    setOnUpdate(fn) {
+        onUsageUpdate = fn;
+    },
+    getPreferredWindow() {
+        return preferredWindow();
+    },
+    pickForFooter(providerId) {
+        const window = pickPlanWindow(peekPlanUsage(providerId), preferredWindow());
+        if (!window)
+            return undefined;
+        return { label: footerPlanLabel(window), usedPercent: window.usedPercent };
+    },
+};
+export function registerUsageServiceBridge(runtime, settingsManager) {
+    activeRuntime = runtime;
+    activeUsageSettings = settingsManager;
+    globalThis[USAGE_SERVICE_BRIDGE_SYMBOL] = usageBridge;
+}
+export function getUsageServiceBridge() {
+    return globalThis[USAGE_SERVICE_BRIDGE_SYMBOL];
+}
+//# sourceMappingURL=usage-service.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"usage-service.js","sourceRoot":"","sources":["../../src/core/usage-service.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAIH,OAAO,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AACrE,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AACvE,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AA6B5D,MAAM,WAAW,GAAG,oCAAoC,CAAC;AAEzD,SAAS,UAAU,CAAC,KAAc,EAAU;IAC3C,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,OAAO,GAAY,KAAK,CAAC;IAC7B,MAAM,IAAI,GAAG,IAAI,GAAG,EAAW,CAAC;IAChC,OAAO,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QACrE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAClB,IAAI,OAAO,YAAY,KAAK,EAAE,CAAC;YAC9B,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC5B,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC;YACxB,SAAS;QACV,CAAC;QACD,MAAM;IACP,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAAA,CACvB;AAED,iFAAiF;AACjF,MAAM,UAAU,kBAAkB,CAAC,KAAc,EAAsB;IACtE,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IAChC,IAAI,sDAAsD,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACzF,iEAAiE;IACjE,IAAI,gDAAgD,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,WAAW,CAAC;IACrF,IAAI,sCAAsC,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,WAAW,CAAC;IAC3E,OAAO,SAAS,CAAC;AAAA,CACjB;AAID,MAAM,QAAQ,GAA2C;IACxD,cAAc,EAAE,mBAAmB;IACnC,aAAa,EAAE,kBAAkB;IACjC,GAAG,EAAE,iBAAiB;IACtB,GAAG,EAAE,iBAAiB;CACtB,CAAC;AAEF,MAAM,CAAC,MAAM,uBAAuB,GAAG,EAAE,GAAG,IAAI,CAAC;AAIjD,MAAM,UAAU,kBAAkB,CAAC,MAAuB,EAAW;IACpE,OAAO,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAAA,CAClC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAuB,EAAW;IACtE,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,KAAK,KAAK,IAAI,CAAC;AAAA,CACzD;AAED,6EAA6E;AAC7E,MAAM,UAAU,cAAc,CAC7B,KAA4B,EAC5B,SAAoC,EACN;IAC9B,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAC7C,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IACpE,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IAC/C,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAChD,IAAI,SAAS,KAAK,IAAI;QAAE,OAAO,KAAK,IAAI,MAAM,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAChF,OAAO,MAAM,IAAI,KAAK,IAAI,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AAAA,CACxD;AAED,MAAM,UAAU,eAAe,CAAC,MAAuB,EAAU;IAChE,IAAI,oBAAoB,CAAC,MAAM,CAAC;QAAE,OAAO,IAAI,CAAC;IAC9C,IAAI,kBAAkB,CAAC,MAAM,CAAC;QAAE,OAAO,IAAI,CAAC;IAC5C,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AAAA,CAC1E;AAED,MAAM,KAAK,GAAG,IAAI,GAAG,EAAqD,CAAC;AAC3E,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAoC,CAAC;AAE7D,6DAA6D;AAC7D,MAAM,UAAU,mBAAmB,CAAC,UAAkB,EAAW;IAChE,OAAO,UAAU,IAAI,QAAQ,CAAC;AAAA,CAC9B;AAED,0FAA0F;AAC1F,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,UAAkB,EAAE,OAAqB,EAAkC;IAC7G,OAAO,CAAC,MAAM,kBAAkB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;AAAA,CAC7D;AAED,4EAA4E;AAC5E,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,UAAkB,EAAE,OAAqB,EAA4B;IAC7G,MAAM,OAAO,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;IACrC,IAAI,CAAC,OAAO;QAAE,OAAO,EAAE,CAAC;IACxB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACrC,IAAI,MAAM,IAAI,MAAM,CAAC,SAAS,GAAG,GAAG;QAAE,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;IAC9F,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACzC,IAAI,OAAO;QAAE,OAAO,OAAO,CAAC;IAC5B,MAAM,IAAI,GAAG,CAAC,KAAK,IAA8B,EAAE,CAAC;QACnD,IAAI,CAAC;YACJ,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC;YACrC,MAAM,MAAM,GAAG,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACjF,KAAK,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,uBAAuB,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1F,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;QACrC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,SAAS,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,SAAS;gBAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;YAC3C,KAAK,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,uBAAuB,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACtF,OAAO,EAAE,CAAC;QACX,CAAC;gBAAS,CAAC;YACV,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAC7B,CAAC;IAAA,CACD,CAAC,EAAE,CAAC;IACL,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;IAC/B,OAAO,IAAI,CAAC;AAAA,CACZ;AAED,iFAAiF;AACjF,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC3C,eAAmC,EACnC,OAAqB,EACS;IAC9B,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,CAAC,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC;IACrG,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAC7C,CAAC,UAAU,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,UAAU,KAAK,eAAe,CACxE,CAAC;IACF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,kBAAkB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAC1G,OAAO;QACN,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;QACxD,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;KAC9E,CAAC;AAAA,CACF;AAED,4FAA4F;AAC5F,MAAM,UAAU,aAAa,CAAC,UAAkB,EAAyB;IACxE,OAAO,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;AAAA,CACvC;AAED,uDAAuD;AACvD,MAAM,UAAU,mBAAmB,GAAS;IAC3C,KAAK,CAAC,KAAK,EAAE,CAAC;IACd,QAAQ,CAAC,KAAK,EAAE,CAAC;AAAA,CACjB;AAED,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E,MAAM,CAAC,MAAM,2BAA2B,GAAG,MAAM,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;AAe7E,IAAI,aAAuC,CAAC;AAC5C,IAAI,mBAAgD,CAAC;AACrD,IAAI,aAAuC,CAAC;AAE5C,SAAS,eAAe,GAA8B;IACrD,OAAO,mBAAmB,EAAE,kBAAkB,EAAE,IAAI,QAAQ,CAAC;AAAA,CAC7D;AAED,MAAM,WAAW,GAAuB;IACvC,IAAI,CAAC,UAAkB,EAAyB;QAC/C,OAAO,aAAa,CAAC,UAAU,CAAC,CAAC;IAAA,CACjC;IACD,QAAQ,CAAC,UAAkB,EAAQ;QAClC,IAAI,CAAC,aAAa,IAAI,CAAC,mBAAmB,CAAC,UAAU,CAAC;YAAE,OAAO;QAC/D,KAAK,YAAY,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;IAAA,CAC3E;IACD,WAAW,CAAC,EAA4B,EAAQ;QAC/C,aAAa,GAAG,EAAE,CAAC;IAAA,CACnB;IACD,kBAAkB,GAA8B;QAC/C,OAAO,eAAe,EAAE,CAAC;IAAA,CACzB;IACD,aAAa,CAAC,UAAkB,EAAiC;QAChE,MAAM,MAAM,GAAG,cAAc,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC;QAC5E,IAAI,CAAC,MAAM;YAAE,OAAO,SAAS,CAAC;QAC9B,OAAO,EAAE,KAAK,EAAE,eAAe,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC;IAAA,CAC3E;CACD,CAAC;AAEF,MAAM,UAAU,0BAA0B,CAAC,OAAqB,EAAE,eAAgC,EAAQ;IACzG,aAAa,GAAG,OAAO,CAAC;IACxB,mBAAmB,GAAG,eAAe,CAAC;IACrC,UAAsC,CAAC,2BAA2B,CAAC,GAAG,WAAW,CAAC;AAAA,CACnF;AAED,MAAM,UAAU,qBAAqB,GAAmC;IACvE,OAAQ,UAAsC,CAAC,2BAA2B,CAAmC,CAAC;AAAA,CAC9G","sourcesContent":["/**\n * Plan/subscription usage service. Per-provider adapters query provider quota\n * endpoints (60-second cache each) so /usage and the footer can show subscription windows\n * alongside session/context totals.\n *\n * Deliberately absent adapters:\n * - anthropic: Anthropic policy prohibits subscription usage tracking in\n *   third-party coding agents (2026-07-18 decision). OAuth Anthropic sessions\n *   fall back to hiding the cost meter.\n * - ollama-cloud: no usage API exists.\n *\n * Adapter/network errors resolve to `undefined` (callers omit the plan UI).\n * Auth failures are different: they return `error` and are not cached as empty.\n */\n\nimport type { ModelRuntime } from \"./model-runtime.ts\";\nimport type { SettingsManager } from \"./settings-manager.ts\";\nimport { fetchKimiPlanUsage } from \"./usage-adapters/kimi-coding.ts\";\nimport { fetchCodexPlanUsage } from \"./usage-adapters/openai-codex.ts\";\nimport { fetchXaiPlanUsage } from \"./usage-adapters/xai.ts\";\nimport { fetchZaiPlanUsage } from \"./usage-adapters/zai.ts\";\n\nexport interface PlanUsageWindow {\n\tlabel: string;\n\tusedPercent: number;\n\t/** Epoch ms when the window resets. */\n\tresetsAt?: number;\n}\n\nexport interface PlanUsage {\n\tprovider: string;\n\tplanLabel?: string;\n\twindows: PlanUsageWindow[];\n}\n\nexport interface PlanUsageResult {\n\t/** First plan, kept for footer and compatibility callers. */\n\tusage?: PlanUsage;\n\t/** Every plan returned by the provider. */\n\tusages?: PlanUsage[];\n\t/** Actionable auth failure. Not cached as an empty success. */\n\terror?: string;\n}\n\nexport interface AllPlanUsageResult {\n\tusages: PlanUsage[];\n\terrors: string[];\n}\n\nconst XAI_RELOGIN = \"xAI login expired. Run /login xai.\";\n\nfunction errorChain(error: unknown): string {\n\tconst parts: string[] = [];\n\tlet current: unknown = error;\n\tconst seen = new Set<unknown>();\n\twhile (current && typeof current === \"object\" && !seen.has(current)) {\n\t\tseen.add(current);\n\t\tif (current instanceof Error) {\n\t\t\tparts.push(current.message);\n\t\t\tcurrent = current.cause;\n\t\t\tcontinue;\n\t\t}\n\t\tbreak;\n\t}\n\treturn parts.join(\" \");\n}\n\n/** Classify adapter/getAuth failures that mean the SuperGrok session is dead. */\nexport function planUsageAuthError(error: unknown): string | undefined {\n\tconst chain = errorChain(error);\n\tif (/timed out|aborted|cancelled|ECONN|ENOTFOUND|network/i.test(chain)) return undefined;\n\t// 403 on billing is often entitlement, not a dead refresh token.\n\tif (/xai billing rejected the session \\(HTTP 401\\)/i.test(chain)) return XAI_RELOGIN;\n\tif (/invalid_grant|refresh token revoked/i.test(chain)) return XAI_RELOGIN;\n\treturn undefined;\n}\n\ntype UsageAdapter = (runtime: ModelRuntime) => Promise<PlanUsage | PlanUsage[] | undefined>;\n\nconst ADAPTERS: Readonly<Record<string, UsageAdapter>> = {\n\t\"openai-codex\": fetchCodexPlanUsage,\n\t\"kimi-coding\": fetchKimiPlanUsage,\n\tzai: fetchZaiPlanUsage,\n\txai: fetchXaiPlanUsage,\n};\n\nexport const PLAN_USAGE_CACHE_TTL_MS = 60 * 1000;\n\nexport type PlanUsageWindowPreference = \"5h\" | \"weekly\";\n\nexport function isWeeklyPlanWindow(window: PlanUsageWindow): boolean {\n\treturn /week/i.test(window.label);\n}\n\nexport function isFiveHourPlanWindow(window: PlanUsageWindow): boolean {\n\treturn /5h/i.test(window.label) || window.label === \"5h\";\n}\n\n/** Preferred window, falling back to the other, skipping Extra/on-demand. */\nexport function pickPlanWindow(\n\tusage: PlanUsage | undefined,\n\tpreferred: PlanUsageWindowPreference,\n): PlanUsageWindow | undefined {\n\tif (!usage?.windows.length) return undefined;\n\tconst usable = usage.windows.filter((w) => !/extra/i.test(w.label));\n\tconst weekly = usable.find(isWeeklyPlanWindow);\n\tconst fiveH = usable.find(isFiveHourPlanWindow);\n\tif (preferred === \"5h\") return fiveH ?? weekly ?? usable[0] ?? usage.windows[0];\n\treturn weekly ?? fiveH ?? usable[0] ?? usage.windows[0];\n}\n\nexport function footerPlanLabel(window: PlanUsageWindow): string {\n\tif (isFiveHourPlanWindow(window)) return \"5h\";\n\tif (isWeeklyPlanWindow(window)) return \"wk\";\n\treturn window.label.length <= 4 ? window.label : window.label.slice(0, 4);\n}\n\nconst cache = new Map<string, { expiresAt: number; value: PlanUsage[] }>();\nconst inflight = new Map<string, Promise<PlanUsageResult>>();\n\n/** Whether a plan-usage adapter exists for this provider. */\nexport function hasPlanUsageAdapter(providerId: string): boolean {\n\treturn providerId in ADAPTERS;\n}\n\n/** Plan usage for a provider, or undefined when unsupported/unavailable. Never throws. */\nexport async function getPlanUsage(providerId: string, runtime: ModelRuntime): Promise<PlanUsage | undefined> {\n\treturn (await getPlanUsageResult(providerId, runtime)).usage;\n}\n\n/** Like getPlanUsage, but surfaces auth failures instead of hiding them. */\nexport async function getPlanUsageResult(providerId: string, runtime: ModelRuntime): Promise<PlanUsageResult> {\n\tconst adapter = ADAPTERS[providerId];\n\tif (!adapter) return {};\n\tconst now = Date.now();\n\tconst cached = cache.get(providerId);\n\tif (cached && cached.expiresAt > now) return { usage: cached.value[0], usages: cached.value };\n\tconst pending = inflight.get(providerId);\n\tif (pending) return pending;\n\tconst task = (async (): Promise<PlanUsageResult> => {\n\t\ttry {\n\t\t\tconst value = await adapter(runtime);\n\t\t\tconst usages = value === undefined ? [] : Array.isArray(value) ? value : [value];\n\t\t\tcache.set(providerId, { expiresAt: Date.now() + PLAN_USAGE_CACHE_TTL_MS, value: usages });\n\t\t\treturn { usage: usages[0], usages };\n\t\t} catch (error) {\n\t\t\tconst authError = planUsageAuthError(error);\n\t\t\tif (authError) return { error: authError };\n\t\t\tcache.set(providerId, { expiresAt: Date.now() + PLAN_USAGE_CACHE_TTL_MS, value: [] });\n\t\t\treturn {};\n\t\t} finally {\n\t\t\tinflight.delete(providerId);\n\t\t}\n\t})();\n\tinflight.set(providerId, task);\n\treturn task;\n}\n\n/** Fetch every stored-credential plan plus the current provider, in parallel. */\nexport async function getAllPlanUsageResults(\n\tcurrentProvider: string | undefined,\n\truntime: ModelRuntime,\n): Promise<AllPlanUsageResult> {\n\tconst stored = new Set((await runtime.listCredentials()).map((credential) => credential.providerId));\n\tconst providers = Object.keys(ADAPTERS).filter(\n\t\t(providerId) => stored.has(providerId) || providerId === currentProvider,\n\t);\n\tconst results = await Promise.all(providers.map((providerId) => getPlanUsageResult(providerId, runtime)));\n\treturn {\n\t\tusages: results.flatMap((result) => result.usages ?? []),\n\t\terrors: results.flatMap((result) => (result.error ? [`${result.error}`] : [])),\n\t};\n}\n\n/** Last fetched usage, including expired cache (footer can show stale while refetching). */\nexport function peekPlanUsage(providerId: string): PlanUsage | undefined {\n\treturn cache.get(providerId)?.value[0];\n}\n\n/** Clear the adapter cache (tests, /login changes). */\nexport function clearPlanUsageCache(): void {\n\tcache.clear();\n\tinflight.clear();\n}\n\n// ---------------------------------------------------------------------------\n// Extension bridge\n// ---------------------------------------------------------------------------\n\nexport const USAGE_SERVICE_BRIDGE_SYMBOL = Symbol.for(\"@lunr/usage-service\");\n\nexport interface FooterPlanSegment {\n\tlabel: string;\n\tusedPercent: number;\n}\n\nexport interface UsageServiceBridge {\n\tpeek(providerId: string): PlanUsage | undefined;\n\tprefetch(providerId: string): void;\n\tgetPreferredWindow(): PlanUsageWindowPreference;\n\tpickForFooter(providerId: string): FooterPlanSegment | undefined;\n\tsetOnUpdate(fn: (() => void) | undefined): void;\n}\n\nlet activeRuntime: ModelRuntime | undefined;\nlet activeUsageSettings: SettingsManager | undefined;\nlet onUsageUpdate: (() => void) | undefined;\n\nfunction preferredWindow(): PlanUsageWindowPreference {\n\treturn activeUsageSettings?.getPlanUsageWindow() ?? \"weekly\";\n}\n\nconst usageBridge: UsageServiceBridge = {\n\tpeek(providerId: string): PlanUsage | undefined {\n\t\treturn peekPlanUsage(providerId);\n\t},\n\tprefetch(providerId: string): void {\n\t\tif (!activeRuntime || !hasPlanUsageAdapter(providerId)) return;\n\t\tvoid getPlanUsage(providerId, activeRuntime).then(() => onUsageUpdate?.());\n\t},\n\tsetOnUpdate(fn: (() => void) | undefined): void {\n\t\tonUsageUpdate = fn;\n\t},\n\tgetPreferredWindow(): PlanUsageWindowPreference {\n\t\treturn preferredWindow();\n\t},\n\tpickForFooter(providerId: string): FooterPlanSegment | undefined {\n\t\tconst window = pickPlanWindow(peekPlanUsage(providerId), preferredWindow());\n\t\tif (!window) return undefined;\n\t\treturn { label: footerPlanLabel(window), usedPercent: window.usedPercent };\n\t},\n};\n\nexport function registerUsageServiceBridge(runtime: ModelRuntime, settingsManager: SettingsManager): void {\n\tactiveRuntime = runtime;\n\tactiveUsageSettings = settingsManager;\n\t(globalThis as Record<symbol, unknown>)[USAGE_SERVICE_BRIDGE_SYMBOL] = usageBridge;\n}\n\nexport function getUsageServiceBridge(): UsageServiceBridge | undefined {\n\treturn (globalThis as Record<symbol, unknown>)[USAGE_SERVICE_BRIDGE_SYMBOL] as UsageServiceBridge | undefined;\n}\n"]}

package/dist/core/user-models.d.ts

@@ -0,0 +1,50 @@
+import type { Api, Model } from "@ashx-j/lunr-ai";
+export declare const USER_MODELS_FILENAME = "user-models.json";
+export interface UserModelEntry {
+    id: string;
+    name: string;
+    api: Api;
+    provider: string;
+    baseUrl: string;
+    reasoning: boolean;
+    input: ("text" | "image")[];
+    cost: Model<Api>["cost"];
+    contextWindow: number;
+    maxTokens: number;
+    compat?: Model<Api>["compat"];
+    thinkingLevelMap?: Model<Api>["thinkingLevelMap"];
+    source: "user";
+}
+export interface UserModelsFile {
+    version: number;
+    models: UserModelEntry[];
+}
+export declare function userModelsPath(agentDir?: string): string;
+export declare function emptyUserModels(): UserModelsFile;
+/** Read user-models.json. Missing or unreadable files yield an empty document (never throws). */
+export declare function readUserModels(path: string): UserModelsFile;
+export declare function writeUserModels(file: UserModelsFile, path: string): void;
+export declare function evictUserModelsOnOfficial(user: UserModelsFile, officialKeys: Iterable<string>): {
+    next: UserModelsFile;
+    evicted: UserModelEntry[];
+};
+export declare function evictUserModelsByProvider(user: UserModelsFile, providerId: string): {
+    next: UserModelsFile;
+    evicted: UserModelEntry[];
+};
+export declare function upsertUserModels(existing: UserModelsFile, rows: readonly UserModelEntry[]): UserModelsFile;
+export declare function userEntryToModel(entry: UserModelEntry): Model<Api>;
+export declare function modelToUserEntry(model: Model<Api>): UserModelEntry;
+export declare class UserModelsStore {
+    private readonly path;
+    private file;
+    constructor(path: string | undefined);
+    list(): UserModelEntry[];
+    keys(): Set<string>;
+    modelsFor(providerId: string): Model<Api>[];
+    evictOfficial(officialKeys: Iterable<string>): UserModelEntry[];
+    evictProvider(providerId: string): UserModelEntry[];
+    upsert(rows: readonly UserModelEntry[]): void;
+    private persist;
+}
+//# sourceMappingURL=user-models.d.ts.map

package/dist/core/user-models.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"user-models.d.ts","sourceRoot":"","sources":["../../src/core/user-models.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAIxD,eAAO,MAAM,oBAAoB,qBAAqB,CAAC;AAEvD,MAAM,WAAW,cAAc;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,GAAG,CAAC;IACT,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,OAAO,CAAC;IACnB,KAAK,EAAE,CAAC,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC;IAC5B,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;IAC9B,gBAAgB,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,kBAAkB,CAAC,CAAC;IAClD,MAAM,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,cAAc;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,cAAc,EAAE,CAAC;CACzB;AAED,wBAAgB,cAAc,CAAC,QAAQ,GAAE,MAAsB,GAAG,MAAM,CAEvE;AA+CD,wBAAgB,eAAe,IAAI,cAAc,CAEhD;AAED,iGAAiG;AACjG,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,CAc3D;AAED,wBAAgB,eAAe,CAAC,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI,CAGxE;AAED,wBAAgB,yBAAyB,CACxC,IAAI,EAAE,cAAc,EACpB,YAAY,EAAE,QAAQ,CAAC,MAAM,CAAC,GAC5B;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,OAAO,EAAE,cAAc,EAAE,CAAA;CAAE,CASrD;AAED,wBAAgB,yBAAyB,CACxC,IAAI,EAAE,cAAc,EACpB,UAAU,EAAE,MAAM,GAChB;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,OAAO,EAAE,cAAc,EAAE,CAAA;CAAE,CAQrD;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,cAAc,EAAE,IAAI,EAAE,SAAS,cAAc,EAAE,GAAG,cAAc,CAI1G;AAED,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,cAAc,GAAG,KAAK,CAAC,GAAG,CAAC,CAelE;AAED,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,cAAc,CAgBlE;AAED,qBAAa,eAAe;IAC3B,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAqB;IAC1C,OAAO,CAAC,IAAI,CAAiB;IAE7B,YAAY,IAAI,EAAE,MAAM,GAAG,SAAS,EAGnC;IAED,IAAI,IAAI,cAAc,EAAE,CAEvB;IAED,IAAI,IAAI,GAAG,CAAC,MAAM,CAAC,CAElB;IAED,SAAS,CAAC,UAAU,EAAE,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,EAAE,CAE1C;IAED,aAAa,CAAC,YAAY,EAAE,QAAQ,CAAC,MAAM,CAAC,GAAG,cAAc,EAAE,CAO9D;IAED,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,cAAc,EAAE,CAOlD;IAED,MAAM,CAAC,IAAI,EAAE,SAAS,cAAc,EAAE,GAAG,IAAI,CAI5C;IAED,OAAO,CAAC,OAAO;CAQf","sourcesContent":["import { existsSync, mkdirSync, readFileSync, writeFileSync } from \"node:fs\";\nimport { dirname, join } from \"node:path\";\nimport type { Api, Model } from \"@ashx-j/lunr-ai\";\nimport { getAgentDir } from \"../config.ts\";\nimport { modelKey } from \"./official-catalog.ts\";\n\nexport const USER_MODELS_FILENAME = \"user-models.json\";\n\nexport interface UserModelEntry {\n\tid: string;\n\tname: string;\n\tapi: Api;\n\tprovider: string;\n\tbaseUrl: string;\n\treasoning: boolean;\n\tinput: (\"text\" | \"image\")[];\n\tcost: Model<Api>[\"cost\"];\n\tcontextWindow: number;\n\tmaxTokens: number;\n\tcompat?: Model<Api>[\"compat\"];\n\tthinkingLevelMap?: Model<Api>[\"thinkingLevelMap\"];\n\tsource: \"user\";\n}\n\nexport interface UserModelsFile {\n\tversion: number;\n\tmodels: UserModelEntry[];\n}\n\nexport function userModelsPath(agentDir: string = getAgentDir()): string {\n\treturn join(agentDir, USER_MODELS_FILENAME);\n}\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n\treturn typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\nfunction asFiniteNumber(value: unknown): number | undefined {\n\tif (typeof value === \"number\" && Number.isFinite(value)) return value;\n\treturn undefined;\n}\n\nfunction parseUserEntry(value: unknown): UserModelEntry | undefined {\n\tif (!isRecord(value)) return undefined;\n\tconst id = typeof value.id === \"string\" ? value.id.trim() : \"\";\n\tconst provider = typeof value.provider === \"string\" ? value.provider.trim() : \"\";\n\tif (!id || !provider) return undefined;\n\tconst parsedInput: (\"text\" | \"image\")[] = Array.isArray(value.input)\n\t\t? value.input.filter((entry): entry is \"text\" | \"image\" => entry === \"text\" || entry === \"image\")\n\t\t: [\"text\"];\n\tconst input: (\"text\" | \"image\")[] = parsedInput.length > 0 ? parsedInput : [\"text\"];\n\tconst cost = isRecord(value.cost)\n\t\t? {\n\t\t\t\tinput: asFiniteNumber(value.cost.input) ?? 0,\n\t\t\t\toutput: asFiniteNumber(value.cost.output) ?? 0,\n\t\t\t\tcacheRead: asFiniteNumber(value.cost.cacheRead) ?? 0,\n\t\t\t\tcacheWrite: asFiniteNumber(value.cost.cacheWrite) ?? 0,\n\t\t\t}\n\t\t: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };\n\treturn {\n\t\tid,\n\t\tname: typeof value.name === \"string\" && value.name.trim() ? value.name : id,\n\t\tapi: typeof value.api === \"string\" && value.api.trim() ? value.api : \"openai-completions\",\n\t\tprovider,\n\t\tbaseUrl: typeof value.baseUrl === \"string\" ? value.baseUrl : \"\",\n\t\treasoning: value.reasoning === true,\n\t\tinput,\n\t\tcost,\n\t\tcontextWindow: asFiniteNumber(value.contextWindow) ?? 128000,\n\t\tmaxTokens: asFiniteNumber(value.maxTokens) ?? 8192,\n\t\tcompat: isRecord(value.compat) ? (value.compat as Model<Api>[\"compat\"]) : undefined,\n\t\tthinkingLevelMap: isRecord(value.thinkingLevelMap)\n\t\t\t? (value.thinkingLevelMap as Model<Api>[\"thinkingLevelMap\"])\n\t\t\t: undefined,\n\t\tsource: \"user\",\n\t};\n}\n\nexport function emptyUserModels(): UserModelsFile {\n\treturn { version: 1, models: [] };\n}\n\n/** Read user-models.json. Missing or unreadable files yield an empty document (never throws). */\nexport function readUserModels(path: string): UserModelsFile {\n\ttry {\n\t\tif (!existsSync(path)) return emptyUserModels();\n\t\tconst parsed: unknown = JSON.parse(readFileSync(path, \"utf-8\"));\n\t\tif (!isRecord(parsed) || !Array.isArray(parsed.models)) return emptyUserModels();\n\t\tconst models: UserModelEntry[] = [];\n\t\tfor (const entry of parsed.models) {\n\t\t\tconst model = parseUserEntry(entry);\n\t\t\tif (model) models.push(model);\n\t\t}\n\t\treturn { version: asFiniteNumber(parsed.version) ?? 1, models };\n\t} catch {\n\t\treturn emptyUserModels();\n\t}\n}\n\nexport function writeUserModels(file: UserModelsFile, path: string): void {\n\tmkdirSync(dirname(path), { recursive: true, mode: 0o700 });\n\twriteFileSync(path, `${JSON.stringify({ version: file.version ?? 1, models: file.models }, null, 2)}\\n`, \"utf-8\");\n}\n\nexport function evictUserModelsOnOfficial(\n\tuser: UserModelsFile,\n\tofficialKeys: Iterable<string>,\n): { next: UserModelsFile; evicted: UserModelEntry[] } {\n\tconst hits = new Set(officialKeys);\n\tconst kept: UserModelEntry[] = [];\n\tconst evicted: UserModelEntry[] = [];\n\tfor (const model of user.models) {\n\t\tif (hits.has(modelKey(model.provider, model.id))) evicted.push(model);\n\t\telse kept.push(model);\n\t}\n\treturn { next: { version: user.version, models: kept }, evicted };\n}\n\nexport function evictUserModelsByProvider(\n\tuser: UserModelsFile,\n\tproviderId: string,\n): { next: UserModelsFile; evicted: UserModelEntry[] } {\n\tconst kept: UserModelEntry[] = [];\n\tconst evicted: UserModelEntry[] = [];\n\tfor (const model of user.models) {\n\t\tif (model.provider === providerId) evicted.push(model);\n\t\telse kept.push(model);\n\t}\n\treturn { next: { version: user.version, models: kept }, evicted };\n}\n\nexport function upsertUserModels(existing: UserModelsFile, rows: readonly UserModelEntry[]): UserModelsFile {\n\tconst byKey = new Map(existing.models.map((model) => [modelKey(model.provider, model.id), model]));\n\tfor (const row of rows) byKey.set(modelKey(row.provider, row.id), { ...row, source: \"user\" });\n\treturn { version: existing.version ?? 1, models: [...byKey.values()] };\n}\n\nexport function userEntryToModel(entry: UserModelEntry): Model<Api> {\n\treturn {\n\t\tid: entry.id,\n\t\tname: entry.name,\n\t\tapi: entry.api,\n\t\tprovider: entry.provider,\n\t\tbaseUrl: entry.baseUrl,\n\t\treasoning: entry.reasoning,\n\t\tinput: entry.input,\n\t\tcost: entry.cost,\n\t\tcontextWindow: entry.contextWindow,\n\t\tmaxTokens: entry.maxTokens,\n\t\tcompat: entry.compat,\n\t\tthinkingLevelMap: entry.thinkingLevelMap,\n\t};\n}\n\nexport function modelToUserEntry(model: Model<Api>): UserModelEntry {\n\treturn {\n\t\tid: model.id,\n\t\tname: model.name,\n\t\tapi: model.api,\n\t\tprovider: model.provider,\n\t\tbaseUrl: model.baseUrl,\n\t\treasoning: model.reasoning,\n\t\tinput: model.input,\n\t\tcost: model.cost,\n\t\tcontextWindow: model.contextWindow,\n\t\tmaxTokens: model.maxTokens,\n\t\tcompat: model.compat,\n\t\tthinkingLevelMap: model.thinkingLevelMap,\n\t\tsource: \"user\",\n\t};\n}\n\nexport class UserModelsStore {\n\tprivate readonly path: string | undefined;\n\tprivate file: UserModelsFile;\n\n\tconstructor(path: string | undefined) {\n\t\tthis.path = path;\n\t\tthis.file = path ? readUserModels(path) : emptyUserModels();\n\t}\n\n\tlist(): UserModelEntry[] {\n\t\treturn this.file.models;\n\t}\n\n\tkeys(): Set<string> {\n\t\treturn new Set(this.file.models.map((model) => modelKey(model.provider, model.id)));\n\t}\n\n\tmodelsFor(providerId: string): Model<Api>[] {\n\t\treturn this.file.models.filter((model) => model.provider === providerId).map(userEntryToModel);\n\t}\n\n\tevictOfficial(officialKeys: Iterable<string>): UserModelEntry[] {\n\t\tconst { next, evicted } = evictUserModelsOnOfficial(this.file, officialKeys);\n\t\tif (evicted.length > 0) {\n\t\t\tthis.file = next;\n\t\t\tthis.persist();\n\t\t}\n\t\treturn evicted;\n\t}\n\n\tevictProvider(providerId: string): UserModelEntry[] {\n\t\tconst { next, evicted } = evictUserModelsByProvider(this.file, providerId);\n\t\tif (evicted.length > 0) {\n\t\t\tthis.file = next;\n\t\t\tthis.persist();\n\t\t}\n\t\treturn evicted;\n\t}\n\n\tupsert(rows: readonly UserModelEntry[]): void {\n\t\tif (rows.length === 0) return;\n\t\tthis.file = upsertUserModels(this.file, rows);\n\t\tthis.persist();\n\t}\n\n\tprivate persist(): void {\n\t\tif (!this.path) return;\n\t\ttry {\n\t\t\twriteUserModels(this.file, this.path);\n\t\t} catch {\n\t\t\t// Persist is best-effort in test / in-memory runtimes.\n\t\t}\n\t}\n}\n"]}