@slkiser/opencode-quota 3.11.1 → 4.0.0

package/dist/lib/quota-providers-remote.js

@@ -0,0 +1,318 @@
+import { createProviderApiKeyResolver, getApiKeyCheckedPaths, getGlobalOpencodeConfigCandidatePaths, } from "./api-key-resolver.js";
+import { sanitizeSingleLineDisplayText } from "./display-sanitize.js";
+import { getAuthPaths, readAuthFile } from "./opencode-auth.js";
+import { REQUEST_TIMEOUT_MS } from "./types.js";
+export const QUOTA_PROVIDER_MAX_BODY_BYTES = 256 * 1024;
+export const QUOTA_PROVIDER_MAX_ACCOUNTING_ROWS = 100;
+export const QUOTA_PROVIDER_CONCURRENCY = 4;
+const RESULT_TYPES = new Set([
+    "quota",
+    "rate_limit",
+    "usage",
+    "spend",
+    "budget",
+    "balance",
+    "status",
+]);
+const ISO_TIMESTAMP_RE = /^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}(?:\.\d+)?(?:Z|[+-]\d{2}:\d{2})$/;
+function isRecord(value) {
+    return Boolean(value) && typeof value === "object" && !Array.isArray(value);
+}
+function hasOnlyKeys(value, allowed) {
+    const allowedSet = new Set(allowed);
+    return Object.keys(value).every((key) => allowedSet.has(key));
+}
+function isIsoTimestamp(value) {
+    return (typeof value === "string" && ISO_TIMESTAMP_RE.test(value) && Number.isFinite(Date.parse(value)));
+}
+function boundedDisplayText(value, maxCodePoints) {
+    if (typeof value !== "string")
+        return null;
+    const normalized = sanitizeSingleLineDisplayText(value);
+    const length = Array.from(normalized).length;
+    return length >= 1 && length <= maxCodePoints ? normalized : null;
+}
+function formatUsd(value) {
+    return `$${value.toFixed(2)}`;
+}
+function accountingMetadata(resultType, observedAtIso) {
+    return {
+        resultType,
+        acquisitionMethod: "remote_api",
+        ownership: "user_configured",
+        authority: "provider_reported",
+        ...(observedAtIso ? { observedAtIso } : {}),
+    };
+}
+export async function resolveQuotaProviderApiKey(source) {
+    const resolver = createProviderApiKeyResolver({
+        envVars: source.apiKeyEnv ? [{ name: source.apiKeyEnv, source: "env" }] : [],
+        providerKeys: [source.providerId],
+        allowedEnvVars: source.apiKeyEnv ? [source.apiKeyEnv] : [],
+        configJsonSource: "opencode.json",
+        configJsoncSource: "opencode.jsonc",
+        getConfigCandidates: getGlobalOpencodeConfigCandidatePaths,
+        auth: {
+            readAuth: readAuthFile,
+            authKeys: [source.providerId],
+            authSource: "auth.json",
+        },
+    });
+    const resolved = await resolver.resolve();
+    return {
+        ...(resolved ? { key: resolved.key } : {}),
+        source: resolved?.source ?? null,
+        checkedPaths: getApiKeyCheckedPaths({
+            envVarNames: source.apiKeyEnv ? [source.apiKeyEnv] : [],
+            getConfigCandidates: getGlobalOpencodeConfigCandidatePaths,
+        }),
+        authPaths: getAuthPaths(),
+    };
+}
+async function readBoundedJson(response) {
+    const contentType = response.headers.get("content-type")?.split(";", 1)[0]?.trim().toLowerCase();
+    if (!contentType ||
+        (contentType !== "application/json" &&
+            !(contentType.startsWith("application/") && contentType.endsWith("+json")))) {
+        throw new Error("Expected a JSON response");
+    }
+    const contentLength = response.headers.get("content-length");
+    if (contentLength) {
+        const parsedLength = Number(contentLength);
+        if (Number.isFinite(parsedLength) && parsedLength > QUOTA_PROVIDER_MAX_BODY_BYTES) {
+            throw new Error("Response exceeded 262144 bytes");
+        }
+    }
+    let bytes;
+    if (response.body) {
+        const reader = response.body.getReader();
+        const chunks = [];
+        let total = 0;
+        try {
+            while (true) {
+                const next = await reader.read();
+                if (next.done)
+                    break;
+                total += next.value.byteLength;
+                if (total > QUOTA_PROVIDER_MAX_BODY_BYTES) {
+                    await reader.cancel();
+                    throw new Error("Response exceeded 262144 bytes");
+                }
+                chunks.push(next.value);
+            }
+        }
+        finally {
+            reader.releaseLock();
+        }
+        bytes = new Uint8Array(total);
+        let offset = 0;
+        for (const chunk of chunks) {
+            bytes.set(chunk, offset);
+            offset += chunk.byteLength;
+        }
+    }
+    else {
+        const buffer = new Uint8Array(await response.arrayBuffer());
+        if (buffer.byteLength > QUOTA_PROVIDER_MAX_BODY_BYTES) {
+            throw new Error("Response exceeded 262144 bytes");
+        }
+        bytes = buffer;
+    }
+    try {
+        return JSON.parse(new TextDecoder().decode(bytes));
+    }
+    catch {
+        throw new Error("Invalid JSON response");
+    }
+}
+function parseAccountingV1(source, body) {
+    if (!isRecord(body) ||
+        !hasOnlyKeys(body, ["version", "entries"]) ||
+        body.version !== "accounting-v1" ||
+        !Array.isArray(body.entries)) {
+        return { success: false, error: "Invalid accounting-v1 response" };
+    }
+    if (body.entries.length < 1 || body.entries.length > QUOTA_PROVIDER_MAX_ACCOUNTING_ROWS) {
+        return { success: false, error: "Invalid accounting-v1 response" };
+    }
+    const entries = [];
+    for (const raw of body.entries) {
+        if (!isRecord(raw) ||
+            !hasOnlyKeys(raw, [
+                "kind",
+                "name",
+                "resultType",
+                "percentRemaining",
+                "value",
+                "label",
+                "right",
+                "resetTimeIso",
+                "observedAtIso",
+            ])) {
+            return { success: false, error: "Invalid accounting-v1 response" };
+        }
+        const name = boundedDisplayText(raw.name, 80);
+        const resultType = typeof raw.resultType === "string" && RESULT_TYPES.has(raw.resultType)
+            ? raw.resultType
+            : null;
+        const label = raw.label === undefined ? undefined : boundedDisplayText(raw.label, 80);
+        const right = raw.right === undefined ? undefined : boundedDisplayText(raw.right, 160);
+        const resetTimeIso = raw.resetTimeIso === undefined
+            ? undefined
+            : isIsoTimestamp(raw.resetTimeIso)
+                ? raw.resetTimeIso
+                : null;
+        const observedAtIso = raw.observedAtIso === undefined
+            ? undefined
+            : isIsoTimestamp(raw.observedAtIso)
+                ? raw.observedAtIso
+                : null;
+        if (!name ||
+            !resultType ||
+            (raw.label !== undefined && label === null) ||
+            (raw.right !== undefined && right === null) ||
+            resetTimeIso === null ||
+            observedAtIso === null) {
+            return { success: false, error: "Invalid accounting-v1 response" };
+        }
+        const common = {
+            accounting: accountingMetadata(resultType, observedAtIso),
+            name: `${source.label} ${name}`,
+            group: source.label,
+            ...(label ? { label } : {}),
+            ...(right ? { right } : {}),
+            ...(resetTimeIso ? { resetTimeIso } : {}),
+        };
+        if (raw.kind === "percent" &&
+            typeof raw.percentRemaining === "number" &&
+            Number.isFinite(raw.percentRemaining) &&
+            raw.percentRemaining <= 100 &&
+            ["quota", "rate_limit", "budget"].includes(resultType) &&
+            raw.value === undefined) {
+            entries.push({ ...common, kind: "percent", percentRemaining: raw.percentRemaining });
+            continue;
+        }
+        const value = boundedDisplayText(raw.value, 160);
+        if (raw.kind === "value" && value !== null && raw.percentRemaining === undefined) {
+            entries.push({ ...common, kind: "value", value });
+            continue;
+        }
+        return { success: false, error: "Invalid accounting-v1 response" };
+    }
+    return { success: true, entries };
+}
+function parseOpenRouterKeyV1(source, body) {
+    if (!isRecord(body) || !isRecord(body.data)) {
+        return { success: false, error: "Invalid openrouter-key-v1 response" };
+    }
+    const data = body.data;
+    const usage = data.usage;
+    const limit = data.limit;
+    const remaining = data.limit_remaining;
+    if (typeof usage !== "number" ||
+        !Number.isFinite(usage) ||
+        usage < 0 ||
+        (limit !== null && (typeof limit !== "number" || !Number.isFinite(limit) || limit < 0)) ||
+        (remaining !== undefined && (typeof remaining !== "number" || !Number.isFinite(remaining)))) {
+        return { success: false, error: "Invalid openrouter-key-v1 response" };
+    }
+    if (typeof limit === "number" && limit > 0) {
+        const remainingValue = typeof remaining === "number" ? remaining : limit - usage;
+        if (remainingValue > limit) {
+            return {
+                success: false,
+                error: "Invalid openrouter-key-v1 response",
+            };
+        }
+        const percentRemaining = (remainingValue / limit) * 100;
+        return {
+            success: true,
+            entries: [
+                {
+                    accounting: accountingMetadata("budget"),
+                    kind: "percent",
+                    name: `${source.label} budget`,
+                    group: source.label,
+                    label: "Budget:",
+                    right: `${formatUsd(usage)}/${formatUsd(limit)}`,
+                    percentRemaining,
+                },
+            ],
+        };
+    }
+    return {
+        success: true,
+        entries: [
+            {
+                accounting: accountingMetadata("spend"),
+                kind: "value",
+                name: `${source.label} spend`,
+                group: source.label,
+                label: "Spend:",
+                value: formatUsd(usage),
+            },
+        ],
+    };
+}
+export async function fetchRemoteQuotaProvider(source, apiKey, requestTimeoutMs) {
+    const timeoutMs = requestTimeoutMs ?? REQUEST_TIMEOUT_MS;
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), timeoutMs);
+    try {
+        const response = await fetch(source.url, {
+            method: "GET",
+            redirect: "manual",
+            signal: controller.signal,
+            headers: {
+                Authorization: `Bearer ${apiKey}`,
+                Accept: "application/json",
+            },
+        });
+        if (response.redirected || (response.status >= 300 && response.status < 400)) {
+            return { success: false, error: "Redirect rejected" };
+        }
+        if (!response.ok) {
+            return { success: false, error: `HTTP ${response.status}` };
+        }
+        const body = await readBoundedJson(response);
+        return source.format === "accounting-v1"
+            ? parseAccountingV1(source, body)
+            : parseOpenRouterKeyV1(source, body);
+    }
+    catch (error) {
+        if (controller.signal.aborted || (error instanceof Error && error.name === "AbortError")) {
+            return {
+                success: false,
+                error: `Request timeout after ${Math.round(timeoutMs / 1000)}s`,
+            };
+        }
+        if (error instanceof Error &&
+            [
+                "Response exceeded 262144 bytes",
+                "Expected a JSON response",
+                "Invalid JSON response",
+            ].includes(error.message)) {
+            return { success: false, error: error.message };
+        }
+        return { success: false, error: "Failed to read accounting data" };
+    }
+    finally {
+        clearTimeout(timeoutId);
+    }
+}
+export async function mapWithConcurrency(values, concurrency, map) {
+    const results = new Array(values.length);
+    let nextIndex = 0;
+    async function worker() {
+        while (true) {
+            const index = nextIndex;
+            nextIndex += 1;
+            if (index >= values.length)
+                return;
+            results[index] = await map(values[index], index);
+        }
+    }
+    await Promise.all(Array.from({ length: Math.min(Math.max(1, concurrency), values.length) }, () => worker()));
+    return results;
+}
+//# sourceMappingURL=quota-providers-remote.js.map

