@thelioo/opencode-balancer 0.1.3 → 0.1.4

package/dist/core/usage/providers/openai.js

@@ -0,0 +1,133 @@
+import { redactUsageError, redactUsagePayload } from "../redact";
+function unavailable(account, message, error) {
+    return {
+        providerID: account.providerID,
+        alias: account.alias,
+        fetchedAt: Date.now(),
+        confidence: "unavailable",
+        message,
+        ...(error ? { error: redactUsageError(error, account) } : {}),
+    };
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function asNumber(value) {
+    return typeof value === "number" && Number.isFinite(value) ? value : undefined;
+}
+function decodeJwtPayload(token) {
+    const payload = token.split(".")[1];
+    if (!payload)
+        return undefined;
+    try {
+        const normalized = payload.replace(/-/g, "+").replace(/_/g, "/");
+        const padded = `${normalized}${"=".repeat((4 - (normalized.length % 4)) % 4)}`;
+        const decoded = JSON.parse(Buffer.from(padded, "base64").toString("utf8"));
+        return isRecord(decoded) ? decoded : undefined;
+    }
+    catch {
+        return undefined;
+    }
+}
+function chatgptAccountID(account) {
+    if (account.auth.type !== "oauth")
+        return undefined;
+    if (account.auth.accountId)
+        return account.auth.accountId;
+    const payload = decodeJwtPayload(account.auth.access);
+    const authClaim = payload?.["https://api.openai.com/auth"];
+    if (!isRecord(authClaim))
+        return undefined;
+    const accountID = authClaim.chatgpt_account_id;
+    return typeof accountID === "string" && accountID.trim() ? accountID : undefined;
+}
+function planName(planType) {
+    if (typeof planType !== "string" || !planType.trim())
+        return undefined;
+    const normalized = planType.trim().toLowerCase();
+    const names = {
+        plus: "ChatGPT Plus",
+        pro: "ChatGPT Pro",
+        team: "ChatGPT Team",
+        enterprise: "ChatGPT Enterprise",
+    };
+    return names[normalized] ?? planType;
+}
+function resetAtMillis(value) {
+    const timestamp = asNumber(value);
+    if (timestamp === undefined || timestamp < 0)
+        return undefined;
+    return timestamp > 1_000_000_000_000 ? timestamp : timestamp * 1000;
+}
+async function refreshChatGPTUsage(account) {
+    if (account.auth.type !== "oauth") {
+        return unavailable(account, "OpenAI ChatGPT usage requires OAuth auth.");
+    }
+    try {
+        const headers = {
+            Authorization: `Bearer ${account.auth.access}`,
+            "User-Agent": "opencode-balancer/0.1.3",
+        };
+        const accountID = chatgptAccountID(account);
+        if (accountID)
+            headers["ChatGPT-Account-Id"] = accountID;
+        const response = await fetch("https://chatgpt.com/backend-api/wham/usage", { headers });
+        if (!response.ok) {
+            return unavailable(account, `OpenAI ChatGPT usage request failed with HTTP ${response.status}.`);
+        }
+        const body = (await response.json());
+        const rateLimit = isRecord(body) && isRecord(body.rate_limit) ? body.rate_limit : undefined;
+        const primary = rateLimit && isRecord(rateLimit.primary_window) ? rateLimit.primary_window : undefined;
+        return {
+            providerID: account.providerID,
+            alias: account.alias,
+            fetchedAt: Date.now(),
+            confidence: "exact",
+            planName: isRecord(body) ? planName(body.plan_type) : undefined,
+            usedPercent: primary ? asNumber(primary.used_percent) : undefined,
+            resetAt: primary ? resetAtMillis(primary.reset_at) : undefined,
+            message: "OpenAI ChatGPT usage fetched.",
+            rawRedacted: redactUsagePayload(body, account),
+        };
+    }
+    catch (error) {
+        return unavailable(account, "OpenAI ChatGPT usage request failed; exact usage is unavailable.", error instanceof Error ? error.message : String(error));
+    }
+}
+export const openaiUsageService = {
+    providerID: "openai",
+    supports(providerID) {
+        return providerID === "openai";
+    },
+    async refreshUsage(account) {
+        if (account.auth.type !== "api") {
+            return refreshChatGPTUsage(account);
+        }
+        const url = new URL("https://api.openai.com/v1/organization/usage/completions");
+        url.searchParams.set("start_time", String(Math.floor((Date.now() - 24 * 60 * 60 * 1000) / 1000)));
+        url.searchParams.set("bucket_width", "1d");
+        try {
+            const response = await fetch(url, {
+                headers: {
+                    Authorization: `Bearer ${account.auth.key}`,
+                },
+            });
+            if (!response.ok) {
+                return unavailable(account, `OpenAI usage request failed with HTTP ${response.status}.`);
+            }
+            const body = await response.json();
+            return {
+                providerID: account.providerID,
+                alias: account.alias,
+                fetchedAt: Date.now(),
+                confidence: "exact",
+                message: "OpenAI organization usage fetched.",
+                rawRedacted: redactUsagePayload(body, account),
+            };
+        }
+        catch (error) {
+            return unavailable(account, "OpenAI usage request failed; exact usage is unavailable.", error instanceof Error ? error.message : String(error));
+        }
+    },
+};
+//# sourceMappingURL=openai.js.map

package/dist/core/usage/providers/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../../src/core/usage/providers/openai.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAGjE,SAAS,WAAW,CAAC,OAAgB,EAAE,OAAe,EAAE,KAAc;IAClE,OAAO;QACH,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,UAAU,EAAE,aAAa;QACzB,OAAO;QACP,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,gBAAgB,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAChE,CAAC;AACN,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC5B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC5B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AACnF,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa;IACnC,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACpC,IAAI,CAAC,OAAO;QAAE,OAAO,SAAS,CAAC;IAE/B,IAAI,CAAC;QACD,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACjE,MAAM,MAAM,GAAG,GAAG,UAAU,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC/E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAY,CAAC;QACtF,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;IACnD,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,SAAS,CAAC;IACrB,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAgB;IACtC,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO;QAAE,OAAO,SAAS,CAAC;IACpD,IAAI,OAAO,CAAC,IAAI,CAAC,SAAS;QAAE,OAAO,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC;IAE1D,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtD,MAAM,SAAS,GAAG,OAAO,EAAE,CAAC,6BAA6B,CAAC,CAAC;IAC3D,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;QAAE,OAAO,SAAS,CAAC;IAE3C,MAAM,SAAS,GAAG,SAAS,CAAC,kBAAkB,CAAC;IAC/C,OAAO,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;AACrF,CAAC;AAED,SAAS,QAAQ,CAAC,QAAiB;IAC/B,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE;QAAE,OAAO,SAAS,CAAC;IACvE,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACjD,MAAM,KAAK,GAA2B;QAClC,IAAI,EAAE,cAAc;QACpB,GAAG,EAAE,aAAa;QAClB,IAAI,EAAE,cAAc;QACpB,UAAU,EAAE,oBAAoB;KACnC,CAAC;IACF,OAAO,KAAK,CAAC,UAAU,CAAC,IAAI,QAAQ,CAAC;AACzC,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACjC,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAClC,IAAI,SAAS,KAAK,SAAS,IAAI,SAAS,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IAC/D,OAAO,SAAS,GAAG,iBAAiB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC;AACxE,CAAC;AAED,KAAK,UAAU,mBAAmB,CAAC,OAAgB;IAC/C,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAChC,OAAO,WAAW,CAAC,OAAO,EAAE,2CAA2C,CAAC,CAAC;IAC7E,CAAC;IAED,IAAI,CAAC;QACD,MAAM,OAAO,GAA2B;YACpC,aAAa,EAAE,UAAU,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE;YAC9C,YAAY,EAAE,yBAAyB;SAC1C,CAAC;QACF,MAAM,SAAS,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,SAAS;YAAE,OAAO,CAAC,oBAAoB,CAAC,GAAG,SAAS,CAAC;QAEzD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,4CAA4C,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACxF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACf,OAAO,WAAW,CAAC,OAAO,EAAE,iDAAiD,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QACrG,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAY,CAAC;QAChD,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QAC5F,MAAM,OAAO,GAAG,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,CAAC;QACvG,OAAO;YACH,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,UAAU,EAAE,OAAO;YACnB,QAAQ,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS;YAC/D,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS;YACjE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS;YAC9D,OAAO,EAAE,+BAA+B;YACxC,WAAW,EAAE,kBAAkB,CAAC,IAAI,EAAE,OAAO,CAAC;SACjD,CAAC;IACN,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,WAAW,CACd,OAAO,EACP,kEAAkE,EAClE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACzD,CAAC;IACN,CAAC;AACL,CAAC;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAyB;IACpD,UAAU,EAAE,QAAQ;IACpB,QAAQ,CAAC,UAAU;QACf,OAAO,UAAU,KAAK,QAAQ,CAAC;IACnC,CAAC;IACD,KAAK,CAAC,YAAY,CAAC,OAAO;QACtB,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YAC9B,OAAO,mBAAmB,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,0DAA0D,CAAC,CAAC;QAChF,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAClG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;QAE3C,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAC9B,OAAO,EAAE;oBACL,aAAa,EAAE,UAAU,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE;iBAC9C;aACJ,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,OAAO,WAAW,CAAC,OAAO,EAAE,yCAAyC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YAC7F,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,OAAO;gBACH,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,UAAU,EAAE,OAAO;gBACnB,OAAO,EAAE,oCAAoC;gBAC7C,WAAW,EAAE,kBAAkB,CAAC,IAAI,EAAE,OAAO,CAAC;aACjD,CAAC;QACN,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,WAAW,CACd,OAAO,EACP,0DAA0D,EAC1D,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACzD,CAAC;QACN,CAAC;IACL,CAAC;CACJ,CAAC"}

package/dist/core/usage/redact.d.ts

@@ -0,0 +1,3 @@
+import type { Account } from "../types";
+export declare function redactUsagePayload(value: unknown, account: Account): unknown;
+export declare function redactUsageError(error: string, account: Account): string;

package/dist/core/usage/redact.js

@@ -0,0 +1,67 @@
+const REDACTED = "[redacted]";
+function accountSecretValues(account) {
+    const values = new Set();
+    if (account.auth.type === "api")
+        values.add(account.auth.key);
+    if (account.auth.type === "oauth") {
+        values.add(account.auth.access);
+        values.add(account.auth.refresh);
+    }
+    if (account.auth.type === "wellknown") {
+        values.add(account.auth.key);
+        values.add(account.auth.token);
+    }
+    values.delete("");
+    return values;
+}
+function keyWords(key) {
+    return key
+        .replace(/([a-z0-9])([A-Z])/g, "$1 $2")
+        .toLowerCase()
+        .split(/[^a-z0-9]+/)
+        .filter(Boolean);
+}
+function isCredentialKey(key) {
+    const lower = key.toLowerCase();
+    const compact = lower.replace(/[^a-z0-9]/g, "");
+    if (lower.includes("api_key") ||
+        compact.includes("apikey") ||
+        lower.includes("access_token") ||
+        compact.includes("accesstoken") ||
+        lower.includes("refresh_token") ||
+        compact.includes("refreshtoken") ||
+        lower.includes("x-api-key") ||
+        compact.includes("xapikey") ||
+        lower.includes("authorization")) {
+        return true;
+    }
+    const words = keyWords(key);
+    return ["token", "key", "secret", "password"].some((credential) => lower === credential || words.includes(credential));
+}
+export function redactUsagePayload(value, account) {
+    return redactValue(value, accountSecretValues(account));
+}
+export function redactUsageError(error, account) {
+    return redactString(error, accountSecretValues(account));
+}
+function redactString(value, secrets) {
+    let redacted = value;
+    for (const secret of [...secrets].sort((a, b) => b.length - a.length)) {
+        redacted = redacted.split(secret).join(REDACTED);
+    }
+    return redacted;
+}
+function redactValue(value, secrets) {
+    if (typeof value === "string")
+        return redactString(value, secrets);
+    if (!value || typeof value !== "object")
+        return value;
+    if (Array.isArray(value))
+        return value.map((item) => redactValue(item, secrets));
+    const redacted = {};
+    for (const [key, child] of Object.entries(value)) {
+        redacted[redactString(key, secrets)] = isCredentialKey(key) ? REDACTED : redactValue(child, secrets);
+    }
+    return redacted;
+}
+//# sourceMappingURL=redact.js.map

package/dist/core/usage/redact.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"redact.js","sourceRoot":"","sources":["../../../src/core/usage/redact.ts"],"names":[],"mappings":"AAEA,MAAM,QAAQ,GAAG,YAAY,CAAC;AAE9B,SAAS,mBAAmB,CAAC,OAAgB;IACzC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;IACjC,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,KAAK;QAAE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC9D,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAChC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnC,CAAC;IACD,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAClB,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,QAAQ,CAAC,GAAW;IACzB,OAAO,GAAG;SACL,OAAO,CAAC,oBAAoB,EAAE,OAAO,CAAC;SACtC,WAAW,EAAE;SACb,KAAK,CAAC,YAAY,CAAC;SACnB,MAAM,CAAC,OAAO,CAAC,CAAC;AACzB,CAAC;AAED,SAAS,eAAe,CAAC,GAAW;IAChC,MAAM,KAAK,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;IAChC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;IAChD,IACI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;QACzB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAC1B,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC9B,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;QAC/B,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC;QAC/B,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC;QAChC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC;QAC3B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC3B,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC,EACjC,CAAC;QACC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC5B,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,EAAE,CAC9D,KAAK,KAAK,UAAU,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,CACrD,CAAC;AACN,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,KAAc,EAAE,OAAgB;IAC/D,OAAO,WAAW,CAAC,KAAK,EAAE,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,KAAa,EAAE,OAAgB;IAC5D,OAAO,YAAY,CAAC,KAAK,EAAE,mBAAmB,CAAC,OAAO,CAAC,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,YAAY,CAAC,KAAa,EAAE,OAAoB;IACrD,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,KAAK,MAAM,MAAM,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;QACpE,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACrD,CAAC;IACD,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,WAAW,CAAC,KAAc,EAAE,OAAoB;IACrD,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACnE,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC;IAEjF,MAAM,QAAQ,GAA4B,EAAE,CAAC;IAC7C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC/C,QAAQ,CAAC,YAAY,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACzG,CAAC;IACD,OAAO,QAAQ,CAAC;AACpB,CAAC"}

