opencode-visual-cache 1.4.0 → 1.5.0

package/dist/tui.js

@@ -13,7 +13,109 @@ import { createElement as _$createElement } from "@opentui/solid";
 import { createMemo, createSignal, createEffect, onMount, onCleanup, Show, untrack } from "solid-js";
 
 // src/_version.ts
-var PLUGIN_VERSION = "1.4.0";
+var PLUGIN_VERSION = "1.5.0";
+
+// src/balance-providers.ts
+var BalanceError = class extends Error {
+};
+var siliconflowProvider = {
+  id: "siliconflow",
+  name: "SiliconFlow",
+  keyPlaceholder: "sk-...",
+  async fetchBalance(apiKey, signal) {
+    const res = await fetch("https://api.siliconflow.cn/v1/user/info", {
+      headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
+      signal
+    });
+    if (!res.ok) {
+      if (res.status === 401) throw new BalanceError("401");
+      if (res.status === 403) throw new BalanceError("403");
+      throw new BalanceError(String(res.status));
+    }
+    const json = await res.json();
+    const total = json.data?.totalBalance ?? json.data?.balance;
+    if (typeof total === "undefined" || total === null) throw new BalanceError("EMPTY");
+    return [{ currency: "CNY", total: String(total) }];
+  }
+};
+var deepseekProvider = {
+  id: "deepseek",
+  name: "DeepSeek",
+  keyPlaceholder: "sk-...",
+  async fetchBalance(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 BalanceError("401");
+      if (res.status === 402 || res.status === 403) throw new BalanceError("403");
+      throw new BalanceError(String(res.status));
+    }
+    const json = await res.json();
+    const infos = json.balance_infos ?? [];
+    if (infos.length === 0) throw new BalanceError("EMPTY");
+    return infos.map((info) => ({
+      currency: info.currency ?? "CNY",
+      total: info.total_balance ?? "0"
+    }));
+  }
+};
+var openrouterProvider = {
+  id: "openrouter",
+  name: "OpenRouter",
+  keyPlaceholder: "sk-or-...",
+  async fetchBalance(apiKey, signal) {
+    const res = await fetch("https://openrouter.ai/api/v1/credits", {
+      headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
+      signal
+    });
+    if (!res.ok) {
+      if (res.status === 401 || res.status === 403) throw new BalanceError("403");
+      throw new BalanceError(String(res.status));
+    }
+    const json = await res.json();
+    const credits = json.data?.total_credits;
+    const usage = json.data?.total_usage;
+    if (typeof credits !== "number" || typeof usage !== "number") throw new BalanceError("EMPTY");
+    return [{ currency: "USD", total: (credits - usage).toFixed(2) }];
+  }
+};
+var moonshotProvider = {
+  id: "moonshot",
+  name: "Moonshot",
+  keyPlaceholder: "sk-...",
+  async fetchBalance(apiKey, signal) {
+    const res = await fetch("https://api.moonshot.cn/v1/users/me/balance", {
+      headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
+      signal
+    });
+    if (!res.ok) {
+      if (res.status === 401) throw new BalanceError("401");
+      if (res.status === 403) throw new BalanceError("403");
+      throw new BalanceError(String(res.status));
+    }
+    const json = await res.json();
+    const balance = json.data?.available_balance;
+    if (typeof balance === "undefined" || balance === null) throw new BalanceError("EMPTY");
+    return [{ currency: "CNY", total: String(balance) }];
+  }
+};
+var balanceProviders = [deepseekProvider, siliconflowProvider, openrouterProvider, moonshotProvider];
+function getBalanceProvider(id) {
+  return balanceProviders.find((p) => p.id === id) ?? balanceProviders[0] ?? deepseekProvider;
+}
+function matchBalanceProvider(providerId) {
+  const id = providerId.toLowerCase();
+  const exact = balanceProviders.find((p) => p.id.toLowerCase() === id);
+  if (exact) return exact;
+  return balanceProviders.find((p) => id.startsWith(p.id.toLowerCase()));
+}
+function maskKey(k) {
+  if (!k) return "";
+  if (k.length <= 10) return k.slice(0, 5) + "*".repeat(Math.max(3, k.length - 5));
+  return k.slice(0, 5) + "*".repeat(Math.max(3, k.length - 10)) + k.slice(-5);
+}
 
 // src/index.tsx
 function charColumns(c) {
@@ -91,7 +193,7 @@ var ZH_T = {
   secModel: "\u6A21\u578B",
   secSkills: "\u5DF2\u52A0\u8F7D\u6280\u80FD",
   balTotal: "\u603B\u4F59\u989D:",
-  balNoKey: "\u672A\u914D\u7F6E API Key",
+  balNoKey: "\u672A\u914D\u7F6E {p} API Key",
   balLoading: "\u67E5\u8BE2\u4E2D...",
   balError: "\u67E5\u8BE2\u5931\u8D25",
   balErr401: "API Key \u65E0\u6548",
@@ -130,7 +232,7 @@ var 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",
@@ -249,33 +351,25 @@ function estimateTokens(text) {
   return Math.max(1, Math.ceil(ascii / asciiPerToken + cjk / 1));
 }
 var BALANCE_POLL_MS = 5 * 60 * 1e3;
-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"
-  }));
-}
 function convertBalance(target, targetRate, amount, from) {
   if (from === target) return amount;
   const fromRate = DEFAULT_RATES[from] ?? 1;
   const usd = from === "USD" ? amount : amount / fromRate;
   return target === "USD" ? usd : usd * targetRate;
 }