package/dist/lib/quota-providers-remote.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-providers-remote.js","sourceRoot":"","sources":["../../src/lib/quota-providers-remote.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,4BAA4B,EAC5B,qBAAqB,EACrB,qCAAqC,GACtC,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,6BAA6B,EAAE,MAAM,uBAAuB,CAAC;AACtE,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,MAAM,CAAC,MAAM,6BAA6B,GAAG,GAAG,GAAG,IAAI,CAAC;AACxD,MAAM,CAAC,MAAM,kCAAkC,GAAG,GAAG,CAAC;AACtD,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC;AAE5C,MAAM,YAAY,GAAG,IAAI,GAAG,CAAuB;IACjD,OAAO;IACP,YAAY;IACZ,OAAO;IACP,OAAO;IACP,QAAQ;IACR,SAAS;IACT,QAAQ;CACT,CAAC,CAAC;AACH,MAAM,gBAAgB,GAAG,sEAAsE,CAAC;AAehG,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,WAAW,CAAC,KAA8B,EAAE,OAA0B;IAC7E,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;IACpC,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;AAChE,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACpC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ,IAAI,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAChG,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAc,EAAE,aAAqB;IAC/D,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC3C,MAAM,UAAU,GAAG,6BAA6B,CAAC,KAAK,CAAC,CAAC;IACxD,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC;IAC7C,OAAO,MAAM,IAAI,CAAC,IAAI,MAAM,IAAI,aAAa,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC;AACpE,CAAC;AAED,SAAS,SAAS,CAAC,KAAa;IAC9B,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;AAChC,CAAC;AAED,SAAS,kBAAkB,CACzB,UAAgC,EAChC,aAAsB;IAEtB,OAAO;QACL,UAAU;QACV,iBAAiB,EAAE,YAAY;QAC/B,SAAS,EAAE,iBAAiB;QAC5B,SAAS,EAAE,mBAAmB;QAC9B,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC5C,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,MAAwC;IAExC,MAAM,QAAQ,GAAG,4BAA4B,CAA0B;QACrE,OAAO,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE;QAC5E,YAAY,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC;QACjC,cAAc,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE;QAC1D,gBAAgB,EAAE,eAAe;QACjC,iBAAiB,EAAE,gBAAgB;QACnC,mBAAmB,EAAE,qCAAqC;QAC1D,IAAI,EAAE;YACJ,QAAQ,EAAE,YAAY;YACtB,QAAQ,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC;YAC7B,UAAU,EAAE,WAAW;SACxB;KACF,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,CAAC;IAC1C,OAAO;QACL,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1C,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,IAAI;QAChC,YAAY,EAAE,qBAAqB,CAAC;YAClC,WAAW,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE;YACvD,mBAAmB,EAAE,qCAAqC;SAC3D,CAAC;QACF,SAAS,EAAE,YAAY,EAAE;KAC1B,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,QAAkB;IAC/C,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACjG,IACE,CAAC,WAAW;QACZ,CAAC,WAAW,KAAK,kBAAkB;YACjC,CAAC,CAAC,WAAW,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAC7E,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IAC7D,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM,YAAY,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC;QAC3C,IAAI,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,6BAA6B,EAAE,CAAC;YAClF,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,IAAI,KAAiB,CAAC;IACtB,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;QAClB,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,MAAM,GAAiB,EAAE,CAAC;QAChC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBACjC,IAAI,IAAI,CAAC,IAAI;oBAAE,MAAM;gBACrB,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;gBAC/B,IAAI,KAAK,GAAG,6BAA6B,EAAE,CAAC;oBAC1C,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;oBACtB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;gBACpD,CAAC;gBACD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;QACD,KAAK,GAAG,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC;QAC9B,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC;QAC7B,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;QAC5D,IAAI,MAAM,CAAC,UAAU,GAAG,6BAA6B,EAAE,CAAC;YACtD,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QACD,KAAK,GAAG,MAAM,CAAC;IACjB,CAAC;IAED,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAY,CAAC;IAChE,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CACxB,MAAwC,EACxC,IAAa;IAEb,IACE,CAAC,QAAQ,CAAC,IAAI,CAAC;QACf,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAC1C,IAAI,CAAC,OAAO,KAAK,eAAe;QAChC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAC5B,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,gCAAgC,EAAE,CAAC;IACrE,CAAC;IACD,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,kCAAkC,EAAE,CAAC;QACxF,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,gCAAgC,EAAE,CAAC;IACrE,CAAC;IAED,MAAM,OAAO,GAAsB,EAAE,CAAC;IACtC,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAC/B,IACE,CAAC,QAAQ,CAAC,GAAG,CAAC;YACd,CAAC,WAAW,CAAC,GAAG,EAAE;gBAChB,MAAM;gBACN,MAAM;gBACN,YAAY;gBACZ,kBAAkB;gBAClB,OAAO;gBACP,OAAO;gBACP,OAAO;gBACP,cAAc;gBACd,eAAe;aAChB,CAAC,EACF,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,gCAAgC,EAAE,CAAC;QACrE,CAAC;QAED,MAAM,IAAI,GAAG,kBAAkB,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC9C,MAAM,UAAU,GACd,OAAO,GAAG,CAAC,UAAU,KAAK,QAAQ,IAAI,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,UAAkC,CAAC;YAC5F,CAAC,CAAE,GAAG,CAAC,UAAmC;YAC1C,CAAC,CAAC,IAAI,CAAC;QACX,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACtF,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QACvF,MAAM,YAAY,GAChB,GAAG,CAAC,YAAY,KAAK,SAAS;YAC5B,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC;gBAChC,CAAC,CAAC,GAAG,CAAC,YAAY;gBAClB,CAAC,CAAC,IAAI,CAAC;QACb,MAAM,aAAa,GACjB,GAAG,CAAC,aAAa,KAAK,SAAS;YAC7B,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,aAAa,CAAC;gBACjC,CAAC,CAAC,GAAG,CAAC,aAAa;gBACnB,CAAC,CAAC,IAAI,CAAC;QAEb,IACE,CAAC,IAAI;YACL,CAAC,UAAU;YACX,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,CAAC;YAC3C,CAAC,GAAG,CAAC,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,CAAC;YAC3C,YAAY,KAAK,IAAI;YACrB,aAAa,KAAK,IAAI,EACtB,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,gCAAgC,EAAE,CAAC;QACrE,CAAC;QAED,MAAM,MAAM,GAAG;YACb,UAAU,EAAE,kBAAkB,CAAC,UAAU,EAAE,aAAa,CAAC;YACzD,IAAI,EAAE,GAAG,MAAM,CAAC,KAAK,IAAI,IAAI,EAAE;YAC/B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3B,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3B,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC1C,CAAC;QAEF,IACE,GAAG,CAAC,IAAI,KAAK,SAAS;YACtB,OAAO,GAAG,CAAC,gBAAgB,KAAK,QAAQ;YACxC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,CAAC;YACrC,GAAG,CAAC,gBAAgB,IAAI,GAAG;YAC3B,CAAC,OAAO,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC;YACtD,GAAG,CAAC,KAAK,KAAK,SAAS,EACvB,CAAC;YACD,OAAO,CAAC,IAAI,CAAC,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC;YACrF,SAAS;QACX,CAAC;QAED,MAAM,KAAK,GAAG,kBAAkB,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QACjD,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,KAAK,IAAI,IAAI,GAAG,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;YACjF,OAAO,CAAC,IAAI,CAAC,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;YAClD,SAAS;QACX,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,gCAAgC,EAAE,CAAC;IACrE,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,oBAAoB,CAC3B,MAAwC,EACxC,IAAa;IAEb,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5C,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,oCAAoC,EAAE,CAAC;IACzE,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;IACvB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;IACzB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;IACzB,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC;IACvC,IACE,OAAO,KAAK,KAAK,QAAQ;QACzB,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC;QACvB,KAAK,GAAG,CAAC;QACT,CAAC,KAAK,KAAK,IAAI,IAAI,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC;QACvF,CAAC,SAAS,KAAK,SAAS,IAAI,CAAC,OAAO,SAAS,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,EAC3F,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,oCAAoC,EAAE,CAAC;IACzE,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;QAC3C,MAAM,cAAc,GAAG,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,GAAG,KAAK,CAAC;QACjF,IAAI,cAAc,GAAG,KAAK,EAAE,CAAC;YAC3B,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,oCAAoC;aAC5C,CAAC;QACJ,CAAC;QACD,MAAM,gBAAgB,GAAG,CAAC,cAAc,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC;QACxD,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE;gBACP;oBACE,UAAU,EAAE,kBAAkB,CAAC,QAAQ,CAAC;oBACxC,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,GAAG,MAAM,CAAC,KAAK,SAAS;oBAC9B,KAAK,EAAE,MAAM,CAAC,KAAK;oBACnB,KAAK,EAAE,SAAS;oBAChB,KAAK,EAAE,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,SAAS,CAAC,KAAK,CAAC,EAAE;oBAChD,gBAAgB;iBACjB;aACF;SACF,CAAC;IACJ,CAAC;IAED,OAAO;QACL,OAAO,EAAE,IAAI;QACb,OAAO,EAAE;YACP;gBACE,UAAU,EAAE,kBAAkB,CAAC,OAAO,CAAC;gBACvC,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,GAAG,MAAM,CAAC,KAAK,QAAQ;gBAC7B,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,KAAK,EAAE,QAAQ;gBACf,KAAK,EAAE,SAAS,CAAC,KAAK,CAAC;aACxB;SACF;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,MAAwC,EACxC,MAAc,EACd,gBAAyB;IAEzB,MAAM,SAAS,GAAG,gBAAgB,IAAI,kBAAkB,CAAC;IACzD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;IAElE,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE;YACvC,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,QAAQ;YAClB,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,MAAM,EAAE;gBACjC,MAAM,EAAE,kBAAkB;aAC3B;SACF,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,UAAU,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,MAAM,GAAG,GAAG,CAAC,EAAE,CAAC;YAC7E,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;QAC9D,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO,MAAM,CAAC,MAAM,KAAK,eAAe;YACtC,CAAC,CAAC,iBAAiB,CAAC,MAAM,EAAE,IAAI,CAAC;YACjC,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,UAAU,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,EAAE,CAAC;YACzF,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,yBAAyB,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG;aAChE,CAAC;QACJ,CAAC;QACD,IACE,KAAK,YAAY,KAAK;YACtB;gBACE,gCAAgC;gBAChC,0BAA0B;gBAC1B,uBAAuB;aACxB,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,EACzB,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;QAClD,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,gCAAgC,EAAE,CAAC;IACrE,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,SAAS,CAAC,CAAC;IAC1B,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,MAAoB,EACpB,WAAmB,EACnB,GAA4C;IAE5C,MAAM,OAAO,GAAG,IAAI,KAAK,CAAI,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5C,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,KAAK,UAAU,MAAM;QACnB,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,KAAK,GAAG,SAAS,CAAC;YACxB,SAAS,IAAI,CAAC,CAAC;YACf,IAAI,KAAK,IAAI,MAAM,CAAC,MAAM;gBAAE,OAAO;YACnC,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,KAAK,CAAE,EAAE,KAAK,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,MAAM,OAAO,CAAC,GAAG,CACf,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC,CAC1F,CAAC;IACF,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/lib/quota-providers.d.ts

