@thelioo/opencode-balancer 0.1.0

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

@@ -0,0 +1 @@
+{"version":3,"file":"fetch-patch.js","sourceRoot":"","sources":["../../src/balancer/fetch-patch.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,oBAAoB,EACpB,kBAAkB,EAClB,WAAW,EACX,iBAAiB,EACjB,mBAAmB,EACnB,WAAW,EACX,uBAAuB,EACvB,UAAU,EACV,aAAa,EACb,eAAe,EACf,QAAQ,EACR,sBAAsB,EACtB,oBAAoB,EACpB,cAAc,EACd,GAAG,EACH,oBAAoB,EACpB,aAAa,EACb,sBAAsB,EACtB,qBAAqB,EACrB,mBAAmB,EACnB,QAAQ,EACR,cAAc,EACd,SAAS,EACT,gBAAgB,EAChB,YAAY,EACZ,WAAW,EACX,aAAa,EACb,SAAS,EACT,KAAK,EACL,UAAU,GACb,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAGhD,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,MAAW;IAC/C,IAAI,KAAK,CAAC,YAAY;QAAE,OAAO;IAC/B,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;IAC1B,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,GAAG,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;QAE5B,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAClD,IACI,gBAAgB;YAChB,CAAC,IAAI,EAAE,MAAM,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,KAAK,MAAM,EAClD,CAAC;YACC,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,IAAI,EAAE,OAAO,KAAK,UAAU,EAAE,CAAC;gBAC/B,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACnC,MAAM,EACN,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAC3D,CAAC;gBACF,MAAM,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,MAAM,EAAE,MAAM,CAAC,CAAC;gBACjE,OAAO,IAAI,QAAQ,CACf,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC,EAC7D;oBACI,MAAM,EAAE,GAAG;oBACX,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;iBAClD,CACJ,CAAC;YACN,CAAC;QACL,CAAC;QAED,IAAI,aAAa,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,QAAQ,KAAK,gBAAgB,EAAE,CAAC;YAC3D,OAAO,sBAAsB,CAAC,MAAM,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,iBAAiB,GAAG,sBAAsB,CAAC,GAAG,CAAC,CAAC;QACtD,IACI,iBAAiB;YACjB,CAAC,IAAI,EAAE,MAAM,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,KAAK,MAAM,EAClD,CAAC;YACC,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,KAAK,GACP,OAAO,IAAI,EAAE,MAAM,EAAE,CAAC,oBAAoB,CAAC,KAAK,QAAQ;gBACpD,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;gBACnD,CAAC,CAAC,EAAE,CAAC;YACb,IAAI,KAAK;gBAAE,KAAK,CAAC,iBAAiB,CAAC,GAAG,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAC;YACjE,IAAI,IAAI,EAAE,MAAM,IAAI,oBAAoB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBACtD,OAAO,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;gBACzC,IAAI,GAAG,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;YACnD,CAAC;YACD,OAAO,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACtC,CAAC;QAED,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,GAAG,CAAC,CAAC;QACpD,IACI,gBAAgB;YAChB,CAAC,IAAI,EAAE,MAAM,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,KAAK,MAAM,EAClD,CAAC;YACC,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,KAAK,CAAC,iBAAiB,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YAC5D,IAAI,QAAQ,CAAC,EAAE,IAAI,KAAK,EAAE,CAAC;gBACvB,UAAU,CAAC,KAAK,IAAI,EAAE;oBAClB,MAAM,IAAI,GAAG,CAAC,MAAM,cAAc,EAAE,CAAC,CAAC,gBAAgB,CAAC,CAAC;oBACxD,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;wBACnB,MAAM,WAAW,CAAC,gBAAgB,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;wBACjD,MAAM,SAAS,CACX,MAAM,EACN,qBAAqB,gBAAgB,IAAI,KAAK,SAAS,EACvD,SAAS,CACZ,CAAC;oBACN,CAAC;oBACD,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;gBACrD,CAAC,EAAE,GAAG,CAAC,CAAC;YACZ,CAAC;YACD,OAAO,QAAQ,CAAC;QACpB,CAAC;QAED,MAAM,eAAe,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAClD,IACI,eAAe;YACf,CAAC,IAAI,EAAE,MAAM,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,KAAK,KAAK,EACjD,CAAC;YACC,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAClD,IAAI,QAAQ,CAAC,EAAE,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;gBAClC,MAAM,eAAe,GACjB,KAAK,CAAC,wBAAwB,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;gBAC7D,IAAI,eAAe,IAAI,GAAG,EAAE,EAAE,CAAC;oBAC3B,MAAM,WAAW,CAAC,eAAe,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;oBACjD,MAAM,SAAS,CACX,MAAM,EACN,aAAa,eAAe,mDAAmD,EAC/E,MAAM,CACT,CAAC;gBACN,CAAC;YACL,CAAC;YACD,OAAO,QAAQ,CAAC;QACpB,CAAC;QAED,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,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACrD,OAAO,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC;QAExC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;YACpB,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,GAAG,iBAAiB,CAC3C,KAAK,EACL,IAAI,EACJ,OAAO,CACV,CAAC;YACF,OAAO,aAAa,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,OAAO,GAAY,OAAO,CAAC,OAAO,CAAC;QACvC,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,OAAO,IAAI,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;YACV,MAAM,cAAc,GAAG,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;YAC5C,kBAAkB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;YAC5C,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,GAAG,iBAAiB,CAC3C,KAAK,EACL,IAAI,EACJ,cAAc,CACjB,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAC1D,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzC,MAAM,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;gBAClD,OAAO,QAAQ,CAAC;YACpB,CAAC;YAED,MAAM,eAAe,CACjB,OAAO,CAAC,UAAU,EAClB,OAAO,CAAC,KAAK,EACb,YAAY,CAAC,QAAQ,CAAC,CACzB,CAAC;YACF,MAAM,IAAI,GAAG,MAAM,oBAAoB,CACnC,OAAO,CAAC,UAAU,EAClB,OAAO,CAAC,KAAK,CAChB,CAAC;YACF,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,CAAC;gBAAE,OAAO,QAAQ,CAAC;YAE3C,OAAO,GAAG,IAAI,CAAC;YACf,MAAM,KAAK,GAAG,MAAM,SAAS,EAAE,CAAC;YAChC,MAAM,QAAQ,GAAG,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YACrD,IAAI,QAAQ,EAAE,CAAC;gBACX,QAAQ,CAAC,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;gBAChC,MAAM,UAAU,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;YACD,MAAM,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAC9D,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;IACL,CAAC,CAA4B,CAAC;AAClC,CAAC"}

package/dist/balancer/http.d.ts

@@ -0,0 +1,40 @@
+import type { Account } from "./types";
+export declare function applyAuthToHeaders(headers: Headers, account: Account): void;
+export declare function cloneRequestInput(input: RequestInfo | URL, init: RequestInit | undefined, headers: Headers): [RequestInfo | URL, RequestInit | undefined];
+export declare function parseUrl(input: RequestInfo | URL): URL | undefined;
+export declare function headersFrom(input: RequestInfo | URL, init?: RequestInit): Headers;
+export declare function parseJsonBody(init?: RequestInit): any;
+export declare function isInternalUrl(url: URL | undefined): boolean;
+export declare function parseAuthSetProvider(url: URL | undefined): string | undefined;
+export declare function parseProviderAuthorize(url: URL | undefined): string | undefined;
+export declare function parseProviderCallback(url: URL | undefined): string | undefined;
+export declare function parseSessionCommand(url: URL | undefined): string | undefined;
+export declare function fakeCommandResponse(sessionID: string, text: string): {
+    info: {
+        id: string;
+        sessionID: string;
+        role: string;
+        time: {
+            created: number;
+        };
+        agent: string;
+        model: {
+            providerID: string;
+            modelID: string;
+        };
+    };
+    parts: {
+        id: string;
+        sessionID: string;
+        messageID: string;
+        type: string;
+        text: string;
+        synthetic: boolean;
+        ignored: boolean;
+        metadata: {
+            opencodeBalancerCommand: boolean;
+        };
+    }[];
+};
+export declare function maybeInjectAliasPrompt(response: Response): Promise<Response>;
+export declare function retryAfterMs(response: Response): number;

package/dist/balancer/http.js

@@ -0,0 +1,199 @@
+import { BALANCER_ALIAS_INPUT, BALANCER_METADATA_KEY, now, } from "./state";
+function getHeaderEntries(headers) {
+    return Array.from(headers.entries()).map(([key, value]) => [key.toLowerCase(), value]);
+}
+export function applyAuthToHeaders(headers, account) {
+    const auth = account.auth;
+    const entries = getHeaderEntries(headers);
+    if (auth.type === "oauth") {
+        if (auth.access)
+            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;
+        if (name === "authorization") {
+            headers.set(name, current.toLowerCase().startsWith("bearer ")
+                ? `Bearer ${auth.key}`
+                : auth.key);
+        }
+        else {
+            headers.set(name, auth.key);
+        }
+        changed = true;
+    }
+    if (!changed)
+        headers.set("authorization", `Bearer ${auth.key}`);
+}
+export function cloneRequestInput(input, init, headers) {
+    if (typeof Request !== "undefined" && input instanceof Request) {
+        return [new Request(input, { ...init, headers }), undefined];
+    }
+    return [input, { ...init, headers }];
+}
+export function parseUrl(input) {
+    try {
+        if (typeof input === "string")
+            return new URL(input, "http://localhost");
+        if (input instanceof URL)
+            return input;
+        if (typeof Request !== "undefined" && input instanceof Request)
+            return new URL(input.url);
+    }
+    catch { }
+    return undefined;
+}
+export 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 requestBodyText(init) {
+    const body = init?.body;
+    if (typeof body === "string")
+        return body;
+    if (body instanceof URLSearchParams)
+        return body.toString();
+    return undefined;
+}
+export function parseJsonBody(init) {
+    const text = requestBodyText(init);
+    if (!text)
+        return undefined;
+    try {
+        return JSON.parse(text);
+    }
+    catch {
+        return undefined;
+    }
+}
+export function isInternalUrl(url) {
+    if (!url)
+        return false;
+    return (url.hostname === "localhost" ||
+        url.hostname === "127.0.0.1" ||
+        url.hostname === "0.0.0.0");
+}
+export function parseAuthSetProvider(url) {
+    if (!url || !isInternalUrl(url))
+        return undefined;
+    const match = url.pathname.match(/^\/auth\/([^/]+)$/);
+    return match ? decodeURIComponent(match[1]) : undefined;
+}
+export function parseProviderAuthorize(url) {
+    if (!url || !isInternalUrl(url))
+        return undefined;
+    const match = url.pathname.match(/^\/provider\/([^/]+)\/oauth\/authorize$/);
+    return match ? decodeURIComponent(match[1]) : undefined;
+}
+export function parseProviderCallback(url) {
+    if (!url || !isInternalUrl(url))
+        return undefined;
+    const match = url.pathname.match(/^\/provider\/([^/]+)\/oauth\/callback$/);
+    return match ? decodeURIComponent(match[1]) : undefined;
+}
+export function parseSessionCommand(url) {
+    if (!url || !isInternalUrl(url))
+        return undefined;
+    const match = url.pathname.match(/^\/session\/([^/]+)\/command$/);
+    return match ? decodeURIComponent(match[1]) : undefined;
+}
+export function fakeCommandResponse(sessionID, text) {
+    const created = now();
+    const suffix = `${created.toString(36)}${Math.random().toString(36).slice(2, 8)}`;
+    const messageID = `msg_${suffix}`;
+    return {
+        info: {
+            id: messageID,
+            sessionID,
+            role: "user",
+            time: { created },
+            agent: "build",
+            model: { providerID: "opencode", modelID: "balancer-command" },
+        },
+        parts: [
+            {
+                id: `prt_${suffix}`,
+                sessionID,
+                messageID,
+                type: "text",
+                text,
+                synthetic: true,
+                ignored: true,
+                metadata: { [BALANCER_METADATA_KEY]: true },
+            },
+        ],
+    };
+}
+export async function maybeInjectAliasPrompt(response) {
+    const contentType = response.headers.get("content-type") ?? "";
+    if (!response.ok || !contentType.includes("application/json"))
+        return response;
+    try {
+        const data = (await response.clone().json());
+        for (const methods of Object.values(data)) {
+            if (!Array.isArray(methods))
+                continue;
+            for (const method of methods) {
+                const prompts = Array.isArray(method.prompts)
+                    ? method.prompts
+                    : [];
+                if (prompts.some((prompt) => prompt?.key === BALANCER_ALIAS_INPUT))
+                    continue;
+                method.prompts = [
+                    ...prompts,
+                    {
+                        type: "text",
+                        key: BALANCER_ALIAS_INPUT,
+                        message: "Alias to save in the balancer (optional)",
+                        placeholder: "work, personal, backup",
+                    },
+                ];
+            }
+        }
+        const headers = new Headers(response.headers);
+        headers.set("content-type", "application/json");
+        return new Response(JSON.stringify(data), {
+            status: response.status,
+            statusText: response.statusText,
+            headers,
+        });
+    }
+    catch {
+        return response;
+    }
+}
+export 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 - now());
+    return 60_000;
+}
+//# sourceMappingURL=http.js.map

