Skip to content

feat: 新增 DeepSeek 余额查询显示 #19

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Hotakus merged 2 commits into from
Aug 8, 2026
Merged

feat: 新增 DeepSeek 余额查询显示 #19

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
586fece
feat: 新增 DeepSeek 余额查询显示
troublewind Aug 8, 2026
b2f7c8e
fix: 修复余额轮询的循环请求、并发竞态与超时问题
troublewind Aug 8, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
255 changes: 253 additions & 2 deletions src/index.tsx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -113,6 +113,14 @@ const ZH_T = {
secDetail: "明细",
secModel: "模型",
secSkills: "已加载技能",
balTotal: "总余额:",
balNoKey: "未配置 API Key",
balLoading: "查询中...",
balError: "查询失败",
balErr401: "API Key 无效",
balErr403: "余额查询被拒绝",
balErrEmpty:"未获取到余额数据",
balErrTimeout: "查询超时",
} as const

const EN_T = {
Expand Down Expand Up @@ -145,6 +153,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",
} as const

// ── color helpers ────────────────────────────────────────────────
Expand Down Expand Up @@ -331,6 +347,64 @@ interface TokenDist {
stepCost: number
}

// ---------------------------------------------------------------------------
// DeepSeek Balance
// ---------------------------------------------------------------------------

interface DeepSeekBalance {
currency: string
total: string
}

interface BalanceState {
status: "idle" | "loading" | "ok" | "error"
data: DeepSeekBalance[] | null
lastFetch: number
error?: string
key?: string // 上次成功/尝试查询所用的 key，用于检测 key 是否更换
}

const BALANCE_POLL_MS = 5 * 60 * 1000 // 5 minutes

async function fetchDeepSeekBalance(apiKey: string, signal?: AbortSignal): Promise<DeepSeekBalance[]> {
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() 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 Error("EMPTY")
return infos.map((info) => ({
currency: info.currency ?? "CNY",
total: info.total_balance ?? "0",
}))
}

/**
* 将余额从来源币种换算为目标币种。
* DEFAULT_RATES 以 USD=1 为基准：先折算为 USD，再换算到目标币种。
*/
function convertBalance(target: string, targetRate: number, amount: number, from: string): number {
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: string): string {
const sym = CURRENCIES[currency]
return sym ?? currency + " "
}