@@ -0,0 +1,77 @@
+export declare const QUOTA_PROVIDER_REMOTE_FORMATS: readonly ["accounting-v1", "openrouter-key-v1"];
+export declare const QUOTA_PROVIDER_MODES: readonly ["remote-api", "local-estimate"];
+export declare const QUOTA_PROVIDER_WINDOW_TYPES: readonly ["utc-day", "rolling"];
+export declare const QUOTA_PROVIDERS_AGGREGATE_ID = "quota-providers";
+export declare const MAINTAINED_LOCAL_ESTIMATE_IDS: readonly ["qwen-code", "alibaba-coding-plan"];
+export declare function isMaintainedQuotaProviderTuning(definition: QuotaProviderDefinition): boolean;
+export declare function customQuotaProviderDefinitions(definitions: readonly QuotaProviderDefinition[]): QuotaProviderDefinition[];
+export declare function resolveQuotaProviderSessionModelIdentity(params: {
+    currentModel: string;
+    currentProviderID?: string;
+}): {
+    providerId: string;
+    modelId: string;
+} | null;
+export declare function selectEligibleQuotaProviderDefinitions(params: {
+    definitions: readonly QuotaProviderDefinition[];
+    availableProviderIds: ReadonlySet<string>;
+    onlyCurrentModel?: boolean;
+    currentModel?: string;
+    currentProviderID?: string;
+}): QuotaProviderDefinition[];
+export type QuotaProviderRemoteFormat = (typeof QUOTA_PROVIDER_REMOTE_FORMATS)[number];
+export type QuotaProviderMode = (typeof QUOTA_PROVIDER_MODES)[number];
+export type QuotaProviderWindowType = (typeof QUOTA_PROVIDER_WINDOW_TYPES)[number];
+interface QuotaProviderDefinitionBase {
+    id: string;
+    /** Effective OpenCode provider id. Input omits this when it is the same as id. */
+    providerId: string;
+    /** Normalized display label; omitted input defaults to id. */
+    label: string;
+    mode: QuotaProviderMode;
+    /** Exact OpenCode model ids, without a provider prefix. Omission covers the provider. */
+    modelIds?: string[];
+}
+export interface RemoteApiQuotaProviderDefinition extends QuotaProviderDefinitionBase {
+    mode: "remote-api";
+    /** Canonical absolute HTTPS URL, or loopback HTTP URL. */
+    url: string;
+    format: QuotaProviderRemoteFormat;
+    apiKeyEnv?: string;
+}
+export interface LocalEstimateWindow {
+    id: string;
+    label: string;
+    type: QuotaProviderWindowType;
+    /** Present only for rolling windows. */
+    durationMinutes?: number;
+    requestLimit: number;
+    usdBudget?: number;
+}
+export interface LocalEstimateQuotaProviderDefinition extends QuotaProviderDefinitionBase {
+    mode: "local-estimate";
+    windows: LocalEstimateWindow[];
+    /**
+     * Exact source model id -> models.dev provider/model fallback.
+     * Automatic models.dev matching always wins.
+     */
+    pricingModelMap?: Record<string, string>;
+}
+export type QuotaProviderDefinition = RemoteApiQuotaProviderDefinition | LocalEstimateQuotaProviderDefinition;
+export interface QuotaProviderValidationIssue {
+    key: string;
+    message: string;
+}
+export type QuotaProvidersValidationResult = {
+    value: QuotaProviderDefinition[];
+    issues: [];
+} | {
+    value?: undefined;
+    issues: QuotaProviderValidationIssue[];
+};
+/** Validate and normalize the complete global-only quotaProviders array atomically. */
+export declare function validateQuotaProviders(value: unknown): QuotaProvidersValidationResult;
+export declare function cloneQuotaProviders(definitions: readonly QuotaProviderDefinition[]): QuotaProviderDefinition[];
+export declare function findQuotaProviderDefinition(definitions: readonly QuotaProviderDefinition[], id: string): QuotaProviderDefinition | undefined;
+export {};
+//# sourceMappingURL=quota-providers.d.ts.map

