opencode-visual-cache 1.3.0 → 1.4.0

package/README.md

@@ -50,7 +50,8 @@
 - **折叠记忆**：折叠状态持久化，重启后保持
 - **语言适配**：自动检测系统语言，支持 `/cache-lang` 运行时切换中/英文，偏好持久化
 - **多币种**：通过 `/cache-currency` 切换货币，费用和节省同步换算
-- **斜杠命令**：`/cache-rate` `/cache-section` `/cache-config` `/cache-lang` 动态配置面板
+- **斜杠命令**：`/cache-session` `/cache-session-back` `/cache-rate` `/cache-section` `/cache-config` `/cache-lang` 动态配置面板
+- **子代理缓存查看**：`/cache-session` 自动扫描并列出子代理，选择一个即可切换面板显示其缓存统计，支持 `/cache-session-back` 返回主会话
 - **已加载技能**：检测 session 中 LLM 调用 `skill` tool 的记录，展示已加载技能名及估算 Token 占用
 
 ---
@@ -100,6 +101,8 @@ npm install -g opencode-visual-cache@latest
 
 | 命令 | 功能 | 使用方式 |
 |------|------|---------|
+| `/cache-session` | 查看子代理缓存统计 | 自动列出子代理供选择，或手动粘贴 Session ID 切换面板数据源 |
+| `/cache-session-back` | 返回主会话统计 | 从子代理缓存视图切回主会话 |
 | `/cache-currency` | 切换货币单位 | 从列表选择货币（USD / CNY / EUR / JPY / GBP / KRW），自动填入默认汇率 |
 | `/cache-rate` | 调整汇率乘数 | 输入自定义汇率（如 `7.2`），用于费用换算 |
 | `/cache-section` | 开关区块与边框 | 独立控制 Token 明细 / 模型与定价 / 估算 Token 分布 / 已加载技能 / 面板边框的显隐 |

package/README_EN.md

@@ -50,7 +50,8 @@ Interested in sub-agent monitoring? Check out [opencode-subagent-magazine](https
 - **Persistent State**: Fold preferences and config remembered across restarts via api.kv
 - **Language**: Auto-detects system locale, with `/cache-lang` for runtime switching between Chinese and English — preference persisted across restarts
 - **Multi-currency**: Switch via `/cache-currency` — costs, savings, and per-million rates convert in real time
-- **Slash Commands**: `/cache-rate` `/cache-section` `/cache-config` `/cache-lang` for live panel configuration
+- **Slash Commands**: `/cache-session` `/cache-session-back` `/cache-rate` `/cache-section` `/cache-config` `/cache-lang` for live panel configuration
+- **Sub-Agent Cache View**: `/cache-session` auto-scans and lists sub-agents; select one to switch the panel stats. Use `/cache-session-back` to return to the main session
 - **Loaded Skills**: Detects `skill` tool calls in the session and displays loaded skill names with estimated token footprint
 
 ---
@@ -100,6 +101,8 @@ The plugin supports slash commands and command palette (`Ctrl + P`) for runtime
 
 | Command | Function | How to use |
 |---------|----------|------------|
+| `/cache-session` | View sub-agent cache stats | Lists sub-agents automatically, or paste a Session ID to switch the panel data source |
+| `/cache-session-back` | Return to main session | Switch back to main session from sub-agent cache view |
 | `/cache-currency` | Switch currency | Pick from a list (USD / CNY / EUR / JPY / GBP / KRW); default exchange rate auto-filled |
 | `/cache-rate` | Adjust exchange rate | Enter a custom rate (e.g. `7.2` for CNY) |
 | `/cache-section` | Toggle sections & border | Independently show/hide Detail, Model & Pricing, Token Distribution, Loaded Skills, or the panel border |

package/dist/_version.d.ts

@@ -1 +1 @@
-export declare const PLUGIN_VERSION = "1.3.0";
+export declare const PLUGIN_VERSION = "1.4.0";

package/dist/_version.js

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

package/dist/index.js

@@ -94,6 +94,14 @@ const ZH_T = {
     secDetail: "明细",
     secModel: "模型",
     secSkills: "已加载技能",
+    balTotal: "总余额:",
+    balNoKey: "未配置 API Key",
+    balLoading: "查询中...",
+    balError: "查询失败",
+    balErr401: "API Key 无效",
+    balErr403: "余额查询被拒绝",
+    balErrEmpty: "未获取到余额数据",
+    balErrTimeout: "查询超时",
 };
 const EN_T = {
     title: "Token Cache",
@@ -125,6 +133,14 @@ const EN_T = {
     secDetail: "Detail",
     secModel: "Model",
     secSkills: "Loaded Skills",
+    balTotal: "Total:",
+    balNoKey: "No API Key set",
+    balLoading: "Fetching...",
+    balError: "Fetch failed",
+    balErr401: "Invalid API Key",
+    balErr403: "Balance request rejected",
+    balErrEmpty: "No balance data",
+    balErrTimeout: "Request timed out",
 };
 // ── color helpers ────────────────────────────────────────────────
 /** Extract { r, g, b } (0–255) from a hex string or RGBA-like object. */
@@ -299,6 +315,44 @@ function estimateTokens(text) {
     const asciiPerToken = jsonLike ? 3.5 : codeLike ? 3.5 : 4;
     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，再换算到目标币种。
+ */
+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;
+}
+/** 货币符号：优先取 /cache-currency 内置映射，未知币种回退为代码。 */
+function balanceSymbol(currency) {
+    const sym = CURRENCIES[currency];
+    return sym ?? currency + " ";
+}
 const CURRENCIES = {
     USD: "$", CNY: "¥", EUR: "€", JPY: "JP¥", GBP: "£", KRW: "₩",
 };
@@ -325,7 +379,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, borderVisible, setBorderVisible, } = props.signals;
+    const { currencySymbol, setCurrencySymbol, exchangeRate, setExchangeRate, langZH, setLangZH, sectionDetail, setSectionDetail, sectionModel, setSectionModel, sectionDist, setSectionDist, sectionSkills, setSectionSkills, sectionBalance, setSectionBalance, balanceRefresh, balanceCurrency, setBalanceCurrency, borderVisible, setBorderVisible, } = props.signals;
     // ── reactive translation (follows langZH signal) ──
     const t = createMemo(() => langZH() ? ZH_T : EN_T);
     // ── scan session messages reactively ──
@@ -352,6 +406,52 @@ function TokenCachePanel(props) {
         hasSkills: false,
     });
     const [refreshTick, setRefreshTick] = createSignal(0);
