@thelioo/opencode-balancer 0.1.0 → 0.1.4

package/dist/server/request-balancer.js

@@ -0,0 +1,43 @@
+import { getAccount, listAccounts, normalizeAlias } from "../core/accounts";
+import { now } from "../core/time";
+export const INTERNAL_REQUEST_HEADER = "x-opencode-balancer-request";
+export const BALANCER_METADATA_KEY = "opencodeBalancerCommand";
+export const RETRYABLE_STATUS = new Set([429, 500, 502, 503, 504, 529]);
+const pendingRequests = new Map();
+export function setPendingRequest(requestID, request) {
+    pendingRequests.set(requestID, request);
+}
+export function takePendingRequest(requestID) {
+    const request = pendingRequests.get(requestID);
+    pendingRequests.delete(requestID);
+    return request;
+}
+export function __testGetPendingRequest(requestID) {
+    return pendingRequests.get(requestID);
+}
+export function __testClearPendingRequests() {
+    pendingRequests.clear();
+}
+export function markRateLimited(db, providerID, alias, retryAfterMs = 60_000) {
+    const account = getAccount(db, providerID, alias);
+    if (!account)
+        return;
+    const timestamp = now();
+    db.query(`UPDATE accounts
+         SET failures = failures + 1,
+             rate_limited_until = ?,
+             updated_at = ?
+         WHERE provider_id = ? AND alias = ?`).run(timestamp + retryAfterMs, timestamp, providerID, account.alias);
+}
+export function chooseFailoverAccount(db, providerID, currentAlias) {
+    const timestamp = now();
+    const normalizedCurrentAlias = normalizeAlias(currentAlias);
+    return listAccounts(db, providerID).find((account) => {
+        if (account.alias === normalizedCurrentAlias)
+            return false;
+        if (account.disabled)
+            return false;
+        return !account.rateLimitedUntil || account.rateLimitedUntil <= timestamp;
+    });
+}
+//# sourceMappingURL=request-balancer.js.map

package/dist/server/request-balancer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-balancer.js","sourceRoot":"","sources":["../../src/server/request-balancer.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAC5E,OAAO,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAGnC,MAAM,CAAC,MAAM,uBAAuB,GAAG,6BAA6B,CAAC;AACrE,MAAM,CAAC,MAAM,qBAAqB,GAAG,yBAAyB,CAAC;AAE/D,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;AAOxE,MAAM,eAAe,GAAG,IAAI,GAAG,EAA0B,CAAC;AAE1D,MAAM,UAAU,iBAAiB,CAAC,SAAiB,EAAE,OAAuB;IACxE,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,SAAiB;IAChD,MAAM,OAAO,GAAG,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC/C,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAClC,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,SAAiB;IACrD,OAAO,eAAe,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,0BAA0B;IACtC,eAAe,CAAC,KAAK,EAAE,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,eAAe,CAC3B,EAAY,EACZ,UAAkB,EAClB,KAAa,EACb,YAAY,GAAG,MAAM;IAErB,MAAM,OAAO,GAAG,UAAU,CAAC,EAAE,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;IAClD,IAAI,CAAC,OAAO;QAAE,OAAO;IAErB,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC;IACxB,EAAE,CAAC,KAAK,CACJ;;;;6CAIqC,CACxC,CAAC,GAAG,CAAC,SAAS,GAAG,YAAY,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;AAC1E,CAAC;AAED,MAAM,UAAU,qBAAqB,CACjC,EAAY,EACZ,UAAkB,EAClB,YAAoB;IAEpB,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC;IACxB,MAAM,sBAAsB,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;IAC5D,OAAO,YAAY,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE;QACjD,IAAI,OAAO,CAAC,KAAK,KAAK,sBAAsB;YAAE,OAAO,KAAK,CAAC;QAC3D,IAAI,OAAO,CAAC,QAAQ;YAAE,OAAO,KAAK,CAAC;QACnC,OAAO,CAAC,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,gBAAgB,IAAI,SAAS,CAAC;IAC9E,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/tui/actions.d.ts