package/dist/balancer/http.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http.js","sourceRoot":"","sources":["../../src/balancer/http.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,oBAAoB,EACpB,qBAAqB,EACrB,GAAG,GACN,MAAM,SAAS,CAAC;AAGjB,SAAS,gBAAgB,CAAC,OAAgB;IACtC,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,MAAM,UAAU,kBAAkB,CAAC,OAAgB,EAAE,OAAgB;IACjE,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAC1B,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAE1C,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QACxB,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACvE,OAAO;IACX,CAAC;IAED,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QAC5B,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;;YACpD,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACvC,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,IAAI,IAAI,KAAK,eAAe,EAAE,CAAC;YAC3B,OAAO,CAAC,GAAG,CACP,IAAI,EACJ,OAAO,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC;gBACvC,CAAC,CAAC,UAAU,IAAI,CAAC,GAAG,EAAE;gBACtB,CAAC,CAAC,IAAI,CAAC,GAAG,CACjB,CAAC;QACN,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;QACD,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,iBAAiB,CAC7B,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,MAAM,UAAU,QAAQ,CAAC,KAAwB;IAC7C,IAAI,CAAC;QACD,IAAI,OAAO,KAAK,KAAK,QAAQ;YACzB,OAAO,IAAI,GAAG,CAAC,KAAK,EAAE,kBAAkB,CAAC,CAAC;QAC9C,IAAI,KAAK,YAAY,GAAG;YAAE,OAAO,KAAK,CAAC;QACvC,IAAI,OAAO,OAAO,KAAK,WAAW,IAAI,KAAK,YAAY,OAAO;YAC1D,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAClC,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;IACV,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAwB,EAAE,IAAkB;IACpE,IAAI,IAAI,EAAE,OAAO;QAAE,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpD,IAAI,OAAO,OAAO,KAAK,WAAW,IAAI,KAAK,YAAY,OAAO;QAC1D,OAAO,IAAI,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACtC,OAAO,IAAI,OAAO,EAAE,CAAC;AACzB,CAAC;AAED,SAAS,eAAe,CAAC,IAAkB;IACvC,MAAM,IAAI,GAAG,IAAI,EAAE,IAAI,CAAC;IACxB,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC1C,IAAI,IAAI,YAAY,eAAe;QAAE,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;IAC5D,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,IAAkB;IAC5C,MAAM,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;IACnC,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,IAAI,CAAC;QACD,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,SAAS,CAAC;IACrB,CAAC;AACL,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,GAAoB;IAC9C,IAAI,CAAC,GAAG;QAAE,OAAO,KAAK,CAAC;IACvB,OAAO,CACH,GAAG,CAAC,QAAQ,KAAK,WAAW;QAC5B,GAAG,CAAC,QAAQ,KAAK,WAAW;QAC5B,GAAG,CAAC,QAAQ,KAAK,SAAS,CAC7B,CAAC;AACN,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,GAAoB;IACrD,IAAI,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IAClD,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;IACtD,OAAO,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAC5D,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,GAAoB;IACvD,IAAI,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IAClD,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAC5E,OAAO,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAC5D,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,GAAoB;IACtD,IAAI,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IAClD,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC3E,OAAO,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAC5D,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,GAAoB;IACpD,IAAI,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IAClD,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;IAClE,OAAO,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAC5D,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,SAAiB,EAAE,IAAY;IAC/D,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC;IACtB,MAAM,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IAClF,MAAM,SAAS,GAAG,OAAO,MAAM,EAAE,CAAC;IAClC,OAAO;QACH,IAAI,EAAE;YACF,EAAE,EAAE,SAAS;YACb,SAAS;YACT,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,EAAE,OAAO,EAAE;YACjB,KAAK,EAAE,OAAO;YACd,KAAK,EAAE,EAAE,UAAU,EAAE,UAAU,EAAE,OAAO,EAAE,kBAAkB,EAAE;SACjE;QACD,KAAK,EAAE;YACH;gBACI,EAAE,EAAE,OAAO,MAAM,EAAE;gBACnB,SAAS;gBACT,SAAS;gBACT,IAAI,EAAE,MAAM;gBACZ,IAAI;gBACJ,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE,IAAI;gBACb,QAAQ,EAAE,EAAE,CAAC,qBAAqB,CAAC,EAAE,IAAI,EAAE;aAC9C;SACJ;KACJ,CAAC;AACN,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,QAAkB;IAC3D,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAC/D,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACzD,OAAO,QAAQ,CAAC;IACpB,IAAI,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,CAG1C,CAAC;QACF,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACxC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;gBAAE,SAAS;YACtC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC3B,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;oBACzC,CAAC,CAAC,MAAM,CAAC,OAAO;oBAChB,CAAC,CAAC,EAAE,CAAC;gBACT,IACI,OAAO,CAAC,IAAI,CACR,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,oBAAoB,CACnD;oBAED,SAAS;gBACb,MAAM,CAAC,OAAO,GAAG;oBACb,GAAG,OAAO;oBACV;wBACI,IAAI,EAAE,MAAM;wBACZ,GAAG,EAAE,oBAAoB;wBACzB,OAAO,EAAE,0CAA0C;wBACnD,WAAW,EAAE,wBAAwB;qBACxC;iBACJ,CAAC;YACN,CAAC;QACL,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC9C,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,kBAAkB,CAAC,CAAC;QAChD,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;YACtC,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,UAAU,EAAE,QAAQ,CAAC,UAAU;YAC/B,OAAO;SACV,CAAC,CAAC;IACP,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,QAAQ,CAAC;IACpB,CAAC;AACL,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,QAAkB;IAC3C,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IACvD,IAAI,CAAC,UAAU;QAAE,OAAO,MAAM,CAAC;IAC/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;IACrE,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,GAAG,EAAE,CAAC,CAAC;IAChE,OAAO,MAAM,CAAC;AAClB,CAAC"}