+    // ── balance state + polling ──────────────────────────────────
+    const [balanceState, setBalanceState] = createSignal({
+        status: "idle", data: null, lastFetch: 0,
+    });
+    // 请求序号：防止定时轮询与手动刷新并发时，慢的旧请求覆盖新结果
+    let balanceSeq = 0;
+    const pollBalance = async () => {
+        const key = props.api.kv.get(`${KV_PREFIX}.ds_key`, "");
+        if (!key) {
+            setBalanceState({ status: "idle", data: null, lastFetch: 0, error: undefined, key: undefined });
+            return;
+        }
+        const now = Date.now();
+        const prev = balanceState();
+        // key 已更换（重新输入）→ 强制重新查询，绕过缓存
+        if (prev.status === "ok" && prev.key === key && now - prev.lastFetch < BALANCE_POLL_MS)
+            return; // cache still fresh
+        const seq = ++balanceSeq;
+        setBalanceState({ ...prev, status: "loading", error: undefined, key });
+        const controller = new AbortController();
+        let timedOut = false;
+        const timer = setTimeout(() => { timedOut = true; controller.abort(); }, 10_000);
+        try {
+            const data = await fetchDeepSeekBalance(key, controller.signal);
+            clearTimeout(timer);
+            if (seq !== balanceSeq)
+                return; // 已被更新的请求取代，丢弃过期结果
+            setBalanceState({ status: "ok", data, lastFetch: Date.now(), error: undefined, key });
+        }
+        catch (err) {
+            clearTimeout(timer);
+            if (seq !== balanceSeq)
+                return;
+            const code = timedOut ? "TIMEOUT" : (err instanceof Error ? err.message : "");
+            // 失败时清空旧数据，避免显示过期余额
+            setBalanceState({ status: "error", data: null, lastFetch: 0, error: code, key });
+        }
+    };
+    // Re-fetch when the API key is (re)configured via /cache-balance-key.
+    // 注意：pollBalance 内部读写 balanceState 信号，若不做 untrack 包裹，
+    // effect 会追踪 balanceState 的变化并与 pollBalance 的 setBalanceState
+    // 形成无限循环（每次重跑都发起新的 fetch 请求）。
+    createEffect(() => {
+        void balanceRefresh();
+        untrack(() => { void pollBalance(); });
+    });
     // ── auto-clear override when the user navigates to a different main session ──
     let lastMainSid = props.sessionId;
     createEffect(() => {
@@ -602,10 +702,14 @@ function TokenCachePanel(props) {
                     setCurrencySymbol(sym);
                 if (typeof rate === "number" && rate > 0)
                     setExchangeRate(rate);
+                const balCur = props.api.kv.get(`${KV_PREFIX}.balance_currency`);
+                if (typeof balCur === "string")
+                    setBalanceCurrency(balCur);
                 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)));
                 setSectionSkills(Boolean(props.api.kv.get(`${KV_PREFIX}.section.skills`, true)));