@@ -0,0 +1,41 @@
+import { refreshAccountUsage } from "../core/usage";
+import type { BalancerTuiState } from "./state";
+type AuthSetApi = {
+    client: {
+        auth: {
+            set: (input: any) => unknown;
+        };
+    };
+    keymap?: {
+        dispatchCommand?: (command: string) => unknown;
+    };
+    ui?: {
+        toast: (input: {
+            variant: "success" | "warning" | "error" | "info";
+            message: string;
+        }) => unknown;
+    };
+};
+type ActivateAccountOptions = {
+    sessionProviderID?: string;
+    openProviderModelPicker?: (providerID: string) => void;
+};
+type ToastApi = {
+    ui: {
+        toast: (input: {
+            variant: "success" | "warning" | "error" | "info";
+            message: string;
+        }) => unknown;
+    };
+};
+type RefreshUsageOptions = {
+    refreshUsage?: typeof refreshAccountUsage;
+    silent?: boolean;
+};
+export declare function savePendingAlias(state: BalancerTuiState, pendingID: string, alias: string): import("../core/types").Account;
+export declare function activateAccount(api: AuthSetApi, state: BalancerTuiState, providerID: string, alias: string, options?: ActivateAccountOptions): Promise<void>;
+export declare function refreshUsageForAccount(api: ToastApi, state: BalancerTuiState, providerID: string, alias: string, options?: RefreshUsageOptions): Promise<void>;
+export declare function removeAccountFromTui(api: ToastApi, state: BalancerTuiState, providerID: string, alias: string): void;
+export declare function renameAccountFromTui(api: ToastApi, state: BalancerTuiState, providerID: string, alias: string, nextAlias: string): import("../core/types").Account;
+export declare function removePendingFromTui(api: ToastApi, state: BalancerTuiState, pendingID: string): void;
+export {};

package/dist/tui/actions.js

@@ -0,0 +1,88 @@
+import { getAccount, getActiveAccount, removeAccount, renameAccount, setActiveAccount } from "../core/accounts";
+import { appendEvent } from "../core/events";
+import { completePendingConnection, removePendingConnection } from "../core/pending";
+import { refreshAccountUsage } from "../core/usage";
+import { saveUsageSnapshot } from "../core/usage/store";
+import { suppressNativeAuthCapture } from "../core/native-auth-suppression";
+export function savePendingAlias(state, pendingID, alias) {
+    const account = completePendingConnection(state.db, pendingID, alias);
+    state.refresh();
+    return account;
+}
+export async function activateAccount(api, state, providerID, alias, options = {}) {
+    const account = setActiveAccount(state.db, providerID, alias) ?? getActiveAccount(state.db, providerID);
+    if (account) {
+        try {
+            suppressNativeAuthCapture(state.db, providerID);
+            await api.client.auth.set({ path: { id: providerID }, body: account.auth });
+        }
+        catch { }
+    }
+    state.refresh();
+    api.ui?.toast({ variant: "success", message: `Activated ${providerID}/${alias}.` });
+    options.openProviderModelPicker?.(providerID);
+}
+export async function refreshUsageForAccount(api, state, providerID, alias, options = {}) {
+    const account = getAccount(state.db, providerID, alias) ?? state.accounts().find((candidate) => {
+        return candidate.providerID === providerID && candidate.alias === alias;
+    });
+    if (!account) {
+        api.ui.toast({ variant: "error", message: `Account not found: ${providerID}/${alias}` });
+        return;
+    }
+    try {
+        const snapshot = await (options.refreshUsage ?? refreshAccountUsage)(account);
+        saveUsageSnapshot(state.db, snapshot);
+        appendEvent(state.db, {
+            type: snapshot.confidence === "unavailable" ? "usage_unavailable" : "usage_refreshed",
+            providerID,
+            alias,
+            message: snapshot.message,
+        });
+        state.refresh();
+        if (!options.silent) {
+            api.ui.toast({
+                variant: snapshot.confidence === "unavailable" ? "warning" : "success",
+                message: snapshot.message,
+            });
+        }
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : "Usage refresh failed.";
+        api.ui.toast({ variant: "error", message });
+    }
+}
+export function removeAccountFromTui(api, state, providerID, alias) {
+    const removed = removeAccount(state.db, providerID, alias);
+    if (!removed) {
+        api.ui.toast({ variant: "error", message: `Account not found: ${providerID}/${alias}` });
+        return;
+    }
+    const message = `Removed account ${providerID}/${alias}.`;
+    appendEvent(state.db, {
+        type: "account_removed",
+        providerID,
+        alias,
+        message,
+    });
+    state.refresh();
+    state.removeAccountView(providerID, alias);
+    api.ui.toast({ variant: "success", message });
+}
+export function renameAccountFromTui(api, state, providerID, alias, nextAlias) {
+    const account = renameAccount(state.db, providerID, alias, nextAlias);
+    state.refresh();
+    api.ui.toast({ variant: "success", message: `Renamed ${providerID}/${alias} to ${account.alias}.` });
+    return account;
+}
+export function removePendingFromTui(api, state, pendingID) {
+    const removed = removePendingConnection(state.db, pendingID);
+    if (!removed) {
+        api.ui.toast({ variant: "error", message: "Pending connection not found." });
+        return;
+    }
+    state.refresh();
+    state.removePendingView(pendingID);
+    api.ui.toast({ variant: "success", message: "Removed pending connection." });
+}
+//# sourceMappingURL=actions.js.map