package/dist/core/usage/store.d.ts

@@ -0,0 +1,4 @@
+import type { Database } from "bun:sqlite";
+import type { ProviderUsageSnapshot } from "./types";
+export declare function saveUsageSnapshot(db: Database, snapshot: ProviderUsageSnapshot): void;
+export declare function getUsageSnapshot(db: Database, providerID: string, alias: string): ProviderUsageSnapshot | undefined;

package/dist/core/usage/store.js

@@ -0,0 +1,31 @@
+export function saveUsageSnapshot(db, snapshot) {
+    const { rawRedacted, ...normalizedSnapshot } = snapshot;
+    const rawRedactedJSON = Object.hasOwn(snapshot, "rawRedacted") ? JSON.stringify(rawRedacted) : null;
+    db.query(`INSERT INTO usage_snapshots (
+            provider_id, alias, fetched_at, confidence, normalized_json, raw_redacted_json, error
+        ) VALUES (?, ?, ?, ?, ?, ?, ?)
+        ON CONFLICT(provider_id, alias) DO UPDATE SET
+            fetched_at = excluded.fetched_at,
+            confidence = excluded.confidence,
+            normalized_json = excluded.normalized_json,
+            raw_redacted_json = excluded.raw_redacted_json,
+            error = excluded.error`).run(snapshot.providerID, snapshot.alias, snapshot.fetchedAt, snapshot.confidence, JSON.stringify(normalizedSnapshot), rawRedactedJSON, snapshot.error ?? null);
+    if (snapshot.confidence === "exact" && snapshot.usedPercent !== undefined && snapshot.usedPercent < 100) {
+        db.query(`UPDATE accounts
+             SET rate_limited_until = NULL,
+                 updated_at = ?
+             WHERE provider_id = ? AND alias = ?`).run(snapshot.fetchedAt, snapshot.providerID, snapshot.alias);
+    }
+}
+export function getUsageSnapshot(db, providerID, alias) {
+    const row = db
+        .query("SELECT normalized_json, raw_redacted_json FROM usage_snapshots WHERE provider_id = ? AND alias = ?")
+        .get(providerID, alias);
+    if (!row)
+        return undefined;
+    const snapshot = JSON.parse(row.normalized_json);
+    if (row.raw_redacted_json !== null)
+        snapshot.rawRedacted = JSON.parse(row.raw_redacted_json);
+    return snapshot;
+}
+//# sourceMappingURL=store.js.map