package/dist/balancer/native.d.ts

@@ -0,0 +1,3 @@
+import type { AuthInfo } from "./types";
+export declare function setNativeAuth(client: any, providerID: string, auth: AuthInfo): Promise<void>;
+export declare function showToast(client: any, message: string, variant?: "info" | "success" | "warning" | "error"): Promise<void>;

package/dist/balancer/native.js

@@ -0,0 +1,16 @@
+import { now, state } from "./state";
+export async function setNativeAuth(client, providerID, auth) {
+    state.suppressAuthCaptureUntil.set(providerID, now() + 2_000);
+    await client.auth
+        .set({
+        path: { id: providerID },
+        body: auth,
+    })
+        .catch(() => undefined);
+}
+export async function showToast(client, message, variant = "info") {
+    await client.tui
+        .showToast({ body: { message, variant, duration: 15_000 } })
+        .catch(() => undefined);
+}
+//# sourceMappingURL=native.js.map

package/dist/balancer/native.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"native.js","sourceRoot":"","sources":["../../src/balancer/native.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAGrC,MAAM,CAAC,KAAK,UAAU,aAAa,CAC/B,MAAW,EACX,UAAkB,EAClB,IAAc;IAEd,KAAK,CAAC,wBAAwB,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,EAAE,GAAG,KAAK,CAAC,CAAC;IAC9D,MAAM,MAAM,CAAC,IAAI;SACZ,GAAG,CAAC;QACD,IAAI,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE;QACxB,IAAI,EAAE,IAAI;KACb,CAAC;SACD,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;AAChC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAC3B,MAAW,EACX,OAAe,EACf,UAAoD,MAAM;IAE1D,MAAM,MAAM,CAAC,GAAG;SACX,SAAS,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,CAAC;SAC3D,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;AAChC,CAAC"}