package/dist/tui/actions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"actions.js","sourceRoot":"","sources":["../../src/tui/actions.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,aAAa,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAChH,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,yBAAyB,EAAE,uBAAuB,EAAE,MAAM,iBAAiB,CAAC;AACrF,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,yBAAyB,EAAE,MAAM,iCAAiC,CAAC;AAiC5E,MAAM,UAAU,gBAAgB,CAAC,KAAuB,EAAE,SAAiB,EAAE,KAAa;IACtF,MAAM,OAAO,GAAG,yBAAyB,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;IACtE,KAAK,CAAC,OAAO,EAAE,CAAC;IAChB,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACjC,GAAe,EACf,KAAuB,EACvB,UAAkB,EAClB,KAAa,EACb,UAAkC,EAAE;IAEpC,MAAM,OAAO,GAAG,gBAAgB,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,EAAE,KAAK,CAAC,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAExG,IAAI,OAAO,EAAE,CAAC;QACV,IAAI,CAAC;YACD,yBAAyB,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAChD,MAAM,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAChF,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;IACd,CAAC;IAED,KAAK,CAAC,OAAO,EAAE,CAAC;IAChB,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,UAAU,IAAI,KAAK,GAAG,EAAE,CAAC,CAAC;IACpF,OAAO,CAAC,uBAAuB,EAAE,CAAC,UAAU,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CACxC,GAAa,EACb,KAAuB,EACvB,UAAkB,EAClB,KAAa,EACb,UAA+B,EAAE;IAEjC,MAAM,OAAO,GAAG,UAAU,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,EAAE,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE;QAC3F,OAAO,SAAS,CAAC,UAAU,KAAK,UAAU,IAAI,SAAS,CAAC,KAAK,KAAK,KAAK,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,OAAO,EAAE,CAAC;QACX,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,sBAAsB,UAAU,IAAI,KAAK,EAAE,EAAE,CAAC,CAAC;QACzF,OAAO;IACX,CAAC;IAED,IAAI,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,YAAY,IAAI,mBAAmB,CAAC,CAAC,OAAO,CAAC,CAAC;QAC9E,iBAAiB,CAAC,KAAK,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QACtC,WAAW,CAAC,KAAK,CAAC,EAAE,EAAE;YAClB,IAAI,EAAE,QAAQ,CAAC,UAAU,KAAK,aAAa,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,iBAAiB;YACrF,UAAU;YACV,KAAK;YACL,OAAO,EAAE,QAAQ,CAAC,OAAO;SAC5B,CAAC,CAAC;QACH,KAAK,CAAC,OAAO,EAAE,CAAC;QAChB,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAClB,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC;gBACT,OAAO,EAAE,QAAQ,CAAC,UAAU,KAAK,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACtE,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC5B,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB,CAAC;QACjF,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;IAChD,CAAC;AACL,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,GAAa,EAAE,KAAuB,EAAE,UAAkB,EAAE,KAAa;IAC1G,MAAM,OAAO,GAAG,aAAa,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;IAC3D,IAAI,CAAC,OAAO,EAAE,CAAC;QACX,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,sBAAsB,UAAU,IAAI,KAAK,EAAE,EAAE,CAAC,CAAC;QACzF,OAAO;IACX,CAAC;IAED,MAAM,OAAO,GAAG,mBAAmB,UAAU,IAAI,KAAK,GAAG,CAAC;IAC1D,WAAW,CAAC,KAAK,CAAC,EAAE,EAAE;QAClB,IAAI,EAAE,iBAAiB;QACvB,UAAU;QACV,KAAK;QACL,OAAO;KACV,CAAC,CAAC;IACH,KAAK,CAAC,OAAO,EAAE,CAAC;IAChB,KAAK,CAAC,iBAAiB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IAC3C,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,GAAa,EAAE,KAAuB,EAAE,UAAkB,EAAE,KAAa,EAAE,SAAiB;IAC7H,MAAM,OAAO,GAAG,aAAa,CAAC,KAAK,CAAC,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IACtE,KAAK,CAAC,OAAO,EAAE,CAAC;IAChB,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,WAAW,UAAU,IAAI,KAAK,OAAO,OAAO,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC;IACrG,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,GAAa,EAAE,KAAuB,EAAE,SAAiB;IAC1F,MAAM,OAAO,GAAG,uBAAuB,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;IAC7D,IAAI,CAAC,OAAO,EAAE,CAAC;QACX,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,+BAA+B,EAAE,CAAC,CAAC;QAC7E,OAAO;IACX,CAAC;IAED,KAAK,CAAC,OAAO,EAAE,CAAC;IAChB,KAAK,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACnC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,6BAA6B,EAAE,CAAC,CAAC;AACjF,CAAC"}

