@thelioo/opencode-balancer 0.1.6 → 0.2.0

package/dist/tui/components/provider-model-dialog.tsx

@@ -2,70 +2,87 @@
 
 import type { TuiPluginApi } from "@opencode-ai/plugin/tui";
 import { setSelectedModel } from "../../core/accounts";
-import { createNativeModelApplier, type NativeModelApplier } from "../native-model-apply";
-import { providerModelOptions, type ProviderModelOption } from "../provider-models";
+import {
+	createNativeModelApplier,
+	type NativeModelApplier,
+} from "../native-model-apply";
+import {
+	type ProviderModelOption,
+	providerModelOptions,
+} from "../provider-models";
 import type { BalancerTuiState } from "../state";
 
 type OpenProviderModelDialogOptions = {
-    // Test seam: replays the chosen model into opencode's native dialog so the
-    // native model bar updates. Defaults to driving opencode via simulated keys.
-    applyNativeSelection?: NativeModelApplier;
-    onSelected?: (model: { providerID: string; modelID: string }) => void;
+	// Test seam: replays the chosen model into opencode's native dialog so the
+	// native model bar updates. Defaults to driving opencode via simulated keys.
+	applyNativeSelection?: NativeModelApplier | false;
+	onComplete?: () => void;
+	onSelected?: (model: { providerID: string; modelID: string }) => void;
 };
 
 export function openProviderModelDialog(
-    api: TuiPluginApi,
-    state: Pick<BalancerTuiState, "db" | "refresh">,
-    providerID: string,
-    options: OpenProviderModelDialogOptions = {},
+	api: TuiPluginApi,
+	state: Pick<BalancerTuiState, "db" | "refresh">,
+	providerID: string,
+	options: OpenProviderModelDialogOptions = {},
 ) {
-    const modelOptions = providerModelOptions(api.state.provider, providerID);
-    const providerName = modelOptions[0]?.providerName ?? providerID;
-    const applyNativeSelection = options.applyNativeSelection ?? createNativeModelApplier(api);
+	const modelOptions = providerModelOptions(api.state.provider, providerID);
+	const providerName = modelOptions[0]?.providerName ?? providerID;
+	const applyNativeSelection =
+		options.applyNativeSelection ?? createNativeModelApplier(api);
 
-    api.ui.dialog.setSize("medium");
-    api.ui.dialog.replace(() => (
-        <api.ui.DialogSelect<ProviderModelOption>
-            title={`Select ${providerName} model`}
-            placeholder={`Search ${providerName} models`}
-            flat
-            skipFilter={false}
-            options={modelOptions.map((option) => ({
-                title: option.title,
-                description: option.providerName,
-                value: option,
-                footer: `${option.providerID}/${option.modelID}`,
-            }))}
-            onSelect={(option) => {
-                const model = {
-                    providerID: option.value.providerID,
-                    modelID: option.value.modelID,
-                };
+	api.ui.dialog.setSize("medium");
+	api.ui.dialog.replace(() => (
+		<api.ui.DialogSelect<ProviderModelOption>
+			flat
+			onSelect={(option) => {
+				const model = {
+					modelID: option.value.modelID,
+					providerID: option.value.providerID,
+				};
 
-                // Persist first: in manual mode this stores the selected model;
-                // priority mode injects its own persistence callback.
-                if (options.onSelected) options.onSelected(model);
-                else setSelectedModel(state.db, model.providerID, model.modelID);
-                state.refresh();
+				// Persist first: in manual mode this stores the selected model;
+				// priority mode injects its own persistence callback.
+				if (options.onSelected) options.onSelected(model);
+				else setSelectedModel(state.db, model.providerID, model.modelID);
+				state.refresh();
 
-                // Close our picker, then replay the choice into opencode's native
-                // model dialog so its bottom model bar reflects the selection.
-                api.ui.dialog.clear();
+				// Close our picker, then replay the choice into opencode's native
+				// model dialog so its bottom model bar reflects the selection.
+				api.ui.dialog.clear();
 
-                void Promise.resolve(applyNativeSelection(model, option.value.title)).then((applied) => {
-                    if (applied) {
-                        api.ui.toast({
-                            variant: "success",
-                            message: `Switched to ${option.value.providerName}/${option.value.title}.`,
-                        });
-                        return;
-                    }
-                    api.ui.toast({
-                        variant: "warning",
-                        message: `Selected ${option.value.providerName}/${option.value.title}; prompts will use it, but opencode's model bar may not have refreshed.`,
-                    });
-                });
-            }}
-        />
-    ));
+				if (applyNativeSelection === false) {
+					options.onComplete?.();
+					return;
+				}
+
+				void Promise.resolve(applyNativeSelection(model, option.value.title))
+					.then((applied) => {
+						if (applied) {
+							api.ui.toast({
+								message: `Switched to ${option.value.providerName}/${option.value.title}.`,
+								variant: "success",
+							});
+							return;
+						}
+						api.ui.toast({
+							message: `Selected ${option.value.providerName}/${option.value.title}; prompts will use it, but opencode's model bar may not have refreshed.`,
+							variant: "warning",
+						});
+					})
+					.finally(() => {
+						options.onComplete?.();
+					});
+			}}
+			options={modelOptions.map((option) => ({
+				description: option.providerName,
+				footer: `${option.providerID}/${option.modelID}`,
+				title: option.title,
+				value: option,
+			}))}
+			placeholder={`Search ${providerName} models`}
+			skipFilter={false}
+			title={`Select ${providerName} model`}
+		/>
+	));
 }

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