+                setSectionBalance(Boolean(props.api.kv.get(`${KV_PREFIX}.section.balance`, true)));
                 const bv = props.api.kv.get(`${KV_PREFIX}.border`, true);
                 setBorderVisible(bv !== false);
                 // Restore language preference
@@ -663,7 +767,8 @@ function TokenCachePanel(props) {
         const unsubMsg = props.api.event.on("message.updated", () => { bumpPartVersion(); setRefreshTick(v => v + 1); });
         const unsubSession = props.api.event.on("session.updated", () => { setRefreshTick(v => v + 1); });
         setRefreshTick(v => v + 1);
-        onCleanup(() => { clearTimeout(partTimer); unsubPart(); unsubMsg(); unsubSession(); });
+        const balanceTimer = setInterval(pollBalance, BALANCE_POLL_MS);
+        onCleanup(() => { clearTimeout(partTimer); clearInterval(balanceTimer); unsubPart(); unsubMsg(); unsubSession(); });
     });
     // ── colours ──
     // Pull from the current theme, auto-desaturate if too punchy,
@@ -732,7 +837,36 @@ 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: (() => {
+                                                        const code = balanceState().error;
+                                                        if (code === "401")
+                                                            return t().balErr401;
+                                                        if (code === "403")
+                                                            return t().balErr403;
+                                                        if (code === "EMPTY")
+                                                            return t().balErrEmpty;
+                                                        if (code === "TIMEOUT")
+                                                            return t().balErrTimeout;
+                                                        return t().balError + (code ? ` (${code})` : "");
+                                                    })() })] }) }), _jsx(Show, { when: balanceState().status === "ok" && balanceState().data, children: (() => {
+                                            const list = balanceState().data;
+                                            const pref = balanceCurrency();
+                                            // 偏好币种是 DeepSeek 原生返回的（CNY/USD）→ 直接显示
+                                            const native = pref ? list.find(x => x.currency === pref) : undefined;
+                                            if (native) {
+                                                return (_jsx("text", { fg: pal().text, children: justify(t().balTotal, balanceSymbol(native.currency) + native.total) }));
+                                            }
+                                            // 非原生币种（EUR/JPY/GBP/KRW…）→ 取第一条余额按汇率换算
+                                            const base = list[0];
+                                            const baseAmt = parseFloat(base.total);
+                                            const converted = Number.isFinite(baseAmt)
+                                                ? convertBalance(pref || base.currency, exchangeRate(), baseAmt, base.currency)
+                                                : baseAmt;
+                                            const shown = pref && base.currency !== pref
+                                                ? converted.toLocaleString("en-US", { maximumFractionDigits: 2 })
+                                                : base.total;
+                                            return (_jsx("text", { fg: pal().text, children: justify(t().balTotal, balanceSymbol(pref || base.currency) + shown) }));
+                                        })() })] })] })] })] }));
 }
 // ---------------------------------------------------------------------------
 // Plugin entry
