@thelioo/opencode-balancer 0.1.3 → 0.1.4

package/dist/server/auth-watcher.js

@@ -0,0 +1,227 @@
+import { existsSync, readFileSync } from "node:fs";
+import { getActiveAccount, listAccounts, updateAccountAuth } from "../core/accounts";
+import { isNativeAuthCaptureSuppressed as isDbNativeAuthCaptureSuppressed } from "../core/native-auth-suppression";
+import { nativeAuthPath } from "../core/path";
+import { createPendingConnection } from "../core/pending";
+import { isNativeAuthCaptureSuppressed, showToast as defaultShowToast } from "./native";
+let started = false;
+function isRecord(value) {
+    return Boolean(value) && typeof value === "object" && !Array.isArray(value);
+}
+function stringMetadata(value) {
+    if (value === undefined)
+        return { ok: true, metadata: undefined };
+    if (!isRecord(value))
+        return { ok: false };
+    const metadata = {};
+    for (const [key, entry] of Object.entries(value)) {
+        if (typeof entry !== "string")
+            return { ok: false };
+        metadata[key] = entry;
+    }
+    return { ok: true, metadata };
+}
+function attachMetadata(auth, value) {
+    const result = stringMetadata(value);
+    if (!result.ok)
+        return undefined;
+    if (result.metadata)
+        auth.metadata = result.metadata;
+    return auth;
+}
+function parseAuthInfo(value) {
+    if (!isRecord(value) || typeof value.type !== "string")
+        return undefined;
+    if (value.type === "api" && typeof value.key === "string") {
+        return attachMetadata({ type: "api", key: value.key }, value.metadata);
+    }
+    if (value.type === "oauth" &&
+        typeof value.refresh === "string" &&
+        typeof value.access === "string" &&
+        typeof value.expires === "number" &&
+        Number.isFinite(value.expires)) {
+        const auth = {
+            type: "oauth",
+            refresh: value.refresh,
+            access: value.access,
+            expires: value.expires,
+        };
+        if (typeof value.accountId === "string")
+            auth.accountId = value.accountId;
+        if (typeof value.enterpriseUrl === "string")
+            auth.enterpriseUrl = value.enterpriseUrl;
+        return attachMetadata(auth, value.metadata);
+    }
+    if (value.type === "wellknown" &&
+        typeof value.key === "string" &&
+        typeof value.token === "string") {
+        return attachMetadata({ type: "wellknown", key: value.key, token: value.token }, value.metadata);
+    }
+    return undefined;
+}
+function parseNativeAuthContent(content) {
+    let raw;
+    try {
+        raw = JSON.parse(content);
+    }
+    catch {
+        return { ok: false };
+    }
+    if (!isRecord(raw))
+        return { ok: false };
+    const authByProvider = {};
+    for (const [providerID, auth] of Object.entries(raw)) {
+        const parsed = parseAuthInfo(auth);
+        if (!parsed)
+            return { ok: false };
+        authByProvider[providerID] = parsed;
+    }
+    return { ok: true, auth: authByProvider };
+}
+export function readNativeAuth() {
+    const authContent = Bun.env.OPENCODE_AUTH_CONTENT;
+    if (authContent !== undefined)
+        return parseNativeAuthContent(authContent);
+    const path = nativeAuthPath();
+    if (!existsSync(path))
+        return { ok: true, auth: {} };
+    try {
+        return parseNativeAuthContent(readFileSync(path, "utf8"));
+    }
+    catch {
+        return { ok: false };
+    }
+}
+function authChanged(previous, current) {
+    return JSON.stringify(previous ?? null) !== JSON.stringify(current);
+}
+function decodeJwtPayload(token) {
+    const payload = token.split(".")[1];
+    if (!payload)
+        return undefined;
+    try {
+        const decoded = JSON.parse(Buffer.from(payload, "base64url").toString("utf8"));
+        return isRecord(decoded) ? decoded : undefined;
+    }
+    catch {
+        return undefined;
+    }
+}
+function oauthAccountID(auth) {
+    if (auth.accountId)
+        return auth.accountId;
+    const claim = decodeJwtPayload(auth.access)?.["https://api.openai.com/auth"];
+    if (!isRecord(claim))
+        return undefined;
+    const accountID = claim.chatgpt_account_id;
+    return typeof accountID === "string" && accountID.length > 0 ? accountID : undefined;
+}
+function sameSavedAuth(saved, current) {
+    if (saved.type !== current.type)
+        return false;
+    if (saved.type === "api" && current.type === "api")
+        return saved.key === current.key;
+    if (saved.type === "wellknown" && current.type === "wellknown")
+        return saved.key === current.key && saved.token === current.token;
+    if (saved.type === "oauth" && current.type === "oauth") {
+        const savedAccountID = oauthAccountID(saved);
+        const currentAccountID = oauthAccountID(current);
+        if (savedAccountID && currentAccountID)
+            return savedAccountID === currentAccountID;
+        return saved.refresh === current.refresh;
+    }
+    return false;
+}
+function saveKnownNativeAuth(db, providerID, auth) {
+    const account = listAccounts(db, providerID).find((candidate) => sameSavedAuth(candidate.auth, auth));
+    if (!account)
+        return false;
+    updateAccountAuth(db, providerID, account.alias, auth);
+    return true;
+}
+export function captureUnknownNativeAuth(db, providerID, auth) {
+    if (saveKnownNativeAuth(db, providerID, auth))
+        return false;
+    createPendingConnection(db, providerID, auth, "auth-file");
+    return true;
+}
+function initialAuthSnapshot(db, authByProvider) {
+    const snapshot = {};
+    for (const [providerID, auth] of Object.entries(authByProvider)) {
+        if (listAccounts(db, providerID).some((account) => sameSavedAuth(account.auth, auth))) {
+            snapshot[providerID] = auth;
+        }
+    }
+    return snapshot;
+}
+function createAuthWatcher(options) {
+    let lastAuthSnapshot = options.initialSnapshot ?? {};
+    let inFlight = false;
+    const notify = options.showToast ?? defaultShowToast;
+    const createPending = options.createPending ?? createPendingConnection;
+    const isSuppressed = options.isSuppressed ?? ((providerID) => {
+        return isNativeAuthCaptureSuppressed(providerID) || isDbNativeAuthCaptureSuppressed(options.db, providerID);
+    });
+    return {
+        async poll() {
+            if (inFlight)
+                return;
+            inFlight = true;
+            try {
+                const result = options.readAuth();
+                if (!result.ok)
+                    return;
+                const next = result.auth;
+                const changed = [];
+                for (const [providerID, auth] of Object.entries(next)) {
+                    if (isSuppressed(providerID)) {
+                        const active = getActiveAccount(options.db, providerID);
+                        if (active && sameSavedAuth(active.auth, auth))
+                            updateAccountAuth(options.db, providerID, active.alias, auth);
+                        lastAuthSnapshot[providerID] = auth;
+                        continue;
+                    }
+                    if (saveKnownNativeAuth(options.db, providerID, auth)) {
+                        lastAuthSnapshot[providerID] = auth;
+                        continue;
+                    }
+                    if (!authChanged(lastAuthSnapshot[providerID], auth))
+                        continue;
+                    changed.push([providerID, auth]);
+                }
+                for (const [providerID, auth] of changed) {
+                    createPending(options.db, providerID, auth, "auth-file");
+                    lastAuthSnapshot[providerID] = auth;
+                    await notify(options.client, `Balancer detected a ${providerID} connection. Save an alias from the TUI sidebar.`, "info");
+                }
+                for (const providerID of Object.keys(lastAuthSnapshot)) {
+                    if (!(providerID in next))
+                        delete lastAuthSnapshot[providerID];
+                }
+            }
+            finally {
+                inFlight = false;
+            }
+        },
+    };
+}
+export function startAuthWatcher(client, db) {
+    if (started)
+        return;
+    started = true;
+    const initial = readNativeAuth();
+    const watcher = createAuthWatcher({
+        db,
+        client,
+        initialSnapshot: initial.ok ? initialAuthSnapshot(db, initial.auth) : {},
+        readAuth: readNativeAuth,
+    });
+    const timer = setInterval(() => void watcher.poll(), 1_000);
+    timer.unref?.();
+}
+export const __testParseNativeAuthContent = parseNativeAuthContent;
+export const __testInitialAuthSnapshot = initialAuthSnapshot;
+export function __testCreateAuthWatcher(options) {
+    return createAuthWatcher({ ...options, client: options.client ?? {} });
+}
+//# sourceMappingURL=auth-watcher.js.map