package/dist/balancer/state.d.ts

@@ -0,0 +1,14 @@
+import type { AuthInfo, PendingRequest } from "./types";
+export declare const INTERNAL_REQUEST_HEADER = "x-opencode-balancer-request";
+export declare const BALANCER_ALIAS_INPUT = "__opencode_balancer_alias";
+export declare const BALANCER_METADATA_KEY = "opencodeBalancerCommand";
+export declare const RETRYABLE_STATUS: Set<number>;
+export declare const state: {
+    fetchPatched: boolean;
+    pendingRequests: Map<string, PendingRequest>;
+    pendingOauthAlias: Map<string, string>;
+    suppressAuthCaptureUntil: Map<string, number>;
+    lastAuthSnapshot: Record<string, AuthInfo>;
+    watchStarted: boolean;
+};
+export declare function now(): number;

package/dist/balancer/state.js

@@ -0,0 +1,16 @@
+export const INTERNAL_REQUEST_HEADER = "x-opencode-balancer-request";
+export const BALANCER_ALIAS_INPUT = "__opencode_balancer_alias";
+export const BALANCER_METADATA_KEY = "opencodeBalancerCommand";
+export const RETRYABLE_STATUS = new Set([429, 500, 502, 503, 504, 529]);
+export const state = {
+    fetchPatched: false,
+    pendingRequests: new Map(),
+    pendingOauthAlias: new Map(),
+    suppressAuthCaptureUntil: new Map(),
+    lastAuthSnapshot: {},
+    watchStarted: false,
+};
+export function now() {
+    return Date.now();
+}
+//# sourceMappingURL=state.js.map