package/dist/core/usage/store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.js","sourceRoot":"","sources":["../../../src/core/usage/store.ts"],"names":[],"mappings":"AAGA,MAAM,UAAU,iBAAiB,CAAC,EAAY,EAAE,QAA+B;IAC3E,MAAM,EAAE,WAAW,EAAE,GAAG,kBAAkB,EAAE,GAAG,QAAQ,CAAC;IACxD,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAEpG,EAAE,CAAC,KAAK,CACJ;;;;;;;;mCAQ2B,CAC9B,CAAC,GAAG,CACD,QAAQ,CAAC,UAAU,EACnB,QAAQ,CAAC,KAAK,EACd,QAAQ,CAAC,SAAS,EAClB,QAAQ,CAAC,UAAU,EACnB,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,EAClC,eAAe,EACf,QAAQ,CAAC,KAAK,IAAI,IAAI,CACzB,CAAC;IAEF,IAAI,QAAQ,CAAC,UAAU,KAAK,OAAO,IAAI,QAAQ,CAAC,WAAW,KAAK,SAAS,IAAI,QAAQ,CAAC,WAAW,GAAG,GAAG,EAAE,CAAC;QACtG,EAAE,CAAC,KAAK,CACJ;;;iDAGqC,CACxC,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;IACnE,CAAC;AACL,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,EAAY,EAAE,UAAkB,EAAE,KAAa;IAC5E,MAAM,GAAG,GAAG,EAAE;SACT,KAAK,CACF,oGAAoG,CACvG;SACA,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IAC5B,IAAI,CAAC,GAAG;QAAE,OAAO,SAAS,CAAC;IAE3B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,eAAe,CAA0B,CAAC;IAC1E,IAAI,GAAG,CAAC,iBAAiB,KAAK,IAAI;QAAE,QAAQ,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IAC7F,OAAO,QAAQ,CAAC;AACpB,CAAC"}