package/dist/tui/actions.ts

@@ -0,0 +1,140 @@
+import { getAccount, getActiveAccount, removeAccount, renameAccount, setActiveAccount } from "../core/accounts";
+import { appendEvent } from "../core/events";
+import { completePendingConnection, removePendingConnection } from "../core/pending";
+import { refreshAccountUsage } from "../core/usage";
+import { saveUsageSnapshot } from "../core/usage/store";
+import { suppressNativeAuthCapture } from "../core/native-auth-suppression";
+import type { BalancerTuiState } from "./state";
+
+type AuthSetApi = {
+    client: {
+        auth: {
+            set: (input: any) => unknown;
+        };
+    };
+    keymap?: {
+        dispatchCommand?: (command: string) => unknown;
+    };
+    ui?: {
+        toast: (input: { variant: "success" | "warning" | "error" | "info"; message: string }) => unknown;
+    };
+};
+
+type ActivateAccountOptions = {
+    sessionProviderID?: string;
+    openProviderModelPicker?: (providerID: string) => void;
+};
+
+type ToastApi = {
+    ui: {
+        toast: (input: { variant: "success" | "warning" | "error" | "info"; message: string }) => unknown;
+    };
+};
+
+type RefreshUsageOptions = {
+    refreshUsage?: typeof refreshAccountUsage;
+    silent?: boolean;
+};
+
+export function savePendingAlias(state: BalancerTuiState, pendingID: string, alias: string) {
+    const account = completePendingConnection(state.db, pendingID, alias);
+    state.refresh();
+    return account;
+}
+
+export async function activateAccount(
+    api: AuthSetApi,
+    state: BalancerTuiState,
+    providerID: string,
+    alias: string,
+    options: ActivateAccountOptions = {},
+) {
+    const account = setActiveAccount(state.db, providerID, alias) ?? getActiveAccount(state.db, providerID);
+
+    if (account) {
+        try {
+            suppressNativeAuthCapture(state.db, providerID);
+            await api.client.auth.set({ path: { id: providerID }, body: account.auth });
+        } catch {}
+    }
+
+    state.refresh();
+    api.ui?.toast({ variant: "success", message: `Activated ${providerID}/${alias}.` });
+    options.openProviderModelPicker?.(providerID);
+}
+
+export async function refreshUsageForAccount(
+    api: ToastApi,
+    state: BalancerTuiState,
+    providerID: string,
+    alias: string,
+    options: RefreshUsageOptions = {},
+) {
+    const account = getAccount(state.db, providerID, alias) ?? state.accounts().find((candidate) => {
+        return candidate.providerID === providerID && candidate.alias === alias;
+    });
+
+    if (!account) {
+        api.ui.toast({ variant: "error", message: `Account not found: ${providerID}/${alias}` });
+        return;
+    }
+
+    try {
+        const snapshot = await (options.refreshUsage ?? refreshAccountUsage)(account);
+        saveUsageSnapshot(state.db, snapshot);
+        appendEvent(state.db, {
+            type: snapshot.confidence === "unavailable" ? "usage_unavailable" : "usage_refreshed",
+            providerID,
+            alias,
+            message: snapshot.message,
+        });
+        state.refresh();
+        if (!options.silent) {
+            api.ui.toast({
+                variant: snapshot.confidence === "unavailable" ? "warning" : "success",
+                message: snapshot.message,
+            });
+        }
+    } catch (error) {
+        const message = error instanceof Error ? error.message : "Usage refresh failed.";
+        api.ui.toast({ variant: "error", message });
+    }
+}
+
+export function removeAccountFromTui(api: ToastApi, state: BalancerTuiState, providerID: string, alias: string) {
+    const removed = removeAccount(state.db, providerID, alias);
+    if (!removed) {
+        api.ui.toast({ variant: "error", message: `Account not found: ${providerID}/${alias}` });
+        return;
+    }
+
+    const message = `Removed account ${providerID}/${alias}.`;
+    appendEvent(state.db, {
+        type: "account_removed",
+        providerID,
+        alias,
+        message,
+    });
+    state.refresh();
+    state.removeAccountView(providerID, alias);
+    api.ui.toast({ variant: "success", message });
+}
+
+export function renameAccountFromTui(api: ToastApi, state: BalancerTuiState, providerID: string, alias: string, nextAlias: string) {
+    const account = renameAccount(state.db, providerID, alias, nextAlias);
+    state.refresh();
+    api.ui.toast({ variant: "success", message: `Renamed ${providerID}/${alias} to ${account.alias}.` });
+    return account;
+}
+
+export function removePendingFromTui(api: ToastApi, state: BalancerTuiState, pendingID: string) {
+    const removed = removePendingConnection(state.db, pendingID);
+    if (!removed) {
+        api.ui.toast({ variant: "error", message: "Pending connection not found." });
+        return;
+    }
+
+    state.refresh();
+    state.removePendingView(pendingID);
+    api.ui.toast({ variant: "success", message: "Removed pending connection." });
+}