package/dist/lib/quota-providers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-providers.d.ts","sourceRoot":"","sources":["../../src/lib/quota-providers.ts"],"names":[],"mappings":"AAMA,eAAO,MAAM,6BAA6B,iDAAkD,CAAC;AAC7F,eAAO,MAAM,oBAAoB,2CAA4C,CAAC;AAC9E,eAAO,MAAM,2BAA2B,iCAAkC,CAAC;AAC3E,eAAO,MAAM,4BAA4B,oBAAoB,CAAC;AAC9D,eAAO,MAAM,6BAA6B,+CAAgD,CAAC;AAE3F,wBAAgB,+BAA+B,CAAC,UAAU,EAAE,uBAAuB,GAAG,OAAO,CAE5F;AAED,wBAAgB,8BAA8B,CAC5C,WAAW,EAAE,SAAS,uBAAuB,EAAE,GAC9C,uBAAuB,EAAE,CAE3B;AAED,wBAAgB,wCAAwC,CAAC,MAAM,EAAE;IAC/D,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,GAAG;IAAE,UAAU,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAajD;AAED,wBAAgB,sCAAsC,CAAC,MAAM,EAAE;IAC7D,WAAW,EAAE,SAAS,uBAAuB,EAAE,CAAC;IAChD,oBAAoB,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IAC1C,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B,GAAG,uBAAuB,EAAE,CAyB5B;AAED,MAAM,MAAM,yBAAyB,GAAG,CAAC,OAAO,6BAA6B,CAAC,CAAC,MAAM,CAAC,CAAC;AACvF,MAAM,MAAM,iBAAiB,GAAG,CAAC,OAAO,oBAAoB,CAAC,CAAC,MAAM,CAAC,CAAC;AACtE,MAAM,MAAM,uBAAuB,GAAG,CAAC,OAAO,2BAA2B,CAAC,CAAC,MAAM,CAAC,CAAC;AAEnF,UAAU,2BAA2B;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,kFAAkF;IAClF,UAAU,EAAE,MAAM,CAAC;IACnB,8DAA8D;IAC9D,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,iBAAiB,CAAC;IACxB,yFAAyF;IACzF,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;CACrB;AAED,MAAM,WAAW,gCAAiC,SAAQ,2BAA2B;IACnF,IAAI,EAAE,YAAY,CAAC;IACnB,0DAA0D;IAC1D,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,yBAAyB,CAAC;IAClC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,mBAAmB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,uBAAuB,CAAC;IAC9B,wCAAwC;IACxC,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,oCAAqC,SAAQ,2BAA2B;IACvF,IAAI,EAAE,gBAAgB,CAAC;IACvB,OAAO,EAAE,mBAAmB,EAAE,CAAC;IAC/B;;;OAGG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC1C;AAED,MAAM,MAAM,uBAAuB,GAC/B,gCAAgC,GAChC,oCAAoC,CAAC;AAEzC,MAAM,WAAW,4BAA4B;IAC3C,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,MAAM,8BAA8B,GACtC;IAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC;IAAC,MAAM,EAAE,EAAE,CAAA;CAAE,GAChD;IAAE,KAAK,CAAC,EAAE,SAAS,CAAC;IAAC,MAAM,EAAE,4BAA4B,EAAE,CAAA;CAAE,CAAC;AAyZlE,uFAAuF;AACvF,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,OAAO,GAAG,8BAA8B,CAuPrF;AAED,wBAAgB,mBAAmB,CACjC,WAAW,EAAE,SAAS,uBAAuB,EAAE,GAC9C,uBAAuB,EAAE,CAE3B;AAED,wBAAgB,2BAA2B,CACzC,WAAW,EAAE,SAAS,uBAAuB,EAAE,EAC/C,EAAE,EAAE,MAAM,GACT,uBAAuB,GAAG,SAAS,CAErC"}