package/dist/core/usage/types.d.ts

@@ -0,0 +1,21 @@
+import type { Account } from "../types";
+export type UsageConfidence = "exact" | "estimated" | "unavailable";
+export type ProviderUsageSnapshot = {
+    providerID: string;
+    alias: string;
+    fetchedAt: number;
+    confidence: UsageConfidence;
+    planName?: string;
+    usedTokens?: number;
+    remainingTokens?: number;
+    usedPercent?: number;
+    resetAt?: number;
+    message: string;
+    rawRedacted?: unknown;
+    error?: string;
+};
+export type ProviderUsageService = {
+    providerID: string;
+    supports(providerID: string): boolean;
+    refreshUsage(account: Account): Promise<ProviderUsageSnapshot>;
+};

package/dist/core/usage/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/core/usage/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/core/usage/types.ts"],"names":[],"mappings":""}

package/dist/index.d.ts

@@ -1,37 +1 @@
-declare const _default: ({ client }: import("@opencode-ai/plugin").PluginInput) => Promise<{
-    config: (cfg: import("@opencode-ai/plugin").Config) => Promise<void>;
-    "command.execute.before": (input: {
-        command: string;
-        sessionID: string;
-        arguments: string;
-    }, output: {
-        parts: import("@opencode-ai/sdk").Part[];
-    }) => Promise<void>;
-    "experimental.chat.messages.transform": (_input: {}, output: {
-        messages: {
-            info: import("@opencode-ai/sdk").Message;
-            parts: import("@opencode-ai/sdk").Part[];
-        }[];
-    }) => Promise<void>;
-    "chat.headers": (input: {
-        sessionID: string;
-        agent: string;
-        model: import("@opencode-ai/sdk").Model;
-        provider: import("@opencode-ai/plugin").ProviderContext;
-        message: import("@opencode-ai/sdk").UserMessage;
-    }, output: {
-        headers: Record<string, string>;
-    }) => Promise<void>;
-    tool: {
-        balancer_command: {
-            description: string;
-            args: {
-                command: import("zod").ZodString;
-            };
-            execute(args: {
-                command: string;
-            }, context: import("@opencode-ai/plugin").ToolContext): Promise<import("@opencode-ai/plugin").ToolResult>;
-        };
-    };
-}>;
-export default _default;
+export { serverPlugin as default } from "./server/index";