package/dist/tui/balancer-bar-sync.d.ts

@@ -0,0 +1,19 @@
+import type { TuiPluginApi } from "@opencode-ai/plugin/tui";
+import type { BalancerTuiState } from "./state";
+import { type NativeModelApplier } from "./native-model-apply";
+export type BalancerBarSyncDeps = {
+    balancingEnabled: () => boolean;
+    dialogOpen: () => boolean;
+    activeSelection: () => {
+        providerID: string;
+        modelID: string;
+    } | undefined;
+    modelTitle: (providerID: string, modelID: string) => string | undefined;
+    apply: NativeModelApplier;
+};
+export declare function createBalancerBarSync(deps: BalancerBarSyncDeps): {
+    maybeSync: () => Promise<boolean>;
+};
+export declare function createTuiBalancerBarSync(api: TuiPluginApi, state: BalancerTuiState): {
+    maybeSync: () => Promise<boolean>;
+};

package/dist/tui/balancer-bar-sync.js

@@ -0,0 +1,45 @@
+import { getBalancingEnabled, resolveActiveSelection } from "../core/priority";
+import { createNativeModelApplier } from "./native-model-apply";
+export function createBalancerBarSync(deps) {
+    let lastApplied;
+    let applying = false;
+    const maybeSync = async () => {
+        if (!deps.balancingEnabled()) {
+            lastApplied = undefined;
+            return false;
+        }
+        if (applying || deps.dialogOpen())
+            return false;
+        const selection = deps.activeSelection();
+        if (!selection)
+            return false;
+        const key = `${selection.providerID}/${selection.modelID}`;
+        if (key === lastApplied)
+            return false;
+        const title = deps.modelTitle(selection.providerID, selection.modelID) ?? selection.modelID;
+        applying = true;
+        try {
+            const applied = await deps.apply(selection, title);
+            if (applied)
+                lastApplied = key;
+            return applied;
+        }
+        finally {
+            applying = false;
+        }
+    };
+    return { maybeSync };
+}
+export function createTuiBalancerBarSync(api, state) {
+    return createBalancerBarSync({
+        balancingEnabled: () => getBalancingEnabled(state.db),
+        dialogOpen: () => api.ui.dialog.open,
+        activeSelection: () => resolveActiveSelection(state.db),
+        modelTitle: (providerID, modelID) => {
+            const provider = api.state.provider.find((item) => item.id === providerID);
+            return provider?.models?.[modelID]?.name;
+        },
+        apply: createNativeModelApplier(api),
+    });
+}
+//# sourceMappingURL=balancer-bar-sync.js.map

