opencode-visual-cache 1.4.0 → 1.5.0

package/dist/index.js

@@ -1,6 +1,7 @@
 import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "@opentui/solid/jsx-runtime";
 import { createMemo, createSignal, createEffect, onMount, onCleanup, Show, untrack } from "solid-js";
 import { PLUGIN_VERSION } from "./_version";
+import { balanceProviders, getBalanceProvider, maskKey, matchBalanceProvider } from "./balance-providers";
 // ── terminal-width helpers ────────────────────────────────────────
 // CJK characters occupy 2 terminal columns; padEnd/padStart count
 // string length (=1 per char), which breaks alignment with mixed text.
@@ -95,7 +96,7 @@ const ZH_T = {
     secModel: "模型",
     secSkills: "已加载技能",
     balTotal: "总余额:",
-    balNoKey: "未配置 API Key",
+    balNoKey: "未配置 {p} API Key",
     balLoading: "查询中...",
     balError: "查询失败",
     balErr401: "API Key 无效",
@@ -134,7 +135,7 @@ const EN_T = {
     secModel: "Model",
     secSkills: "Loaded Skills",
     balTotal: "Total:",
-    balNoKey: "No API Key set",
+    balNoKey: "{p} API Key not set",
     balLoading: "Fetching...",
     balError: "Fetch failed",
     balErr401: "Invalid API Key",
@@ -316,27 +317,6 @@ function estimateTokens(text) {
     return Math.max(1, Math.ceil(ascii / asciiPerToken + cjk / 1.0));
 }
 const BALANCE_POLL_MS = 5 * 60 * 1000; // 5 minutes
-async function fetchDeepSeekBalance(apiKey, signal) {
-    const res = await fetch("https://api.deepseek.com/user/balance", {
-        headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
-        signal,
-    });
-    if (!res.ok) {
-        if (res.status === 401)
-            throw new Error("401");
-        if (res.status === 402 || res.status === 403)
-            throw new Error("403");
-        throw new Error(String(res.status));
-    }
-    const json = await res.json();
-    const infos = json.balance_infos ?? [];
-    if (infos.length === 0)
-        throw new Error("EMPTY");
-    return infos.map((info) => ({
-        currency: info.currency ?? "CNY",
-        total: info.total_balance ?? "0",
-    }));
-}
 /**
  * 将余额从来源币种换算为目标币种。
  * DEFAULT_RATES 以 USD=1 为基准：先折算为 USD，再换算到目标币种。
@@ -348,6 +328,29 @@ function convertBalance(target, targetRate, amount, from) {
     const usd = from === "USD" ? amount : amount / fromRate;
     return target === "USD" ? usd : usd * targetRate;
 }
+/**
+ * 从 OpenCode 已认证的 provider 读取 API key 作为余额查询的自动兜底。
+ * 匹配复用前缀逻辑：先精确匹配 id，再前缀匹配（如 moonshotai-cn → moonshot）。
+ * key 来源：auth.json（provider.key）或配置（provider.options.apiKey）。
+ * 仅当手动配置的 key 缺失时使用；读取失败或未匹配返回空串。
+ */
+function findOpencodeKey(api, provider) {
+    try {
+        const provs = api.state.provider;
+        // 大小写不敏感：精确匹配 id，否则前缀匹配（如 moonshotai-cn → moonshot）
+        const id = provider.id.toLowerCase();
+        const hit = provs.find((p) => p.id.toLowerCase() === id) ?? provs.find((p) => p.id.toLowerCase().startsWith(id));
+        if (!hit)
+            return "";
+        const k = typeof hit.key === "string" ? hit.key : "";
+        if (k)
+            return k;
+        return typeof hit.options?.apiKey === "string" ? hit.options.apiKey : "";
+    }
+    catch {
+        return "";
+    }
+}
 /** 货币符号：优先取 /cache-currency 内置映射，未知币种回退为代码。 */
 function balanceSymbol(currency) {
     const sym = CURRENCIES[currency];
@@ -379,7 +382,7 @@ function TokenCachePanel(props) {
     const [skillsOpen, setSkillsOpen] = createSignal(true);
     let boxEl;
     // ── shared signals (de-structured so internal code is unchanged) ──
-    const { currencySymbol, setCurrencySymbol, exchangeRate, setExchangeRate, langZH, setLangZH, sectionDetail, setSectionDetail, sectionModel, setSectionModel, sectionDist, setSectionDist, sectionSkills, setSectionSkills, sectionBalance, setSectionBalance, balanceRefresh, balanceCurrency, setBalanceCurrency, borderVisible, setBorderVisible, } = props.signals;
+    const { currencySymbol, setCurrencySymbol, exchangeRate, setExchangeRate, langZH, setLangZH, sectionDetail, setSectionDetail, sectionModel, setSectionModel, sectionDist, setSectionDist, sectionSkills, setSectionSkills, sectionBalance, setSectionBalance, balanceRefresh, balanceProviderId, setBalanceProviderId, autoBalance, setAutoBalance, balanceCurrency, setBalanceCurrency, borderVisible, setBorderVisible, } = props.signals;
     // ── reactive translation (follows langZH signal) ──
     const t = createMemo(() => langZH() ? ZH_T : EN_T);
     // ── scan session messages reactively ──
@@ -412,8 +415,13 @@ function TokenCachePanel(props) {
     });
     // 请求序号：防止定时轮询与手动刷新并发时，慢的旧请求覆盖新结果
     let balanceSeq = 0;
+    // 当前 provider 显示名
+    const providerName = createMemo(() => getBalanceProvider(balanceProviderId()).name);
     const pollBalance = async () => {
-        const key = props.api.kv.get(`${KV_PREFIX}.ds_key`, "");
+        const provider = getBalanceProvider(balanceProviderId());
+        // 手动配置的 key 优先；缺失时自动复用 OpenCode 已认证的 key（auth.json / config）
+        const key = props.api.kv.get(`${KV_PREFIX}.balance.${provider.id}.key`, "")
+            || findOpencodeKey(props.api, provider);
         if (!key) {
             setBalanceState({ status: "idle", data: null, lastFetch: 0, error: undefined, key: undefined });
             return;
@@ -429,7 +437,7 @@ function TokenCachePanel(props) {
         let timedOut = false;
         const timer = setTimeout(() => { timedOut = true; controller.abort(); }, 10_000);
         try {
-            const data = await fetchDeepSeekBalance(key, controller.signal);
+            const data = await provider.fetchBalance(key, controller.signal);
             clearTimeout(timer);
             if (seq !== balanceSeq)
                 return; // 已被更新的请求取代，丢弃过期结果
@@ -452,6 +460,39 @@ function TokenCachePanel(props) {
         void balanceRefresh();
         untrack(() => { void pollBalance(); });
     });
+    // 自动切换当前会话的 provider（前缀匹配）。手动切换会关闭此行为。
+    // 直接追踪 messages 取最后一条 assistant 消息的 providerID——
+    // 不依赖 session.model 的响应式更新（模型切换时该链路可能不触发重算）。
+    createEffect(() => {
+        if (!autoBalance())
+            return;
+        const sid = props.signals.overrideSessionId() ?? props.sessionId;
+        const msgs = props.api.state.session.messages(sid);
+        let pid = "";
+        for (let i = msgs.length - 1; i >= 0; i--) {
+            const m = msgs[i];
+            if (m.role === "assistant" && m.providerID) {
+                pid = m.providerID;
+                break;
+            }
+        }
+        // 会话尚无 assistant 消息（新会话 / 刚切换模型未对话 / 消息未加载）
+        // → 回退到会话级模型元数据，反映当前正在使用的 provider
+        if (!pid) {
+            try {
+                const session = props.api.state.session.get(sid);
+                pid = session?.model?.providerID ?? "";
+            }
+            catch { /* ignore */ }
+        }
+        if (!pid)
+            return;
+        const hit = matchBalanceProvider(pid);
+        if (hit && hit.id !== balanceProviderId()) {
+            setBalanceProviderId(hit.id);
+            props.signals.setBalanceRefresh(props.signals.balanceRefresh() + 1);
+        }
+    });
     // ── auto-clear override when the user navigates to a different main session ──
     let lastMainSid = props.sessionId;
     createEffect(() => {
@@ -705,6 +746,25 @@ function TokenCachePanel(props) {
                 const balCur = props.api.kv.get(`${KV_PREFIX}.balance_currency`);
                 if (typeof balCur === "string")
                     setBalanceCurrency(balCur);
+                // Restore balance provider (fall back to default when unknown)
+                const savedProvider = props.api.kv.get(`${KV_PREFIX}.balance.provider`);
+                if (typeof savedProvider === "string" && balanceProviders.some((p) => p.id === savedProvider)) {
+                    setBalanceProviderId(savedProvider);
+                }
+                // Restore auto-switch (default on)
+                const savedAuto = props.api.kv.get(`${KV_PREFIX}.balance.auto`);
+                if (typeof savedAuto === "boolean")
+                    setAutoBalance(savedAuto);
+                // Migrate legacy DeepSeek key (cache_panel.ds_key → cache_panel.balance.deepseek.key)
+                const legacyKey = props.api.kv.get(`${KV_PREFIX}.ds_key`, "");
+                if (legacyKey) {
+                    const dsKey = props.api.kv.get(`${KV_PREFIX}.balance.deepseek.key`, "");
+                    if (!dsKey)
+                        props.api.kv.set(`${KV_PREFIX}.balance.deepseek.key`, legacyKey);
+                    props.api.kv.set(`${KV_PREFIX}.ds_key`, "");
+                }
+                // 恢复的 provider 可能与默认值不同，强制重新查询
+                props.signals.setBalanceRefresh(props.signals.balanceRefresh() + 1);
                 setSectionDetail(Boolean(props.api.kv.get(`${KV_PREFIX}.section.detail`, true)));
                 setSectionModel(Boolean(props.api.kv.get(`${KV_PREFIX}.section.model`, true)));
                 setSectionDist(Boolean(props.api.kv.get(`${KV_PREFIX}.section.dist`, true)));
@@ -837,7 +897,7 @@ function TokenCachePanel(props) {
                                                 const maxLabel = Math.max(4, panelWidth() - gutter() - rightW - 1);
                                                 const label = truncateVisual(sk.name, maxLabel);
                                                 return (_jsx("text", { fg: pal().muted, children: justify(label, fmt(sk.tokens), t().tok) }));
-                                            }) })] }) }), _jsxs(Show, { when: sectionBalance(), children: [_jsx("text", { fg: pal().muted, children: sep() }), _jsx(Show, { when: balanceState().status === "idle", children: _jsxs("text", { fg: pal().muted, children: [_jsx("span", { style: { fg: pal().muted }, children: "> " }), _jsx("span", { children: t().balNoKey })] }) }), _jsx(Show, { when: balanceState().status === "loading", children: _jsxs("text", { fg: pal().muted, children: [_jsx("span", { style: { fg: pal().muted }, children: "> " }), _jsx("span", { children: t().balLoading })] }) }), _jsx(Show, { when: balanceState().status === "error", children: _jsxs("text", { fg: pal().error, children: [_jsx("span", { style: { fg: pal().muted }, children: "> " }), _jsx("span", { children: (() => {
+                                            }) })] }) }), _jsxs(Show, { when: sectionBalance(), children: [_jsx("text", { fg: pal().muted, children: sep() }), _jsx(Show, { when: balanceState().status === "idle", children: _jsxs("text", { fg: pal().muted, children: [_jsx("span", { style: { fg: pal().muted }, children: "> " }), _jsx("span", { children: t().balNoKey.replace("{p}", providerName()) })] }) }), _jsx(Show, { when: balanceState().status === "loading", children: _jsxs("text", { fg: pal().muted, children: [_jsx("span", { style: { fg: pal().muted }, children: "> " }), _jsx("span", { children: t().balLoading })] }) }), _jsx(Show, { when: balanceState().status === "error", children: _jsxs("text", { fg: pal().error, children: [_jsx("span", { style: { fg: pal().muted }, children: "> " }), _jsx("span", { children: (() => {
                                                         const code = balanceState().error;
                                                         if (code === "401")
                                                             return t().balErr401;
@@ -900,6 +960,8 @@ const tui = async (api) => {
     const [sectionSkills, setSectionSkills] = createSignal(true);
     const [sectionBalance, setSectionBalance] = createSignal(true);
     const [balanceRefresh, setBalanceRefresh] = createSignal(0);
+    const [balanceProviderId, setBalanceProviderId] = createSignal("deepseek");
+    const [autoBalance, setAutoBalance] = createSignal(true);
     const [balanceCurrency, setBalanceCurrency] = createSignal("");
     const [borderVisible, setBorderVisible] = createSignal(true);
     const [langZH, setLangZH] = createSignal(LANG_ZH);
@@ -914,6 +976,8 @@ const tui = async (api) => {
         sectionSkills, setSectionSkills,
         sectionBalance, setSectionBalance,
         balanceRefresh, setBalanceRefresh,
+        balanceProviderId, setBalanceProviderId,
+        autoBalance, setAutoBalance,
         balanceCurrency, setBalanceCurrency,
         borderVisible, setBorderVisible,
         overrideSessionId, setOverrideSessionId,
@@ -921,6 +985,46 @@ const tui = async (api) => {
     api.slots.register(createSidebarSlot(api, signals));
     // ── slash commands for runtime config ──
     const KV_PREFIX = "cache_panel";
+    /** 菜单中 provider 选项标题：标注 key 来源（手动配置 / OpenCode 自动复用 / 未配置）。 */
+    const providerOptionTitle = (p, current) => {
+        const zh = langZH();
+        const hasManual = !!api.kv.get(`${KV_PREFIX}.balance.${p.id}.key`, "");
+        const hasAuto = !hasManual && !!findOpencodeKey(api, p);
+        const mark = hasManual
+            ? (zh ? "（用户 key）" : " (user key)")
+            : hasAuto
+                ? (zh ? "（OpenCode）" : " (OpenCode)")
+                : (zh ? "（未配置）" : " (not set)");
+        return p.name + mark + (current && p.id === current ? " *" : "");
+    };
+    /** 弹出指定 provider 的 API Key 输入框（脱敏预填；空清除 / 含 * 保留原 key / 新 key 实时刷新）。 */
+    const promptBalanceKey = (dialog, provider) => {
+        const zh = langZH();
+        const current = api.kv.get(`${KV_PREFIX}.balance.${provider.id}.key`, "");
+        const masked = maskKey(current);
+        dialog?.replace(() => (_jsx(api.ui.DialogPrompt, { title: provider.name, description: () => _jsx("text", { children: zh ? `输入 ${provider.name} API Key 以显示账户余额（留空清除）` : `Enter your ${provider.name} API key to show account balance (leave empty to clear)` }), placeholder: provider.keyPlaceholder ?? "sk-...", value: masked, onConfirm: (val) => {
+                const input = val.trim();
+                let key;
+                if (input === "") {
+                    key = "";
+                }
+                else if (input.includes("*")) {
+                    key = current;
+                }
+                else {
+                    key = input;
+                }
+                api.kv.set(`${KV_PREFIX}.balance.${provider.id}.key`, key);
+                setBalanceRefresh(v => v + 1);
+                if (key) {
+                    api.ui.toast({ message: zh ? "API Key 已保存，正在查询余额..." : "API Key saved, fetching balance..." });
+                }
+                else {
+                    api.ui.toast({ message: zh ? "API Key 已清除" : "API Key cleared" });
+                }
+                dialog?.clear();
+            }, onCancel: () => dialog?.clear() })));
+    };
     api.command?.register(() => [
         {
             title: "Cache: Set Currency",
@@ -980,7 +1084,7 @@ const tui = async (api) => {
                         { title: `Model & Pricing [${modelOn ? "ON" : "OFF"}]`, value: "model" },
                         { title: `Token Dist.     [${distOn ? "ON" : "OFF"}]`, value: "dist" },
                         { title: `Loaded Skills   [${skillsOn ? "ON" : "OFF"}]`, value: "skills" },
-                        { title: `DS Balance      [${balanceOn ? "ON" : "OFF"}]`, value: "balance" },
+                        { title: `Balance         [${balanceOn ? "ON" : "OFF"}]`, value: "balance" },
                         { title: `Panel Border    [${borderOn ? "ON" : "OFF"}]`, value: "border" },
                     ], onSelect: (opt) => {
                         if (opt.value === "border") {
@@ -1050,47 +1154,79 @@ const tui = async (api) => {
             },
         },
         {
-            title: "Cache: Set DeepSeek API Key",
-            value: "cache.balance.key",
-            description: "Set or update the DeepSeek API key for balance display",
-            slash: { name: "cache-balance-key" },
+            title: "Cache: Switch Balance Provider",
+            value: "cache.balance",
+            description: "切换余额提供商 / 自动切换当前会话提供商 | Switch balance provider / auto-switch session provider",
+            slash: { name: "cache-balance" },
             onSelect: (dialog) => {
                 const zh = langZH();
-                const current = api.kv.get(`${KV_PREFIX}.ds_key`, "");
-                // 已保存的 key 以脱敏形式预填：保留 "sk-" 前缀 + 头 5 尾 5 字符，中间用 * 填充
-                const maskKey = (k) => {
-                    if (!k)
-                        return "";
-                    const prefix = k.startsWith("sk-") ? "sk-" : "";
-                    const body = prefix ? k.slice(3) : k;
-                    if (body.length <= 10)
-                        return prefix + body.slice(0, 5) + "*".repeat(Math.max(3, body.length - 5));
-                    return prefix + body.slice(0, 5) + "*".repeat(Math.max(3, body.length - 10)) + body.slice(-5);
-                };
-                const masked = maskKey(current);
-                dialog?.replace(() => (_jsx(api.ui.DialogPrompt, { title: zh ? "DeepSeek API Key" : "DeepSeek API Key", description: () => _jsx("text", { children: zh ? "输入 DeepSeek API Key 以显示账户余额（留空清除）" : "Enter your DeepSeek API key to show account balance (leave empty to clear)" }), placeholder: "sk-...", value: masked, onConfirm: (val) => {
-                        const input = val.trim();
-                        // 空 → 清除；含 * （脱敏占位符残留）→ 视为未修改，保留原 key；否则为新 key
-                        let key;
-                        if (input === "") {
-                            key = "";
-                        }
-                        else if (input.includes("*")) {
-                            key = current;
-                        }
-                        else {
-                            key = input;
-                        }
-                        api.kv.set(`${KV_PREFIX}.ds_key`, key);
-                        setBalanceRefresh(v => v + 1);
-                        if (key) {
-                            api.ui.toast({ message: zh ? "API Key 已保存，正在查询余额..." : "API Key saved, fetching balance..." });
+                const current = signals.balanceProviderId();
+                const auto = signals.autoBalance();
+                const autoLabel = auto
+                    ? (zh ? "自动切换提供商 [开]" : "Auto-switch provider [ON]")
+                    : (zh ? "自动切换提供商 [关]" : "Auto-switch provider [OFF]");
+                dialog?.replace(() => (_jsx(api.ui.DialogSelect, { title: zh ? "余额提供商 / 自动切换" : "Balance Provider / Auto-switch", options: [
+                        {
+                            title: autoLabel,
+                            value: "__auto__",
+                        },
+                        ...balanceProviders.map((p) => ({
+                            title: providerOptionTitle(p, current),
+                            value: p.id,
+                        })),
+                    ], onSelect: (opt) => {
+                        if (opt.value === "__auto__") {
+                            const next = !auto;
+                            api.kv.set(`${KV_PREFIX}.balance.auto`, next);
+                            signals.setAutoBalance(next);
+                            api.ui.toast({ message: zh ? `自动切换余额提供商: ${next ? "开" : "关"}` : `Auto-switch balance provider: ${next ? "ON" : "OFF"}` });
+                            dialog?.clear();
                         }
                         else {
-                            api.ui.toast({ message: zh ? "API Key 已清除" : "API Key cleared" });
+                            const provider = getBalanceProvider(opt.value);
+                            // 手动切换会关闭自动切换
+                            api.kv.set(`${KV_PREFIX}.balance.provider`, provider.id);
+                            api.kv.set(`${KV_PREFIX}.balance.auto`, false);
+                            signals.setBalanceProviderId(provider.id);
+                            signals.setAutoBalance(false);
+                            // 切换后立即按新 provider 刷新显示（无 key 时显示 idle，避免残留上一 provider 余额）
+                            signals.setBalanceRefresh(signals.balanceRefresh() + 1);
+                            const hasKey = !!api.kv.get(`${KV_PREFIX}.balance.${provider.id}.key`, "");
+                            if (!hasKey) {
+                                // 未配置 key → 进入设置流程（对话框保持打开等待输入）
+                                promptBalanceKey(dialog, provider);
+                            }
+                            else {
+                                api.ui.toast({ message: zh ? `余额提供商: ${provider.name}（自动切换已关闭）` : `Balance provider: ${provider.name} (auto-switch off)` });
+                                dialog?.clear();
+                            }
                         }
-                        dialog?.clear();
-                    }, onCancel: () => dialog?.clear() })));
+                    } })));
+            },
+        },
+        {
+            title: "Cache: Set Balance API Key",
+            value: "cache.balance.key",
+            description: "Select a provider and set its API key for balance display",
+            slash: { name: "cache-balance-key" },
+            onSelect: (dialog) => {
+                const zh = langZH();
+                // 步骤 1：选择 provider
+                dialog?.replace(() => (_jsx(api.ui.DialogSelect, { title: zh ? "选择余额提供商" : "Select Balance Provider", options: balanceProviders.map((p) => ({
+                        title: providerOptionTitle(p),
+                        value: p.id,
+                    })), onSelect: (opt) => {
+                        const provider = getBalanceProvider(opt.value);
+                        // 手动指定 provider 会关闭自动切换
+                        api.kv.set(`${KV_PREFIX}.balance.provider`, provider.id);
+                        api.kv.set(`${KV_PREFIX}.balance.auto`, false);
+                        signals.setBalanceProviderId(provider.id);
+                        signals.setAutoBalance(false);
+                        // 切换后立即刷新显示（防止取消输入时残留上一 provider 的余额）
+                        signals.setBalanceRefresh(signals.balanceRefresh() + 1);
+                        // 步骤 2：输入 key
+                        promptBalanceKey(dialog, provider);
+                    } })));
             },
         },
         {