package/dist/index.js

@@ -1,61 +1,2 @@
-import { tool } from "@opencode-ai/plugin";
-import { BALANCER_METADATA_KEY, INTERNAL_REQUEST_HEADER, getActiveAccount, setNativeAuth, showToast, state, } from "./balancer/core";
-import { startAuthWatcher } from "./balancer/auth-watcher";
-import { runBalancerCommand } from "./balancer/commands";
-import { installFetchPatch } from "./balancer/fetch-patch";
-export default (async ({ client }) => {
-    await installFetchPatch(client);
-    startAuthWatcher(client);
-    return {
-        config: async (cfg) => {
-            cfg.command = {
-                ...(cfg.command ?? {}),
-                balancer: {
-                    description: "Manage multiple accounts per provider",
-                    template: "Run the internal balancer command with these arguments: $ARGUMENTS",
-                },
-            };
-        },
-        "command.execute.before": async (input, output) => {
-            if (input.command !== "balancer")
-                return;
-            const result = await runBalancerCommand(client, input.arguments);
-            output.parts.length = 0;
-            await showToast(client, result, "info");
-            throw new Error(`[balancer]\n${result}`);
-        },
-        "experimental.chat.messages.transform": async (_input, output) => {
-            output.messages = output.messages.filter((message) => {
-                return !message.parts.some((part) => {
-                    return part?.metadata?.[BALANCER_METADATA_KEY] === true;
-                });
-            });
-        },
-        "chat.headers": async (input, output) => {
-            const providerID = input.model.providerID;
-            const account = await getActiveAccount(providerID);
-            if (!account)
-                return;
-            if (account.auth.type === "oauth")
-                await setNativeAuth(client, providerID, account.auth);
-            const requestID = crypto.randomUUID();
-            state.pendingRequests.set(requestID, {
-                providerID,
-                account,
-            });
-            output.headers[INTERNAL_REQUEST_HEADER] = requestID;
-        },
-        tool: {
-            balancer_command: tool({
-                description: "Run an opencode account balancer command. Use for managing provider account aliases.",
-                args: {
-                    command: tool.schema
-                        .string()
-                        .describe("Command, e.g. 'list', 'alias work', 'use anthropic personal'."),
-                },
-                execute: async (args) => runBalancerCommand(client, args.command),
-            }),
-        },
-    };
-});
+export { serverPlugin as default } from "./server/index";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAe,MAAM,qBAAqB,CAAC;AACxD,OAAO,EACH,qBAAqB,EACrB,uBAAuB,EACvB,gBAAgB,EAChB,aAAa,EACb,SAAS,EACT,KAAK,GACR,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAE3D,eAAe,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;IACjC,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAChC,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAEzB,OAAO;QACH,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YAClB,GAAG,CAAC,OAAO,GAAG;gBACV,GAAG,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;gBACtB,QAAQ,EAAE;oBACN,WAAW,EAAE,uCAAuC;oBACpD,QAAQ,EACJ,oEAAoE;iBAC3E;aACJ,CAAC;QACN,CAAC;QAED,wBAAwB,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YAC9C,IAAI,KAAK,CAAC,OAAO,KAAK,UAAU;gBAAE,OAAO;YACzC,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,MAAM,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;YACjE,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;YACxB,MAAM,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,eAAe,MAAM,EAAE,CAAC,CAAC;QAC7C,CAAC;QAED,sCAAsC,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE;YAC7D,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE;gBACjD,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAS,EAAE,EAAE;oBACrC,OAAO,IAAI,EAAE,QAAQ,EAAE,CAAC,qBAAqB,CAAC,KAAK,IAAI,CAAC;gBAC5D,CAAC,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;QACP,CAAC;QAED,cAAc,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACpC,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC;YAC1C,MAAM,OAAO,GAAG,MAAM,gBAAgB,CAAC,UAAU,CAAC,CAAC;YACnD,IAAI,CAAC,OAAO;gBAAE,OAAO;YAErB,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO;gBAC7B,MAAM,aAAa,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAE1D,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;YACtC,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE;gBACjC,UAAU;gBACV,OAAO;aACV,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,GAAG,SAAS,CAAC;QACxD,CAAC;QAED,IAAI,EAAE;YACF,gBAAgB,EAAE,IAAI,CAAC;gBACnB,WAAW,EACP,sFAAsF;gBAC1F,IAAI,EAAE;oBACF,OAAO,EAAE,IAAI,CAAC,MAAM;yBACf,MAAM,EAAE;yBACR,QAAQ,CACL,+DAA+D,CAClE;iBACR;gBACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CACpB,kBAAkB,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC;aAC/C,CAAC;SACL;KACJ,CAAC;AACN,CAAC,CAAkB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,IAAI,OAAO,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/server/auth-watcher.d.ts