package/dist/tui/balancer-bar-sync.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"balancer-bar-sync.js","sourceRoot":"","sources":["../../src/tui/balancer-bar-sync.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAE/E,OAAO,EAAE,wBAAwB,EAA2B,MAAM,sBAAsB,CAAC;AAUzF,MAAM,UAAU,qBAAqB,CAAC,IAAyB;IAC3D,IAAI,WAA+B,CAAC;IACpC,IAAI,QAAQ,GAAG,KAAK,CAAC;IAErB,MAAM,SAAS,GAAG,KAAK,IAAI,EAAE;QACzB,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,CAAC;YAC3B,WAAW,GAAG,SAAS,CAAC;YACxB,OAAO,KAAK,CAAC;QACjB,CAAC;QACD,IAAI,QAAQ,IAAI,IAAI,CAAC,UAAU,EAAE;YAAE,OAAO,KAAK,CAAC;QAEhD,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACzC,IAAI,CAAC,SAAS;YAAE,OAAO,KAAK,CAAC;QAE7B,MAAM,GAAG,GAAG,GAAG,SAAS,CAAC,UAAU,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;QAC3D,IAAI,GAAG,KAAK,WAAW;YAAE,OAAO,KAAK,CAAC;QAEtC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,UAAU,EAAE,SAAS,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC;QAC5F,QAAQ,GAAG,IAAI,CAAC;QAChB,IAAI,CAAC;YACD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACnD,IAAI,OAAO;gBAAE,WAAW,GAAG,GAAG,CAAC;YAC/B,OAAO,OAAO,CAAC;QACnB,CAAC;gBAAS,CAAC;YACP,QAAQ,GAAG,KAAK,CAAC;QACrB,CAAC;IACL,CAAC,CAAC;IAEF,OAAO,EAAE,SAAS,EAAE,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,GAAiB,EAAE,KAAuB;IAC/E,OAAO,qBAAqB,CAAC;QACzB,gBAAgB,EAAE,GAAG,EAAE,CAAC,mBAAmB,CAAC,KAAK,CAAC,EAAE,CAAC;QACrD,UAAU,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI;QACpC,eAAe,EAAE,GAAG,EAAE,CAAC,sBAAsB,CAAC,KAAK,CAAC,EAAE,CAAC;QACvD,UAAU,EAAE,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE;YAChC,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,UAAU,CAAC,CAAC;YAC3E,OAAO,QAAQ,EAAE,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC;QAC7C,CAAC;QACD,KAAK,EAAE,wBAAwB,CAAC,GAAG,CAAC;KACvC,CAAC,CAAC;AACP,CAAC"}

package/dist/tui/balancer-bar-sync.ts

@@ -0,0 +1,56 @@
+import type { TuiPluginApi } from "@opencode-ai/plugin/tui";
+import { getBalancingEnabled, resolveActiveSelection } from "../core/priority";
+import type { BalancerTuiState } from "./state";
+import { createNativeModelApplier, type NativeModelApplier } from "./native-model-apply";
+
+export type BalancerBarSyncDeps = {
+    balancingEnabled: () => boolean;
+    dialogOpen: () => boolean;
+    activeSelection: () => { providerID: string; modelID: string } | undefined;
+    modelTitle: (providerID: string, modelID: string) => string | undefined;
+    apply: NativeModelApplier;
+};
+
+export function createBalancerBarSync(deps: BalancerBarSyncDeps) {
+    let lastApplied: string | undefined;
+    let applying = false;
+
+    const maybeSync = async () => {
+        if (!deps.balancingEnabled()) {
+            lastApplied = undefined;
+            return false;
+        }
+        if (applying || deps.dialogOpen()) return false;
+
+        const selection = deps.activeSelection();
+        if (!selection) return false;
+
+        const key = `${selection.providerID}/${selection.modelID}`;
+        if (key === lastApplied) return false;
+
+        const title = deps.modelTitle(selection.providerID, selection.modelID) ?? selection.modelID;
+        applying = true;
+        try {
+            const applied = await deps.apply(selection, title);
+            if (applied) lastApplied = key;
+            return applied;
+        } finally {
+            applying = false;
+        }
+    };
+
+    return { maybeSync };
+}
+
+export function createTuiBalancerBarSync(api: TuiPluginApi, state: BalancerTuiState) {
+    return createBalancerBarSync({
+        balancingEnabled: () => getBalancingEnabled(state.db),
+        dialogOpen: () => api.ui.dialog.open,
+        activeSelection: () => resolveActiveSelection(state.db),
+        modelTitle: (providerID, modelID) => {
+            const provider = api.state.provider.find((item) => item.id === providerID);
+            return provider?.models?.[modelID]?.name;
+        },
+        apply: createNativeModelApplier(api),
+    });
+}

package/dist/tui/components/alias-dialog.d.ts

@@ -0,0 +1,4 @@
+/** @jsxImportSource @opentui/solid */
+import type { TuiPluginApi } from "@opencode-ai/plugin/tui";
+import type { BalancerTuiState } from "../state";
+export declare function openAliasDialog(api: TuiPluginApi, state: BalancerTuiState, pendingID: string): boolean;