package/dist/balancer/state.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"state.js","sourceRoot":"","sources":["../../src/balancer/state.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,uBAAuB,GAAG,6BAA6B,CAAC;AACrE,MAAM,CAAC,MAAM,oBAAoB,GAAG,2BAA2B,CAAC;AAChE,MAAM,CAAC,MAAM,qBAAqB,GAAG,yBAAyB,CAAC;AAC/D,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;AAExE,MAAM,CAAC,MAAM,KAAK,GAAG;IACjB,YAAY,EAAE,KAAK;IACnB,eAAe,EAAE,IAAI,GAAG,EAA0B;IAClD,iBAAiB,EAAE,IAAI,GAAG,EAAkB;IAC5C,wBAAwB,EAAE,IAAI,GAAG,EAAkB;IACnD,gBAAgB,EAAE,EAA8B;IAChD,YAAY,EAAE,KAAK;CACtB,CAAC;AAEF,MAAM,UAAU,GAAG;IACf,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC;AACtB,CAAC"}

package/dist/balancer/storage.d.ts

@@ -0,0 +1,8 @@
+import type { AuthInfo, Store } from "./types";
+export declare function storePath(): string;
+export declare function authPath(): string;
+export declare function readStore(): Promise<Store>;
+export declare function writeStore(store: Store): Promise<void>;
+export declare function readNativeAuth(): Promise<Record<string, AuthInfo>>;
+export declare function isAuthInfo(value: unknown): value is AuthInfo;
+export declare function captureAuth(providerID: string, auth: AuthInfo, source: NonNullable<Store["lastCaptured"]>["source"]): Promise<Store>;