@@ -0,0 +1,31 @@
+import type { Database } from "bun:sqlite";
+import type { AuthInfo } from "../core/types";
+type NativeAuthReadResult = {
+    ok: true;
+    auth: Record<string, AuthInfo>;
+} | {
+    ok: false;
+};
+type ToastVariant = "info" | "success" | "warning" | "error";
+type AuthWatcherOptions = {
+    db: Database;
+    client: any;
+    initialSnapshot?: Record<string, AuthInfo>;
+    readAuth: () => NativeAuthReadResult;
+    showToast?: (client: any, message: string, variant: ToastVariant) => Promise<void>;
+    createPending?: (db: Database, providerID: string, auth: AuthInfo, source: "auth-file") => unknown;
+    isSuppressed?: (providerID: string) => boolean;
+};
+declare function parseNativeAuthContent(content: string): NativeAuthReadResult;
+export declare function readNativeAuth(): NativeAuthReadResult;
+export declare function captureUnknownNativeAuth(db: Database, providerID: string, auth: AuthInfo): boolean;
+declare function initialAuthSnapshot(db: Database, authByProvider: Record<string, AuthInfo>): Record<string, AuthInfo>;
+export declare function startAuthWatcher(client: any, db: Database): void;
+export declare const __testParseNativeAuthContent: typeof parseNativeAuthContent;
+export declare const __testInitialAuthSnapshot: typeof initialAuthSnapshot;
+export declare function __testCreateAuthWatcher(options: Omit<AuthWatcherOptions, "client"> & {
+    client?: any;
+}): {
+    poll(): Promise<void>;
+};
+export {};