package/dist/tui/components/alias-dialog.js

@@ -0,0 +1,57 @@
+import { jsxs as _jsxs, jsx as _jsx } from "@opentui/solid/jsx-runtime";
+import { normalizeAlias } from "../../core/accounts";
+import { claimPendingPrompt, pendingPromptGroupID, releasePendingPrompt } from "../../core/pending";
+import { savePendingAlias } from "../actions";
+function errorMessage(error) {
+    return error instanceof Error && error.message ? error.message : "Failed to save alias";
+}
+const openPendingDialogs = new Set();
+export function openAliasDialog(api, state, pendingID) {
+    const dialogID = pendingPromptGroupID(state.db, pendingID) ?? pendingID;
+    if (openPendingDialogs.has(dialogID)) {
+        if ("open" in api.ui.dialog && !api.ui.dialog.open)
+            openPendingDialogs.delete(dialogID);
+        else
+            return false;
+    }
+    let claimed = claimPendingPrompt(state.db, pendingID);
+    if (!claimed) {
+        releasePendingPrompt(state.db, pendingID);
+        claimed = claimPendingPrompt(state.db, pendingID);
+    }
+    if (!claimed)
+        return false;
+    const pending = state.pending().find((item) => item.id === pendingID) ?? claimed;
+    const providerID = pending?.providerID ?? "provider";
+    openPendingDialogs.add(dialogID);
+    const close = () => {
+        openPendingDialogs.delete(dialogID);
+        api.ui.dialog.clear();
+    };
+    try {
+        api.ui.dialog.setSize("medium");
+        api.ui.dialog.replace(() => (_jsx(api.ui.DialogPrompt, { title: "Save pending connection", placeholder: "account alias", description: () => (_jsxs("text", { fg: api.theme.current.textMuted, wrapMode: "none", children: ["Choose an alias for ", providerID, "/", pending?.authType ?? "auth", "."] })), onCancel: close, onConfirm: async (value) => {
+                const alias = normalizeAlias(value);
+                if (!alias) {
+                    api.ui.toast({ variant: "error", message: "Alias is required" });
+                    return;
+                }
+                try {
+                    const account = savePendingAlias(state, pendingID, alias);
+                    close();
+                    api.ui.toast({ variant: "success", message: `Saved ${account.providerID}/${account.alias}` });
+                }
+                catch (error) {
+                    openPendingDialogs.delete(dialogID);
+                    api.ui.toast({ variant: "error", message: errorMessage(error) });
+                }
+            } })));
+        return true;
+    }
+    catch {
+        openPendingDialogs.delete(dialogID);
+        releasePendingPrompt(state.db, pendingID);
+        return false;
+    }
+}
+//# sourceMappingURL=alias-dialog.js.map

package/dist/tui/components/alias-dialog.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"alias-dialog.js","sourceRoot":"","sources":["../../../src/tui/components/alias-dialog.tsx"],"names":[],"mappings":";AAGA,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AACpG,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAG9C,SAAS,YAAY,CAAC,KAAc;IAChC,OAAO,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,sBAAsB,CAAC;AAC5F,CAAC;AAED,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAU,CAAC;AAE7C,MAAM,UAAU,eAAe,CAAC,GAAiB,EAAE,KAAuB,EAAE,SAAiB;IACzF,MAAM,QAAQ,GAAG,oBAAoB,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,IAAI,SAAS,CAAC;IACxE,IAAI,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnC,IAAI,MAAM,IAAI,GAAG,CAAC,EAAE,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI;YAAE,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;;YACnF,OAAO,KAAK,CAAC;IACtB,CAAC;IACD,IAAI,OAAO,GAAG,kBAAkB,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;IACtD,IAAI,CAAC,OAAO,EAAE,CAAC;QACX,oBAAoB,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAC1C,OAAO,GAAG,kBAAkB,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;IACtD,CAAC;IACD,IAAI,CAAC,OAAO;QAAE,OAAO,KAAK,CAAC;IAC3B,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,SAAS,CAAC,IAAI,OAAO,CAAC;IACjF,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,UAAU,CAAC;IACrD,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAEjC,MAAM,KAAK,GAAG,GAAG,EAAE;QACf,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACpC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC,CAAC;IAEF,IAAI,CAAC;QACD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAChC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CACxB,KAAC,GAAG,CAAC,EAAE,CAAC,YAAY,IAChB,KAAK,EAAC,yBAAyB,EAC/B,WAAW,EAAC,eAAe,EAC3B,WAAW,EAAE,GAAG,EAAE,CAAC,CACf,gBAAM,EAAE,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,EAAC,MAAM,qCAC7B,UAAU,OAAG,OAAO,EAAE,QAAQ,IAAI,MAAM,SAC1D,CACV,EACD,QAAQ,EAAE,KAAK,EACf,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;gBACvB,MAAM,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;gBACpC,IAAI,CAAC,KAAK,EAAE,CAAC;oBACT,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAAC;oBACjE,OAAO;gBACX,CAAC;gBAED,IAAI,CAAC;oBACD,MAAM,OAAO,GAAG,gBAAgB,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;oBAC1D,KAAK,EAAE,CAAC;oBACR,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAClG,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;oBACpC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACrE,CAAC;YACL,CAAC,GACH,CACL,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IAChB,CAAC;IAAC,MAAM,CAAC;QACL,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACpC,oBAAoB,CAAC,KAAK,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAC1C,OAAO,KAAK,CAAC;IACjB,CAAC;AACL,CAAC"}