package/dist/server/auth-watcher.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth-watcher.js","sourceRoot":"","sources":["../../src/server/auth-watcher.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrF,OAAO,EAAE,6BAA6B,IAAI,+BAA+B,EAAE,MAAM,iCAAiC,CAAC;AACnH,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,uBAAuB,EAAE,MAAM,iBAAiB,CAAC;AAE1D,OAAO,EAAE,6BAA6B,EAAE,SAAS,IAAI,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAExF,IAAI,OAAO,GAAG,KAAK,CAAC;AA2BpB,SAAS,QAAQ,CAAC,KAAc;IAC5B,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IAClC,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,EAAE,EAAE,EAAE,IAAa,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;IAC3E,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,EAAE,EAAE,KAAc,EAAE,CAAC;IAEpD,MAAM,QAAQ,GAA2B,EAAE,CAAC;IAC5C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC/C,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,OAAO,EAAE,EAAE,EAAE,KAAc,EAAE,CAAC;QAC7D,QAAQ,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IAC1B,CAAC;IACD,OAAO,EAAE,EAAE,EAAE,IAAa,EAAE,QAAQ,EAAE,CAAC;AAC3C,CAAC;AAED,SAAS,cAAc,CAAqB,IAAO,EAAE,KAAc;IAC/D,MAAM,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;IACrC,IAAI,CAAC,MAAM,CAAC,EAAE;QAAE,OAAO,SAAS,CAAC;IACjC,IAAI,MAAM,CAAC,QAAQ;QAAE,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;IACrD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACjC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAEzE,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,IAAI,OAAO,KAAK,CAAC,GAAG,KAAK,QAAQ,EAAE,CAAC;QACxD,OAAO,cAAc,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,CAAC,GAAG,EAAE,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;IAC3E,CAAC;IAED,IACI,KAAK,CAAC,IAAI,KAAK,OAAO;QACtB,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QACjC,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ;QAChC,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QACjC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,EAChC,CAAC;QACC,MAAM,IAAI,GAAa;YACnB,IAAI,EAAE,OAAO;YACb,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,OAAO,EAAE,KAAK,CAAC,OAAO;SACzB,CAAC;QACF,IAAI,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ;YAAE,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;QAC1E,IAAI,OAAO,KAAK,CAAC,aAAa,KAAK,QAAQ;YAAE,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,aAAa,CAAC;QACtF,OAAO,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IAED,IACI,KAAK,CAAC,IAAI,KAAK,WAAW;QAC1B,OAAO,KAAK,CAAC,GAAG,KAAK,QAAQ;QAC7B,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ,EACjC,CAAC;QACC,OAAO,cAAc,CACjB,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,EAAE,KAAK,CAAC,GAAG,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,EACzD,KAAK,CAAC,QAAQ,CACjB,CAAC;IACN,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAS,sBAAsB,CAAC,OAAe;IAC3C,IAAI,GAAY,CAAC;IACjB,IAAI,CAAC;QACD,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;IAEzC,MAAM,cAAc,GAA6B,EAAE,CAAC;IACpD,KAAK,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACnD,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;QAClC,cAAc,CAAC,UAAU,CAAC,GAAG,MAAM,CAAC;IACxC,CAAC;IACD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,cAAc;IAC1B,MAAM,WAAW,GAAG,GAAG,CAAC,GAAG,CAAC,qBAAqB,CAAC;IAClD,IAAI,WAAW,KAAK,SAAS;QAAE,OAAO,sBAAsB,CAAC,WAAW,CAAC,CAAC;IAE1E,MAAM,IAAI,GAAG,cAAc,EAAE,CAAC;IAC9B,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;IAErD,IAAI,CAAC;QACD,OAAO,sBAAsB,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;IAC9D,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;IACzB,CAAC;AACL,CAAC;AAED,SAAS,WAAW,CAAC,QAA8B,EAAE,OAAiB;IAClE,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,IAAI,IAAI,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;AACxE,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;IAC/B,IAAI,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAY,CAAC;QAC1F,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,cAAc,CAAC,IAA0C;IAC9D,IAAI,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC,SAAS,CAAC;IAC1C,MAAM,KAAK,GAAG,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,6BAA6B,CAAC,CAAC;IAC7E,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACvC,MAAM,SAAS,GAAG,KAAK,CAAC,kBAAkB,CAAC;IAC3C,OAAO,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;AACzF,CAAC;AAED,SAAS,aAAa,CAAC,KAAe,EAAE,OAAiB;IACrD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,CAAC,IAAI;QAAE,OAAO,KAAK,CAAC;IAC9C,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,IAAI,OAAO,CAAC,IAAI,KAAK,KAAK;QAAE,OAAO,KAAK,CAAC,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC;IACrF,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;QAAE,OAAO,KAAK,CAAC,GAAG,KAAK,OAAO,CAAC,GAAG,IAAI,KAAK,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK,CAAC;IAClI,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QACrD,MAAM,cAAc,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,gBAAgB,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;QACjD,IAAI,cAAc,IAAI,gBAAgB;YAAE,OAAO,cAAc,KAAK,gBAAgB,CAAC;QACnF,OAAO,KAAK,CAAC,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC;IAC7C,CAAC;IACD,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAS,mBAAmB,CAAC,EAAY,EAAE,UAAkB,EAAE,IAAc;IACzE,MAAM,OAAO,GAAG,YAAY,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IACtG,IAAI,CAAC,OAAO;QAAE,OAAO,KAAK,CAAC;IAC3B,iBAAiB,CAAC,EAAE,EAAE,UAAU,EAAE,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACvD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,EAAY,EAAE,UAAkB,EAAE,IAAc;IACrF,IAAI,mBAAmB,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC;QAAE,OAAO,KAAK,CAAC;IAC5D,uBAAuB,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;IAC3D,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,mBAAmB,CAAC,EAAY,EAAE,cAAwC;IAC/E,MAAM,QAAQ,GAA6B,EAAE,CAAC;IAC9C,KAAK,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QAC9D,IAAI,YAAY,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,aAAa,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC;YACpF,QAAQ,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC;QAChC,CAAC;IACL,CAAC;IACD,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,iBAAiB,CAAC,OAA2B;IAClD,IAAI,gBAAgB,GAAG,OAAO,CAAC,eAAe,IAAI,EAAE,CAAC;IACrD,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,IAAI,gBAAgB,CAAC;IACrD,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,uBAAuB,CAAC;IACvE,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,CAAC,CAAC,UAAkB,EAAE,EAAE;QACjE,OAAO,6BAA6B,CAAC,UAAU,CAAC,IAAI,+BAA+B,CAAC,OAAO,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAChH,CAAC,CAAC,CAAC;IAEH,OAAO;QACH,KAAK,CAAC,IAAI;YACN,IAAI,QAAQ;gBAAE,OAAO;YACrB,QAAQ,GAAG,IAAI,CAAC;YAChB,IAAI,CAAC;gBACD,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;gBAClC,IAAI,CAAC,MAAM,CAAC,EAAE;oBAAE,OAAO;gBAEvB,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;gBACzB,MAAM,OAAO,GAA8B,EAAE,CAAC;gBAC9C,KAAK,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;oBACpD,IAAI,YAAY,CAAC,UAAU,CAAC,EAAE,CAAC;wBAC3B,MAAM,MAAM,GAAG,gBAAgB,CAAC,OAAO,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;wBACxD,IAAI,MAAM,IAAI,aAAa,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC;4BAAE,iBAAiB,CAAC,OAAO,CAAC,EAAE,EAAE,UAAU,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;wBAC9G,gBAAgB,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC;wBACpC,SAAS;oBACb,CAAC;oBACD,IAAI,mBAAmB,CAAC,OAAO,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,EAAE,CAAC;wBACpD,gBAAgB,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC;wBACpC,SAAS;oBACb,CAAC;oBACD,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC;wBAAE,SAAS;oBAC/D,OAAO,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC;gBACrC,CAAC;gBAED,KAAK,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,OAAO,EAAE,CAAC;oBACvC,aAAa,CAAC,OAAO,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;oBACzD,gBAAgB,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC;oBACpC,MAAM,MAAM,CACR,OAAO,CAAC,MAAM,EACd,uBAAuB,UAAU,kDAAkD,EACnF,MAAM,CACT,CAAC;gBACN,CAAC;gBAED,KAAK,MAAM,UAAU,IAAI,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC;oBACrD,IAAI,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC;wBAAE,OAAO,gBAAgB,CAAC,UAAU,CAAC,CAAC;gBACnE,CAAC;YACL,CAAC;oBAAS,CAAC;gBACP,QAAQ,GAAG,KAAK,CAAC;YACrB,CAAC;QACL,CAAC;KACJ,CAAC;AACN,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAW,EAAE,EAAY;IACtD,IAAI,OAAO;QAAE,OAAO;IACpB,OAAO,GAAG,IAAI,CAAC;IACf,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;IACjC,MAAM,OAAO,GAAG,iBAAiB,CAAC;QAC9B,EAAE;QACF,MAAM;QACN,eAAe,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,mBAAmB,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE;QACxE,QAAQ,EAAE,cAAc;KAC3B,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,KAAK,OAAO,CAAC,IAAI,EAAE,EAAE,KAAK,CAAC,CAAC;IAC3D,KAAgC,CAAC,KAAK,EAAE,EAAE,CAAC;AAChD,CAAC;AAED,MAAM,CAAC,MAAM,4BAA4B,GAAG,sBAAsB,CAAC;AAEnE,MAAM,CAAC,MAAM,yBAAyB,GAAG,mBAAmB,CAAC;AAE7D,MAAM,UAAU,uBAAuB,CACnC,OAA8D;IAE9D,OAAO,iBAAiB,CAAC,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC,CAAC;AAC3E,CAAC"}