package/dist/balancer/storage.js

@@ -0,0 +1,92 @@
+import { now } from "./state";
+function joinPath(...parts) {
+    return parts
+        .filter(Boolean)
+        .join("/")
+        .replace(/\/+/g, "/");
+}
+function homeDir() {
+    return Bun.env.HOME || "/";
+}
+function xdgConfigHome() {
+    return Bun.env.XDG_CONFIG_HOME || joinPath(homeDir(), ".config");
+}
+function xdgDataHome() {
+    return Bun.env.XDG_DATA_HOME || joinPath(homeDir(), ".local", "share");
+}
+function configDir() {
+    return Bun.env.OPENCODE_CONFIG_DIR || joinPath(xdgConfigHome(), "opencode");
+}
+function dataDir() {
+    return joinPath(xdgDataHome(), "opencode");
+}
+export function storePath() {
+    return joinPath(configDir(), "balancer-accounts.json");
+}
+export function authPath() {
+    return joinPath(dataDir(), "auth.json");
+}
+function emptyStore() {
+    return { version: 1, providers: {} };
+}
+async function readJsonFile(path, fallback) {
+    try {
+        const file = Bun.file(path);
+        if (!(await file.exists()))
+            return fallback;
+        return JSON.parse(await file.text());
+    }
+    catch {
+        return fallback;
+    }
+}
+async function writeJsonFile(path, value) {
+    await Bun.write(path, `${JSON.stringify(value, null, 2)}\n`, {
+        mode: 0o600,
+    });
+}
+export async function readStore() {
+    const parsed = await readJsonFile(storePath(), emptyStore());
+    if (parsed.version !== 1 || !parsed.providers)
+        return emptyStore();
+    return parsed;
+}
+export async function writeStore(store) {
+    await writeJsonFile(storePath(), store);
+}
+export async function readNativeAuth() {
+    if (Bun.env.OPENCODE_AUTH_CONTENT) {
+        try {
+            return JSON.parse(Bun.env.OPENCODE_AUTH_CONTENT);
+        }
+        catch {
+            return {};
+        }
+    }
+    return readJsonFile(authPath(), {});
+}
+export function isAuthInfo(value) {
+    if (!value || typeof value !== "object")
+        return false;
+    const type = value.type;
+    if (type === "api")
+        return typeof value.key === "string";
+    if (type === "oauth") {
+        const item = value;
+        return (typeof item.refresh === "string" &&
+            typeof item.access === "string" &&
+            typeof item.expires === "number");
+    }
+    if (type === "wellknown") {
+        const item = value;
+        return typeof item.key === "string" && typeof item.token === "string";
+    }
+    return false;
+}
+export async function captureAuth(providerID, auth, source) {
+    const store = await readStore();
+    store.lastCaptured = { providerID, auth, capturedAt: now(), source };
+    await writeStore(store);
+    return store;
+}
+//# sourceMappingURL=storage.js.map