// ---------------------------------------------------------------------------
// Sidebar component
// ---------------------------------------------------------------------------
Expand All @@ -353,6 +427,14 @@ interface PanelSignals {
setSectionDist: (v: boolean) => void
sectionSkills: () => boolean
setSectionSkills: (v: boolean) => void
sectionBalance: () => boolean
setSectionBalance: (v: boolean) => void
/** Increment to force a DeepSeek balance re-fetch. */
balanceRefresh: () => number
setBalanceRefresh: (v: number) => void
/** Preferred currency code for balance display (CNY / USD / …). Empty = first entry. */
balanceCurrency: () => string
setBalanceCurrency: (v: string) => void
borderVisible: () => boolean
setBorderVisible: (v: boolean) => void
/** When set, the panel renders stats for this session instead of the main one. */
Expand Down Expand Up @@ -404,6 +486,9 @@ function TokenCachePanel(props: {
sectionModel, setSectionModel,
sectionDist, setSectionDist,
sectionSkills, setSectionSkills,
sectionBalance, setSectionBalance,
balanceRefresh,
balanceCurrency, setBalanceCurrency,
borderVisible, setBorderVisible,
} = props.signals

Expand Down Expand Up @@ -437,6 +522,48 @@ function TokenCachePanel(props: {
})
const [refreshTick, setRefreshTick] = createSignal(0)

// ── balance state + polling ──────────────────────────────────
const [balanceState, setBalanceState] = createSignal<BalanceState>({
status: "idle", data: null, lastFetch: 0,
})
// 请求序号：防止定时轮询与手动刷新并发时，慢的旧请求覆盖新结果
let balanceSeq = 0

const pollBalance = async () => {
const key = props.api.kv.get<string>(`${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() })
})
Comment on lines +562 to +565

@Hotakus Hotakus Aug 8, 2026

Copy link
Copy Markdown
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

感谢你提交的新功能添加，但是我注意到，在 pollBalance() 中，引用了 balanceState()，balanceState() 的状态变化是否会导致 effect 被循环请求？

@troublewind troublewind Aug 8, 2026

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

感谢指出问题！已自查修复并推送。

根因：createEffect 直接调用 async 的 pollBalance()，而 async 函数 await 前的同步段读取并写入了 balanceState 信号，被 effect 追踪后与 setBalanceState 形成无限循环，且每轮重跑都发起新请求。

本次修复：

  • 用 untrack 包裹 pollBalance，使 effect 仅追踪 balanceRefresh（key 变更时才重新查询），循环消除
  • 移除 onMount 中重复的首次请求
  • 增加请求序号，防止定时轮询与手动刷新并发时旧请求覆盖新结果
  • 失败时清空旧数据，避免显示过期余额
  • AbortController 10s 超时，超时显示"查询超时"

麻烦再次 review ，谢谢！


// ── auto-clear override when the user navigates to a different main session ──
let lastMainSid = props.sessionId
createEffect(() => {
Expand Down Expand Up @@ -634,10 +761,13 @@ function TokenCachePanel(props: {
const rate = props.api.kv.get<number>(`${KV_PREFIX}.rate`)
if (typeof sym === "string") setCurrencySymbol(sym)
if (typeof rate === "number" && rate > 0) setExchangeRate(rate)
const balCur = props.api.kv.get<string>(`${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<boolean>(`${KV_PREFIX}.border`, true)
setBorderVisible(bv !== false)
// Restore language preference
Expand Down Expand Up @@ -692,7 +822,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 ──
Expand Down Expand Up @@ -977,6 +1108,65 @@ function TokenCachePanel(props: {
</Show>
</Show>
</Show>

{/* ── DeepSeek balance (single line) ── */}
<Show when={sectionBalance()}>
<text fg={pal().muted}>{sep()}</text>
<Show when={balanceState().status === "idle"}>
<text fg={pal().muted}>
<span style={{ fg: pal().muted }}>{"> "}</span>
<span>{t().balNoKey}</span>
</text>
</Show>
<Show when={balanceState().status === "loading"}>
<text fg={pal().muted}>
<span style={{ fg: pal().muted }}>{"> "}</span>
<span>{t().balLoading}</span>
</text>
</Show>
<Show when={balanceState().status === "error"}>
<text fg={pal().error}>
<span style={{ fg: pal().muted }}>{"> "}</span>
<span>{(() => {
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})` : "")
})()}</span>
</text>
</Show>
<Show when={balanceState().status === "ok" && balanceState().data}>
{(() => {
const list = balanceState().data!
const pref = balanceCurrency()
// 偏好币种是 DeepSeek 原生返回的（CNY/USD）→ 直接显示
const native = pref ? list.find(x => x.currency === pref) : undefined
if (native) {
return (
<text fg={pal().text}>
{justify(t().balTotal, balanceSymbol(native.currency) + native.total)}
</text>
)
}
// 非原生币种（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 (
<text fg={pal().text}>
{justify(t().balTotal, balanceSymbol(pref || base.currency) + shown)}
</text>
)
})()}
</Show>
</Show>
</Show>
</Show>
</box>
Expand Down Expand Up @@ -1022,6 +1212,9 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
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<string | undefined>(undefined)
Expand All @@ -1034,6 +1227,9 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
sectionModel, setSectionModel,
sectionDist, setSectionDist,
sectionSkills, setSectionSkills,
sectionBalance, setSectionBalance,
balanceRefresh, setBalanceRefresh,
balanceCurrency, setBalanceCurrency,
borderVisible, setBorderVisible,
overrideSessionId, setOverrideSessionId,
}
Expand Down Expand Up @@ -1061,6 +1257,9 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
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}` })
Expand Down Expand Up @@ -1105,6 +1304,7 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
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(() => (
<api.ui.DialogSelect
Expand All @@ -1114,6 +1314,7 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
{ 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) => {
Expand All @@ -1130,6 +1331,7 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
if (opt.value === "model") signals.setSectionModel(!cur)
if (opt.value === "dist") 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()
Expand All @@ -1150,9 +1352,10 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
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()
Expand Down Expand Up @@ -1183,6 +1386,54 @@ const tui: TuiPlugin = async (api: TuiPluginApi) => {
))
},
},
{
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<string>(`${KV_PREFIX}.ds_key`, "")
// 已保存的 key 以脱敏形式预填：保留 "sk-" 前缀 + 头 5 尾 5 字符，中间用 * 填充
const maskKey = (k: string): string => {
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(() => (
<api.ui.DialogPrompt
title={zh ? "DeepSeek API Key" : "DeepSeek API Key"}
description={() => <text>{zh ? "输入 DeepSeek API Key 以显示账户余额（留空清除）" : "Enter your DeepSeek API key to show account balance (leave empty to clear)"}</text>}
placeholder="sk-..."
value={masked}
onConfirm={(val) => {
const input = val.trim()
// 空 → 清除；含 * （脱敏占位符残留）→ 视为未修改，保留原 key；否则为新 key
let key: string
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",
Expand Down
Loading