+function findOpencodeKey(api, provider) {
+  try {
+    const provs = api.state.provider;
+    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 "";
+  }
+}
 function balanceSymbol(currency) {
   const sym = CURRENCIES[currency];
   return sym ?? currency + " ";
@@ -330,6 +424,10 @@ function TokenCachePanel(props) {
     sectionBalance,
     setSectionBalance,
     balanceRefresh,
+    balanceProviderId,
+    setBalanceProviderId,
+    autoBalance,
+    setAutoBalance,
     balanceCurrency,
     setBalanceCurrency,
     borderVisible,
@@ -388,8 +486,10 @@ function TokenCachePanel(props) {
     lastFetch: 0
   });
   let balanceSeq = 0;
+  const providerName = createMemo(() => getBalanceProvider(balanceProviderId()).name);
   const pollBalance = async () => {
-    const key = props.api.kv.get(`${KV_PREFIX}.ds_key`, "");
+    const provider = getBalanceProvider(balanceProviderId());
+    const key = props.api.kv.get(`${KV_PREFIX}.balance.${provider.id}.key`, "") || findOpencodeKey(props.api, provider);
     if (!key) {
       setBalanceState({
         status: "idle",
@@ -417,7 +517,7 @@ function TokenCachePanel(props) {
       controller.abort();
     }, 1e4);
     try {
-      const data2 = await fetchDeepSeekBalance(key, controller.signal);
+      const data2 = await provider.fetchBalance(key, controller.signal);
       clearTimeout(timer);
       if (seq !== balanceSeq) return;
       setBalanceState({
@@ -446,6 +546,32 @@ function TokenCachePanel(props) {
       void pollBalance();
     });
   });
+  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;
+      }
+    }
+    if (!pid) {
+      try {
+        const session = props.api.state.session.get(sid);
+        pid = session?.model?.providerID ?? "";
+      } catch {
+      }
+    }
+    if (!pid) return;
+    const hit = matchBalanceProvider(pid);
+    if (hit && hit.id !== balanceProviderId()) {
+      setBalanceProviderId(hit.id);
+      props.signals.setBalanceRefresh(props.signals.balanceRefresh() + 1);
+    }
+  });
   let lastMainSid = props.sessionId;
   createEffect(() => {
     const sid = props.sessionId;
@@ -512,7 +638,7 @@ function TokenCachePanel(props) {
     const freshTotal = input + read, sessionHitRate = freshTotal > 0 ? read / freshTotal * 100 : 0;
     const model = mid.split("/").pop() ?? mid, hasPricing = inputRate > 0 || cacheReadRate > 0 || cacheWriteRate > 0;
     const hasTrendData = prevMsgHitRate >= 0 && lastMsgHitRate >= 0;
-    const trend = hasTrendData ? lastMsgHitRate - prevMsgHitRate : 0, providerName = pid || "";
+    const trend = hasTrendData ? lastMsgHitRate - prevMsgHitRate : 0, providerName2 = pid || "";
     const distData = untrack(() => {
       let dist = {
         system: 0,
@@ -637,7 +763,7 @@ function TokenCachePanel(props) {
       hasData: read > 0 || write > 0 || input > 0 || output > 0 || cost > 0,
       trend,
       hasTrendData,
-      providerName,
+      providerName: providerName2,
       sessionHitRate,
       dist: distData.finalDist,
       hasDistData: distData.finalHasDist,
@@ -689,6 +815,19 @@ function TokenCachePanel(props) {
         if (typeof rate === "number" && rate > 0) setExchangeRate(rate);
         const balCur = props.api.kv.get(`${KV_PREFIX}.balance_currency`);
         if (typeof balCur === "string") setBalanceCurrency(balCur);
+        const savedProvider = props.api.kv.get(`${KV_PREFIX}.balance.provider`);
+        if (typeof savedProvider === "string" && balanceProviders.some((p) => p.id === savedProvider)) {
+          setBalanceProviderId(savedProvider);
+        }
+        const savedAuto = props.api.kv.get(`${KV_PREFIX}.balance.auto`);
+        if (typeof savedAuto === "boolean") setAutoBalance(savedAuto);
+        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`, "");
+        }
+        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)));
@@ -1428,7 +1567,7 @@ function TokenCachePanel(props) {
                     _$insertNode(_el$52, _el$53);
                     _$insertNode(_el$52, _el$55);
                     _$insertNode(_el$53, _$createTextNode(`> `));
-                    _$insert(_el$55, () => t().balNoKey);
+                    _$insert(_el$55, () => t().balNoKey.replace("{p}", providerName()));
                     _$effect((_p$) => {
                       var _v$9 = pal().muted, _v$0 = {
                         fg: pal().muted
@@ -1585,6 +1724,8 @@ var 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);
@@ -1608,6 +1749,10 @@ var tui = async (api) => {
     setSectionBalance,
     balanceRefresh,
     setBalanceRefresh,
+    balanceProviderId,
+    setBalanceProviderId,
+    autoBalance,
+    setAutoBalance,
     balanceCurrency,
     setBalanceCurrency,
     borderVisible,
@@ -1617,6 +1762,56 @@ var tui = async (api) => {
   };
   api.slots.register(createSidebarSlot(api, signals));
   const KV_PREFIX = "cache_panel";
+  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 ? "\uFF08\u7528\u6237 key\uFF09" : " (user key)" : hasAuto ? zh ? "\uFF08OpenCode\uFF09" : " (OpenCode)" : zh ? "\uFF08\u672A\u914D\u7F6E\uFF09" : " (not set)";
+    return p.name + mark + (current && p.id === current ? " *" : "");
+  };
+  const promptBalanceKey = (dialog, provider) => {
+    const zh = langZH();
+    const current = api.kv.get(`${KV_PREFIX}.balance.${provider.id}.key`, "");
+    const masked = maskKey(current);
+    dialog?.replace(() => _$createComponent(api.ui.DialogPrompt, {
+      get title() {
+        return provider.name;
+      },
+      description: () => (() => {
+        var _el$93 = _$createElement("text");
+        _$insert(_el$93, () => zh ? `\u8F93\u5165 ${provider.name} API Key \u4EE5\u663E\u793A\u8D26\u6237\u4F59\u989D\uFF08\u7559\u7A7A\u6E05\u9664\uFF09` : `Enter your ${provider.name} API key to show account balance (leave empty to clear)`);
+        return _el$93;
+      })(),
+      get placeholder() {
+        return 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 \u5DF2\u4FDD\u5B58\uFF0C\u6B63\u5728\u67E5\u8BE2\u4F59\u989D..." : "API Key saved, fetching balance..."
+          });
+        } else {
+          api.ui.toast({
+            message: zh ? "API Key \u5DF2\u6E05\u9664" : "API Key cleared"
+          });
+        }
+        dialog?.clear();
+      },
+      onCancel: () => dialog?.clear()
+    }));
+  };
   api.command?.register(() => [{
     title: "Cache: Set Currency",
     value: "cache.currency",
@@ -1660,9 +1855,9 @@ var tui = async (api) => {
       dialog?.replace(() => _$createComponent(api.ui.DialogPrompt, {
         title: "Exchange Rate",
         description: () => (() => {
-          var _el$93 = _$createElement("text");
-          _$insertNode(_el$93, _$createTextNode(`Enter the exchange rate from USD to your currency (e.g. 7.2 for CNY)`));
-          return _el$93;
+          var _el$94 = _$createElement("text");
+          _$insertNode(_el$94, _$createTextNode(`Enter the exchange rate from USD to your currency (e.g. 7.2 for CNY)`));
+          return _el$94;
         })(),
         placeholder: "1.0",
         get value() {
@@ -1710,7 +1905,7 @@ var tui = async (api) => {
           title: `Loaded Skills   [${skillsOn ? "ON" : "OFF"}]`,
           value: "skills"
         }, {
-          title: `DS Balance      [${balanceOn ? "ON" : "OFF"}]`,
+          title: `Balance         [${balanceOn ? "ON" : "OFF"}]`,
           value: "balance"
         }, {
           title: `Panel Border    [${borderOn ? "ON" : "OFF"}]`,
@@ -1793,56 +1988,83 @@ var tui = async (api) => {
       }));
     }
   }, {
-    title: "Cache: Set DeepSeek API Key",
-    value: "cache.balance.key",
-    description: "Set or update the DeepSeek API key for balance display",
+    title: "Cache: Switch Balance Provider",
+    value: "cache.balance",
+    description: "\u5207\u6362\u4F59\u989D\u63D0\u4F9B\u5546 / \u81EA\u52A8\u5207\u6362\u5F53\u524D\u4F1A\u8BDD\u63D0\u4F9B\u5546 | Switch balance provider / auto-switch session provider",
     slash: {
-      name: "cache-balance-key"
+      name: "cache-balance"
     },
     onSelect: (dialog) => {
       const zh = langZH();
-      const current = api.kv.get(`${KV_PREFIX}.ds_key`, "");
-      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(() => _$createComponent(api.ui.DialogPrompt, {
-        title: zh ? "DeepSeek API Key" : "DeepSeek API Key",
-        description: () => (() => {
-          var _el$95 = _$createElement("text");
-          _$insert(_el$95, zh ? "\u8F93\u5165 DeepSeek API Key \u4EE5\u663E\u793A\u8D26\u6237\u4F59\u989D\uFF08\u7559\u7A7A\u6E05\u9664\uFF09" : "Enter your DeepSeek API key to show account balance (leave empty to clear)");
-          return _el$95;
-        })(),
-        placeholder: "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}.ds_key`, key);
-          setBalanceRefresh((v) => v + 1);
-          if (key) {
+      const current = signals.balanceProviderId();
+      const auto = signals.autoBalance();
+      const autoLabel = auto ? zh ? "\u81EA\u52A8\u5207\u6362\u63D0\u4F9B\u5546 [\u5F00]" : "Auto-switch provider [ON]" : zh ? "\u81EA\u52A8\u5207\u6362\u63D0\u4F9B\u5546 [\u5173]" : "Auto-switch provider [OFF]";
+      dialog?.replace(() => _$createComponent(api.ui.DialogSelect, {
+        title: zh ? "\u4F59\u989D\u63D0\u4F9B\u5546 / \u81EA\u52A8\u5207\u6362" : "Balance Provider / Auto-switch",
+        get options() {
+          return [{
+            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 ? "API Key \u5DF2\u4FDD\u5B58\uFF0C\u6B63\u5728\u67E5\u8BE2\u4F59\u989D..." : "API Key saved, fetching balance..."
+              message: zh ? `\u81EA\u52A8\u5207\u6362\u4F59\u989D\u63D0\u4F9B\u5546: ${next ? "\u5F00" : "\u5173"}` : `Auto-switch balance provider: ${next ? "ON" : "OFF"}`
             });
+            dialog?.clear();
           } else {
-            api.ui.toast({
-              message: zh ? "API Key \u5DF2\u6E05\u9664" : "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);
+            signals.setBalanceRefresh(signals.balanceRefresh() + 1);
+            const hasKey = !!api.kv.get(`${KV_PREFIX}.balance.${provider.id}.key`, "");
+            if (!hasKey) {
+              promptBalanceKey(dialog, provider);
+            } else {
+              api.ui.toast({
+                message: zh ? `\u4F59\u989D\u63D0\u4F9B\u5546: ${provider.name}\uFF08\u81EA\u52A8\u5207\u6362\u5DF2\u5173\u95ED\uFF09` : `Balance provider: ${provider.name} (auto-switch off)`
+              });
+              dialog?.clear();
+            }
           }
-          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();
+      dialog?.replace(() => _$createComponent(api.ui.DialogSelect, {
+        title: zh ? "\u9009\u62E9\u4F59\u989D\u63D0\u4F9B\u5546" : "Select Balance Provider",
+        get options() {
+          return balanceProviders.map((p) => ({
+            title: providerOptionTitle(p),
+            value: p.id
+          }));
         },
-        onCancel: () => dialog?.clear()
+        onSelect: (opt) => {
+          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);
+          signals.setBalanceRefresh(signals.balanceRefresh() + 1);
+          promptBalanceKey(dialog, provider);
+        }
       }));
     }
   }, {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-visual-cache",
-  "version": "1.4.0",
+  "version": "1.5.0",
   "description": "OpenCode TUI plugin displaying real-time token cache hit rate in the sidebar",
   "type": "module",
   "types": "dist/index.d.ts",

package/src/_version.ts

@@ -1,2 +1,2 @@
 // auto-generated
-export const PLUGIN_VERSION="1.4.0";
+export const PLUGIN_VERSION="1.5.0";

package/src/balance-providers.ts

@@ -0,0 +1,153 @@
+// ---------------------------------------------------------------------------
+// Balance providers — pluggable account-balance query adapters.
+// ---------------------------------------------------------------------------
+
+/** 归一化后的余额条目——显示层与具体 provider 解耦。 */
+export interface BalanceEntry {
+  currency: string   // 原生币种（CNY/USD…），复用现有汇率换算
+  total: string      // 余额字符串
+}
+
+/** provider 统一错误：message 即错误码（401/403/EMPTY/…），显示层直接展示。 */
+export class BalanceError extends Error {}
+
+/** 可插拔的余额 provider 适配器。 */
+export interface BalanceProvider {
+  id: string                    // 唯一标识，同时用作 KV key 命名空间
+  name: string                  // 显示名（专有名词，无需 i18n）
+  keyPlaceholder?: string       // key 输入框占位（如 "sk-..."）
+  fetchBalance(apiKey: string, signal?: AbortSignal): Promise<BalanceEntry[]>
+}
+
+const siliconflowProvider: BalanceProvider = {
+  id: "siliconflow",
+  name: "SiliconFlow",
+  keyPlaceholder: "sk-...",
+  async fetchBalance(apiKey, signal) {
+    // 国内站 api.siliconflow.cn（CNY）；国际站为 api.siliconflow.com（USD）
+    const res = await fetch("https://api.siliconflow.cn/v1/user/info", {
+      headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
+      signal,
+    })
+    if (!res.ok) {
+      if (res.status === 401) throw new BalanceError("401")
+      if (res.status === 403) throw new BalanceError("403")
+      throw new BalanceError(String(res.status))
+    }
+    const json = await res.json() as {
+      status?: boolean
+      data?: {
+        balance?: string | number
+        chargeBalance?: string | number
+        totalBalance?: string | number
+      }
+    }
+    // totalBalance 为总余额（含充值+赠送），缺失时回退 balance
+    const total = json.data?.totalBalance ?? json.data?.balance
+    if (typeof total === "undefined" || total === null) throw new BalanceError("EMPTY")
+    return [{ currency: "CNY", total: String(total) }]
+  },
+}
+
+const deepseekProvider: BalanceProvider = {
+  id: "deepseek",
+  name: "DeepSeek",
+  keyPlaceholder: "sk-...",
+  async fetchBalance(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 BalanceError("401")
+      if (res.status === 402 || res.status === 403) throw new BalanceError("403")
+      throw new BalanceError(String(res.status))
+    }
+    const json = await res.json() as {
+      is_available?: boolean
+      balance_infos?: { currency: string; total_balance: string; granted_balance: string; topped_up_balance: string }[]
+    }
+    const infos = json.balance_infos ?? []
+    if (infos.length === 0) throw new BalanceError("EMPTY")
+    return infos.map((info) => ({
+      currency: info.currency ?? "CNY",
+      total: info.total_balance ?? "0",
+    }))
+  },
+}
+
+const openrouterProvider: BalanceProvider = {
+  id: "openrouter",
+  name: "OpenRouter",
+  keyPlaceholder: "sk-or-...",
+  async fetchBalance(apiKey, signal) {
+    // 官方文档标注需 Management key，实测普通 API key 亦可查询账户余额
+    const res = await fetch("https://openrouter.ai/api/v1/credits", {
+      headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
+      signal,
+    })
+    if (!res.ok) {
+      if (res.status === 401 || res.status === 403) throw new BalanceError("403")
+      throw new BalanceError(String(res.status))
+    }
+    const json = await res.json() as {
+      data?: { total_credits?: number; total_usage?: number }
+    }
+    const credits = json.data?.total_credits
+    const usage = json.data?.total_usage
+    if (typeof credits !== "number" || typeof usage !== "number") throw new BalanceError("EMPTY")
+    // 剩余额度 = 充值总额 - 已用
+    return [{ currency: "USD", total: (credits - usage).toFixed(2) }]
+  },
+}
+
+const moonshotProvider: BalanceProvider = {
+  id: "moonshot",
+  name: "Moonshot",
+  keyPlaceholder: "sk-...",
+  async fetchBalance(apiKey, signal) {
+    // 国内站 api.moonshot.cn（CNY）；国际站 api.moonshot.ai（USD）
+    const res = await fetch("https://api.moonshot.cn/v1/users/me/balance", {
+      headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
+      signal,
+    })
+    if (!res.ok) {
+      if (res.status === 401) throw new BalanceError("401")
+      if (res.status === 403) throw new BalanceError("403")
+      throw new BalanceError(String(res.status))
+    }
+    const json = await res.json() as {
+      data?: { available_balance?: string | number }
+    }
+    const balance = json.data?.available_balance
+    if (typeof balance === "undefined" || balance === null) throw new BalanceError("EMPTY")
+    return [{ currency: "CNY", total: String(balance) }]
+  },
+}
+
+/** 已注册的 provider 列表（按需追加新适配器）。 */
+export const balanceProviders: BalanceProvider[] = [deepseekProvider, siliconflowProvider, openrouterProvider, moonshotProvider]
+
+/** 按 id 取 provider；未知 id 回退到第一个。 */
+export function getBalanceProvider(id: string): BalanceProvider {
+  return balanceProviders.find((p) => p.id === id) ?? balanceProviders[0] ?? deepseekProvider
+}
+
+/**
+ * 按 OpenCode providerID 匹配余额 provider。
+ * 先精确匹配，再按前缀匹配（如 moonshotai-cn → moonshot）；未命中返回 undefined。
+ * 比较不区分大小写，容忍 providerID 的大小写变体。
+ */
+export function matchBalanceProvider(providerId: string): BalanceProvider | undefined {
+  const id = providerId.toLowerCase()
+  const exact = balanceProviders.find((p) => p.id.toLowerCase() === id)
+  if (exact) return exact
+  return balanceProviders.find((p) => id.startsWith(p.id.toLowerCase()))
+}
+
+/** key 脱敏：保留头 5 尾 5 字符，中间用 * 填充。 */
+export function maskKey(k: string): string {
+  if (!k) return ""
+  if (k.length <= 10) return k.slice(0, 5) + "*".repeat(Math.max(3, k.length - 5))
+  return k.slice(0, 5) + "*".repeat(Math.max(3, k.length - 10)) + k.slice(-5)
+}