package/dist/balancer/storage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage.js","sourceRoot":"","sources":["../../src/balancer/storage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,MAAM,SAAS,CAAC;AAG9B,SAAS,QAAQ,CAAC,GAAG,KAAe;IAChC,OAAO,KAAK;SACP,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,GAAG,CAAC;SACT,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,OAAO;IACZ,OAAO,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC;AAC/B,CAAC;AAED,SAAS,aAAa;IAClB,OAAO,GAAG,CAAC,GAAG,CAAC,eAAe,IAAI,QAAQ,CAAC,OAAO,EAAE,EAAE,SAAS,CAAC,CAAC;AACrE,CAAC;AAED,SAAS,WAAW;IAChB,OAAO,GAAG,CAAC,GAAG,CAAC,aAAa,IAAI,QAAQ,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;AAC3E,CAAC;AAED,SAAS,SAAS;IACd,OAAO,GAAG,CAAC,GAAG,CAAC,mBAAmB,IAAI,QAAQ,CAAC,aAAa,EAAE,EAAE,UAAU,CAAC,CAAC;AAChF,CAAC;AAED,SAAS,OAAO;IACZ,OAAO,QAAQ,CAAC,WAAW,EAAE,EAAE,UAAU,CAAC,CAAC;AAC/C,CAAC;AAED,MAAM,UAAU,SAAS;IACrB,OAAO,QAAQ,CAAC,SAAS,EAAE,EAAE,wBAAwB,CAAC,CAAC;AAC3D,CAAC;AAED,MAAM,UAAU,QAAQ;IACpB,OAAO,QAAQ,CAAC,OAAO,EAAE,EAAE,WAAW,CAAC,CAAC;AAC5C,CAAC;AAED,SAAS,UAAU;IACf,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;AACzC,CAAC;AAED,KAAK,UAAU,YAAY,CAAI,IAAY,EAAE,QAAW;IACpD,IAAI,CAAC;QACD,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,IAAI,CAAC,CAAC,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;YAAE,OAAO,QAAQ,CAAC;QAC5C,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAM,CAAC;IAC9C,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,QAAQ,CAAC;IACpB,CAAC;AACL,CAAC;AAED,KAAK,UAAU,aAAa,CAAC,IAAY,EAAE,KAAc;IACrD,MAAM,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;QACzD,IAAI,EAAE,KAAK;KACP,CAAC,CAAC;AACd,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS;IAC3B,MAAM,MAAM,GAAG,MAAM,YAAY,CAAQ,SAAS,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IACpE,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS;QAAE,OAAO,UAAU,EAAE,CAAC;IACnE,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,KAAY;IACzC,MAAM,aAAa,CAAC,SAAS,EAAE,EAAE,KAAK,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc;IAChC,IAAI,GAAG,CAAC,GAAG,CAAC,qBAAqB,EAAE,CAAC;QAChC,IAAI,CAAC;YACD,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,qBAAqB,CAG9C,CAAC;QACN,CAAC;QAAC,MAAM,CAAC;YACL,OAAO,EAAE,CAAC;QACd,CAAC;IACL,CAAC;IACD,OAAO,YAAY,CAA2B,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;AAClE,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,KAAc;IACrC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC;IACtD,MAAM,IAAI,GAAI,KAA4B,CAAC,IAAI,CAAC;IAChD,IAAI,IAAI,KAAK,KAAK;QACd,OAAO,OAAQ,KAA2B,CAAC,GAAG,KAAK,QAAQ,CAAC;IAChE,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;QACnB,MAAM,IAAI,GAAG,KAIZ,CAAC;QACF,OAAO,CACH,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ;YAChC,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ;YAC/B,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CACnC,CAAC;IACN,CAAC;IACD,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;QACvB,MAAM,IAAI,GAAG,KAA2C,CAAC;QACzD,OAAO,OAAO,IAAI,CAAC,GAAG,KAAK,QAAQ,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC;IAC1E,CAAC;IACD,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAC7B,UAAkB,EAClB,IAAc,EACd,MAAoD;IAEpD,MAAM,KAAK,GAAG,MAAM,SAAS,EAAE,CAAC;IAChC,KAAK,CAAC,YAAY,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,MAAM,EAAE,CAAC;IACrE,MAAM,UAAU,CAAC,KAAK,CAAC,CAAC;IACxB,OAAO,KAAK,CAAC;AACjB,CAAC"}