package/dist/tui/components/alias-dialog.tsx

@@ -0,0 +1,72 @@
+/** @jsxImportSource @opentui/solid */
+
+import type { TuiPluginApi } from "@opencode-ai/plugin/tui";
+import { normalizeAlias } from "../../core/accounts";
+import { claimPendingPrompt, pendingPromptGroupID, releasePendingPrompt } from "../../core/pending";
+import { savePendingAlias } from "../actions";
+import type { BalancerTuiState } from "../state";
+
+function errorMessage(error: unknown) {
+    return error instanceof Error && error.message ? error.message : "Failed to save alias";
+}
+
+const openPendingDialogs = new Set<string>();
+
+export function openAliasDialog(api: TuiPluginApi, state: BalancerTuiState, pendingID: string) {
+    const dialogID = pendingPromptGroupID(state.db, pendingID) ?? pendingID;
+    if (openPendingDialogs.has(dialogID)) {
+        if ("open" in api.ui.dialog && !api.ui.dialog.open) openPendingDialogs.delete(dialogID);
+        else return false;
+    }
+    let claimed = claimPendingPrompt(state.db, pendingID);
+    if (!claimed) {
+        releasePendingPrompt(state.db, pendingID);
+        claimed = claimPendingPrompt(state.db, pendingID);
+    }
+    if (!claimed) return false;
+    const pending = state.pending().find((item) => item.id === pendingID) ?? claimed;
+    const providerID = pending?.providerID ?? "provider";
+    openPendingDialogs.add(dialogID);
+
+    const close = () => {
+        openPendingDialogs.delete(dialogID);
+        api.ui.dialog.clear();
+    };
+
+    try {
+        api.ui.dialog.setSize("medium");
+        api.ui.dialog.replace(() => (
+            <api.ui.DialogPrompt
+                title="Save pending connection"
+                placeholder="account alias"
+                description={() => (
+                    <text fg={api.theme.current.textMuted} wrapMode="none">
+                        Choose an alias for {providerID}/{pending?.authType ?? "auth"}.
+                    </text>
+                )}
+                onCancel={close}
+                onConfirm={async (value) => {
+                    const alias = normalizeAlias(value);
+                    if (!alias) {
+                        api.ui.toast({ variant: "error", message: "Alias is required" });
+                        return;
+                    }
+
+                    try {
+                        const account = savePendingAlias(state, pendingID, alias);
+                        close();
+                        api.ui.toast({ variant: "success", message: `Saved ${account.providerID}/${account.alias}` });
+                    } catch (error) {
+                        openPendingDialogs.delete(dialogID);
+                        api.ui.toast({ variant: "error", message: errorMessage(error) });
+                    }
+                }}
+            />
+        ));
+        return true;
+    } catch {
+        openPendingDialogs.delete(dialogID);
+        releasePendingPrompt(state.db, pendingID);
+        return false;
+    }
+}

package/dist/tui/components/dashboard.d.ts

@@ -0,0 +1,12 @@
+/** @jsxImportSource @opentui/solid */
+import type { TuiPluginApi } from "@opencode-ai/plugin/tui";
+import type { BalancerTuiState } from "../state";
+export declare function Dashboard(props: {
+    api: TuiPluginApi;
+    state: BalancerTuiState;
+    onBack: () => void;
+    openPriority: () => void;
+    openConnect: () => void;
+    renameAccount: (providerID: string, alias: string) => void;
+    removeAccount: (providerID: string, alias: string) => void;
+}): any;