@@ -6,35 +6,42 @@ import { renameAccountFromTui } from "../actions";
 import type { BalancerTuiState } from "../state";
 
 function errorMessage(error: unknown) {
-    return error instanceof Error && error.message ? error.message : "Failed to rename account";
+	return error instanceof Error && error.message
+		? error.message
+		: "Failed to rename account";
 }
 
-export function openRenameDialog(api: TuiPluginApi, state: BalancerTuiState, providerID: string, alias: string) {
-    api.ui.dialog.setSize("medium");
-    api.ui.dialog.replace(() => (
-        <api.ui.DialogPrompt
-            title="Rename account"
-            placeholder="account alias"
-            description={() => (
-                <text fg={api.theme.current.textMuted} wrapMode="none">
-                    Choose a new alias for {providerID}/{alias}.
-                </text>
-            )}
-            onCancel={() => api.ui.dialog.clear()}
-            onConfirm={(value) => {
-                const nextAlias = normalizeAlias(value);
-                if (!nextAlias) {
-                    api.ui.toast({ variant: "error", message: "Alias is required" });
-                    return;
-                }
+export function openRenameDialog(
+	api: TuiPluginApi,
+	state: BalancerTuiState,
+	providerID: string,
+	alias: string,
+) {
+	api.ui.dialog.setSize("medium");
+	api.ui.dialog.replace(() => (
+		<api.ui.DialogPrompt
+			description={() => (
+				<text fg={api.theme.current.textMuted} wrapMode="none">
+					Choose a new alias for {providerID}/{alias}.
+				</text>
+			)}
+			onCancel={() => api.ui.dialog.clear()}
+			onConfirm={(value) => {
+				const nextAlias = normalizeAlias(value);
+				if (!nextAlias) {
+					api.ui.toast({ message: "Alias is required", variant: "error" });
+					return;
+				}
 
-                try {
-                    renameAccountFromTui(api, state, providerID, alias, nextAlias);
-                    api.ui.dialog.clear();
-                } catch (error) {
-                    api.ui.toast({ variant: "error", message: errorMessage(error) });
-                }
-            }}
-        />
-    ));
+				try {
+					renameAccountFromTui(api, state, providerID, alias, nextAlias);
+					api.ui.dialog.clear();
+				} catch (error) {
+					api.ui.toast({ message: errorMessage(error), variant: "error" });
+				}
+			}}
+			placeholder="account alias"
+			title="Rename account"
+		/>
+	));
 }

package/dist/tui/components/sidebar.tsx

@@ -4,94 +4,118 @@ import type { TuiPluginApi } from "@opencode-ai/plugin/tui";
 import { createSignal, For, onCleanup, Show } from "solid-js";
 import { listAccounts } from "../../core/accounts";
 import { getBalancingEnabled } from "../../core/priority";
-import { getUsageSnapshot } from "../../core/usage/store";
 import type { Account } from "../../core/types";
+import { getUsageSnapshot } from "../../core/usage/store";
 import type { ProviderUsageSnapshot } from "../../core/usage/types";
 import type { BalancerTuiState } from "../state";
 import { UsageSnapshotBar } from "./usage-display";
 
 export type BalancerSidebarProps = {
-    api: TuiPluginApi;
-    state: BalancerTuiState;
-    openDashboard: () => void;
-    activateAccount: (providerID: string, alias: string) => void;
+	api: TuiPluginApi;
+	state: BalancerTuiState;
+	openDashboard: () => void;
+	activateAccount: (providerID: string, alias: string) => void;
 };
 
 export function BalancerSidebar(props: BalancerSidebarProps) {
-    const theme = () => props.api.theme.current;
-    const [accounts, setAccounts] = createSignal<Account[]>(listAccounts(props.state.db));
-    const [usage, setUsage] = createSignal<Record<string, ProviderUsageSnapshot | undefined>>({});
-    const [balancingEnabled, setBalancingEnabled] = createSignal(getBalancingEnabled(props.state.db));
-    const refreshSidebar = () => {
-        const nextAccounts = listAccounts(props.state.db);
-        setAccounts(nextAccounts);
-        setBalancingEnabled(getBalancingEnabled(props.state.db));
-        setUsage(
-            Object.fromEntries(
-                nextAccounts.map((account) => [
-                    `${account.providerID}/${account.alias}`,
-                    getUsageSnapshot(props.state.db, account.providerID, account.alias),
-                ]),
-            ),
-        );
-    };
-    refreshSidebar();
-    const timer = setInterval(refreshSidebar, 500);
-    onCleanup(() => clearInterval(timer));
-    const Button = (buttonProps: { label: string; onClick: () => void }) => (
-        <box onMouseUp={buttonProps.onClick} paddingLeft={0} paddingRight={0}>
-            <text fg={theme().accent} wrapMode="none" truncate>
-                [ {buttonProps.label} ]
-            </text>
-        </box>
-    );
+	const theme = () => props.api.theme.current;
+	const [accounts, setAccounts] = createSignal<Account[]>(
+		listAccounts(props.state.db),
+	);
+	const [usage, setUsage] = createSignal<
+		Record<string, ProviderUsageSnapshot | undefined>
+	>({});
+	const [balancingEnabled, setBalancingEnabled] = createSignal(
+		getBalancingEnabled(props.state.db),
+	);
+	const refreshSidebar = () => {
+		const nextAccounts = listAccounts(props.state.db);
+		setAccounts(nextAccounts);
+		setBalancingEnabled(getBalancingEnabled(props.state.db));
+		setUsage(
+			Object.fromEntries(
+				nextAccounts.map((account) => [
+					`${account.providerID}/${account.alias}`,
+					getUsageSnapshot(props.state.db, account.providerID, account.alias),
+				]),
+			),
+		);
+	};
+	refreshSidebar();
+	const timer = setInterval(refreshSidebar, 500);
+	onCleanup(() => clearInterval(timer));
+	const Button = (buttonProps: { label: string; onClick: () => void }) => (
+		<box onMouseUp={buttonProps.onClick} paddingLeft={0} paddingRight={0}>
+			<text fg={theme().accent} truncate wrapMode="none">
+				[ {buttonProps.label} ]
+			</text>
+		</box>
+	);
 
-    return (
-        <box flexDirection="column" gap={1}>
-            <box flexDirection="column" gap={0}>
-                <text fg={theme().text} wrapMode="none">
-                    Balancer
-                </text>
-                <text fg={balancingEnabled() ? theme().success : theme().textMuted} wrapMode="none">
-                    {balancingEnabled() ? "ON" : "OFF"}
-                </text>
-                <Button label="dashboard" onClick={() => props.openDashboard()} />
-                <text fg={theme().textMuted} wrapMode="none">
-                    ctrl+b
-                </text>
-            </box>
+	return (
+		<box flexDirection="column" gap={1}>
+			<box flexDirection="column" gap={0}>
+				<text fg={theme().text} wrapMode="none">
+					Balancer
+				</text>
+				<text
+					fg={balancingEnabled() ? theme().success : theme().textMuted}
+					wrapMode="none"
+				>
+					{balancingEnabled() ? "ON" : "OFF"}
+				</text>
+				<Button label="dashboard" onClick={() => props.openDashboard()} />
+				<text fg={theme().textMuted} wrapMode="none">
+					ctrl+b
+				</text>
+			</box>
 
-            <box flexDirection="column" gap={0}>
-                <text fg={theme().textMuted} wrapMode="none">
-                    Accounts
-                </text>
-                <Show
-                    when={accounts().length > 0}
-                    fallback={
-                        <text fg={theme().textMuted} wrapMode="none">
-                            none
-                        </text>
-                    }
-                >
-                    <For each={accounts()}>
-                        {(account) => (
-                            <box flexDirection="column" gap={0} onMouseUp={() => props.activateAccount(account.providerID, account.alias)}>
-                                <text fg={theme().text} wrapMode="none" truncate>
-                                    <span style={{ fg: theme().primary }}>{account.providerID}</span>/
-                                    <span style={{ fg: account.disabled ? theme().textMuted : theme().text }}>
-                                        {account.alias}
-                                    </span>{" "}
-                                    <span style={{ fg: theme().textMuted }}>({account.authType})</span>
-                                </text>
-                                <UsageSnapshotBar
-                                    theme={theme()}
-                                    snapshot={usage()[`${account.providerID}/${account.alias}`]}
-                                />
-                            </box>
-                        )}
-                    </For>
-                </Show>
-            </box>
-        </box>
-    );
+			<box flexDirection="column" gap={0}>
+				<text fg={theme().textMuted} wrapMode="none">
+					Accounts
+				</text>
+				<Show
+					fallback={
+						<text fg={theme().textMuted} wrapMode="none">
+							none
+						</text>
+					}
+					when={accounts().length > 0}
+				>
+					<For each={accounts()}>
+						{(account) => (
+							<box
+								flexDirection="column"
+								gap={0}
+								onMouseUp={() =>
+									props.activateAccount(account.providerID, account.alias)
+								}
+							>
+								<text fg={theme().text} truncate wrapMode="none">
+									<span style={{ fg: theme().primary }}>
+										{account.providerID}
+									</span>
+									/
+									<span
+										style={{
+											fg: account.disabled ? theme().textMuted : theme().text,
+										}}
+									>
+										{account.alias}
+									</span>{" "}
+									<span style={{ fg: theme().textMuted }}>
+										({account.authType})
+									</span>
+								</text>
+								<UsageSnapshotBar
+									snapshot={usage()[`${account.providerID}/${account.alias}`]}
+									theme={theme()}
+								/>
+							</box>
+						)}
+					</For>
+				</Show>
+			</box>
+		</box>
+	);
 }

package/dist/tui/components/status-indicator.tsx

@@ -3,76 +3,95 @@
 import type { TuiPluginApi } from "@opencode-ai/plugin/tui";
 import { createSignal, onCleanup } from "solid-js";
 import { getActiveAccount, getSelectedAccount } from "../../core/accounts";
-import { getBalancingEnabled, resolveActiveSelection, type ActiveSelection } from "../../core/priority";
+import {
+	type ActiveSelection,
+	getBalancingEnabled,
+	resolveActiveSelection,
+} from "../../core/priority";
 import type { Account } from "../../core/types";
 import { getUsageSnapshot } from "../../core/usage/store";
-import { formatBalancerStatus } from "../status-format";
 import type { BalancerTuiState } from "../state";
+import { formatBalancerStatus } from "../status-format";
 import { formatUsageBar } from "../usage-format";
 import { snapshotPercent } from "./usage-display";
 
 export type BalancerStatusIndicatorProps = {
-    api: TuiPluginApi;
-    state: BalancerTuiState;
-    providerID?: string | (() => string | undefined);
+	api: TuiPluginApi;
+	state: BalancerTuiState;
+	providerID?: string | (() => string | undefined);
 };
 
 export function BalancerStatusIndicator(props: BalancerStatusIndicatorProps) {
-    const [selected, setSelected] = createSignal<Account | undefined>();
-    const [sessionActive, setSessionActive] = createSignal<Account | undefined>();
-    const [balancing, setBalancing] = createSignal<ActiveSelection | undefined>();
-    const [usage, setUsage] = createSignal<string | undefined>();
-    const providerID = () => (typeof props.providerID === "function" ? props.providerID() : props.providerID);
-    const refresh = () => {
-        const currentProviderID = providerID();
-        const nextSelected = getSelectedAccount(props.state.db);
-        const nextSessionActive = currentProviderID ? getActiveAccount(props.state.db, currentProviderID) : undefined;
-        const nextBalancing = getBalancingEnabled(props.state.db)
-            ? resolveActiveSelection(props.state.db, undefined, currentProviderID)
-            : undefined;
-        const usageAccount = nextBalancing?.account ?? nextSelected ?? nextSessionActive;
-        setSelected(nextSelected);
-        setSessionActive(nextSessionActive);
-        setBalancing(nextBalancing);
-        setUsage(
-            usageAccount
-                ? formatUsageBar(
-                      snapshotPercent(getUsageSnapshot(props.state.db, usageAccount.providerID, usageAccount.alias)),
-                      4,
-                  )
-                : undefined,
-        );
-    };
-    refresh();
-    const timer = setInterval(refresh, 500);
-    onCleanup(() => clearInterval(timer));
+	const [selected, setSelected] = createSignal<Account | undefined>();
+	const [sessionActive, setSessionActive] = createSignal<Account | undefined>();
+	const [balancing, setBalancing] = createSignal<ActiveSelection | undefined>();
+	const [usage, setUsage] = createSignal<string | undefined>();
+	const providerID = () =>
+		typeof props.providerID === "function"
+			? props.providerID()
+			: props.providerID;
+	const refresh = () => {
+		const currentProviderID = providerID();
+		const nextSelected = getSelectedAccount(props.state.db);
+		const nextSessionActive = currentProviderID
+			? getActiveAccount(props.state.db, currentProviderID)
+			: undefined;
+		const nextBalancing = getBalancingEnabled(props.state.db)
+			? resolveActiveSelection(props.state.db, undefined, currentProviderID)
+			: undefined;
+		const usageAccount =
+			nextBalancing?.account ?? nextSelected ?? nextSessionActive;
+		setSelected(nextSelected);
+		setSessionActive(nextSessionActive);
+		setBalancing(nextBalancing);
+		setUsage(
+			usageAccount
+				? formatUsageBar(
+						snapshotPercent(
+							getUsageSnapshot(
+								props.state.db,
+								usageAccount.providerID,
+								usageAccount.alias,
+							),
+						),
+						4,
+					)
+				: undefined,
+		);
+	};
+	refresh();
+	const timer = setInterval(refresh, 500);
+	onCleanup(() => clearInterval(timer));
 
-    const status = () => {
-        return formatBalancerStatus({
-            selected: selected(),
-            sessionActive: sessionActive(),
-            sessionProviderID: providerID(),
-            balancing: balancing()
-                ? {
-                      providerID: balancing()!.providerID,
-                      alias: balancing()!.account.alias,
-                      modelID: balancing()!.modelID,
-                  }
-                : undefined,
-            usage: usage(),
-        });
-    };
+	const status = () => {
+		return formatBalancerStatus({
+			balancing: balancing()
+				? {
+						alias: balancing()!.account.alias,
+						modelID: balancing()!.modelID,
+						providerID: balancing()!.providerID,
+					}
+				: undefined,
+			selected: selected(),
+			sessionActive: sessionActive(),
+			sessionProviderID: providerID(),
+			usage: usage(),
+		});
+	};
 
-    const color = () => {
-        if (balancing()) return props.api.theme.current.success;
-        if (providerID()) return sessionActive() ? props.api.theme.current.success : props.api.theme.current.textMuted;
-        if (!selected()) return props.api.theme.current.textMuted;
-        return props.api.theme.current.success;
-    };
+	const color = () => {
+		if (balancing()) return props.api.theme.current.success;
+		if (providerID())
+			return sessionActive()
+				? props.api.theme.current.success
+				: props.api.theme.current.textMuted;
+		if (!selected()) return props.api.theme.current.textMuted;
+		return props.api.theme.current.success;
+	};
 
-    return (
-        <text fg={color()} wrapMode="none" truncate>
-            {status()}
-        </text>
-    );
+	return (
+		<text fg={color()} truncate wrapMode="none">
+			{status()}
+		</text>
+	);
 }

package/dist/tui/components/usage-bar.tsx

@@ -3,11 +3,22 @@
 import type { TuiThemeCurrent } from "@opencode-ai/plugin/tui";
 import { formatUsageBar, truncateMiddle } from "../usage-format";
 
-export function UsageBar(props: { theme: TuiThemeCurrent; percent?: number; label: string; muted?: boolean }) {
-    const color = () => (props.muted || props.percent === undefined ? props.theme.textMuted : props.theme.primary);
-    return (
-        <text fg={color()} wrapMode="none" overflow="hidden" truncate>
-            {formatUsageBar(props.percent, 8)} <span style={{ fg: props.theme.textMuted }}>{truncateMiddle(props.label, 34)}</span>
-        </text>
-    );
+export function UsageBar(props: {
+	theme: TuiThemeCurrent;
+	percent?: number;
+	label: string;
+	muted?: boolean;
+}) {
+	const color = () =>
+		props.muted || props.percent === undefined
+			? props.theme.textMuted
+			: props.theme.primary;
+	return (
+		<text fg={color()} overflow="hidden" truncate wrapMode="none">
+			{formatUsageBar(props.percent, 8)}{" "}
+			<span style={{ fg: props.theme.textMuted }}>
+				{truncateMiddle(props.label, 34)}
+			</span>
+		</text>
+	);
 }