package/dist/balancer/types.d.ts

@@ -0,0 +1,44 @@
+export type AuthInfo = {
+    type: "api";
+    key: string;
+    metadata?: Record<string, string>;
+} | {
+    type: "oauth";
+    refresh: string;
+    access: string;
+    expires: number;
+    accountId?: string;
+    enterpriseUrl?: string;
+} | {
+    type: "wellknown";
+    key: string;
+    token: string;
+};
+export type Account = {
+    alias: string;
+    providerID: string;
+    auth: AuthInfo;
+    createdAt: number;
+    updatedAt: number;
+    lastUsedAt?: number;
+    rateLimitedUntil?: number;
+    failures?: number;
+};
+export type ProviderState = {
+    active?: string;
+    accounts: Record<string, Account>;
+};
+export type Store = {
+    version: 1;
+    providers: Record<string, ProviderState>;
+    lastCaptured?: {
+        providerID: string;
+        auth: AuthInfo;
+        capturedAt: number;
+        source: "auth-file" | "http" | "oauth-callback";
+    };
+};
+export type PendingRequest = {
+    providerID: string;
+    account?: Account;
+};

package/dist/balancer/types.js

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

package/dist/balancer/types.js.map

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

package/dist/index.d.ts

@@ -0,0 +1,37 @@
+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;

package/dist/index.js

@@ -0,0 +1,61 @@
+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),
+            }),
+        },
+    };
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +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"}

package/package.json

@@ -0,0 +1,39 @@
+{
+    "name": "@thelioo/opencode-balancer",
+    "version": "0.1.0",
+    "description": "Account balancer plugin for opencode.",
+    "type": "module",
+    "main": "./dist/index.js",
+    "types": "./dist/index.d.ts",
+    "exports": {
+        ".": {
+            "types": "./dist/index.d.ts",
+            "default": "./dist/index.js"
+        }
+    },
+    "files": [
+        "dist",
+        "README.md",
+        "INSTALL.txt",
+        "LICENSE"
+    ],
+    "scripts": {
+        "build": "tsc --project tsconfig.json",
+        "check": "tsc --noEmit --project tsconfig.json",
+        "prepublishOnly": "npm run build"
+    },
+    "keywords": [
+        "opencode",
+        "opencode-plugin",
+        "balancer",
+        "accounts"
+    ],
+    "license": "MIT",
+    "dependencies": {
+        "@opencode-ai/plugin": "^1.14.51"
+    },
+    "devDependencies": {
+        "@types/bun": "^1.3.14",
+        "typescript": "^5.9.3"
+    }
+}