@@ -764,6 +898,9 @@ const tui = async (api) => {
     const [sectionModel, setSectionModel] = createSignal(true);
     const [sectionDist, setSectionDist] = createSignal(true);
     const [sectionSkills, setSectionSkills] = createSignal(true);
+    const [sectionBalance, setSectionBalance] = createSignal(true);
+    const [balanceRefresh, setBalanceRefresh] = createSignal(0);
+    const [balanceCurrency, setBalanceCurrency] = createSignal("");
     const [borderVisible, setBorderVisible] = createSignal(true);
     const [langZH, setLangZH] = createSignal(LANG_ZH);
     const [overrideSessionId, setOverrideSessionId] = createSignal(undefined);
@@ -775,6 +912,9 @@ const tui = async (api) => {
         sectionModel, setSectionModel,
         sectionDist, setSectionDist,
         sectionSkills, setSectionSkills,
+        sectionBalance, setSectionBalance,
+        balanceRefresh, setBalanceRefresh,
+        balanceCurrency, setBalanceCurrency,
         borderVisible, setBorderVisible,
         overrideSessionId, setOverrideSessionId,
     };
@@ -796,6 +936,9 @@ const tui = async (api) => {
                         const defRate = DEFAULT_RATES[opt.value] ?? 1;
                         api.kv.set(`${KV_PREFIX}.currency`, sym);
                         api.kv.set(`${KV_PREFIX}.rate`, defRate);
+                        // 同步余额显示币种偏好：CNY/USD 原生直显，其余币种按汇率换算
+                        api.kv.set(`${KV_PREFIX}.balance_currency`, opt.value);
+                        signals.setBalanceCurrency(opt.value);
                         signals.setCurrencySymbol(sym);
                         signals.setExchangeRate(defRate);
                         api.ui.toast({ message: `Currency: ${opt.value} (${sym}), rate: ${defRate}` });
@@ -830,12 +973,14 @@ const tui = async (api) => {
                 const modelOn = Boolean(api.kv.get(`${KV_PREFIX}.section.model`, true));
                 const distOn = Boolean(api.kv.get(`${KV_PREFIX}.section.dist`, true));
                 const skillsOn = Boolean(api.kv.get(`${KV_PREFIX}.section.skills`, true));
+                const balanceOn = Boolean(api.kv.get(`${KV_PREFIX}.section.balance`, true));
                 const borderOn = Boolean(api.kv.get(`${KV_PREFIX}.border`, true));
                 dialog?.replace(() => (_jsx(api.ui.DialogSelect, { title: "Toggle Section", options: [
                         { title: `Token Detail    [${detailOn ? "ON" : "OFF"}]`, value: "detail" },
                         { 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: `Panel Border    [${borderOn ? "ON" : "OFF"}]`, value: "border" },
                     ], onSelect: (opt) => {
                         if (opt.value === "border") {
@@ -856,6 +1001,8 @@ const tui = async (api) => {
                                 signals.setSectionDist(!cur);
                             if (opt.value === "skills")
                                 signals.setSectionSkills(!cur);
+                            if (opt.value === "balance")
+                                signals.setSectionBalance(!cur);
                             api.ui.toast({ message: `${opt.value} section ${!cur ? "shown" : "hidden"}` });
                         }
                         dialog?.clear();
@@ -874,9 +1021,10 @@ const tui = async (api) => {
                 const model = Boolean(api.kv.get(`${KV_PREFIX}.section.model`, true));
                 const dist = Boolean(api.kv.get(`${KV_PREFIX}.section.dist`, true));
                 const skills = Boolean(api.kv.get(`${KV_PREFIX}.section.skills`, true));
+                const balance = Boolean(api.kv.get(`${KV_PREFIX}.section.balance`, true));
                 api.ui.toast({
                     title: "Cache Panel Config",
-                    message: `Currency: ${sym}  |  Rate: ${rate}  |  Detail: ${detail ? "ON" : "OFF"}  |  Model: ${model ? "ON" : "OFF"}  |  Dist: ${dist ? "ON" : "OFF"}  |  Skills: ${skills ? "ON" : "OFF"}`,
+                    message: `Currency: ${sym}  |  Rate: ${rate}  |  Detail: ${detail ? "ON" : "OFF"}  |  Model: ${model ? "ON" : "OFF"}  |  Dist: ${dist ? "ON" : "OFF"}  |  Skills: ${skills ? "ON" : "OFF"}  |  Balance: ${balance ? "ON" : "OFF"}`,
                     duration: 8000,
                 });
                 dialog?.clear();
@@ -901,6 +1049,50 @@ 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" },
+            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..." });
+                        }
+                        else {
+                            api.ui.toast({ message: zh ? "API Key 已清除" : "API Key cleared" });
+                        }
+                        dialog?.clear();
+                    }, onCancel: () => dialog?.clear() })));
+            },
+        },
         {
             title: "Cache: Debug Skills Detection",
             value: "cache.debug-skills",