package/dist/server/commands.d.ts

@@ -0,0 +1,2 @@
+import type { Database } from "bun:sqlite";
+export declare function runFallbackBalancerCommand(db: Database, raw: string): string;

package/dist/server/commands.js

@@ -0,0 +1,46 @@
+import { getActiveAccount, listAccounts, setActiveAccount } from "../core/accounts";
+import { listPendingConnections } from "../core/pending";
+function parseArgs(input) {
+    return input.match(/(?:[^\s"]+|"[^"]*")+/g)?.map((x) => x.replace(/^"|"$/g, "")) ?? [];
+}
+export function runFallbackBalancerCommand(db, raw) {
+    const [action = "help", ...args] = parseArgs(raw);
+    if (["help", "--help", "-h"].includes(action)) {
+        return [
+            "Balancer is TUI-first. Use fallback commands only for compatibility or troubleshooting.",
+            "Fallback commands: list, status, use <provider> <alias>, active <provider>.",
+        ].join("\n");
+    }
+    if (action === "list") {
+        const accounts = listAccounts(db);
+        if (accounts.length === 0)
+            return "No accounts saved.";
+        return accounts.map((account) => `${account.providerID}/${account.alias}`).join("\n");
+    }
+    if (action === "status") {
+        return `accounts=${listAccounts(db).length} pending=${listPendingConnections(db).length}`;
+    }
+    if (action === "use") {
+        const [providerID, alias] = args;
+        if (!providerID || !alias)
+            return "Usage: /balancer use <provider> <alias>";
+        try {
+            const account = setActiveAccount(db, providerID, alias);
+            return `Active account changed to ${providerID}/${account.alias}`;
+        }
+        catch (error) {
+            return error instanceof Error ? error.message : "Account not found.";
+        }
+    }
+    if (action === "active") {
+        const [providerID] = args;
+        if (!providerID)
+            return "Usage: /balancer active <provider>";
+        const account = getActiveAccount(db, providerID);
+        if (!account)
+            return "No active account for provider.";
+        return `${providerID}/${account.alias}`;
+    }
+    return `Unknown command: ${action}. Use /balancer help.`;
+}
+//# sourceMappingURL=commands.js.map

package/dist/server/commands.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"commands.js","sourceRoot":"","sources":["../../src/server/commands.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpF,OAAO,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAEzD,SAAS,SAAS,CAAC,KAAa;IAC5B,OAAO,KAAK,CAAC,KAAK,CAAC,uBAAuB,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;AAC3F,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,EAAY,EAAE,GAAW;IAChE,MAAM,CAAC,MAAM,GAAG,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;IAElD,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5C,OAAO;YACH,yFAAyF;YACzF,6EAA6E;SAChF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjB,CAAC;IAED,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;QACpB,MAAM,QAAQ,GAAG,YAAY,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,oBAAoB,CAAC;QACvD,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1F,CAAC;IAED,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;QACtB,OAAO,YAAY,YAAY,CAAC,EAAE,CAAC,CAAC,MAAM,YAAY,sBAAsB,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC;IAC9F,CAAC;IAED,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;QACnB,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,GAAG,IAAI,CAAC;QACjC,IAAI,CAAC,UAAU,IAAI,CAAC,KAAK;YAAE,OAAO,yCAAyC,CAAC;QAC5E,IAAI,CAAC;YACD,MAAM,OAAO,GAAG,gBAAgB,CAAC,EAAE,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;YACxD,OAAO,6BAA6B,UAAU,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,oBAAoB,CAAC;QACzE,CAAC;IACL,CAAC;IAED,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;QACtB,MAAM,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC;QAC1B,IAAI,CAAC,UAAU;YAAE,OAAO,oCAAoC,CAAC;QAC7D,MAAM,OAAO,GAAG,gBAAgB,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QACjD,IAAI,CAAC,OAAO;YAAE,OAAO,iCAAiC,CAAC;QACvD,OAAO,GAAG,UAAU,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;IAC5C,CAAC;IAED,OAAO,oBAAoB,MAAM,uBAAuB,CAAC;AAC7D,CAAC"}

package/dist/server/fetch-patch.d.ts

@@ -0,0 +1,3 @@
+import type { Database } from "bun:sqlite";
+export declare function __testResetFetchPatch(): void;
+export declare function installFetchPatch(db: Database, client: any): void;

package/dist/server/fetch-patch.js

@@ -0,0 +1,118 @@
+import { setActiveAccount } from "../core/accounts";
+import { getBalancingEnabled } from "../core/priority";
+import { showToast, setNativeAuth } from "./native";
+import { chooseFailoverAccount, INTERNAL_REQUEST_HEADER, markRateLimited, RETRYABLE_STATUS, takePendingRequest, } from "./request-balancer";
+let fetchPatched = false;
+function headerEntries(headers) {
+    return Array.from(headers.entries()).map(([key, value]) => [key.toLowerCase(), value]);
+}
+function applyAuthToHeaders(headers, account, options = {}) {
+    const auth = account.auth;
+    const entries = headerEntries(headers);
+    if (auth.type === "oauth") {
+        if (options.preserveExistingOAuth && headers.has("authorization"))
+            return;
+        headers.set("authorization", `Bearer ${auth.access}`);
+        return;
+    }
+    if (auth.type === "wellknown") {
+        if (headers.has("authorization")) {
+            headers.set("authorization", `Bearer ${auth.token}`);
+        }
+        else {
+            headers.set(auth.key, auth.token);
+        }
+        return;
+    }
+    const authHeaderNames = [
+        "authorization",
+        "x-api-key",
+        "api-key",
+        "x-goog-api-key",
+        "x-stainless-api-key",
+        "anthropic-api-key",
+        "cohere-api-key",
+    ];
+    let changed = false;
+    for (const name of authHeaderNames) {
+        const current = entries.find(([key]) => key === name)?.[1];
+        if (!current)
+            continue;
+        headers.set(name, name === "authorization" && current.toLowerCase().startsWith("bearer ")
+            ? `Bearer ${auth.key}`
+            : auth.key);
+        changed = true;
+    }
+    if (!changed)
+        headers.set("authorization", `Bearer ${auth.key}`);
+}
+export function __testResetFetchPatch() {
+    fetchPatched = false;
+}
+function headersFrom(input, init) {
+    if (init?.headers)
+        return new Headers(init.headers);
+    if (typeof Request !== "undefined" && input instanceof Request) {
+        return new Headers(input.headers);
+    }
+    return new Headers();
+}
+function cloneRequestInput(input, init, headers) {
+    if (typeof Request !== "undefined" && input instanceof Request) {
+        return [new Request(input, { ...init, headers }), undefined];
+    }
+    return [input, { ...init, headers }];
+}
+function retryAfterMs(response) {
+    const retryAfter = response.headers.get("retry-after");
+    if (!retryAfter)
+        return 60_000;
+    const seconds = Number(retryAfter);
+    if (Number.isFinite(seconds))
+        return Math.max(1_000, seconds * 1000);
+    const date = Date.parse(retryAfter);
+    if (Number.isFinite(date))
+        return Math.max(1_000, date - Date.now());
+    return 60_000;
+}
+export function installFetchPatch(db, client) {
+    if (fetchPatched)
+        return;
+    fetchPatched = true;
+    const originalFetch = globalThis.fetch.bind(globalThis);
+    globalThis.fetch = (async (input, init) => {
+        const headers = headersFrom(input, init);
+        const requestID = headers.get(INTERNAL_REQUEST_HEADER);
+        if (!requestID)
+            return originalFetch(input, init);
+        const pending = takePendingRequest(requestID);
+        headers.delete(INTERNAL_REQUEST_HEADER);
+        if (!pending?.account) {
+            const [nextInput, nextInit] = cloneRequestInput(input, init, headers);
+            return originalFetch(nextInput, nextInit);
+        }
+        let account = pending.account;
+        const maxAttempts = 3;
+        for (let attempt = 0; attempt < maxAttempts; attempt++) {
+            const attemptHeaders = new Headers(headers);
+            applyAuthToHeaders(attemptHeaders, account, { preserveExistingOAuth: attempt === 0 });
+            const [nextInput, nextInit] = cloneRequestInput(input, init, attemptHeaders);
+            const response = await originalFetch(nextInput, nextInit);
+            if (!RETRYABLE_STATUS.has(response.status))
+                return response;
+            markRateLimited(db, account.providerID, account.alias, retryAfterMs(response));
+            if (!getBalancingEnabled(db))
+                return response;
+            if (attempt === maxAttempts - 1)
+                return response;
+            const next = chooseFailoverAccount(db, account.providerID, account.alias);
+            if (!next)
+                return response;
+            account = setActiveAccount(db, next.providerID, next.alias);
+            await setNativeAuth(client, account.providerID, account.auth, db);
+            await showToast(client, `Balancer: ${pending.providerID}/${pending.account.alias} is rate limited. Switching to ${account.alias}.`, "warning");
+        }
+        throw new Error("Balancer retry loop exited unexpectedly");
+    });
+}
+//# sourceMappingURL=fetch-patch.js.map

package/dist/server/fetch-patch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fetch-patch.js","sourceRoot":"","sources":["../../src/server/fetch-patch.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAEvD,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACpD,OAAO,EACH,qBAAqB,EACrB,uBAAuB,EACvB,eAAe,EACf,gBAAgB,EAChB,kBAAkB,GACrB,MAAM,oBAAoB,CAAC;AAE5B,IAAI,YAAY,GAAG,KAAK,CAAC;AAEzB,SAAS,aAAa,CAAC,OAAgB;IACnC,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CACpC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,KAAK,CAAU,CACxD,CAAC;AACN,CAAC;AAED,SAAS,kBAAkB,CAAC,OAAgB,EAAE,OAAgB,EAAE,UAA+C,EAAE;IAC7G,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAC1B,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IAEvC,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QACxB,IAAI,OAAO,CAAC,qBAAqB,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;YAAE,OAAO;QAC1E,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACtD,OAAO;IACX,CAAC;IAED,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QAC5B,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACtC,CAAC;QACD,OAAO;IACX,CAAC;IAED,MAAM,eAAe,GAAG;QACpB,eAAe;QACf,WAAW;QACX,SAAS;QACT,gBAAgB;QAChB,qBAAqB;QACrB,mBAAmB;QACnB,gBAAgB;KACnB,CAAC;IAEF,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,KAAK,MAAM,IAAI,IAAI,eAAe,EAAE,CAAC;QACjC,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC3D,IAAI,CAAC,OAAO;YAAE,SAAS;QACvB,OAAO,CAAC,GAAG,CACP,IAAI,EACJ,IAAI,KAAK,eAAe,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC;YACnE,CAAC,CAAC,UAAU,IAAI,CAAC,GAAG,EAAE;YACtB,CAAC,CAAC,IAAI,CAAC,GAAG,CACjB,CAAC;QACF,OAAO,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,IAAI,CAAC,OAAO;QAAE,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;AACrE,CAAC;AAED,MAAM,UAAU,qBAAqB;IACjC,YAAY,GAAG,KAAK,CAAC;AACzB,CAAC;AAED,SAAS,WAAW,CAAC,KAAwB,EAAE,IAAkB;IAC7D,IAAI,IAAI,EAAE,OAAO;QAAE,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpD,IAAI,OAAO,OAAO,KAAK,WAAW,IAAI,KAAK,YAAY,OAAO,EAAE,CAAC;QAC7D,OAAO,IAAI,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC;IACD,OAAO,IAAI,OAAO,EAAE,CAAC;AACzB,CAAC;AAED,SAAS,iBAAiB,CACtB,KAAwB,EACxB,IAA6B,EAC7B,OAAgB;IAEhB,IAAI,OAAO,OAAO,KAAK,WAAW,IAAI,KAAK,YAAY,OAAO,EAAE,CAAC;QAC7D,OAAO,CAAC,IAAI,OAAO,CAAC,KAAK,EAAE,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,CAAC,CAAC;IACjE,CAAC;IACD,OAAO,CAAC,KAAK,EAAE,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;AACzC,CAAC;AAED,SAAS,YAAY,CAAC,QAAkB;IACpC,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IACvD,IAAI,CAAC,UAAU;QAAE,OAAO,MAAM,CAAC;IAE/B,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;IACnC,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAAC,CAAC;IAErE,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACpC,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAErE,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,EAAY,EAAE,MAAW;IACvD,IAAI,YAAY;QAAE,OAAO;IACzB,YAAY,GAAG,IAAI,CAAC;IACpB,MAAM,aAAa,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAExD,UAAU,CAAC,KAAK,GAAG,CAAC,KAAK,EAAE,KAAwB,EAAE,IAAkB,EAAE,EAAE;QACvE,MAAM,OAAO,GAAG,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACzC,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;QACvD,IAAI,CAAC,SAAS;YAAE,OAAO,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAElD,MAAM,OAAO,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAC9C,OAAO,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;YACpB,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,GAAG,iBAAiB,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;YACtE,OAAO,aAAa,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC9B,MAAM,WAAW,GAAG,CAAC,CAAC;QACtB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;YACrD,MAAM,cAAc,GAAG,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;YAC5C,kBAAkB,CAAC,cAAc,EAAE,OAAO,EAAE,EAAE,qBAAqB,EAAE,OAAO,KAAK,CAAC,EAAE,CAAC,CAAC;YACtF,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,GAAG,iBAAiB,CAAC,KAAK,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC;YAC7E,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAC1D,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAAE,OAAO,QAAQ,CAAC;YAE5D,eAAe,CAAC,EAAE,EAAE,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,KAAK,EAAE,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC/E,IAAI,CAAC,mBAAmB,CAAC,EAAE,CAAC;gBAAE,OAAO,QAAQ,CAAC;YAC9C,IAAI,OAAO,KAAK,WAAW,GAAG,CAAC;gBAAE,OAAO,QAAQ,CAAC;YAEjD,MAAM,IAAI,GAAG,qBAAqB,CAAC,EAAE,EAAE,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YAC1E,IAAI,CAAC,IAAI;gBAAE,OAAO,QAAQ,CAAC;YAE3B,OAAO,GAAG,gBAAgB,CAAC,EAAE,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAClE,MAAM,SAAS,CACX,MAAM,EACN,aAAa,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,OAAO,CAAC,KAAK,kCAAkC,OAAO,CAAC,KAAK,GAAG,EAC1G,SAAS,CACZ,CAAC;QACN,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAC/D,CAAC,CAA4B,CAAC;AAClC,CAAC"}

package/dist/server/index.d.ts

@@ -0,0 +1,8 @@
+import { type Config, type Hooks } from "@opencode-ai/plugin";
+import type { Database } from "bun:sqlite";
+export declare function configureFallbackCommand(cfg: Config): void;
+export declare function createServerHooks({ db, client }: {
+    db: Database;
+    client: any;
+}): Hooks;
+export declare const serverPlugin: ({ client }: import("@opencode-ai/plugin").PluginInput) => Promise<Hooks>;

package/dist/server/index.js

@@ -0,0 +1,94 @@
+import { tool } from "@opencode-ai/plugin";
+import { getActiveAccount, getSelectedAccount, getSelectedModel, setActiveAccount } from "../core/accounts";
+import { openBalancerDatabase } from "../core/database";
+import { storePath } from "../core/path";
+import { getBalancingEnabled, resolveActiveSelection } from "../core/priority";
+import { migrate } from "../core/schema";
+import { runFallbackBalancerCommand } from "./commands";
+import { installFetchPatch } from "./fetch-patch";
+import { setNativeAuth, showToast } from "./native";
+import { BALANCER_METADATA_KEY, INTERNAL_REQUEST_HEADER, setPendingRequest, } from "./request-balancer";
+export function configureFallbackCommand(cfg) {
+    if (!cfg.command?.balancer)
+        return;
+}
+function runSafeFallbackBalancerCommand(db, raw) {
+    try {
+        return runFallbackBalancerCommand(db, raw);
+    }
+    catch (error) {
+        return error instanceof Error ? error.message : "Balancer command failed.";
+    }
+}
+export function createServerHooks({ db, client }) {
+    return {
+        config: async (cfg) => {
+            configureFallbackCommand(cfg);
+        },
+        "command.execute.before": async (input, output) => {
+            if (input.command !== "balancer")
+                return;
+            const result = runSafeFallbackBalancerCommand(db, input.arguments);
+            output.parts.length = 0;
+            await showToast(client, result.split("\n")[0] ?? 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.message": async (_input, output) => {
+            // Balancing on: the priority list decides the provider/model for every
+            // message (recomputed per message -> failover and recovery for free).
+            if (getBalancingEnabled(db)) {
+                const selection = resolveActiveSelection(db, undefined, output.message.model?.providerID);
+                if (!selection)
+                    return;
+                setActiveAccount(db, selection.providerID, selection.account.alias);
+                output.message.model = { providerID: selection.providerID, modelID: selection.modelID };
+                return;
+            }
+            // Balancing off: keep opencode's native choice; only fill when missing.
+            if (output.message.model?.providerID && output.message.model?.modelID)
+                return;
+            const selected = getSelectedAccount(db);
+            if (!selected)
+                return;
+            const model = getSelectedModel(db, selected.providerID);
+            if (!model)
+                return;
+            output.message.model = { providerID: model.providerID, modelID: model.modelID };
+        },
+        "chat.headers": async (input, output) => {
+            const providerID = input.model.providerID;
+            const account = getActiveAccount(db, providerID);
+            if (!account)
+                return;
+            if (account.auth.type === "oauth") {
+                await setNativeAuth(client, providerID, account.auth, db);
+            }
+            const requestID = crypto.randomUUID();
+            setPendingRequest(requestID, { providerID, account });
+            output.headers[INTERNAL_REQUEST_HEADER] = requestID;
+        },
+        tool: {
+            balancer_command: tool({
+                description: "Run fallback account balancer commands.",
+                args: {
+                    command: tool.schema.string().describe("Command arguments for /balancer."),
+                },
+                execute: async (args) => runSafeFallbackBalancerCommand(db, args.command),
+            }),
+        },
+    };
+}
+export const serverPlugin = (async ({ client }) => {
+    const db = openBalancerDatabase(storePath());
+    migrate(db);
+    installFetchPatch(db, client);
+    return createServerHooks({ db, client });
+});
+//# sourceMappingURL=index.js.map

package/dist/server/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/server/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAwC,MAAM,qBAAqB,CAAC;AAEjF,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAC5G,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAC/E,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,0BAA0B,EAAE,MAAM,YAAY,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AACpD,OAAO,EACH,qBAAqB,EACrB,uBAAuB,EACvB,iBAAiB,GACpB,MAAM,oBAAoB,CAAC;AAE5B,MAAM,UAAU,wBAAwB,CAAC,GAAW;IAChD,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,QAAQ;QAAE,OAAO;AACvC,CAAC;AAED,SAAS,8BAA8B,CAAC,EAAY,EAAE,GAAW;IAC7D,IAAI,CAAC;QACD,OAAO,0BAA0B,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;IAC/C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,0BAA0B,CAAC;IAC/E,CAAC;AACL,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,EAAE,EAAE,EAAE,MAAM,EAAiC;IAC3E,OAAO;QACH,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YAClB,wBAAwB,CAAC,GAAG,CAAC,CAAC;QAClC,CAAC;QAED,wBAAwB,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YAC9C,IAAI,KAAK,CAAC,OAAO,KAAK,UAAU;gBAAE,OAAO;YACzC,MAAM,MAAM,GAAG,8BAA8B,CAAC,EAAE,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;YACnE,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;YACxB,MAAM,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,MAAM,EAAE,MAAM,CAAC,CAAC;YACjE,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,MAAM,EAAE,MAAM,EAAE,EAAE;YACrC,uEAAuE;YACvE,sEAAsE;YACtE,IAAI,mBAAmB,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC1B,MAAM,SAAS,GAAG,sBAAsB,CAAC,EAAE,EAAE,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;gBAC1F,IAAI,CAAC,SAAS;oBAAE,OAAO;gBACvB,gBAAgB,CAAC,EAAE,EAAE,SAAS,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBACpE,MAAM,CAAC,OAAO,CAAC,KAAK,GAAG,EAAE,UAAU,EAAE,SAAS,CAAC,UAAU,EAAE,OAAO,EAAE,SAAS,CAAC,OAAO,EAAE,CAAC;gBACxF,OAAO;YACX,CAAC;YAED,wEAAwE;YACxE,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,UAAU,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO;gBAAE,OAAO;YAE9E,MAAM,QAAQ,GAAG,kBAAkB,CAAC,EAAE,CAAC,CAAC;YACxC,IAAI,CAAC,QAAQ;gBAAE,OAAO;YACtB,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;YACxD,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,CAAC,OAAO,CAAC,KAAK,GAAG,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;QACpF,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,gBAAgB,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YACjD,IAAI,CAAC,OAAO;gBAAE,OAAO;YAErB,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAChC,MAAM,aAAa,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC9D,CAAC;YAED,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;YACtC,iBAAiB,CAAC,SAAS,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC;YACtD,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,GAAG,SAAS,CAAC;QACxD,CAAC;QAED,IAAI,EAAE;YACF,gBAAgB,EAAE,IAAI,CAAC;gBACnB,WAAW,EAAE,yCAAyC;gBACtD,IAAI,EAAE;oBACF,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kCAAkC,CAAC;iBAC7E;gBACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,8BAA8B,CAAC,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC;aAC5E,CAAC;SACL;KACJ,CAAC;AACN,CAAC;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;IAC9C,MAAM,EAAE,GAAG,oBAAoB,CAAC,SAAS,EAAE,CAAC,CAAC;IAC7C,OAAO,CAAC,EAAE,CAAC,CAAC;IACZ,iBAAiB,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IAE9B,OAAO,iBAAiB,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;AAC7C,CAAC,CAAkB,CAAC"}

package/dist/server/native.d.ts

@@ -0,0 +1,6 @@
+import type { Database } from "bun:sqlite";
+import type { AuthInfo } from "../core/types";
+export declare function suppressNativeAuthCapture(providerID: string, durationMs?: number): void;
+export declare function isNativeAuthCaptureSuppressed(providerID: string): boolean;
+export declare function setNativeAuth(client: any, providerID: string, auth: AuthInfo, db?: Database): Promise<void>;
+export declare function showToast(client: any, message: string, variant?: "info" | "success" | "warning" | "error"): Promise<void>;

package/dist/server/native.js

@@ -0,0 +1,35 @@
+import { now } from "../core/time";
+import { isNativeConnectInProgress } from "../core/native-connect";
+import { suppressNativeAuthCapture as suppressDbNativeAuthCapture } from "../core/native-auth-suppression";
+const suppressAuthCaptureUntil = new Map();
+export function suppressNativeAuthCapture(providerID, durationMs = 2_000) {
+    suppressAuthCaptureUntil.set(providerID, now() + durationMs);
+}
+export function isNativeAuthCaptureSuppressed(providerID) {
+    const suppressedUntil = suppressAuthCaptureUntil.get(providerID) ?? 0;
+    if (suppressedUntil <= now()) {
+        suppressAuthCaptureUntil.delete(providerID);
+        return false;
+    }
+    return true;
+}
+export async function setNativeAuth(client, providerID, auth, db) {
+    if (db && isNativeConnectInProgress(db))
+        return;
+    suppressNativeAuthCapture(providerID);
+    if (db)
+        suppressDbNativeAuthCapture(db, providerID);
+    try {
+        await client.auth.set({ path: { id: providerID }, body: auth });
+    }
+    catch { }
+}
+export async function showToast(client, message, variant = "info") {
+    try {
+        await client.tui.showToast({
+            body: { message, variant, duration: 15_000 },
+        });
+    }
+    catch { }
+}
+//# sourceMappingURL=native.js.map

package/dist/server/native.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"native.js","sourceRoot":"","sources":["../../src/server/native.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AACnC,OAAO,EAAE,yBAAyB,EAAE,MAAM,wBAAwB,CAAC;AACnE,OAAO,EAAE,yBAAyB,IAAI,2BAA2B,EAAE,MAAM,iCAAiC,CAAC;AAE3G,MAAM,wBAAwB,GAAG,IAAI,GAAG,EAAkB,CAAC;AAE3D,MAAM,UAAU,yBAAyB,CAAC,UAAkB,EAAE,UAAU,GAAG,KAAK;IAC5E,wBAAwB,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,EAAE,GAAG,UAAU,CAAC,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,UAAkB;IAC5D,MAAM,eAAe,GAAG,wBAAwB,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IACtE,IAAI,eAAe,IAAI,GAAG,EAAE,EAAE,CAAC;QAC3B,wBAAwB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAC5C,OAAO,KAAK,CAAC;IACjB,CAAC;IACD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CAC/B,MAAW,EACX,UAAkB,EAClB,IAAc,EACd,EAAa;IAEb,IAAI,EAAE,IAAI,yBAAyB,CAAC,EAAE,CAAC;QAAE,OAAO;IAChD,yBAAyB,CAAC,UAAU,CAAC,CAAC;IACtC,IAAI,EAAE;QAAE,2BAA2B,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IACpD,IAAI,CAAC;QACD,MAAM,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACpE,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;AACd,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAC3B,MAAW,EACX,OAAe,EACf,UAAoD,MAAM;IAE1D,IAAI,CAAC;QACD,MAAM,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC;YACvB,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE;SAC/C,CAAC,CAAC;IACP,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;AACd,CAAC"}

package/dist/server/request-balancer.d.ts

@@ -0,0 +1,16 @@
+import type { Database } from "bun:sqlite";
+import type { Account } from "../core/types";
+export declare const INTERNAL_REQUEST_HEADER = "x-opencode-balancer-request";
+export declare const BALANCER_METADATA_KEY = "opencodeBalancerCommand";
+export declare const RETRYABLE_STATUS: Set<number>;
+type PendingRequest = {
+    providerID: string;
+    account?: Account;
+};
+export declare function setPendingRequest(requestID: string, request: PendingRequest): void;
+export declare function takePendingRequest(requestID: string): PendingRequest | undefined;
+export declare function __testGetPendingRequest(requestID: string): PendingRequest | undefined;
+export declare function __testClearPendingRequests(): void;
+export declare function markRateLimited(db: Database, providerID: string, alias: string, retryAfterMs?: number): void;
+export declare function chooseFailoverAccount(db: Database, providerID: string, currentAlias: string): Account | undefined;
+export {};