opencode-visual-cache 1.6.1 → 1.7.0-beta.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (62) hide show
  1. package/LICENSE +21 -21
  2. package/README.md +247 -247
  3. package/README_EN.md +247 -247
  4. package/dist/_version.d.ts +1 -1
  5. package/dist/_version.js +1 -1
  6. package/dist/core/color.d.ts +37 -0
  7. package/dist/core/color.js +108 -0
  8. package/dist/core/currency.d.ts +17 -0
  9. package/dist/core/currency.js +43 -0
  10. package/dist/core/estimate.d.ts +1 -0
  11. package/dist/core/estimate.js +40 -0
  12. package/dist/core/format.d.ts +15 -0
  13. package/dist/core/format.js +90 -0
  14. package/dist/core/index.d.ts +5 -0
  15. package/dist/core/index.js +5 -0
  16. package/dist/core/types.d.ts +17 -0
  17. package/dist/core/types.js +1 -0
  18. package/dist/index.js +159 -815
  19. package/dist/panel/TokenCachePanel.d.ts +10 -0
  20. package/dist/panel/TokenCachePanel.js +549 -0
  21. package/dist/panel/panel-api.d.ts +119 -0
  22. package/dist/panel/panel-api.js +1 -0
  23. package/dist/tui.js +349 -264
  24. package/dist/v2/commands.d.ts +10 -0
  25. package/dist/v2/commands.js +490 -0
  26. package/dist/v2/data.d.ts +27 -0
  27. package/dist/v2/data.js +103 -0
  28. package/dist/v2/index.d.ts +4 -0
  29. package/dist/v2/index.js +163 -0
  30. package/dist/v2/sidebar.d.ts +6 -0
  31. package/dist/v2/sidebar.js +36 -0
  32. package/dist/v2/status.d.ts +14 -0
  33. package/dist/v2/status.js +83 -0
  34. package/dist/v2/theme.d.ts +8 -0
  35. package/dist/v2/theme.js +16 -0
  36. package/dist/v2/types.d.ts +219 -0
  37. package/dist/v2/types.js +6 -0
  38. package/dist/v2/v2-panel-api.d.ts +8 -0
  39. package/dist/v2/v2-panel-api.js +176 -0
  40. package/dist/v2.js +2941 -0
  41. package/package.json +70 -67
  42. package/src/_version.ts +1 -1
  43. package/src/balance-providers.ts +153 -153
  44. package/src/core/color.ts +108 -0
  45. package/src/core/currency.ts +46 -0
  46. package/src/core/estimate.ts +36 -0
  47. package/src/core/format.ts +81 -0
  48. package/src/core/index.ts +5 -0
  49. package/src/core/types.ts +17 -0
  50. package/src/i18n.ts +380 -380
  51. package/src/index.tsx +1035 -1980
  52. package/src/panel/TokenCachePanel.tsx +776 -0
  53. package/src/panel/panel-api.ts +104 -0
  54. package/src/server.ts +10 -10
  55. package/src/v2/commands.ts +471 -0
  56. package/src/v2/data.ts +112 -0
  57. package/src/v2/index.tsx +184 -0
  58. package/src/v2/sidebar.tsx +69 -0
  59. package/src/v2/status.tsx +96 -0
  60. package/src/v2/theme.ts +19 -0
  61. package/src/v2/types.ts +159 -0
  62. package/src/v2/v2-panel-api.ts +177 -0
package/src/index.tsx CHANGED
@@ -1,1980 +1,1035 @@
1
- /** @jsxImportSource @opentui/solid */
2
-
3
- import type { JSX } from "@opentui/solid"
4
- import type {
5
- TuiPlugin,
6
- TuiPluginApi,
7
- TuiSlotContext,
8
- TuiSlotPlugin,
9
- TuiPluginModule,
10
- TuiThemeCurrent,
11
- TuiDialogStack,
12
- TuiPromptRef,
13
- } from "@opencode-ai/plugin/tui"
14
- import type { UserMessage, AssistantMessage, Message } from "@opencode-ai/sdk"
15
- import type {
16
- Part,
17
- TextPart,
18
- ToolPart,
19
- FilePart,
20
- ReasoningPart,
21
- } from "@opencode-ai/sdk/v2"
22
- import { createMemo, createSignal, createEffect, onMount, onCleanup, Show, untrack } from "solid-js"
23
- import { PLUGIN_VERSION } from "./_version"
24
- import { balanceProviders, getBalanceProvider, maskKey, matchBalanceProvider, type BalanceEntry, type BalanceProvider } from "./balance-providers"
25
- import { LANG_META, createT, detectLang, type LangCode } from "./i18n"
26
-
27
- // ---------------------------------------------------------------------------
28
- // Helpers
29
- // ---------------------------------------------------------------------------
30
-
31
- // Bun / Node globals — available at runtime in the OpenCode TUI process
32
- declare const process: { env: Record<string, string | undefined> } | undefined
33
-
34
- // ── terminal-width helpers ────────────────────────────────────────
35
- // CJK characters occupy 2 terminal columns; padEnd/padStart count
36
- // string length (=1 per char), which breaks alignment with mixed text.
37
-
38
- function charColumns(c: string): number {
39
- const code = c.codePointAt(0) ?? 0
40
- if (code < 0x20) return 0 // control
41
- if (code < 0x7F) return 1 // ASCII
42
- if (code < 0xA0) return 0 // C1 controls
43
- // East-Asian wide / fullwidth ranges
44
- if ((code >= 0x1100 && code <= 0x115F) || // Hangul Jamo
45
- (code >= 0x2E80 && code <= 0xA4CF) || // CJK Radicals … Yi
46
- (code >= 0xAC00 && code <= 0xD7A3) || // Hangul
47
- (code >= 0xF900 && code <= 0xFAFF) || // CJK Compat
48
- (code >= 0xFE10 && code <= 0xFE6F) || // Vertical / Compat
49
- (code >= 0xFF01 && code <= 0xFF60) || // Fullwidth
50
- (code >= 0xFFE0 && code <= 0xFFE6) || // Fullwidth signs
51
- (code >= 0x1F300 && code <= 0x1F64F) || // Misc Symbols (emoji)
52
- (code >= 0x20000 && code <= 0x3FFFD)) // SIP / TIP
53
- return 2
54
- return 1
55
- }
56
-
57
- function visualWidth(s: string): number {
58
- let w = 0; for (const c of s) w += charColumns(c); return w
59
- }
60
-
61
- function visualPadEnd(s: string, cols: number): string {
62
- const pad = cols - visualWidth(s)
63
- return pad > 0 ? s + " ".repeat(pad) : s
64
- }
65
-
66
- /** Truncate `s` to fit within `maxCols` visual columns, appending "…" when cut. */
67
- function truncateVisual(s: string, maxCols: number): string {
68
- if (visualWidth(s) <= maxCols) return s
69
- let result = "", w = 0
70
- for (const c of s) {
71
- const cw = charColumns(c)
72
- if (w + cw > maxCols - 1) { result += "\u2026"; break }
73
- result += c; w += cw
74
- }
75
- return result
76
- }
77
-
78
- // ── language ──────────────────────────────────────────────────────
79
- // 语言初始化:环境变量 CACHE_TUI_LANG 覆盖 → 否则按系统 locale 自动检测。
80
- // 用户通过 /cache-lang 设置的偏好会在 KV 就绪后优先覆盖(见 tui() 内恢复逻辑)。
81
-
82
- const DEBUG_LANG = typeof process !== "undefined" ? process.env?.CACHE_TUI_LANG : undefined
83
- const INIT_LANG: LangCode = DEBUG_LANG !== undefined && LANG_META.some((m) => m.code === DEBUG_LANG)
84
- ? (DEBUG_LANG as LangCode)
85
- : detectLang()
86
-
87
- // ── color helpers ────────────────────────────────────────────────
88
-
89
- /** Extract { r, g, b } (0–255) from a hex string or RGBA-like object. */
90
- function rgb(raw: unknown): { r: number; g: number; b: number } | null {
91
- if (typeof raw === "string" && raw.startsWith("#")) {
92
- const h = raw.slice(1)
93
- return {
94
- r: parseInt(h.slice(0, 2), 16),
95
- g: parseInt(h.slice(2, 4), 16),
96
- b: parseInt(h.slice(4, 6), 16),
97
- }
98
- }
99
- if (raw && typeof raw === "object") {
100
- const o = raw as Record<string, unknown>
101
- if (typeof o.r === "number" && typeof o.g === "number" && typeof o.b === "number") {
102
- // RGBA channels may be 0-1 floats; detect and upscale.
103
- const scale = o.r > 1 || o.g > 1 || o.b > 1 ? 1 : 255
104
- return {
105
- r: Math.round(o.r * scale),
106
- g: Math.round(o.g * scale),
107
- b: Math.round(o.b * scale),
108
- }
109
- }
110
- }
111
- return null
112
- }
113
-
114
- /** HSL saturation of an RGB color (0–1). */
115
- function saturation(r: number, g: number, b: number): number {
116
- const max = Math.max(r, g, b) / 255
117
- const min = Math.min(r, g, b) / 255
118
- const delta = max - min
119
- if (delta === 0) return 0
120
- const L = (max + min) / 2
121
- return L <= 0.5 ? delta / (max + min) : delta / (2 - max - min)
122
- }
123
-
124
- /**
125
- * If the colour's saturation exceeds `maxSat`, pull it toward grey
126
- * until saturation drops to maxSat. Returns a hex string.
127
- */
128
- function desaturateTo(raw: unknown, maxSat: number, fallback: string): string {
129
- const c = rgb(raw)
130
- if (!c) return fallback
131
- const sat = saturation(c.r, c.g, c.b)
132
- if (sat <= maxSat) {
133
- // already muted — return as hex
134
- return "#" + [c.r, c.g, c.b].map((v) => v.toString(16).padStart(2, "0")).join("")
135
- }
136
- /**
137
- * Binary search for the optimal grey-mix ratio α (0…1).
138
- *
139
- * 12 iterations 1/2^12 1/4096 resolution. The downstream RGB
140
- * channels are only 0–255 (8 bit), so 8 iterations (1/256) would
141
- * technically suffice; 12 is intentionally over-budget — the extra
142
- * precision costs almost nothing and guarantees the saturation probe
143
- * converges to within a fraction of an 8‑bit step, eliminating
144
- * colour banding in edge cases.
145
- */
146
- // Bt.601 luma (perceptual brightness used as the grey anchor)
147
- const luma = c.r * 0.299 + c.g * 0.587 + c.b * 0.114
148
- let lo = 0, hi = 1
149
- for (let i = 0; i < 12; i++) {
150
- const mid = (lo + hi) / 2
151
- const nr = Math.round(c.r + (luma - c.r) * mid)
152
- const ng = Math.round(c.g + (luma - c.g) * mid)
153
- const nb = Math.round(c.b + (luma - c.b) * mid)
154
- if (saturation(nr, ng, nb) > maxSat) lo = mid
155
- else hi = mid
156
- }
157
- const nr = Math.round(c.r + (luma - c.r) * hi)
158
- const ng = Math.round(c.g + (luma - c.g) * hi)
159
- const nb = Math.round(c.b + (luma - c.b) * hi)
160
- return "#" + [nr, ng, nb].map((v) => Math.max(0, Math.min(255, v)).toString(16).padStart(2, "0")).join("")
161
- }
162
-
163
- /** Darken a hex colour by multiplying each channel by `factor` (0–1). */
164
- function dimColor(hex: string, factor = 0.5): string {
165
- const c = rgb(hex)
166
- if (!c) return hex
167
- const r = Math.round(c.r * factor)
168
- const g = Math.round(c.g * factor)
169
- const b = Math.round(c.b * factor)
170
- return "#" + [r, g, b].map((v) => Math.max(0, Math.min(255, v)).toString(16).padStart(2, "0")).join("")
171
- }
172
-
173
- // Morandi fallbacks used when a theme colour cannot be resolved
174
- const FALLBACK = {
175
- primary: "#8B9DAF",
176
- text: "#C5C5BB",
177
- muted: "#7A7A72",
178
- success: "#9CAF8B",
179
- warning: "#C5B88D",
180
- error: "#B08A8A",
181
- border: "#6B6B63",
182
- } as const
183
-
184
- /**
185
- * Desaturation ceiling for the Morandi-style palette.
186
- *
187
- * Morandi colours float around 0.15–0.30 saturation in HSL space.
188
- * 0.28 sits near the upper end of that range: it strips the aggressive
189
- * punch from high-saturation themes (Dracula, Solarized …) while
190
- * preserving enough colour identity that green / orange / red hit-rate
191
- * coding stays distinguishable.
192
- *
193
- * Lower more grey, harder to tell colours apart.
194
- * Higher → bright themes bleed through and defeat the muted look.
195
- */
196
- const MAX_SAT = 0.28
197
-
198
- function progressBar(percent: number, width: number): string {
199
- const clamped = Math.max(0, Math.min(100, percent))
200
- const filled = Math.round((clamped / 100) * width)
201
- const empty = Math.max(0, width - filled)
202
- return "\u2588".repeat(filled) + "\u2591".repeat(empty)
203
- }
204
-
205
- function fmt(n: number): string {
206
- if (n >= 1_000_000) return (n / 1_000_000).toFixed(1) + "M"
207
- if (n >= 10_000) return (n / 1_000).toFixed(1) + "K"
208
- return n.toLocaleString("en-US")
209
- }
210
-
211
- function num(v: unknown): number {
212
- return typeof v === "number" && Number.isFinite(v) ? v : 0
213
- }
214
-
215
- function fmtCost(n: number, symbol = "$", rate = 1): string {
216
- const v = n * rate
217
- if (v >= 1) return symbol + v.toFixed(2)
218
- if (v >= 0.01) return symbol + v.toFixed(3)
219
- return symbol + v.toFixed(4)
220
- }
221
-
222
- // ── token estimation ──
223
- // Character-based BPE approximation. Default ratios (~4 ASCII or ~1.5 CJK
224
- // chars per token) work well for natural language but systematically
225
- // under-count tokens in JSON and source code where every punctuation mark
226
- // tends to be its own token. Detect these cases and tighten the ratio.
227
- // See: GPT-4 / Claude tokenizer behaviour with structured text.
228
-
229
- function estimateTokens(text: string): number {
230
- if (!text || text.length === 0) return 0
231
- let ascii = 0
232
- let cjk = 0
233
- for (const c of text) {
234
- const code = c.codePointAt(0) ?? 0
235
- if (code >= 0x4E00 && code <= 0x9FFF) cjk++ // CJK Unified
236
- else if (code >= 0x3040 && code <= 0x30FF) cjk++ // Hiragana/Katakana
237
- else if (code >= 0xAC00 && code <= 0xD7A3) cjk++ // Hangul
238
- else if (code >= 0x1100 && code <= 0x11FF) cjk++ // Hangul Jamo
239
- else if (code >= 0x2E80 && code <= 0x2EFF) cjk++ // CJK Radicals
240
- else ascii++
241
- }
242
-
243
- // Real BPE tokenizers (cl100k_base, o200k_base) average ~3.5-4.0
244
- // ASCII chars/token for both JSON and source code — close to prose.
245
- // The old 2.0 / 2.5 ratios matched minified-JS extremes, not typical
246
- // payloads, and systematically over-estimated token counts.
247
- const trimmed = text.trimStart()
248
- // Strip markdown code-fence prefix so that ```json … is detected as JSON
249
- const strippedFence = trimmed.replace(/^\x60{3}\w*\s*\n?/, "")
250
- const jsonLike = (strippedFence.startsWith("{") || strippedFence.startsWith("["))
251
- && /"[^"]+"\s*:/.test(text)
252
- const codeLike = !jsonLike
253
- && /```|^import |^export |^function |^const |^let |^var |^class |^interface |^type |^def |^fn |^pub |^use |^mod |^package /m.test(text)
254
-
255
- const asciiPerToken = jsonLike ? 3.5 : codeLike ? 3.5 : 4
256
- return Math.max(1, Math.ceil(ascii / asciiPerToken + cjk / 1.0))
257
- }
258
-
259
- interface TokenDist {
260
- system: number // UserMessage.system
261
- user: number // user message text/file parts
262
- agent: number // task tool input prompt/description (sub-agent delegation)
263
- toolCall: number // ToolPart.input (actual tool params)
264
- toolResult: number // ToolPart completed output / error
265
- output: number // AssistantMessage.tokens.output (API exact, reasoning excluded)
266
- reasoning: number // AssistantMessage.tokens.reasoning (API exact)
267
- apiOutput: number // StepFinishPart.tokens.output (API exact, preferred)
268
- apiInput: number // API exact total input context (input + cache read + cache write)
269
- stepCost: number // last step-finish part cost (USD) in the current round
270
- stepCount: number // step-finish parts count across the current round (parentID chain)
271
- }
272
-
273
- // ---------------------------------------------------------------------------
274
- // Balance state
275
- // ---------------------------------------------------------------------------
276
-
277
- interface BalanceState {
278
- status: "idle" | "loading" | "ok" | "error"
279
- data: BalanceEntry[] | null
280
- lastFetch: number
281
- error?: string
282
- key?: string // 上次成功/尝试查询所用的 key,用于检测 key 是否更换
283
- }
284
-
285
- const BALANCE_POLL_MS = 5 * 60 * 1000 // 5 minutes
286
-
287
- /**
288
- * 将余额从来源币种换算为目标币种。
289
- * DEFAULT_RATES USD=1 为基准:先折算为 USD,再换算到目标币种。
290
- */
291
- function convertBalance(target: string, targetRate: number, amount: number, from: string): number {
292
- if (from === target) return amount
293
- const fromRate = DEFAULT_RATES[from] ?? 1
294
- const usd = from === "USD" ? amount : amount / fromRate
295
- return target === "USD" ? usd : usd * targetRate
296
- }
297
-
298
- /**
299
- * 从 OpenCode 已认证的 provider 读取 API key 作为余额查询的自动兜底。
300
- * 匹配复用前缀逻辑:先精确匹配 id,再前缀匹配(如 moonshotai-cn → moonshot)。
301
- * key 来源:auth.json(provider.key)或配置(provider.options.apiKey)。
302
- * 仅当手动配置的 key 缺失时使用;读取失败或未匹配返回空串。
303
- */
304
- function findOpencodeKey(api: TuiPluginApi, provider: BalanceProvider): string {
305
- try {
306
- const provs = api.state.provider as unknown as Array<{ id: string; key?: string; options?: { apiKey?: string } }>
307
- // 大小写不敏感:精确匹配 id,否则前缀匹配(如 moonshotai-cn moonshot)
308
- const id = provider.id.toLowerCase()
309
- const hit = provs.find((p) => p.id.toLowerCase() === id) ?? provs.find((p) => p.id.toLowerCase().startsWith(id))
310
- if (!hit) return ""
311
- const k = typeof hit.key === "string" ? hit.key : ""
312
- if (k) return k
313
- return typeof hit.options?.apiKey === "string" ? hit.options.apiKey : ""
314
- } catch {
315
- return ""
316
- }
317
- }
318
-
319
- /** 货币符号:优先取 /cache-currency 内置映射,未知币种回退为代码。 */
320
- function balanceSymbol(currency: string): string {
321
- const sym = CURRENCIES[currency]
322
- return sym ?? currency + " "
323
- }
324
-
325
- /** 紧凑数字缩写(底部状态栏用):1234 "1.2K",1234567 → "1.2M"。 */
326
- function fmtCompact(n: number): string {
327
- if (n >= 1e6) return (n / 1e6).toFixed(1) + "M"
328
- if (n >= 1e3) return (n / 1e3).toFixed(1) + "K"
329
- return String(Math.round(n))
330
- }
331
-
332
- /** 余额数值格式化:≥1 0 显示固定 2 位小数;小额(<1)保留精度(最多 6 位),避免抹成 0.00。 */
333
- function formatBalanceAmount(total: string): string {
334
- const n = parseFloat(total)
335
- if (!Number.isFinite(n)) return total
336
- if (n === 0 || n >= 1) return n.toLocaleString("en-US", { minimumFractionDigits: 2, maximumFractionDigits: 2 })
337
- return n.toLocaleString("en-US", { maximumFractionDigits: 6 })
338
- }
339
-
340
- /**
341
- * 将余额列表格式化为单行文本。
342
- * 优先直接显示偏好币种(CNY/USD…);偏好币种为换算币种时按汇率折算第一条余额。
343
- */
344
- function formatBalanceText(list: BalanceEntry[], pref: string, rate: number): string {
345
- const native = pref ? list.find((x) => x.currency === pref) : undefined
346
- if (native) return balanceSymbol(native.currency) + formatBalanceAmount(native.total)
347
- const base = list[0]
348
- const baseAmt = parseFloat(base.total)
349
- const converted = Number.isFinite(baseAmt)
350
- ? convertBalance(pref || base.currency, rate, baseAmt, base.currency)
351
- : baseAmt
352
- const shown = pref && base.currency !== pref
353
- ? converted.toLocaleString("en-US", { maximumFractionDigits: 2 })
354
- : formatBalanceAmount(base.total)
355
- return balanceSymbol(pref || base.currency) + shown
356
- }
357
-
358
- // ---------------------------------------------------------------------------
359
- // Sidebar component
360
- // ---------------------------------------------------------------------------
361
-
362
- /** Signals shared between the TUI component and slash commands.
363
- * Created in the `tui` function scope so they do not survive module reload —
364
- * the component re-creates them on mount and restores user config from kv. */
365
- interface PanelSignals {
366
- currencySymbol: () => string
367
- setCurrencySymbol: (v: string) => void
368
- exchangeRate: () => number
369
- setExchangeRate: (v: number) => void
370
- langCode: () => LangCode
371
- setLangCode: (v: LangCode) => void
372
- sectionDetail: () => boolean
373
- setSectionDetail: (v: boolean) => void
374
- sectionModel: () => boolean
375
- setSectionModel: (v: boolean) => void
376
- sectionDist: () => boolean
377
- setSectionDist: (v: boolean) => void
378
- sectionSkills: () => boolean
379
- setSectionSkills: (v: boolean) => void
380
- sectionBalance: () => boolean
381
- setSectionBalance: (v: boolean) => void
382
- /** Bottom status bar (prompt hint line) visibility. */
383
- sectionBottom: () => boolean
384
- setSectionBottom: (v: boolean) => void
385
- /** Increment to force a balance re-fetch. */
386
- balanceRefresh: () => number
387
- setBalanceRefresh: (v: number) => void
388
- /** Currently selected balance provider id (e.g. "deepseek"). */
389
- balanceProviderId: () => string
390
- setBalanceProviderId: (v: string) => void
391
- /** Auto-switch to the session's provider for balance display. Manual switch disables it. */
392
- autoBalance: () => boolean
393
- setAutoBalance: (v: boolean) => void
394
- /** True when the session's provider has no balance adapter (auto mode). Suppresses balance polling. */
395
- balanceUnsupported: () => boolean
396
- setBalanceUnsupported: (v: boolean) => void
397
- /** Shared balance query state — single source of truth for sidebar and bottom bar. */
398
- balanceState: () => BalanceState
399
- /** Preferred currency code for balance display (CNY / USD / …). Empty = first entry. */
400
- balanceCurrency: () => string
401
- setBalanceCurrency: (v: string) => void
402
- borderVisible: () => boolean
403
- setBorderVisible: (v: boolean) => void
404
- /** When set, the panel renders stats for this session instead of the main one. */
405
- overrideSessionId: () => string | undefined
406
- setOverrideSessionId: (v: string | undefined) => void
407
- }
408
-
409
- const CURRENCIES: Record<string, string> = {
410
- USD: "$", CNY: "¥", EUR: "€", JPY: "JP¥", GBP: "£", KRW: "₩",
411
- }
412
- /** Approximate USD exchange rates — used as defaults when switching currency.
413
- * Users can override via /cache-rate. Last updated 2026-05. */
414
- const DEFAULT_RATES: Record<string, number> = {
415
- USD: 1, CNY: 7.2, EUR: 0.92, JPY: 150, GBP: 0.79, KRW: 1350,
416
- }
417
-
418
- const MIN_PANEL_WIDTH = 20
419
- const DEFAULT_PANEL_WIDTH = 26
420
-
421
- /** ── layout measurement constants (visual columns) ── */
422
- const LABEL_GAP = 1 // label(如 "Hit")后面的空格
423
- const BAR_BRACKETS = 2 // "[" + "]" 包围进度条
424
- const BAR_GAP = 1 // "]" 后面的空格
425
- const PCT_FIXED_WIDTH = 5 // "XX.X%" 固定 5 字符宽度
426
- const HEADER_PREFIX = 2 // 折叠态标题行:▶/▼ 图标 + 后面的空格
427
- const UNIT_GAP = 1 // 计量单位前的空格(如 "tok")
428
-
429
-
430
- function TokenCachePanel(props: {
431
- theme: TuiThemeCurrent
432
- api: TuiPluginApi
433
- sessionId: string
434
- signals: PanelSignals
435
- }): JSX.Element {
436
- const [panelWidth, setPanelWidth] = createSignal(DEFAULT_PANEL_WIDTH)
437
- const [open, setOpen] = createSignal(true)
438
- const [detailOpen, setDetailOpen] = createSignal(true)
439
- const [modelOpen, setModelOpen] = createSignal(true)
440
- const [distOpen, setDistOpen] = createSignal(false)
441
- const [skillsOpen, setSkillsOpen] = createSignal(true)
442
- let boxEl: any
443
-
444
- // ── shared signals (de-structured so internal code is unchanged) ──
445
- const {
446
- currencySymbol, setCurrencySymbol,
447
- exchangeRate, setExchangeRate,
448
- langCode,
449
- sectionDetail, setSectionDetail,
450
- sectionModel, setSectionModel,
451
- sectionDist, setSectionDist,
452
- sectionSkills, setSectionSkills,
453
- sectionBalance, setSectionBalance,
454
- balanceRefresh,
455
- balanceProviderId, setBalanceProviderId,
456
- autoBalance, setAutoBalance,
457
- balanceUnsupported, setBalanceUnsupported,
458
- balanceState,
459
- balanceCurrency, setBalanceCurrency,
460
- borderVisible, setBorderVisible,
461
- } = props.signals
462
-
463
- // ── reactive translation (follows langCode signal) ──
464
- const t = createT(() => langCode())
465
-
466
- // ── scan session messages reactively ──
467
- // SolidJS createMemo re-evaluates whenever the underlying
468
- // api.state.session state changes — no event listener needed.
469
-
470
- // ── distribution cache ────────────────────────────────────────
471
- // When data() re-computes before api.state.part() is warm (e.g. after
472
- // a view switch), hasDistData flips to false and the distribution
473
- // block disappears. Keep the last valid snapshot so the UI stays
474
- // stable until the next successful computation arrives.
475
- const [lastDist, setLastDist] = createSignal<TokenDist>({
476
- system: 0, user: 0, agent: 0, toolCall: 0, toolResult: 0,
477
- output: 0, reasoning: 0, apiOutput: 0, apiInput: 0, stepCost: 0, stepCount: 0,
478
- })
479
- const [lastHasDist, setLastHasDist] = createSignal(false)
480
-
481
- const [dataSignal, setDataSignal] = createSignal<any>({
482
- hitRate: 0, read: 0, write: 0, freshInput: 0, output: 0,
483
- cost: 0, saved: 0, model: "", inputRate: 0, cacheReadRate: 0, cacheWriteRate: 0,
484
- hasPricing: false, hasData: false, trend: 0, hasTrendData: false,
485
- providerName: "", sessionHitRate: 0,
486
- dist: { system: 0, user: 0, agent: 0, toolCall: 0, toolResult: 0, output: 0, reasoning: 0, apiOutput: 0, apiInput: 0, stepCost: 0, stepCount: 0 },
487
- hasDistData: false,
488
- skills: [] as { name: string; tokens: number }[],
489
- hasSkills: false,
490
- })
491
- const [refreshTick, setRefreshTick] = createSignal(0)
492
-
493
- // 当前 provider 显示名(余额查询状态为共享信号,见 PanelSignals.balanceState)
494
- const providerName = createMemo(() => getBalanceProvider(balanceProviderId()).name)
495
-
496
- // 自动切换当前会话的 provider(前缀匹配)。手动切换会关闭此行为。
497
- // 直接追踪 messages 取最后一条 assistant 消息的 providerID——
498
- // 不依赖 session.model 的响应式更新(模型切换时该链路可能不触发重算)。
499
- createEffect(() => {
500
- if (!autoBalance()) return
501
- const sid = props.signals.overrideSessionId() ?? props.sessionId
502
- const msgs = props.api.state.session.messages(sid) as Message[]
503
- let pid = ""
504
- for (let i = msgs.length - 1; i >= 0; i--) {
505
- const m = msgs[i]
506
- if (m.role === "assistant" && (m as AssistantMessage).providerID) {
507
- pid = (m as AssistantMessage).providerID
508
- break
509
- }
510
- }
511
- // 会话尚无 assistant 消息(新会话 / 刚切换模型未对话 / 消息未加载)
512
- // 回退到会话级模型元数据,反映当前正在使用的 provider
513
- if (!pid) {
514
- try {
515
- const session = props.api.state.session.get(sid)
516
- pid = session?.model?.providerID ?? ""
517
- } catch { /* ignore */ }
518
- }
519
- if (!pid) return
520
- const hit = matchBalanceProvider(pid)
521
- if (hit) {
522
- setBalanceUnsupported(false)
523
- if (hit.id !== balanceProviderId()) {
524
- setBalanceProviderId(hit.id)
525
- props.signals.setBalanceRefresh(props.signals.balanceRefresh() + 1)
526
- }
527
- } else {
528
- // 当前提供商没有余额适配器 标记不支持,余额显示 N/A 并停止轮询
529
- setBalanceUnsupported(true)
530
- }
531
- })
532
-
533
- // ── auto-clear override when the user navigates to a different main session ──
534
- let lastMainSid = props.sessionId
535
- createEffect(() => {
536
- const sid = props.sessionId
537
- if (sid !== lastMainSid) {
538
- lastMainSid = sid
539
- if (props.signals.overrideSessionId()) {
540
- props.signals.setOverrideSessionId(undefined)
541
- props.api.kv.set(`${KV_PREFIX}.session`, "")
542
- }
543
- }
544
- })
545
-
546
- createEffect(() => {
547
- const sid = props.signals.overrideSessionId() ?? props.sessionId
548
- void refreshTick()
549
- void partVersion()
550
-
551
- // 自然追踪 messages provider(SDK 数据就绪时自动重新执行)
552
- const msgs = props.api.state.session.messages(sid) as Message[]
553
- const session = typeof props.api.state.session.get === "function"
554
- ? props.api.state.session.get(sid)
555
- : undefined
556
-
557
- // 累计值优先使用 Session 聚合字段(数据库级,不受 sync 层 limit:100 截断)
558
- // 若字段不存在(旧版本 SDK),降级到消息遍历累加
559
- let input = session?.tokens?.input ?? 0
560
- let read = session?.tokens?.cache?.read ?? 0
561
- let write = session?.tokens?.cache?.write ?? 0
562
- let output = session?.tokens?.output ?? 0
563
- let cost = session?.cost ?? 0
564
- let pid = session?.model?.providerID ?? ""
565
- let mid = session?.model?.id ?? ""
566
-
567
- const fallbackTokens = session?.tokens == null
568
- const fallbackCost = session?.cost == null
569
- const fallbackModel = !pid || !mid
570
-
571
- let prevMsgHitRate = -1, lastMsgHitRate = -1
572
- for (const msg of msgs) {
573
- if (msg.role !== "assistant") continue
574
- const tok = (msg as AssistantMessage).tokens; if (!tok) continue
575
- const mit = num(tok.input) + num(tok.cache?.read) + num(tok.cache?.write), mrt = num(tok.cache?.read)
576
- if (mit > 0) { prevMsgHitRate = lastMsgHitRate; lastMsgHitRate = (mrt / mit) * 100 }
577
- if (fallbackTokens) {
578
- input += num(tok.input); read += num(tok.cache?.read); write += num(tok.cache?.write); output += num(tok.output)
579
- }
580
- if (fallbackCost) {
581
- cost += num((msg as AssistantMessage).cost)
582
- }
583
- if (fallbackModel && (msg as AssistantMessage).providerID && (msg as AssistantMessage).modelID) {
584
- pid = (msg as AssistantMessage).providerID; mid = (msg as AssistantMessage).modelID
585
- }
586
- }
587
- let saved = 0, inputRate = 0, cacheReadRate = 0, cacheWriteRate = 0
588
- if (read > 0 && pid && mid && Array.isArray(props.api.state.provider)) for (const provider of props.api.state.provider) {
589
- if (provider.id !== pid) continue
590
- const model = provider.models[mid]; if (!model?.cost) continue
591
- inputRate = num(model.cost.input); cacheReadRate = num(model.cost.cache?.read); cacheWriteRate = num(model.cost.cache?.write)
592
- if (inputRate > cacheReadRate) saved = (read * (inputRate - cacheReadRate)) / 1_000_000
593
- break
594
- }
595
- const hitRate = lastMsgHitRate >= 0 ? lastMsgHitRate : 0
596
- // 总命中率分母含缓存写(业界口径:read / (input+read+write))
597
- const freshTotal = input + read + write, sessionHitRate = freshTotal > 0 ? (read / freshTotal) * 100 : 0
598
- const model = mid.split("/").pop() ?? mid, hasPricing = inputRate > 0 || cacheReadRate > 0 || cacheWriteRate > 0
599
- const hasTrendData = prevMsgHitRate >= 0 && lastMsgHitRate >= 0
600
- const trend = hasTrendData ? lastMsgHitRate - prevMsgHitRate : 0, providerName = pid || ""
601
-
602
- // untrack 只包裹已知触发死锁的 API
603
- const distData = untrack(() => {
604
- let dist: TokenDist = { system: 0, user: 0, agent: 0, toolCall: 0, toolResult: 0, output: 0, reasoning: 0, apiOutput: 0, apiInput: 0, stepCost: 0, stepCount: 0 }
605
- let hasDistData = false
606
- const loadedSkills = new Map<string, { name: string; tokens: number }>()
607
- try {
608
- const cfg = props.api.state.config as Record<string, unknown>
609
- const agentName = String(session?.agent ?? (cfg as any)?.default_agent ?? "build")
610
- const agents = cfg?.agent as Record<string, unknown> | undefined
611
- const agentCfg = agents?.[agentName] as Record<string, unknown> | undefined
612
- const sysPrompt = typeof agentCfg?.prompt === "string" ? agentCfg.prompt : ""
613
- if (sysPrompt) dist.system = estimateTokens(sysPrompt)
614
- let lastAssMsg: AssistantMessage | undefined
615
- for (const msg of msgs) {
616
- if (msg.role === "user") {
617
- const um = msg as UserMessage; if (um.system) dist.system += estimateTokens(um.system)
618
- let parts: readonly Part[] = []; try { parts = props.api.state.part(msg.id) } catch {}
619
- for (const p of parts) {
620
- if (p.type === "text" && !(p as any).synthetic && !(p as any).ignored) dist.user += estimateTokens((p as any).text)
621
- else if (p.type === "file") { const fp = p as any; if (fp.source?.text?.value) dist.user += estimateTokens(fp.source.text.value) }
622
- }
623
- } else if (msg.role === "assistant") {
624
- const am = msg as AssistantMessage
625
- dist.output += num(am.tokens?.output)
626
- dist.reasoning += num(am.tokens?.reasoning)
627
- let parts: readonly Part[] = []; try { parts = props.api.state.part(msg.id) } catch {}
628
- for (const p of parts) {
629
- if (p.type === "tool") {
630
- const tp = p as any; let rawInput = ""
631
- try { rawInput = tp.state.raw ?? (tp.state.input != null ? JSON.stringify(tp.state.input) : "") } catch {}
632
- if (rawInput) dist.toolCall += estimateTokens(rawInput)
633
- // 子代理委托(task 工具):任务描述计入子代理指令(1.15.x 无 subtask part)
634
- if (tp.tool === "task" && tp.state?.input) {
635
- const ti = tp.state.input
636
- const prompt = typeof ti.prompt === "string" ? ti.prompt : ""
637
- const desc = typeof ti.description === "string" ? ti.description : ""
638
- dist.agent += estimateTokens(prompt || desc)
639
- }
640
- if (tp.state.status === "completed") { const c = tp.state; if (c.output) dist.toolResult += estimateTokens(c.output) }
641
- else if (tp.state.status === "error") { const e = tp.state; if (e.error) dist.toolResult += estimateTokens(e.error) }
642
- if (tp.tool === "skill" && tp.state.status === "completed") {
643
- // TUI SDK strips tool metadata — extract skill name from well-known output format.
644
- // Cross-validated against api.client.app.skills() when available.
645
- let name: string | undefined = tp.state.metadata?.name
646
- if (typeof name !== "string") {
647
- const m = typeof tp.state.output === "string"
648
- ? tp.state.output.match(/^#{1,2}\s*Skill:\s*(.+)/m)
649
- : null
650
- if (m) name = m[1].trim()
651
- }
652
- if (typeof name === "string") {
653
- const tokens = typeof tp.state.output === "string" ? estimateTokens(tp.state.output) : 0
654
- const existing = loadedSkills.get(name)
655
- if (!existing || existing.tokens < tokens) {
656
- loadedSkills.set(name, { name, tokens })
657
- }
658
- }
659
- }
660
- } else if (p.type === "subtask") { const sub = p as any; dist.agent += estimateTokens(sub.prompt || sub.description || "") }
661
- }
662
- }
663
- }
664
- // 从后往前找最后一条有 token 数据的 assistant 消息(避免取到 streaming 中未填充的消息)
665
- for (let i = msgs.length - 1; i >= 0; i--) {
666
- if (msgs[i].role !== "assistant") continue
667
- const tok = (msgs[i] as AssistantMessage).tokens
668
- if (tok && ((tok.input ?? 0) > 0 || (tok.cache?.read ?? 0) > 0 || (tok.cache?.write ?? 0) > 0)) { lastAssMsg = msgs[i] as AssistantMessage; break }
669
- }
670
- // 取最后一条有数据消息的总输入(含缓存读/写)作为当前 context 大小
671
- dist.apiInput = num(lastAssMsg?.tokens?.input) + num(lastAssMsg?.tokens?.cache?.read) + num(lastAssMsg?.tokens?.cache?.write)
672
- dist.apiOutput = num(lastAssMsg?.tokens?.output)
673
- // 本回合(最后一条有数据消息所在的 parentID 链)的 API 调用次数与末次成本。
674
- // opencode 将回合内每次工具调用循环拆为独立 assistant 消息(各含 1 个 step-finish),
675
- // 故按 parentID 链聚合统计,而非单条消息。
676
- if (lastAssMsg) {
677
- const roundParent = (lastAssMsg as AssistantMessage).parentID
678
- let lastCost: number | undefined
679
- for (let i = msgs.length - 1; i >= 0; i--) {
680
- const m = msgs[i]
681
- if (m.role !== "assistant") continue
682
- if ((m as AssistantMessage).parentID !== roundParent) break
683
- let parts: readonly Part[] = []; try { parts = props.api.state.part(m.id) } catch {}
684
- for (const p of parts) {
685
- if (p.type !== "step-finish") continue
686
- dist.stepCount++
687
- const sc = (p as { cost?: unknown }).cost
688
- if (lastCost === undefined && typeof sc === "number" && Number.isFinite(sc)) lastCost = sc
689
- }
690
- }
691
- if (lastCost !== undefined) dist.stepCost = lastCost
692
- }
693
- hasDistData = dist.system + dist.user + dist.agent + dist.toolCall + dist.toolResult > 0 || dist.apiOutput > 0 || dist.apiInput > 0 || dist.reasoning > 0
694
- } catch {}
695
- const finalDist = hasDistData ? dist : lastDist(), finalHasDist = hasDistData || lastHasDist()
696
- const skills = [...loadedSkills.values()]
697
- return { finalDist, finalHasDist, skills }
698
- })
699
-
700
- setDataSignal({
701
- hitRate, read, write, freshInput: input, output, cost, saved, model,
702
- inputRate, cacheReadRate, cacheWriteRate, hasPricing,
703
- hasData: read > 0 || write > 0 || input > 0 || output > 0 || cost > 0,
704
- trend, hasTrendData, providerName, sessionHitRate,
705
- dist: distData.finalDist, hasDistData: distData.finalHasDist,
706
- skills: distData.skills, hasSkills: distData.skills.length > 0,
707
- })
708
- })
709
-
710
- const data = createMemo(() => {
711
- return dataSignal()
712
- })
713
-
714
- // Persist the last valid distribution so that data() can fall back
715
- // to it while api.state.part() is re-hydrating after a view switch.
716
- createEffect(() => {
717
- const d = data()
718
- if (d.hasDistData) {
719
- setLastDist({ ...d.dist })
720
- setLastHasDist(true)
721
- // Also persist across component remounts (view switches)
722
- try { props.api.kv.set(`${KV_PREFIX}.dist_snapshot`, { ...d.dist }) } catch {}
723
- }
724
- })
725
-
726
- // ── token distribution (in-process via api.state.part) ──
727
- const [partVersion, setPartVersion] = createSignal(0)
728
-
729
- // Persist fold state to api.kv
730
- const KV_PREFIX = "cache_panel"
731
- const persistFold = (key: string, val: boolean) => {
732
- try { props.api.kv.set(`${KV_PREFIX}.${key}`, val) } catch {}
733
- }
734
-
735
- onMount(() => {
736
- // Reset panelWidth on (re)mount so the layout uses a clean
737
- // default until onSizeChange measures the live box dimensions.
738
- setPanelWidth(DEFAULT_PANEL_WIDTH)
739
-
740
- // Restore fold state from persisted storage (non-critical — fire and forget)
741
- try {
742
- setOpen(Boolean(props.api.kv.get(`${KV_PREFIX}.open`, false)))
743
- setDetailOpen(Boolean(props.api.kv.get(`${KV_PREFIX}.detail`, true)))
744
- setModelOpen(Boolean(props.api.kv.get(`${KV_PREFIX}.model`, true)))
745
- setDistOpen(Boolean(props.api.kv.get(`${KV_PREFIX}.dist`, false)))
746
- setSkillsOpen(Boolean(props.api.kv.get(`${KV_PREFIX}.skills`, true)))
747
- } catch {}
748
-
749
- // Restore user config (currency, rate, section visibility).
750
- // Try synchronously first (kv is usually ready on mount), fall back to
751
- // polling if the module was reloaded and kv hasn't initialised yet.
752
- const doRestore = () => {
753
- try {
754
- const sym = props.api.kv.get<string>(`${KV_PREFIX}.currency`)
755
- const rate = props.api.kv.get<number>(`${KV_PREFIX}.rate`)
756
- if (typeof sym === "string") setCurrencySymbol(sym)
757
- if (typeof rate === "number" && rate > 0) setExchangeRate(rate)
758
- const balCur = props.api.kv.get<string>(`${KV_PREFIX}.balance_currency`)
759
- if (typeof balCur === "string") setBalanceCurrency(balCur)
760
- // Restore balance provider (fall back to default when unknown)
761
- const savedProvider = props.api.kv.get<string>(`${KV_PREFIX}.balance.provider`)
762
- if (typeof savedProvider === "string" && balanceProviders.some((p) => p.id === savedProvider)) {
763
- setBalanceProviderId(savedProvider)
764
- setBalanceUnsupported(false)
765
- }
766
- // Restore auto-switch (default on)
767
- const savedAuto = props.api.kv.get<boolean>(`${KV_PREFIX}.balance.auto`)
768
- if (typeof savedAuto === "boolean") setAutoBalance(savedAuto)
769
- // Migrate legacy DeepSeek key (cache_panel.ds_key → cache_panel.balance.deepseek.key)
770
- const legacyKey = props.api.kv.get<string>(`${KV_PREFIX}.ds_key`, "")
771
- if (legacyKey) {
772
- const dsKey = props.api.kv.get<string>(`${KV_PREFIX}.balance.deepseek.key`, "")
773
- if (!dsKey) props.api.kv.set(`${KV_PREFIX}.balance.deepseek.key`, legacyKey)
774
- props.api.kv.set(`${KV_PREFIX}.ds_key`, "")
775
- }
776
- // 恢复的 provider 可能与默认值不同,强制重新查询
777
- props.signals.setBalanceRefresh(props.signals.balanceRefresh() + 1)
778
- setSectionDetail(Boolean(props.api.kv.get(`${KV_PREFIX}.section.detail`, true)))
779
- setSectionModel(Boolean(props.api.kv.get(`${KV_PREFIX}.section.model`, true)))
780
- setSectionDist(Boolean(props.api.kv.get(`${KV_PREFIX}.section.dist`, true)))
781
- setSectionSkills(Boolean(props.api.kv.get(`${KV_PREFIX}.section.skills`, true)))
782
- setSectionBalance(Boolean(props.api.kv.get(`${KV_PREFIX}.section.balance`, true)))
783
- const bv = props.api.kv.get<boolean>(`${KV_PREFIX}.border`, true)
784
- setBorderVisible(bv !== false)
785
- // Restore distribution snapshot so the token distribution block
786
- // doesn't blank out while api.state.part() re-hydrates.
787
- const cachedDist = props.api.kv.get<TokenDist>(`${KV_PREFIX}.dist_snapshot`)
788
- if (cachedDist) {
789
- setLastDist(cachedDist)
790
- setLastHasDist(true)
791
- }
792
- } catch {
793
- // kv read failed — signals stay at defaults
794
- }
795
- // Re-measure panel width after config signals have settled
796
- if (boxEl && typeof boxEl.width === "number" && boxEl.width > 0) {
797
- setPanelWidth(Math.max(MIN_PANEL_WIDTH, boxEl.width))
798
- }
799
- }
800
-
801
- if (props.api.kv.ready) {
802
- doRestore()
803
- } else {
804
- // Poll kv.ready with a 1-second timeout to avoid infinite busy-wait
805
- // on platforms where kv initialisation may be delayed (Linux single-thread
806
- // mode, session switch storms, etc.).
807
- const MAX_POLL = 100
808
- let tries = 0
809
- const pollRestore = () => {
810
- if (!props.api.kv.ready) {
811
- if (++tries > MAX_POLL) { doRestore(); return }
812
- setTimeout(pollRestore, 10)
813
- return
814
- }
815
- doRestore()
816
- }
817
- pollRestore()
818
- }
819
-
820
- // Debounce partVersion updates so that event bursts during session
821
- // switching / streaming don't cause data() to re-compute on every
822
- // single event (up to hundreds per second on Linux single-thread).
823
- let partTimer: ReturnType<typeof setTimeout> | undefined
824
- const bumpPartVersion = () => {
825
- clearTimeout(partTimer)
826
- partTimer = setTimeout(() => setPartVersion((v) => v + 1), 100)
827
- }
828
- const unsubPart = props.api.event.on("message.part.updated", () => { bumpPartVersion(); setRefreshTick(v => v + 1) })
829
- const unsubMsg = props.api.event.on("message.updated", () => { bumpPartVersion(); setRefreshTick(v => v + 1) })
830
- const unsubSession = props.api.event.on("session.updated", () => { setRefreshTick(v => v + 1) })
831
- setRefreshTick(v => v + 1)
832
- onCleanup(() => { clearTimeout(partTimer); unsubPart(); unsubMsg(); unsubSession() })
833
- })
834
-
835
- // ── colours ──
836
- // Pull from the current theme, auto-desaturate if too punchy,
837
- // fall back to Morandi when a key is missing from the theme.
838
- const pal = createMemo(() => {
839
- const t = props.theme as Record<string, unknown>
840
- const sat = (k: string, fb: string) => desaturateTo(t[k], MAX_SAT, fb)
841
- return {
842
- primary: sat("primary", FALLBACK.primary),
843
- text: sat("text", FALLBACK.text),
844
- muted: sat("textMuted", FALLBACK.muted),
845
- success: sat("success", FALLBACK.success),
846
- warning: sat("warning", FALLBACK.warning),
847
- error: sat("error", FALLBACK.error),
848
- border: sat("border", FALLBACK.border),
849
- }
850
- })
851
-
852
- const hitColor = createMemo(() => {
853
- const r = data().hitRate
854
- if (r >= 85) return pal().success
855
- if (r >= 70) return pal().warning
856
- return pal().error
857
- })
858
-
859
- /** Horizontal space eaten by border (1+1 when visible) + padding (2+2 when visible). */
860
- const gutter = createMemo(() => borderVisible() ? 6 : 0)
861
-
862
- const sep = createMemo(() => "\u2500".repeat(Math.max(1, panelWidth() - gutter())))
863
- function trendLabel(t: number): string {
864
- // |t| < 0.05 视为无变化:避免显示 "↑0.0%" 的矛盾(箭头存在但数值截断为零)
865
- if (Math.abs(t) < 0.05) return "-"
866
- return (t > 0 ? "\u2191" : "\u2193") + Math.abs(t).toFixed(1) + "%"
867
- }
868
-
869
- const barW = createMemo(() => {
870
- const trendSpace = data().hasTrendData ? LABEL_GAP + visualWidth(trendLabel(data().trend)) : 0
871
- const overhead = visualWidth(t("hit")) + LABEL_GAP + BAR_BRACKETS + BAR_GAP + PCT_FIXED_WIDTH + trendSpace + gutter()
872
- return Math.max(3, panelWidth() - overhead)
873
- })
874
- const bar = createMemo(() => progressBar(data().hitRate, barW()))
875
- const pct = createMemo(() => (Math.floor(data().hitRate * 10) / 10).toFixed(1) + "%")
876
-
877
- // When border visibility changes the box dimensions shift, which
878
- // may not reliably trigger onSizeChange across (re)mount cycles.
879
- // Force panelWidth to resync with the live box after every change.
880
- createEffect(() => {
881
- borderVisible()
882
- if (boxEl && typeof boxEl.width === "number" && boxEl.width > 0) {
883
- const w = Math.max(MIN_PANEL_WIDTH, boxEl.width)
884
- setPanelWidth((prev) => (prev === w ? prev : w))
885
- }
886
- })
887
-
888
- // left-align label, right-align value — auto-fill space between
889
- const justify = (label: string, value: string, unit = ""): string => {
890
- const gauge = panelWidth() - gutter()
891
- const used = visualWidth(label) + visualWidth(value) + (unit ? visualWidth(unit) + UNIT_GAP : 0)
892
- const gap = Math.max(1, gauge - used)
893
- return label + " ".repeat(gap) + value + (unit ? " " + unit : "")
894
- }
895
-
896
- return (
897
- <box
898
- border={borderVisible()}
899
- {...(borderVisible() ? { borderColor: pal().border } : {})}
900
- paddingTop={0}
901
- paddingBottom={0}
902
- paddingLeft={borderVisible() ? 2 : 0}
903
- paddingRight={borderVisible() ? 2 : 0}
904
- flexDirection="column"
905
- gap={0}
906
- ref={boxEl}
907
- onSizeChange={() => {
908
- // boxEl.width may be undefined before the first measurement — guard with 0
909
- const w = boxEl ? Math.max(MIN_PANEL_WIDTH, boxEl.width ?? 0) : DEFAULT_PANEL_WIDTH
910
- setPanelWidth((prev) => (prev === w ? prev : w))
911
- }}
912
- >
913
- {/* collapsible header */}
914
- <text onMouseUp={() => setOpen((o) => { const n = !o; persistFold("open", n); return n })}>
915
- <span style={{ fg: pal().muted }}>{open() ? "\u25bc " : "\u25b6 "}</span>
916
- <span style={{ fg: pal().primary }}>
917
- <b>{t("title")}</b>
918
- <Show when={open()}>
919
- <span style={{ fg: dimColor(pal().muted, 0.75) }}> v{PLUGIN_VERSION}</span>
920
- </Show>
921
- </span>
922
- <Show when={!open() && data().hasData}>
923
- <Show when={data().hasTrendData}>
924
- <span>
925
- {" ".repeat(Math.max(1, panelWidth() - gutter() - HEADER_PREFIX - visualWidth(t("title")) - visualWidth(pct() + " " + t("hitFolded") + " " + trendLabel(data().trend))))}
926
- </span>
927
- <span style={{ fg: hitColor() }}>{pct()} {t("hitFolded")}</span>
928
- <span style={{ fg: Math.abs(data().trend) >= 0.05 ? (data().trend > 0 ? pal().success : pal().error) : pal().text }}>
929
- {" "}{trendLabel(data().trend)}
930
- </span>
931
- </Show>
932
- <Show when={!data().hasTrendData}>
933
- <span>
934
- {" ".repeat(Math.max(1, panelWidth() - gutter() - HEADER_PREFIX - visualWidth(t("title")) - visualWidth(pct() + " " + t("hitFolded"))))}
935
- </span>
936
- <span style={{ fg: hitColor() }}>{pct()} {t("hitFolded")}</span>
937
- </Show>
938
- </Show>
939
- </text>
940
-
941
- <Show when={open()}>
942
- <Show when={props.signals.overrideSessionId()}>
943
- {(() => {
944
- const prefix = " \u21b3 " + t("subPrefix")
945
- const maxSidW = Math.max(6, panelWidth() - visualWidth(prefix))
946
- return (
947
- <text>
948
- <span style={{ fg: pal().muted }}>{prefix}</span>
949
- <span style={{ fg: pal().text }}>{truncateVisual(props.signals.overrideSessionId()!, maxSidW)}</span>
950
- </text>
951
- )
952
- })()}
953
- </Show>
954
- <Show when={data().hasData} fallback={
955
- <>
956
- <text fg={pal().muted}>{sep()}</text>
957
- <text>
958
- <span style={{ fg: pal().muted }}>{"> "}</span>
959
- <span style={{ fg: pal().muted }}>{t("noData")}</span>
960
- </text>
961
- </>
962
- }>
963
- <text fg={pal().muted}>{sep()}</text>
964
-
965
- {/* hit rate + bar — inline to avoid box spacing */}
966
- <text>
967
- <span style={{ fg: pal().text }}>{t("hit")} </span>
968
- <span style={{ fg: hitColor() }}>[{bar()}] </span>
969
- <span style={{ fg: pal().text }}>{pct()}</span>
970
- <Show when={data().hasTrendData}>
971
- <span style={{ fg: Math.abs(data().trend) >= 0.05 ? (data().trend > 0 ? pal().success : pal().error) : pal().text }}>
972
- {" "}{trendLabel(data().trend)}
973
- </span>
974
- </Show>
975
- </text>
976
-
977
- {/* session cumulative hit rate */}
978
- <text fg={pal().muted}>
979
- {justify(t("totalHit"), (Math.floor(data().sessionHitRate * 10) / 10).toFixed(1) + "%")}
980
- </text>
981
-
982
- {/* ── detail section (collapsible, default open) ── */}
983
- <Show when={sectionDetail()}>
984
- <text onMouseUp={() => setDetailOpen((o) => { const n = !o; persistFold("detail", n); return n })}>
985
- <span style={{ fg: pal().muted }}>{detailOpen() ? "\u25bc " : "\u25b6 "}</span>
986
- <span style={{ fg: pal().primary }}><b>{t("secDetail")}</b></span>
987
- <span style={{ fg: pal().muted }}>{sep().slice(visualWidth((detailOpen() ? "\u25bc " : "\u25b6 ") + t("secDetail")))}</span>
988
- </text>
989
-
990
- <Show when={detailOpen()}>
991
- <Show when={data().read > 0}>
992
- <text fg={pal().muted}>
993
- {justify(t("read"), fmt(data().read), t("tok"))}
994
- </text>
995
- </Show>
996
- <Show when={data().write > 0}>
997
- <text fg={pal().muted}>
998
- {justify(t("write"), fmt(data().write), t("tok"))}
999
- </text>
1000
- </Show>
1001
- {/* 未命中 = 新鲜输入 + 缓存写(两者都未从缓存命中) */}
1002
- <text fg={pal().muted}>
1003
- {justify(t("miss"), fmt(data().freshInput + data().write), t("tok"))}
1004
- </text>
1005
- <text fg={pal().muted}>
1006
- {justify(t("out"), fmt(data().output), t("tok"))}
1007
- </text>
1008
- {/* 本回合多次 API 调用时才显示调用次数与末次成本(单次调用不占行) */}
1009
- <Show when={data().dist.stepCount >= 2}>
1010
- <text fg={pal().muted}>
1011
- {justify(t("stepsCount", { n: data().dist.stepCount }), fmtCost(data().dist.stepCost, currencySymbol(), exchangeRate()))}
1012
- </text>
1013
- </Show>
1014
- <Show when={data().saved > 0}>
1015
- <text>
1016
- <span style={{ fg: pal().muted }}>{t("saved")}</span>
1017
- <span>{" ".repeat(Math.max(1, panelWidth() - gutter() - visualWidth(t("saved")) - visualWidth("~" + fmtCost(data().saved, currencySymbol(), exchangeRate()))))}</span>
1018
- <span style={{ fg: pal().success }}>~{fmtCost(data().saved, currencySymbol(), exchangeRate())}</span>
1019
- </text>
1020
- </Show>
1021
- </Show>
1022
- </Show>
1023
-
1024
- {/* ── model section (collapsible, default open) ── */}
1025
- <Show when={sectionModel()}>
1026
- {<text onMouseUp={() => setModelOpen((o) => { const n = !o; persistFold("model", n); return n })}>
1027
- <span style={{ fg: pal().muted }}>{modelOpen() ? "\u25bc " : "\u25b6 "}</span>
1028
- <span style={{ fg: pal().primary }}><b>{t("secModel")}</b></span>
1029
- <span style={{ fg: pal().muted }}>{sep().slice(visualWidth((modelOpen() ? "\u25bc " : "\u25b6 ") + t("secModel")))}</span>
1030
- </text>}
1031
-
1032
- <Show when={modelOpen()}>
1033
- <text fg={pal().text}>
1034
- {justify(t("cost"), fmtCost(data().cost, currencySymbol(), exchangeRate()))}
1035
- </text>
1036
- <Show when={data().providerName}>
1037
- <text fg={pal().muted}>
1038
- {justify(t("provider"), data().providerName)}
1039
- </text>
1040
- </Show>
1041
- <text fg={pal().muted}>
1042
- {justify(t("model"), data().model)}
1043
- </text>
1044
- <Show when={data().hasPricing}>
1045
- <text fg={pal().muted}>
1046
- {justify(t("rate"), currencySymbol() + (data().inputRate * exchangeRate()).toFixed(2) + "/M " + t("inputRate"))}
1047
- </text>
1048
- <Show when={data().cacheReadRate > 0}>
1049
- <text fg={pal().muted}>
1050
- {justify("", currencySymbol() + (data().cacheReadRate * exchangeRate()).toFixed(2) + "/M " + t("cacheRate"))}
1051
- </text>
1052
- </Show>
1053
- <Show when={data().cacheWriteRate > 0}>
1054
- <text fg={pal().muted}>
1055
- {justify("", currencySymbol() + (data().cacheWriteRate * exchangeRate()).toFixed(2) + "/M " + t("writeRate"))}
1056
- </text>
1057
- </Show>
1058
- </Show>
1059
- </Show>
1060
- </Show>
1061
-
1062
- {/* ── token distribution (collapsible, default closed) ── */}
1063
- <Show when={sectionDist()}>
1064
- <Show when={data().hasDistData}>
1065
- {<text onMouseUp={() => setDistOpen((o) => { const n = !o; persistFold("dist", n); return n })}>
1066
- <span style={{ fg: pal().muted }}>{distOpen() ? "\u25bc " : "\u25b6 "}</span>
1067
- <span style={{ fg: pal().primary }}><b>{t("distTitle")}</b></span>
1068
- <span style={{ fg: pal().muted }}>{sep().slice(visualWidth((distOpen() ? "\u25bc " : "\u25b6 ") + t("distTitle")))}</span>
1069
- </text>}
1070
- <Show when={distOpen()}>
1071
- <Show when={data().dist.system > 0}>
1072
- <text fg={pal().muted}>
1073
- {justify(t("distSys"), fmt(data().dist.system), t("tok"))}
1074
- </text>
1075
- </Show>
1076
- <Show when={data().dist.user > 0}>
1077
- <text fg={pal().muted}>
1078
- {justify(t("distUser"), fmt(data().dist.user), t("tok"))}
1079
- </text>
1080
- </Show>
1081
- <Show when={data().dist.agent > 0}>
1082
- <text fg={pal().muted}>
1083
- {justify(t("distAgent"), fmt(data().dist.agent), t("tok"))}
1084
- </text>
1085
- </Show>
1086
- <Show when={data().dist.toolCall > 0}>
1087
- <text fg={pal().muted}>
1088
- {justify(t("distTool"), fmt(data().dist.toolCall), t("tok"))}
1089
- </text>
1090
- </Show>
1091
- <Show when={data().dist.toolResult > 0}>
1092
- <text fg={pal().muted}>
1093
- {justify(t("distRes"), fmt(data().dist.toolResult), t("tok"))}
1094
- </text>
1095
- </Show>
1096
- <Show when={data().dist.reasoning > 0}>
1097
- <text fg={pal().muted}>
1098
- {justify(t("distReason"), fmt(data().dist.reasoning), t("tok"))}
1099
- </text>
1100
- </Show>
1101
- </Show>
1102
- </Show>
1103
- </Show>
1104
-
1105
- {/* ── loaded skills (collapsible, default open) ── */}
1106
- <Show when={sectionSkills()}>
1107
- <Show when={data().hasSkills}>
1108
- {<text onMouseUp={() => setSkillsOpen((o) => { const n = !o; persistFold("skills", n); return n })}>
1109
- <span style={{ fg: pal().muted }}>{skillsOpen() ? "\u25bc " : "\u25b6 "}</span>
1110
- <span style={{ fg: pal().primary }}><b>{t("secSkills")}</b></span>
1111
- <span style={{ fg: pal().muted }}> ({data().skills.length})</span>
1112
- <span style={{ fg: pal().muted }}>{sep().slice(visualWidth((skillsOpen() ? "\u25bc " : "\u25b6 ") + t("secSkills") + ` (${data().skills.length})`))}</span>
1113
- </text>}
1114
- <Show when={skillsOpen()}>
1115
- {data().skills.map((sk: { name: string; tokens: number }) => {
1116
- const rightW = visualWidth(fmt(sk.tokens)) + UNIT_GAP + visualWidth(t("tok"))
1117
- const maxLabel = Math.max(4, panelWidth() - gutter() - rightW - 1)
1118
- const label = truncateVisual(sk.name, maxLabel)
1119
- return (
1120
- <text fg={pal().muted}>
1121
- {justify(label, fmt(sk.tokens), t("tok"))}
1122
- </text>
1123
- )
1124
- })}
1125
- </Show>
1126
- </Show>
1127
- </Show>
1128
-
1129
- {/* ── provider balance (single line) ── */}
1130
- <Show when={sectionBalance()}>
1131
- <text fg={pal().muted}>{sep()}</text>
1132
- <Show when={balanceUnsupported()}>
1133
- <text fg={pal().muted}>
1134
- <span style={{ fg: pal().muted }}>{"> "}</span>
1135
- <span>{t("balUnsupported")}</span>
1136
- </text>
1137
- </Show>
1138
- <Show when={!balanceUnsupported()}>
1139
- <Show when={balanceState().status === "idle"}>
1140
- <text fg={pal().muted}>
1141
- <span style={{ fg: pal().muted }}>{"> "}</span>
1142
- <span>{t("balNoKey", { p: providerName() })}</span>
1143
- </text>
1144
- </Show>
1145
- <Show when={balanceState().status === "loading"}>
1146
- <text fg={pal().muted}>
1147
- <span style={{ fg: pal().muted }}>{"> "}</span>
1148
- <span>{t("balLoading")}</span>
1149
- </text>
1150
- </Show>
1151
- <Show when={balanceState().status === "error"}>
1152
- <text fg={pal().error}>
1153
- <span style={{ fg: pal().muted }}>{"> "}</span>
1154
- <span>{(() => {
1155
- const code = balanceState().error
1156
- if (code === "401") return t("balErr401")
1157
- if (code === "403") return t("balErr403")
1158
- if (code === "EMPTY") return t("balErrEmpty")
1159
- if (code === "TIMEOUT") return t("balErrTimeout")
1160
- return t("balError") + (code ? ` (${code})` : "")
1161
- })()}</span>
1162
- </text>
1163
- </Show>
1164
- <Show when={balanceState().status === "ok" && balanceState().data}>
1165
- <text fg={pal().text}>
1166
- {justify(t("balTotal"), formatBalanceText(balanceState().data!, balanceCurrency(), exchangeRate()))}
1167
- </text>
1168
- </Show>
1169
- </Show>
1170
- </Show>
1171
- </Show>
1172
- </Show>
1173
- </box>
1174
- )
1175
- }
1176
-
1177
- // ---------------------------------------------------------------------------
1178
- // Plugin entry
1179
- // ---------------------------------------------------------------------------
1180
-
1181
- /**
1182
- * 输入框 hint 行(session_prompt slot 的 hint):单行显示 路径 · 命中率 · 余额 · Tokens。
1183
- * 通过 ui.Prompt 的 hint prop 注入——宿主右侧的 token/commands 提示自动保留,
1184
- * 三合一信息与路径同行显示在中间位置。
1185
- */
1186
- function BottomStatusBar(props: { api: TuiPluginApi; signals: PanelSignals; sessionId: string }): JSX.Element {
1187
- const KV_PREFIX = "cache_panel"
1188
- const t = createT(() => props.signals.langCode())
1189
-
1190
- const sid = props.sessionId
1191
-
1192
- // ── 命中率(单条口径:最后一条有 token 的 assistant 消息)+ token 汇总 ──
1193
- const stats = createMemo(() => {
1194
- const id = sid
1195
- if (!id) return null
1196
- const msgs = props.api.state.session.messages(id) as Message[]
1197
- const session = typeof props.api.state.session.get === "function"
1198
- ? props.api.state.session.get(id)
1199
- : undefined
1200
- let input = session?.tokens?.input ?? 0
1201
- let read = session?.tokens?.cache?.read ?? 0
1202
- let write = session?.tokens?.cache?.write ?? 0
1203
- // 旧 SDK 无 session 聚合字段 → 遍历消息累加(与侧边栏 fallback 一致)
1204
- if (session?.tokens == null) {
1205
- for (const m of msgs) {
1206
- if (m.role !== "assistant") continue
1207
- const tk = (m as AssistantMessage).tokens
1208
- if (!tk) continue
1209
- input += num(tk.input)
1210
- read += num(tk.cache?.read)
1211
- write += num(tk.cache?.write)
1212
- }
1213
- }
1214
- // 从后往前取最后两条有 token 数据的 assistant 消息 → 单条命中率 + 趋势
1215
- // 分母含缓存写(业界口径:read / (input+read+write))
1216
- let hitRate = -1, prevHitRate = -1
1217
- for (let i = msgs.length - 1; i >= 0; i--) {
1218
- const m = msgs[i]
1219
- if (m.role !== "assistant") continue
1220
- const tk = (m as AssistantMessage).tokens
1221
- if (!tk) continue
1222
- const mit = num(tk.input) + num(tk.cache?.read) + num(tk.cache?.write)
1223
- const mrt = num(tk.cache?.read)
1224
- if (mit <= 0) continue
1225
- const rate = (mrt / mit) * 100
1226
- if (hitRate < 0) { hitRate = rate; continue }
1227
- prevHitRate = rate
1228
- break
1229
- }
1230
- return { hitRate, prevHitRate, input, read, write }
1231
- })
1232
-
1233
- // 余额查询状态为共享信号(PanelSignals.balanceState),由 tui() 统一轮询
1234
-
1235
- // 自动切换 provider(跟随当前会话模型;幂等,与侧边栏共享信号)
1236
- createEffect(() => {
1237
- if (!props.signals.autoBalance()) return
1238
- const id = sid
1239
- if (!id) return
1240
- const msgs = props.api.state.session.messages(id) as Message[]
1241
- let pid = ""
1242
- for (let i = msgs.length - 1; i >= 0; i--) {
1243
- const m = msgs[i]
1244
- if (m.role === "assistant" && (m as AssistantMessage).providerID) { pid = (m as AssistantMessage).providerID; break }
1245
- }
1246
- if (!pid) {
1247
- try { pid = props.api.state.session.get(id)?.model?.providerID ?? "" } catch {}
1248
- }
1249
- if (!pid) return
1250
- const hit = matchBalanceProvider(pid)
1251
- if (hit) {
1252
- props.signals.setBalanceUnsupported(false)
1253
- if (hit.id !== props.signals.balanceProviderId()) {
1254
- props.signals.setBalanceProviderId(hit.id)
1255
- props.signals.setBalanceRefresh(props.signals.balanceRefresh() + 1)
1256
- }
1257
- } else {
1258
- // 当前提供商没有余额适配器 → 标记不支持,余额显示 N/A 并停止轮询
1259
- props.signals.setBalanceUnsupported(true)
1260
- }
1261
- })
1262
-
1263
- // ── 主题色(与侧边栏同口径)──
1264
- const pal = createMemo(() => {
1265
- const th = props.api.theme.current as Record<string, unknown>
1266
- const sat = (k: string, fb: string) => desaturateTo(th[k], MAX_SAT, fb)
1267
- return {
1268
- text: sat("text", FALLBACK.text),
1269
- muted: sat("textMuted", FALLBACK.muted),
1270
- success: sat("success", FALLBACK.success),
1271
- warning: sat("warning", FALLBACK.warning),
1272
- error: sat("error", FALLBACK.error),
1273
- }
1274
- })
1275
-
1276
- const hitColor = createMemo(() => {
1277
- const r = stats()?.hitRate ?? -1
1278
- if (r >= 85) return pal().success
1279
- if (r >= 70) return pal().warning
1280
- return pal().error
1281
- })
1282
-
1283
- // 命中率趋势:最后一条与上一条的差值;|Δ| < 0.05 视为无变化(null = 不显示)
1284
- const trend = createMemo(() => {
1285
- const s = stats()
1286
- if (!s || s.prevHitRate < 0 || s.hitRate < 0) return null
1287
- const d = s.hitRate - s.prevHitRate
1288
- return Math.abs(d) < 0.05 ? null : d
1289
- })
1290
-
1291
- const balanceText = createMemo(() => {
1292
- const s = props.signals.balanceState()
1293
- if (s.status === "ok" && s.data) return formatBalanceText(s.data, props.signals.balanceCurrency(), props.signals.exchangeRate())
1294
- if (s.status === "loading") return "\u2026"
1295
- if (s.status === "error") return "\u26a0"
1296
- return "-"
1297
- })
1298
-
1299
- // 路径显示(替换宿主默认 hint 左侧的 cwd 文本)
1300
- const directory = createMemo(() => {
1301
- try { return props.api.state.path.directory } catch { return "" }
1302
- })
1303
-
1304
- // 恢复显隐偏好(默认显示);关闭时回退为仅显示路径,与宿主默认 hint 行一致
1305
- onMount(() => {
1306
- try {
1307
- const v = props.api.kv.get<boolean>(`${KV_PREFIX}.section.bottom`, true)
1308
- props.signals.setSectionBottom(v !== false)
1309
- } catch {}
1310
- })
1311
-
1312
- return (
1313
- <Show when={props.signals.sectionBottom()} fallback={<text fg={pal().muted}>{directory()}</text>}>
1314
- <box marginLeft={1} flexGrow={1} flexShrink={0} flexDirection="row" justifyContent="space-between">
1315
- <text fg={pal().muted}>{directory()}</text>
1316
- <box flexDirection="row">
1317
- <text>
1318
- <span style={{ fg: pal().muted }}>{t("barHit")} </span>
1319
- <span style={{ fg: hitColor() }}>{(stats()?.hitRate ?? -1) >= 0 ? (Math.floor(stats()!.hitRate * 10) / 10).toFixed(1) + "%" : "--"}</span>
1320
- <Show when={trend() !== null}>
1321
- <span style={{ fg: trend()! > 0 ? pal().success : pal().error }}>
1322
- {" " + (trend()! > 0 ? "\u2191" : "\u2193") + Math.abs(trend()!).toFixed(1) + "%"}
1323
- </span>
1324
- </Show>
1325
- <span style={{ fg: pal().muted }}>{" \u00b7 " + t("barTok") + " "}</span>
1326
- <span style={{ fg: pal().text }}>
1327
- {stats() ? fmtCompact(stats()!.input + stats()!.read + stats()!.write) : "--"}
1328
- </span>
1329
- <Show when={!props.signals.balanceUnsupported()}>
1330
- <span style={{ fg: pal().muted }}>{" \u00b7 " + t("barBal") + " "}</span>
1331
- <span style={{ fg: pal().text }}>{balanceText()}</span>
1332
- </Show>
1333
- </text>
1334
- <text fg={pal().muted}>{" \u00b7 "}</text>
1335
- </box>
1336
- </box>
1337
- </Show>
1338
- )
1339
- }
1340
-
1341
- function createSidebarSlot(api: TuiPluginApi, signals: PanelSignals): TuiSlotPlugin {
1342
- let lastSlotSid = ""
1343
- return {
1344
- order: 55,
1345
- slots: {
1346
- sidebar_content(ctx: TuiSlotContext, input: { session_id: string }): JSX.Element {
1347
- // ── auto-clear override when the user navigates to a different main session ──
1348
- if (input.session_id !== lastSlotSid) {
1349
- lastSlotSid = input.session_id
1350
- if (signals.overrideSessionId()) {
1351
- signals.setOverrideSessionId(undefined)
1352
- api.kv.set("cache_panel.session", "")
1353
- }
1354
- }
1355
- return (
1356
- <TokenCachePanel
1357
- theme={ctx.theme.current}
1358
- api={api}
1359
- sessionId={input.session_id}
1360
- signals={signals}
1361
- />
1362
- )
1363
- },
1364
- },
1365
- }
1366
- }
1367
-
1368
- const tui: TuiPlugin = async (api: TuiPluginApi) => {
1369
- // ── shared panel signals ──────────────────────────────────────
1370
- const [currencySymbol, setCurrencySymbol] = createSignal("$")
1371
- const [exchangeRate, setExchangeRate] = createSignal(1)
1372
- const [sectionDetail, setSectionDetail] = createSignal(true)
1373
- const [sectionModel, setSectionModel] = createSignal(true)
1374
- const [sectionDist, setSectionDist] = createSignal(true)
1375
- const [sectionSkills, setSectionSkills] = createSignal(true)
1376
- const [sectionBalance, setSectionBalance] = createSignal(true)
1377
- const [sectionBottom, setSectionBottom] = createSignal(true)
1378
- const [balanceRefresh, setBalanceRefresh] = createSignal(0)
1379
- const [balanceProviderId, setBalanceProviderId] = createSignal("deepseek")
1380
- const [autoBalance, setAutoBalance] = createSignal(true)
1381
- const [balanceUnsupported, setBalanceUnsupported] = createSignal(false)
1382
- const [balanceCurrency, setBalanceCurrency] = createSignal("")
1383
- const [borderVisible, setBorderVisible] = createSignal(true)
1384
- const [langCode, setLangCode] = createSignal<LangCode>(INIT_LANG)
1385
- const [overrideSessionId, setOverrideSessionId] = createSignal<string | undefined>(undefined)
1386
-
1387
- // ── 余额查询状态(共享):侧边栏与底部栏读同一份数据,
1388
- // 避免重复请求导致两处余额不一致 ──
1389
- const [balanceState, setBalanceState] = createSignal<BalanceState>({
1390
- status: "idle", data: null, lastFetch: 0,
1391
- })
1392
- // 请求序号:防止定时轮询与手动刷新并发时,慢的旧请求覆盖新结果
1393
- let balanceSeq = 0
1394
-
1395
- const signals: PanelSignals = {
1396
- currencySymbol, setCurrencySymbol,
1397
- exchangeRate, setExchangeRate,
1398
- langCode, setLangCode,
1399
- sectionDetail, setSectionDetail,
1400
- sectionModel, setSectionModel,
1401
- sectionDist, setSectionDist,
1402
- sectionSkills, setSectionSkills,
1403
- sectionBalance, setSectionBalance,
1404
- sectionBottom, setSectionBottom,
1405
- balanceRefresh, setBalanceRefresh,
1406
- balanceProviderId, setBalanceProviderId,
1407
- autoBalance, setAutoBalance,
1408
- balanceUnsupported, setBalanceUnsupported,
1409
- balanceState,
1410
- balanceCurrency, setBalanceCurrency,
1411
- borderVisible, setBorderVisible,
1412
- overrideSessionId, setOverrideSessionId,
1413
- }
1414
-
1415
- api.slots.register(createSidebarSlot(api, signals))
1416
-
1417
- // 输入框 hint 行(session_prompt slot,replace 模式):
1418
- // 用宿主同一 Prompt 组件重渲染输入框,仅替换 hint 行左侧——
1419
- // 在路径与右侧 token/commands 提示之间插入 命中率 · 余额 · Tokens。
1420
- api.slots.register({
1421
- order: 55,
1422
- slots: {
1423
- session_prompt(
1424
- _ctx: TuiSlotContext,
1425
- input: {
1426
- session_id: string
1427
- visible?: boolean
1428
- disabled?: boolean
1429
- on_submit?: () => void
1430
- ref?: (ref: TuiPromptRef | undefined) => void
1431
- },
1432
- ): JSX.Element {
1433
- return (
1434
- <api.ui.Prompt
1435
- sessionID={input.session_id}
1436
- visible={input.visible}
1437
- disabled={input.disabled}
1438
- onSubmit={input.on_submit}
1439
- ref={input.ref}
1440
- hint={<BottomStatusBar api={api} signals={signals} sessionId={input.session_id} />}
1441
- />
1442
- )
1443
- },
1444
- },
1445
- })
1446
-
1447
- // ── slash commands for runtime config ──
1448
- const KV_PREFIX = "cache_panel"
1449
-
1450
- // ── 语言偏好恢复:KV 就绪后优先用户设置(/cache-lang),覆盖自动识别 ──
1451
- const restoreLang = () => {
1452
- try {
1453
- const saved = api.kv.get<string>(`${KV_PREFIX}.lang`)
1454
- if (saved && LANG_META.some((m) => m.code === saved)) setLangCode(saved as LangCode)
1455
- } catch {}
1456
- }
1457
- if (api.kv.ready) {
1458
- restoreLang()
1459
- } else {
1460
- const langTimer = setInterval(() => {
1461
- if (api.kv.ready) { clearInterval(langTimer); restoreLang() }
1462
- }, 10)
1463
- api.lifecycle.onDispose(() => clearInterval(langTimer))
1464
- }
1465
-
1466
- const pollBalance = async () => {
1467
- const provider = getBalanceProvider(balanceProviderId())
1468
- // 手动配置的 key 优先;缺失时自动复用 OpenCode 已认证的 key(auth.json / config)
1469
- const key = api.kv.get<string>(`${KV_PREFIX}.balance.${provider.id}.key`, "")
1470
- || findOpencodeKey(api, provider)
1471
- if (balanceUnsupported()) { setBalanceState({ status: "idle", data: null, lastFetch: 0, error: undefined, key: undefined }); return }
1472
- if (!key) { setBalanceState({ status: "idle", data: null, lastFetch: 0, error: undefined, key: undefined }); return }
1473
- const now = Date.now()
1474
- const prev = balanceState()
1475
- // key 已更换(重新输入)→ 强制重新查询,绕过缓存
1476
- if (prev.status === "ok" && prev.key === key && now - prev.lastFetch < BALANCE_POLL_MS) return // cache still fresh
1477
- const seq = ++balanceSeq
1478
- setBalanceState({ ...prev, status: "loading", error: undefined, key })
1479
- const controller = new AbortController()
1480
- let timedOut = false
1481
- const timer = setTimeout(() => { timedOut = true; controller.abort() }, 10_000)
1482
- try {
1483
- const data = await provider.fetchBalance(key, controller.signal)
1484
- clearTimeout(timer)
1485
- if (seq !== balanceSeq) return // 已被更新的请求取代,丢弃过期结果
1486
- setBalanceState({ status: "ok", data, lastFetch: Date.now(), error: undefined, key })
1487
- } catch (err) {
1488
- clearTimeout(timer)
1489
- if (seq !== balanceSeq) return
1490
- const code = timedOut ? "TIMEOUT" : (err instanceof Error ? err.message : "")
1491
- // 失败时清空旧数据,避免显示过期余额
1492
- setBalanceState({ status: "error", data: null, lastFetch: 0, error: code, key })
1493
- }
1494
- }
1495
-
1496
- // Re-fetch when the API key is (re)configured via /cache-balance-key.
1497
- // 注意:pollBalance 内部读写 balanceState 信号,若不做 untrack 包裹,
1498
- // effect 会追踪 balanceState 的变化并与 pollBalance 的 setBalanceState
1499
- // 形成无限循环(每次重跑都发起新的 fetch 请求)。
1500
- createEffect(() => {
1501
- void balanceRefresh()
1502
- untrack(() => { void pollBalance() })
1503
- })
1504
-
1505
- // 定时轮询(5 分钟);随插件生命周期清理
1506
- const balanceTimer = setInterval(pollBalance, BALANCE_POLL_MS)
1507
- api.lifecycle.onDispose(() => clearInterval(balanceTimer))
1508
-
1509
- /** 菜单中 provider 选项标题:标注 key 来源(手动配置 / OpenCode 自动复用 / 未配置)。 */
1510
- const providerOptionTitle = (p: BalanceProvider, current?: string) => {
1511
- const t = createT(() => langCode())
1512
- const hasManual = !!api.kv.get<string>(`${KV_PREFIX}.balance.${p.id}.key`, "")
1513
- const hasAuto = !hasManual && !!findOpencodeKey(api, p)
1514
- const mark = hasManual
1515
- ? t("keyUser")
1516
- : hasAuto
1517
- ? t("keyOpenCode")
1518
- : t("keyNotSet")
1519
- return p.name + mark + (current && p.id === current ? " *" : "")
1520
- }
1521
-
1522
- /** 弹出指定 provider 的 API Key 输入框(脱敏预填;空清除 / 含 * 保留原 key / 新 key 实时刷新)。 */
1523
- const promptBalanceKey = (dialog: TuiDialogStack | undefined, provider: BalanceProvider) => {
1524
- const t = createT(() => langCode())
1525
- const current = api.kv.get<string>(`${KV_PREFIX}.balance.${provider.id}.key`, "")
1526
- const masked = maskKey(current)
1527
- dialog?.replace(() => (
1528
- <api.ui.DialogPrompt
1529
- title={provider.name}
1530
- description={() => <text>{t("balKeyPrompt", { p: provider.name })}</text>}
1531
- placeholder={provider.keyPlaceholder ?? "sk-..."}
1532
- value={masked}
1533
- onConfirm={(val) => {
1534
- const input = val.trim()
1535
- let key: string
1536
- if (input === "") {
1537
- key = ""
1538
- } else if (input.includes("*")) {
1539
- key = current
1540
- } else {
1541
- key = input
1542
- }
1543
- api.kv.set(`${KV_PREFIX}.balance.${provider.id}.key`, key)
1544
- setBalanceRefresh(v => v + 1)
1545
- if (key) {
1546
- api.ui.toast({ message: t("keySaved") })
1547
- } else {
1548
- api.ui.toast({ message: t("keyCleared") })
1549
- }
1550
- dialog?.clear()
1551
- }}
1552
- onCancel={() => dialog?.clear()}
1553
- />
1554
- ))
1555
- }
1556
-
1557
- api.command?.register(() => [
1558
- {
1559
- title: "Cache: Set Currency",
1560
- value: "cache.currency",
1561
- description: "Change the currency unit for cost display",
1562
- slash: { name: "cache-currency" },
1563
- onSelect: (dialog) => {
1564
- dialog?.replace(() => (
1565
- <api.ui.DialogSelect
1566
- title="Select Currency"
1567
- options={Object.entries(CURRENCIES).map(([code, sym]) => ({
1568
- title: `${code} (${sym})`,
1569
- value: code,
1570
- }))}
1571
- onSelect={(opt) => {
1572
- const t = createT(() => langCode())
1573
- const sym = CURRENCIES[opt.value] ?? "$"
1574
- const defRate = DEFAULT_RATES[opt.value] ?? 1
1575
- api.kv.set(`${KV_PREFIX}.currency`, sym)
1576
- api.kv.set(`${KV_PREFIX}.rate`, defRate)
1577
- // 同步余额显示币种偏好:CNY/USD 原生直显,其余币种按汇率换算
1578
- api.kv.set(`${KV_PREFIX}.balance_currency`, opt.value)
1579
- signals.setBalanceCurrency(opt.value)
1580
- signals.setCurrencySymbol(sym)
1581
- signals.setExchangeRate(defRate)
1582
- api.ui.toast({ message: t("currencySet", { v: opt.value, s: sym, r: defRate }) })
1583
- dialog?.clear()
1584
- }}
1585
- />
1586
- ))
1587
- },
1588
- },
1589
- {
1590
- title: "Cache: Set Exchange Rate",
1591
- value: "cache.rate",
1592
- description: "Set the exchange rate multiplier for the selected currency",
1593
- slash: { name: "cache-rate" },
1594
- onSelect: (dialog) => {
1595
- dialog?.replace(() => (
1596
- <api.ui.DialogPrompt
1597
- title="Exchange Rate"
1598
- description={() => <text>Enter the exchange rate from USD to your currency (e.g. 7.2 for CNY)</text>}
1599
- placeholder="1.0"
1600
- value={String(api.kv.get<number>(`${KV_PREFIX}.rate`, 1))}
1601
- onConfirm={(val) => {
1602
- const t = createT(() => langCode())
1603
- const n = parseFloat(val)
1604
- if (n > 0) {
1605
- api.kv.set(`${KV_PREFIX}.rate`, n)
1606
- signals.setExchangeRate(n)
1607
- api.ui.toast({ message: t("rateSet", { r: n }) })
1608
- }
1609
- dialog?.clear()
1610
- }}
1611
- />
1612
- ))
1613
- },
1614
- },
1615
- {
1616
- title: "Cache: Toggle Section",
1617
- value: "cache.section",
1618
- description: "Show or hide a sidebar section",
1619
- slash: { name: "cache-section" },
1620
- onSelect: (dialog) => {
1621
- const t = createT(() => langCode())
1622
- const detailOn = Boolean(api.kv.get(`${KV_PREFIX}.section.detail`, true))
1623
- const modelOn = Boolean(api.kv.get(`${KV_PREFIX}.section.model`, true))
1624
- const distOn = Boolean(api.kv.get(`${KV_PREFIX}.section.dist`, true))
1625
- const skillsOn = Boolean(api.kv.get(`${KV_PREFIX}.section.skills`, true))
1626
- const balanceOn = Boolean(api.kv.get(`${KV_PREFIX}.section.balance`, true))
1627
- const bottomOn = Boolean(api.kv.get(`${KV_PREFIX}.section.bottom`, true))
1628
- const borderOn = Boolean(api.kv.get(`${KV_PREFIX}.border`, true))
1629
- const labels: Record<string, string> = {
1630
- detail: t("secDetail"),
1631
- model: t("secModel"),
1632
- dist: t("distTitle"),
1633
- skills: t("secSkills"),
1634
- balance: t("secBalance"),
1635
- bottom: t("secBottom"),
1636
- border: t("secBorder"),
1637
- }
1638
- const optTitle = (label: string, on: boolean) => `${visualPadEnd(label, 15)}[${on ? "ON" : "OFF"}]`
1639
- dialog?.replace(() => (
1640
- <api.ui.DialogSelect
1641
- title={t("secToggle")}
1642
- options={[
1643
- { title: optTitle(labels.detail, detailOn), value: "detail" },
1644
- { title: optTitle(labels.model, modelOn), value: "model" },
1645
- { title: optTitle(labels.dist, distOn), value: "dist" },
1646
- { title: optTitle(labels.skills, skillsOn), value: "skills" },
1647
- { title: optTitle(labels.balance, balanceOn), value: "balance" },
1648
- { title: optTitle(labels.bottom, bottomOn), value: "bottom" },
1649
- { title: optTitle(labels.border, borderOn), value: "border" },
1650
- ]}
1651
- onSelect={(opt) => {
1652
- if (opt.value === "border") {
1653
- const cur = Boolean(api.kv.get(`${KV_PREFIX}.border`, true))
1654
- api.kv.set(`${KV_PREFIX}.border`, !cur)
1655
- signals.setBorderVisible(!cur)
1656
- api.ui.toast({ message: !cur ? t("borderShown") : t("borderHidden") })
1657
- } else {
1658
- const key = `${KV_PREFIX}.section.${opt.value}`
1659
- const cur = Boolean(api.kv.get(key, true))
1660
- api.kv.set(key, !cur)
1661
- if (opt.value === "detail") signals.setSectionDetail(!cur)
1662
- if (opt.value === "model") signals.setSectionModel(!cur)
1663
- if (opt.value === "dist") signals.setSectionDist(!cur)
1664
- if (opt.value === "skills") signals.setSectionSkills(!cur)
1665
- if (opt.value === "balance") signals.setSectionBalance(!cur)
1666
- if (opt.value === "bottom") signals.setSectionBottom(!cur)
1667
- const name = labels[opt.value] ?? opt.value
1668
- api.ui.toast({ message: t(!cur ? "sectionShown" : "sectionHidden", { s: name }) })
1669
- }
1670
- dialog?.clear()
1671
- }}
1672
- />
1673
- ))
1674
- },
1675
- },
1676
- {
1677
- title: "Cache: Show Config",
1678
- value: "cache.config",
1679
- description: "Display the current plugin configuration",
1680
- slash: { name: "cache-config" },
1681
- onSelect: (dialog) => {
1682
- const t = createT(() => langCode())
1683
- const sym = api.kv.get<string>(`${KV_PREFIX}.currency`) ?? "$"
1684
- const rate = api.kv.get<number>(`${KV_PREFIX}.rate`) ?? 1
1685
- const detail = Boolean(api.kv.get(`${KV_PREFIX}.section.detail`, true))
1686
- const model = Boolean(api.kv.get(`${KV_PREFIX}.section.model`, true))
1687
- const dist = Boolean(api.kv.get(`${KV_PREFIX}.section.dist`, true))
1688
- const skills = Boolean(api.kv.get(`${KV_PREFIX}.section.skills`, true))
1689
- const balance = Boolean(api.kv.get(`${KV_PREFIX}.section.balance`, true))
1690
- const bottom = Boolean(api.kv.get(`${KV_PREFIX}.section.bottom`, true))
1691
- const on = (v: boolean) => v ? "ON" : "OFF"
1692
- api.ui.toast({
1693
- title: t("panelConfigTitle"),
1694
- message: t("panelConfigMsg", {
1695
- c: sym, r: rate,
1696
- d: on(detail), m: on(model),
1697
- t: on(dist), k: on(skills),
1698
- b: on(balance), f: on(bottom),
1699
- }),
1700
- duration: 8000,
1701
- })
1702
- dialog?.clear()
1703
- },
1704
- },
1705
- {
1706
- title: "Cache: Switch Language",
1707
- value: "cache.lang",
1708
- description: "Switch between Chinese and English display",
1709
- slash: { name: "cache-lang" },
1710
- onSelect: (dialog) => {
1711
- const t = createT(() => langCode())
1712
- const cur = langCode()
1713
- dialog?.replace(() => (
1714
- <api.ui.DialogSelect
1715
- title={t("langTitle")}
1716
- options={LANG_META.map((m) => ({
1717
- title: `${visualPadEnd(m.label, 9)}${cur === m.code ? "\u2713" : ""}`,
1718
- value: m.code,
1719
- }))}
1720
- onSelect={(opt) => {
1721
- const code = opt.value as LangCode
1722
- api.kv.set(`${KV_PREFIX}.lang`, code)
1723
- setLangCode(code)
1724
- api.ui.toast({ message: t("langSwitched") })
1725
- dialog?.clear()
1726
- }}
1727
- />
1728
- ))
1729
- },
1730
- },
1731
- {
1732
- title: "Cache: Switch Balance Provider",
1733
- value: "cache.balance",
1734
- description: "切换余额提供商 / 自动切换当前会话提供商 | Switch balance provider / auto-switch session provider",
1735
- slash: { name: "cache-balance" },
1736
- onSelect: (dialog) => {
1737
- const t = createT(() => langCode())
1738
- const current = signals.balanceProviderId()
1739
- const auto = signals.autoBalance()
1740
- const autoLabel = `${t("autoSwitchOpt")} [${auto ? "ON" : "OFF"}]`
1741
- dialog?.replace(() => (
1742
- <api.ui.DialogSelect
1743
- title={t("balProvTitle")}
1744
- options={[
1745
- {
1746
- title: autoLabel,
1747
- value: "__auto__",
1748
- },
1749
- ...balanceProviders.map((p) => ({
1750
- title: providerOptionTitle(p, current),
1751
- value: p.id,
1752
- })),
1753
- ]}
1754
- onSelect={(opt) => {
1755
- if (opt.value === "__auto__") {
1756
- const next = !auto
1757
- api.kv.set(`${KV_PREFIX}.balance.auto`, next)
1758
- signals.setAutoBalance(next)
1759
- api.ui.toast({ message: next ? t("autoSwitchOn") : t("autoSwitchOff") })
1760
- dialog?.clear()
1761
- } else {
1762
- const provider = getBalanceProvider(opt.value)
1763
- // 手动切换会关闭自动切换
1764
- api.kv.set(`${KV_PREFIX}.balance.provider`, provider.id)
1765
- api.kv.set(`${KV_PREFIX}.balance.auto`, false)
1766
- signals.setBalanceProviderId(provider.id)
1767
- signals.setAutoBalance(false)
1768
- signals.setBalanceUnsupported(false)
1769
- // 切换后立即按新 provider 刷新显示(无 key 时显示 idle,避免残留上一 provider 余额)
1770
- signals.setBalanceRefresh(signals.balanceRefresh() + 1)
1771
- const hasKey = !!api.kv.get<string>(`${KV_PREFIX}.balance.${provider.id}.key`, "")
1772
- if (!hasKey) {
1773
- // 未配置 key → 进入设置流程(对话框保持打开等待输入)
1774
- promptBalanceKey(dialog, provider)
1775
- } else {
1776
- api.ui.toast({ message: t("providerManual", { p: provider.name }) })
1777
- dialog?.clear()
1778
- }
1779
- }
1780
- }}
1781
- />
1782
- ))
1783
- },
1784
- },
1785
- {
1786
- title: "Cache: Set Balance API Key",
1787
- value: "cache.balance.key",
1788
- description: "Select a provider and set its API key for balance display",
1789
- slash: { name: "cache-balance-key" },
1790
- onSelect: (dialog) => {
1791
- const t = createT(() => langCode())
1792
- // 步骤 1:选择 provider
1793
- dialog?.replace(() => (
1794
- <api.ui.DialogSelect
1795
- title={t("balSelectTitle")}
1796
- options={balanceProviders.map((p) => ({
1797
- title: providerOptionTitle(p),
1798
- value: p.id,
1799
- }))}
1800
- onSelect={(opt) => {
1801
- const provider = getBalanceProvider(opt.value)
1802
- // 手动指定 provider 会关闭自动切换
1803
- api.kv.set(`${KV_PREFIX}.balance.provider`, provider.id)
1804
- api.kv.set(`${KV_PREFIX}.balance.auto`, false)
1805
- signals.setBalanceProviderId(provider.id)
1806
- signals.setAutoBalance(false)
1807
- // 切换后立即刷新显示(防止取消输入时残留上一 provider 的余额)
1808
- signals.setBalanceRefresh(signals.balanceRefresh() + 1)
1809
- // 步骤 2:输入 key
1810
- promptBalanceKey(dialog, provider)
1811
- }}
1812
- />
1813
- ))
1814
- },
1815
- },
1816
- {
1817
- title: "Cache: Debug Skills Detection",
1818
- value: "cache.debug-skills",
1819
- description: "Dump all tool parts found in the current session for skill detection debugging",
1820
- slash: { name: "cache-debug-skills" },
1821
- onSelect: () => {
1822
- const t = createT(() => langCode())
1823
- const rt = api.route.current
1824
- if (rt.name !== "session" || !rt.params) {
1825
- api.ui.toast({ message: t("runInSession"), variant: "warning" })
1826
- return
1827
- }
1828
- const sid = String(rt.params.sessionID)
1829
- const msgs = api.state.session.messages(sid)
1830
- const byTool: Record<string, number> = {}
1831
- const skillParts: string[] = []
1832
- for (const msg of msgs) {
1833
- if (msg.role !== "assistant") continue
1834
- let parts: readonly any[] = []
1835
- try { parts = api.state.part(msg.id) } catch {}
1836
- for (const p of parts) {
1837
- if (p.type === "tool") {
1838
- const t = String(p.tool ?? "?")
1839
- byTool[t] = (byTool[t] ?? 0) + 1
1840
- if (t === "skill") {
1841
- const meta = p.state?.metadata
1842
- const rootMeta = p.metadata
1843
- skillParts.push(`state.metadata=${JSON.stringify(meta)} | root.metadata=${JSON.stringify(rootMeta)} | state.title="${p.state?.title}" | state.output[:80]="${String(p.state?.output ?? "").slice(0, 80)}"`)
1844
- }
1845
- }
1846
- }
1847
- }
1848
- const summary = Object.entries(byTool).map(([k, v]) => `${k}: ${v}`).join(" | ")
1849
- const extra = skillParts.length > 0 ? "\n\nSkill parts:\n" + skillParts.join("\n") : "\n\n⚠ No skill tool parts found — AI may be reading SKILL.md instead. Try: 'Use the skill tool to load karpathy-guidelines'"
1850
- api.ui.toast({
1851
- title: `Tool Summary (${Object.keys(byTool).length} types)`,
1852
- message: summary + extra,
1853
- duration: 15000,
1854
- })
1855
- },
1856
- },
1857
- {
1858
- title: "Cache: Sub-Agent Stats",
1859
- value: "cache.session",
1860
- description: "View token cache statistics for a sub-agent by session ID",
1861
- slash: { name: "cache-session" },
1862
- onSelect: (dialog) => {
1863
- // ── 扫描当前主 session 的子代理 session ID 列表 ──
1864
- const rt = api.route.current
1865
- const parentSid = rt.name === "session" && rt.params ? String(rt.params.sessionID) : ""
1866
- const SUBAGENT_TOOLS = new Set(["task", "delegate", "call_omo_agent"])
1867
-
1868
- interface ChildEntry { title: string; value: string; description: string }
1869
- const children: ChildEntry[] = []
1870
- if (parentSid) {
1871
- try {
1872
- const msgs = api.state.session.messages(parentSid)
1873
- for (const msg of msgs) {
1874
- if (msg.role !== "assistant") continue
1875
- let parts: readonly Part[] = []
1876
- try { parts = api.state.part(msg.id) } catch {}
1877
- for (const p of parts) {
1878
- if (p.type !== "tool") continue
1879
- const tool = String((p as ToolPart).tool ?? "")
1880
- if (!SUBAGENT_TOOLS.has(tool)) continue
1881
- const st = (p as any).state as Record<string, unknown> | undefined
1882
- const stMeta = st?.metadata as Record<string, unknown> | undefined
1883
- const subSid = stMeta?.session_id ?? stMeta?.sessionId
1884
- if (!subSid) continue
1885
- const sidStr = String(subSid)
1886
- const input = st?.input as Record<string, unknown> | undefined
1887
- const agent = String((p as any).subagent_type ?? input?.subagent_type ?? input?.category ?? tool)
1888
- const prompt = String(input?.prompt ?? "")
1889
- const desc = input?.description ? String(input.description) : ""
1890
- const title = desc || prompt.replace(/\n/g, " ").replace(/\s+/g, " ").trim().slice(0, 40) || agent
1891
- children.push({ title, value: sidStr, description: `${agent} · ${sidStr.slice(0, 24)}…` })
1892
- }
1893
- }
1894
- } catch {}
1895
- }
1896
-
1897
- // 去重
1898
- const seen = new Set<string>()
1899
- const unique = children.filter(c => { if (seen.has(c.value)) return false; seen.add(c.value); return true })
1900
-
1901
- if (unique.length > 0) {
1902
- // ── 有子代理 → DialogSelect 列表选择 ──
1903
- const t = createT(() => langCode())
1904
- const currentSid = signals.overrideSessionId() ?? api.kv.get<string>(`${KV_PREFIX}.session`, "")
1905
- const options = unique.map((c, i) => ({
1906
- title: `${i + 1}. ${c.title}`,
1907
- value: c.value,
1908
- description: c.description,
1909
- }))
1910
- // 首尾各放一个"回到主会话",长列表时顶部底部均可直达
1911
- const backValue = "__main__"
1912
- const backTitle = `\u2500 ${t("backToMainTitle")}`
1913
- options.unshift({ title: backTitle, value: backValue, description: "" })
1914
- options.push({ title: backTitle, value: backValue, description: "" })
1915
- const currentIdx = currentSid ? options.findIndex(o => o.value === currentSid) : -1
1916
- dialog?.replace(() => (
1917
- <api.ui.DialogSelect
1918
- title={t("subSelectTitle")}
1919
- options={options}
1920
- current={currentIdx >= 0 ? options[currentIdx].value : undefined}
1921
- onSelect={(opt) => {
1922
- if (opt.value === backValue) {
1923
- signals.setOverrideSessionId(undefined)
1924
- api.kv.set(`${KV_PREFIX}.session`, "")
1925
- api.ui.toast({ message: t("backToMain") })
1926
- } else {
1927
- signals.setOverrideSessionId(opt.value)
1928
- api.kv.set(`${KV_PREFIX}.session`, opt.value)
1929
- api.ui.toast({ message: t("subAgentSwitched", { s: opt.value.slice(0, 24) + "\u2026" }) })
1930
- }
1931
- dialog?.clear()
1932
- }}
1933
- />
1934
- ))
1935
- } else {
1936
- // ── 无子代理 → DialogPrompt 手动粘贴 ──
1937
- const t = createT(() => langCode())
1938
- dialog?.replace(() => (
1939
- <api.ui.DialogPrompt
1940
- title={signals.overrideSessionId() ? t("subSwitchTitle") : t("subViewTitle")}
1941
- description={() => <text>{t("subNoFound")}</text>}
1942
- placeholder="ses_..."
1943
- value={signals.overrideSessionId() ?? api.kv.get<string>(`${KV_PREFIX}.session`, "") ?? ""}
1944
- onConfirm={(val) => {
1945
- const sid = val.trim()
1946
- if (sid) {
1947
- signals.setOverrideSessionId(sid)
1948
- api.kv.set(`${KV_PREFIX}.session`, sid)
1949
- api.ui.toast({ message: t("subAgentSwitched", { s: sid.slice(0, 24) + "\u2026" }) })
1950
- }
1951
- dialog?.clear()
1952
- }}
1953
- onCancel={() => dialog?.clear()}
1954
- />
1955
- ))
1956
- }
1957
- },
1958
- },
1959
- {
1960
- title: "Cache: Back to Main",
1961
- value: "cache.session.back",
1962
- description: "Return to main session stats",
1963
- slash: { name: "cache-session-back" },
1964
- onSelect: (dialog) => {
1965
- const t = createT(() => langCode())
1966
- signals.setOverrideSessionId(undefined)
1967
- api.kv.set(`${KV_PREFIX}.session`, "")
1968
- api.ui.toast({ message: t("backToMain") })
1969
- dialog?.clear()
1970
- },
1971
- },
1972
- ])
1973
- }
1974
-
1975
- const mod: TuiPluginModule & { id: string } = {
1976
- id: "opencode-visual-cache",
1977
- tui,
1978
- }
1979
-
1980
- export default mod
1
+ /** @jsxImportSource @opentui/solid */
2
+
3
+ import type { JSX } from "@opentui/solid"
4
+ import type {
5
+ TuiPlugin,
6
+ TuiPluginApi,
7
+ TuiSlotContext,
8
+ TuiSlotPlugin,
9
+ TuiPluginModule,
10
+ TuiThemeCurrent,
11
+ TuiDialogStack,
12
+ TuiPromptRef,
13
+ SequenceBindingLike,
14
+ } from "@opencode-ai/plugin/tui"
15
+ import type { UserMessage, AssistantMessage, Message } from "@opencode-ai/sdk"
16
+ import type {
17
+ Part,
18
+ TextPart,
19
+ ToolPart,
20
+ FilePart,
21
+ ReasoningPart,
22
+ } from "@opencode-ai/sdk/v2"
23
+ import { createMemo, createSignal, createEffect, onMount, onCleanup, Show, For, untrack } from "solid-js"
24
+ import { PLUGIN_VERSION } from "./_version"
25
+ import { balanceProviders, getBalanceProvider, maskKey, matchBalanceProvider, type BalanceEntry, type BalanceProvider } from "./balance-providers"
26
+ import { LANG_META, createT, detectLang, type LangCode } from "./i18n"
27
+ import {
28
+ MAX_SAT, FALLBACK, CURRENCIES, DEFAULT_RATES,
29
+ charColumns, visualWidth, visualPadEnd, truncateVisual, progressBar,
30
+ fmt, num, fmtCost, fmtCompact, formatBalanceAmount, formatBalanceText,
31
+ balanceSymbol, convertBalance, estimateTokens,
32
+ rgb, saturation, desaturateTo, dimColor,
33
+ type TokenDist,
34
+ } from "./core"
35
+ import { TokenCachePanel } from "./panel/TokenCachePanel"
36
+ import type { PanelSignals, BalanceState } from "./panel/panel-api"
37
+
38
+ // ---------------------------------------------------------------------------
39
+ // Helpers
40
+ // ---------------------------------------------------------------------------
41
+
42
+ // Bun / Node globals available at runtime in the OpenCode TUI process
43
+ declare const process: { env: Record<string, string | undefined> } | undefined
44
+ // ── language ──────────────────────────────────────────────────────
45
+ // 语言初始化:环境变量 CACHE_TUI_LANG 覆盖 否则按系统 locale 自动检测。
46
+ // 用户通过 /cache-lang 设置的偏好会在 KV 就绪后优先覆盖(见 tui() 内恢复逻辑)。
47
+
48
+ const DEBUG_LANG = typeof process !== "undefined" ? process.env?.CACHE_TUI_LANG : undefined
49
+ const INIT_LANG: LangCode = DEBUG_LANG !== undefined && LANG_META.some((m) => m.code === DEBUG_LANG)
50
+ ? (DEBUG_LANG as LangCode)
51
+ : detectLang()
52
+
53
+ // ---------------------------------------------------------------------------
54
+ // Balance state
55
+ // ---------------------------------------------------------------------------
56
+
57
+
58
+ const BALANCE_POLL_MS = 5 * 60 * 1000 // 5 minutes
59
+
60
+ /**
61
+ * OpenCode 已认证的 provider 读取 API key 作为余额查询的自动兜底。
62
+ * 匹配复用前缀逻辑:先精确匹配 id,再前缀匹配(如 moonshotai-cn → moonshot)。
63
+ * key 来源:auth.json(provider.key)或配置(provider.options.apiKey)。
64
+ * 仅当手动配置的 key 缺失时使用;读取失败或未匹配返回空串。
65
+ */
66
+ function findOpencodeKey(api: TuiPluginApi, provider: BalanceProvider): string {
67
+ try {
68
+ const provs = api.state.provider as unknown as Array<{ id: string; key?: string; options?: { apiKey?: string } }>
69
+ // 大小写不敏感:精确匹配 id,否则前缀匹配(如 moonshotai-cn moonshot)
70
+ const id = provider.id.toLowerCase()
71
+ const hit = provs.find((p) => p.id.toLowerCase() === id) ?? provs.find((p) => p.id.toLowerCase().startsWith(id))
72
+ if (!hit) return ""
73
+ const k = typeof hit.key === "string" ? hit.key : ""
74
+ if (k) return k
75
+ return typeof hit.options?.apiKey === "string" ? hit.options.apiKey : ""
76
+ } catch {
77
+ return ""
78
+ }
79
+ }
80
+
81
+ // ---------------------------------------------------------------------------
82
+ // Sidebar component
83
+ // ---------------------------------------------------------------------------
84
+
85
+ /** Signals shared between the TUI component and slash commands.
86
+ * Created in the `tui` function scope so they do not survive module reload —
87
+ * the component re-creates them on mount and restores user config from kv. */
88
+
89
+ const MIN_PANEL_WIDTH = 20
90
+ const DEFAULT_PANEL_WIDTH = 26
91
+
92
+ /** ── layout measurement constants (visual columns) ── */
93
+ const LABEL_GAP = 1 // label(如 "Hit")后面的空格
94
+ const BAR_BRACKETS = 2 // "[" + "]" 包围进度条
95
+ const BAR_GAP = 1 // "]" 后面的空格
96
+ const PCT_FIXED_WIDTH = 5 // "XX.X%" 固定 5 字符宽度
97
+ const HEADER_PREFIX = 2 // 折叠态标题行:▶/▼ 图标 + 后面的空格
98
+ const UNIT_GAP = 1 // 计量单位前的空格(如 "tok")
99
+
100
+
101
+
102
+
103
+ // ---------------------------------------------------------------------------
104
+ // Plugin entry
105
+ // ---------------------------------------------------------------------------
106
+
107
+ /**
108
+ * 路径截断的固定开销(列数):仅宿主布局常量,不含任何内容宽度——
109
+ * 统计宽度、宿主 usage、commands 快捷键均为运行时动态计算。
110
+ * - marginLeft=1 + space-between 余量 ≈ 2 列
111
+ */
112
+ const PATH_CHROME = 2
113
+ /**
114
+ * 路径可用宽度低于此列数时整体隐藏(极窄终端下路径信息价值太低,
115
+ * 残留的 "E:\Work…" 反而挤压右侧统计与 commands,直接让位更干净)。
116
+ */
117
+ const HIDE_PATH_BELOW = 14
118
+
119
+ /**
120
+ * 从宿主 keymap 动态读取命令的快捷键显示文本(与宿主 Prompt 同源),
121
+ * 取不到时回退到传入的默认文本。
122
+ */
123
+ function keyShortcut(api: TuiPluginApi, command: string, fallback: string): string {
124
+ try {
125
+ const binds = api.tuiConfig.keybinds.get(command)
126
+ const seq = binds?.map((b) => ({ key: b.key }))
127
+ const s = api.keys.formatBindings(seq as unknown as SequenceBindingLike[])
128
+ return s || fallback
129
+ } catch {
130
+ return fallback
131
+ }
132
+ }
133
+
134
+ /**
135
+ * 输入框 hint 行(session_prompt slot 的 hint):单行显示 路径 · 命中率 · 余额 · Tokens。
136
+ * 通过 ui.Prompt 的 hint prop 注入——宿主右侧的 token/commands 提示自动保留,
137
+ * 三合一信息与路径同行显示在中间位置。
138
+ */
139
+ function BottomStatusBar(props: { api: TuiPluginApi; signals: PanelSignals; sessionId: string }): JSX.Element {
140
+ const KV_PREFIX = "cache_panel"
141
+ const t = createT(() => props.signals.langCode())
142
+
143
+ const sid = props.sessionId
144
+
145
+ // ── 命中率(单条口径:最后一条有 token 的 assistant 消息)+ token 汇总 ──
146
+ const stats = createMemo(() => {
147
+ const id = sid
148
+ if (!id) return null
149
+ const msgs = props.api.state.session.messages(id) as Message[]
150
+ const session = typeof props.api.state.session.get === "function"
151
+ ? props.api.state.session.get(id)
152
+ : undefined
153
+ let input = session?.tokens?.input ?? 0
154
+ let read = session?.tokens?.cache?.read ?? 0
155
+ let write = session?.tokens?.cache?.write ?? 0
156
+ // 旧 SDK 无 session 聚合字段 → 遍历消息累加(与侧边栏 fallback 一致)
157
+ if (session?.tokens == null) {
158
+ for (const m of msgs) {
159
+ if (m.role !== "assistant") continue
160
+ const tk = (m as AssistantMessage).tokens
161
+ if (!tk) continue
162
+ input += num(tk.input)
163
+ read += num(tk.cache?.read)
164
+ write += num(tk.cache?.write)
165
+ }
166
+ }
167
+ // 从后往前取最后两条有 token 数据的 assistant 消息 → 单条命中率 + 趋势
168
+ // 分母含缓存写(业界口径:read / (input+read+write)
169
+ let hitRate = -1, prevHitRate = -1
170
+ for (let i = msgs.length - 1; i >= 0; i--) {
171
+ const m = msgs[i]
172
+ if (m.role !== "assistant") continue
173
+ const tk = (m as AssistantMessage).tokens
174
+ if (!tk) continue
175
+ const mit = num(tk.input) + num(tk.cache?.read) + num(tk.cache?.write)
176
+ const mrt = num(tk.cache?.read)
177
+ if (mit <= 0) continue
178
+ const rate = (mrt / mit) * 100
179
+ if (hitRate < 0) { hitRate = rate; continue }
180
+ prevHitRate = rate
181
+ break
182
+ }
183
+ return { hitRate, prevHitRate, input, read, write }
184
+ })
185
+
186
+ // 余额查询状态为共享信号(PanelSignals.balanceState),由 tui() 统一轮询
187
+
188
+ // 自动切换 provider(跟随当前会话模型;幂等,与侧边栏共享信号)
189
+ createEffect(() => {
190
+ if (!props.signals.autoBalance()) return
191
+ const id = sid
192
+ if (!id) return
193
+ const msgs = props.api.state.session.messages(id) as Message[]
194
+ let pid = ""
195
+ for (let i = msgs.length - 1; i >= 0; i--) {
196
+ const m = msgs[i]
197
+ if (m.role === "assistant" && (m as AssistantMessage).providerID) { pid = (m as AssistantMessage).providerID; break }
198
+ }
199
+ if (!pid) {
200
+ try { pid = props.api.state.session.get(id)?.model?.providerID ?? "" } catch {}
201
+ }
202
+ if (!pid) return
203
+ const hit = matchBalanceProvider(pid)
204
+ if (hit) {
205
+ props.signals.setBalanceUnsupported(false)
206
+ if (hit.id !== props.signals.balanceProviderId()) {
207
+ props.signals.setBalanceProviderId(hit.id)
208
+ props.signals.setBalanceRefresh(props.signals.balanceRefresh() + 1)
209
+ }
210
+ } else {
211
+ // 当前提供商没有余额适配器 标记不支持,余额显示 N/A 并停止轮询
212
+ props.signals.setBalanceUnsupported(true)
213
+ }
214
+ })
215
+
216
+ // ── 主题色(与侧边栏同口径)──
217
+ const pal = createMemo(() => {
218
+ const th = props.api.theme.current as Record<string, unknown>
219
+ const sat = (k: string, fb: string) => desaturateTo(th[k], MAX_SAT, fb)
220
+ return {
221
+ text: sat("text", FALLBACK.text),
222
+ muted: sat("textMuted", FALLBACK.muted),
223
+ success: sat("success", FALLBACK.success),
224
+ warning: sat("warning", FALLBACK.warning),
225
+ error: sat("error", FALLBACK.error),
226
+ }
227
+ })
228
+
229
+ const hitColor = createMemo(() => {
230
+ const r = stats()?.hitRate ?? -1
231
+ if (r >= 85) return pal().success
232
+ if (r >= 70) return pal().warning
233
+ return pal().error
234
+ })
235
+
236
+ // 命中率趋势:最后一条与上一条的差值;|Δ| < 0.05 视为无变化(null = 不显示)
237
+ const trend = createMemo(() => {
238
+ const s = stats()
239
+ if (!s || s.prevHitRate < 0 || s.hitRate < 0) return null
240
+ const d = s.hitRate - s.prevHitRate
241
+ return Math.abs(d) < 0.05 ? null : d
242
+ })
243
+
244
+ const balanceText = createMemo(() => {
245
+ const s = props.signals.balanceState()
246
+ if (s.status === "ok" && s.data) return formatBalanceText(s.data, props.signals.balanceCurrency(), props.signals.exchangeRate())
247
+ if (s.status === "loading") return "\u2026"
248
+ if (s.status === "error") return "\u26a0"
249
+ return "-"
250
+ })
251
+
252
+ // 路径显示(替换宿主默认 hint 左侧的 cwd 文本)
253
+ const directory = createMemo(() => {
254
+ try { return props.api.state.path.directory } catch { return "" }
255
+ })
256
+
257
+ // 终端宽度信号:初始读取渲染器,窗口 resize 时更新(宿主不约束 hint 行宽度,
258
+ // 路径截断必须基于终端宽度手动计算)。
259
+ // CliRenderer 继承的 EventEmitter 因项目未装 @types/node 类型不可见,
260
+ // 用最小接口声明补齐 resize 事件的 on/off。
261
+ interface ResizeEmitter {
262
+ on(event: "resize", cb: () => void): unknown
263
+ off(event: "resize", cb: () => void): unknown
264
+ }
265
+ const [termW, setTermW] = createSignal(props.api.renderer.terminalWidth)
266
+ // resize 事件(主通道);事件接口若在插件环境不可用则跳过,由轮询兜底
267
+ createEffect(() => {
268
+ const r = props.api.renderer as unknown as ResizeEmitter
269
+ if (typeof r.on !== "function" || typeof r.off !== "function") return
270
+ const onResize = () => setTermW(props.api.renderer.terminalWidth)
271
+ r.on("resize", onResize)
272
+ onCleanup(() => r.off("resize", onResize))
273
+ })
274
+ // 轮询兜底:事件通道若在插件环境不可用,定期同步终端宽度(值不变时不触发更新)
275
+ createEffect(() => {
276
+ const timer = setInterval(() => setTermW(props.api.renderer.terminalWidth), 500)
277
+ onCleanup(() => clearInterval(timer))
278
+ })
279
+
280
+ // 统计部分分段(单一数据源):量宽拼接 text,渲染逐段着色,避免双源漂移
281
+ const statsSegs = createMemo<{ text: string; color: string | undefined }[]>(() => {
282
+ const s = stats()
283
+ const hr = s && s.hitRate >= 0 ? (Math.floor(s.hitRate * 10) / 10).toFixed(1) + "%" : "--"
284
+ const segs: { text: string; color: string | undefined }[] = [
285
+ { text: t("barHit") + " ", color: pal().muted },
286
+ { text: hr, color: hitColor() },
287
+ ]
288
+ const tr = trend()
289
+ if (tr !== null) {
290
+ segs.push({ text: " " + (tr > 0 ? "\u2191" : "\u2193") + Math.abs(tr).toFixed(1) + "%", color: tr > 0 ? pal().success : pal().error })
291
+ }
292
+ segs.push({ text: " \u00b7 " + t("barTok") + " ", color: pal().muted })
293
+ segs.push({ text: s ? fmtCompact(s.input + s.read + s.write) : "--", color: pal().text })
294
+ if (!props.signals.balanceUnsupported()) {
295
+ segs.push({ text: " \u00b7 " + t("barBal") + " ", color: pal().muted })
296
+ segs.push({ text: balanceText(), color: pal().text })
297
+ }
298
+ segs.push({ text: " \u00b7 ", color: pal().muted })
299
+ return segs
300
+ })
301
+ const statsW = createMemo(() => {
302
+ let w = 0
303
+ for (const sg of statsSegs()) w += visualWidth(sg.text)
304
+ return w
305
+ })
306
+
307
+ // 宿主右侧 usage 文本复刻(1.18.16 Prompt 口径):
308
+ // 最后一条 output>0 的 assistant 消息 → tokens 合计格式化 + 模型 context limit 百分比 + session 累计费用
309
+ const sessionCost = createMemo(() => {
310
+ try { return num(props.api.state.session.get(sid)?.cost) } catch { return 0 }
311
+ })
312
+ const usageText = createMemo(() => {
313
+ const id = sid
314
+ if (!id) return ""
315
+ const msgs = props.api.state.session.messages(id) as Message[]
316
+ let last: AssistantMessage | undefined
317
+ for (let i = msgs.length - 1; i >= 0; i--) {
318
+ const m = msgs[i]
319
+ if (m.role !== "assistant") continue
320
+ const tk = (m as AssistantMessage).tokens
321
+ if (tk && num(tk.output) > 0) { last = m as AssistantMessage; break }
322
+ }
323
+ if (!last) return ""
324
+ const tk = last.tokens
325
+ if (!tk) return ""
326
+ const tokens = num(tk.input) + num(tk.output) + num(tk.reasoning) + num(tk.cache?.read) + num(tk.cache?.write)
327
+ if (tokens <= 0) return ""
328
+ let pct = ""
329
+ try {
330
+ const p = props.api.state.provider.find((x) => x.id === last.providerID)
331
+ const limit = p?.models?.[last.modelID]?.limit?.context
332
+ if (typeof limit === "number" && limit > 0) pct = ` (${Math.round((tokens / limit) * 100)}%)`
333
+ } catch {}
334
+ const context = fmtCompact(tokens) + pct
335
+ const cost = sessionCost()
336
+ return cost > 0 ? context + " \u00b7 " + fmtCost(cost) : context
337
+ })
338
+
339
+ // 宿主 1513 行右侧文本:usage(有数据)或 "快捷键 agents"(无数据)+ commands,
340
+ // 快捷键从宿主 keymap 动态读取
341
+ const rightText = createMemo(() => {
342
+ const cmds = keyShortcut(props.api, "command.palette.show", "ctrl+p") + " commands"
343
+ const u = usageText()
344
+ if (u) return u + " " + cmds
345
+ return keyShortcut(props.api, "agent.cycle", "") + " agents " + cmds
346
+ })
347
+ const rightW = createMemo(() => visualWidth(rightText()))
348
+
349
+ // 输入框实际宽度 = 终端宽度 - 侧边栏(可见时 42) - 边距 4
350
+ // (与宿主 session 布局 contentWidth 口径一致;侧边栏可见性由本面板挂载状态驱动)
351
+ const inputW = createMemo(() => termW() - (props.signals.sidebarVisible() ? 42 : 0) - 4)
352
+
353
+ // 路径可用宽度 = 输入框宽度 - 统计宽度(精确) - 宿主右侧宽度(动态) - 布局开销;
354
+ // 低于 HIDE_PATH_BELOW 时整体隐藏路径(宽度归零),把空间让给统计与 commands
355
+ const dirDisplay = createMemo(() => {
356
+ const avail = inputW() - statsW() - rightW() - PATH_CHROME
357
+ if (avail < HIDE_PATH_BELOW) return ""
358
+ return truncateVisual(directory(), avail)
359
+ })
360
+ // 状态栏关闭(仅路径)时同样在极窄条件下隐藏路径
361
+ const dirFallback = createMemo(() => {
362
+ const avail = inputW() - rightW() - PATH_CHROME
363
+ if (avail < HIDE_PATH_BELOW) return ""
364
+ return truncateVisual(directory(), avail)
365
+ })
366
+
367
+ // 恢复显隐偏好(默认显示);关闭时回退为仅显示路径,与宿主默认 hint 行一致
368
+ onMount(() => {
369
+ try {
370
+ const v = props.api.kv.get<boolean>(`${KV_PREFIX}.section.bottom`, true)
371
+ props.signals.setSectionBottom(v !== false)
372
+ } catch {}
373
+ })
374
+
375
+ return (
376
+ <Show
377
+ when={props.signals.sectionBottom()}
378
+ fallback={<text fg={pal().muted}>{dirFallback()}</text>}
379
+ >
380
+ <box marginLeft={1} flexGrow={1} flexShrink={0} flexDirection="row" justifyContent="space-between">
381
+ <text fg={pal().muted}>{dirDisplay()}</text>
382
+ <box flexDirection="row">
383
+ <text>
384
+ <For each={statsSegs()}>
385
+ {(sg) => <span style={{ fg: sg.color }}>{sg.text}</span>}
386
+ </For>
387
+ </text>
388
+ </box>
389
+ </box>
390
+ </Show>
391
+ )
392
+ }
393
+ function createSidebarSlot(api: TuiPluginApi, signals: PanelSignals): TuiSlotPlugin {
394
+ let lastSlotSid = ""
395
+ return {
396
+ order: 55,
397
+ slots: {
398
+ sidebar_content(ctx: TuiSlotContext, input: { session_id: string }): JSX.Element {
399
+ // ── auto-clear override when the user navigates to a different main session ──
400
+ if (input.session_id !== lastSlotSid) {
401
+ lastSlotSid = input.session_id
402
+ if (signals.overrideSessionId()) {
403
+ signals.setOverrideSessionId(undefined)
404
+ api.kv.set("cache_panel.session", "")
405
+ }
406
+ }
407
+ return (
408
+ <TokenCachePanel
409
+ theme={ctx.theme.current}
410
+ api={api}
411
+ sessionId={input.session_id}
412
+ signals={signals}
413
+ />
414
+ )
415
+ },
416
+ },
417
+ }
418
+ }
419
+
420
+ const tui: TuiPlugin = async (api: TuiPluginApi) => {
421
+ // ── shared panel signals ──────────────────────────────────────
422
+ const [currencySymbol, setCurrencySymbol] = createSignal("$")
423
+ const [exchangeRate, setExchangeRate] = createSignal(1)
424
+ const [sectionDetail, setSectionDetail] = createSignal(true)
425
+ const [sectionModel, setSectionModel] = createSignal(true)
426
+ const [sectionDist, setSectionDist] = createSignal(true)
427
+ const [sectionSkills, setSectionSkills] = createSignal(true)
428
+ const [sectionBalance, setSectionBalance] = createSignal(true)
429
+ const [sectionBottom, setSectionBottom] = createSignal(true)
430
+ const [balanceRefresh, setBalanceRefresh] = createSignal(0)
431
+ const [balanceProviderId, setBalanceProviderId] = createSignal("deepseek")
432
+ const [autoBalance, setAutoBalance] = createSignal(true)
433
+ const [balanceUnsupported, setBalanceUnsupported] = createSignal(false)
434
+ const [balanceCurrency, setBalanceCurrency] = createSignal("")
435
+ const [borderVisible, setBorderVisible] = createSignal(true)
436
+ const [langCode, setLangCode] = createSignal<LangCode>(INIT_LANG)
437
+ const [overrideSessionId, setOverrideSessionId] = createSignal<string | undefined>(undefined)
438
+ // 侧边栏可见性(由 TokenCachePanel 挂载状态驱动):可见时宿主输入框宽度 = 终端宽 - 42 - 4
439
+ const [sidebarVisible, setSidebarVisible] = createSignal(false)
440
+
441
+ // ── 余额查询状态(共享):侧边栏与底部栏读同一份数据,
442
+ // 避免重复请求导致两处余额不一致 ──
443
+ const [balanceState, setBalanceState] = createSignal<BalanceState>({
444
+ status: "idle", data: null, lastFetch: 0,
445
+ })
446
+ // 请求序号:防止定时轮询与手动刷新并发时,慢的旧请求覆盖新结果
447
+ let balanceSeq = 0
448
+
449
+ const signals: PanelSignals = {
450
+ currencySymbol, setCurrencySymbol,
451
+ exchangeRate, setExchangeRate,
452
+ langCode, setLangCode,
453
+ sectionDetail, setSectionDetail,
454
+ sectionModel, setSectionModel,
455
+ sectionDist, setSectionDist,
456
+ sectionSkills, setSectionSkills,
457
+ sectionBalance, setSectionBalance,
458
+ sectionBottom, setSectionBottom,
459
+ balanceRefresh, setBalanceRefresh,
460
+ balanceProviderId, setBalanceProviderId,
461
+ autoBalance, setAutoBalance,
462
+ balanceUnsupported, setBalanceUnsupported,
463
+ balanceState,
464
+ balanceCurrency, setBalanceCurrency,
465
+ borderVisible, setBorderVisible,
466
+ overrideSessionId, setOverrideSessionId,
467
+ sidebarVisible, setSidebarVisible,
468
+ }
469
+
470
+ api.slots.register(createSidebarSlot(api, signals))
471
+
472
+ // 输入框 hint 行(session_prompt slot,replace 模式):
473
+ // 用宿主同一 Prompt 组件重渲染输入框,仅替换 hint 行左侧——
474
+ // 在路径与右侧 token/commands 提示之间插入 命中率 · 余额 · Tokens。
475
+ api.slots.register({
476
+ order: 55,
477
+ slots: {
478
+ session_prompt(
479
+ _ctx: TuiSlotContext,
480
+ input: {
481
+ session_id: string
482
+ visible?: boolean
483
+ disabled?: boolean
484
+ on_submit?: () => void
485
+ ref?: (ref: TuiPromptRef | undefined) => void
486
+ },
487
+ ): JSX.Element {
488
+ return (
489
+ <api.ui.Prompt
490
+ sessionID={input.session_id}
491
+ visible={input.visible}
492
+ disabled={input.disabled}
493
+ onSubmit={input.on_submit}
494
+ ref={input.ref}
495
+ hint={<BottomStatusBar api={api} signals={signals} sessionId={input.session_id} />}
496
+ />
497
+ )
498
+ },
499
+ },
500
+ })
501
+
502
+ // ── slash commands for runtime config ──
503
+ const KV_PREFIX = "cache_panel"
504
+
505
+ // ── 语言偏好恢复:KV 就绪后优先用户设置(/cache-lang),覆盖自动识别 ──
506
+ const restoreLang = () => {
507
+ try {
508
+ const saved = api.kv.get<string>(`${KV_PREFIX}.lang`)
509
+ if (saved && LANG_META.some((m) => m.code === saved)) setLangCode(saved as LangCode)
510
+ } catch {}
511
+ }
512
+ if (api.kv.ready) {
513
+ restoreLang()
514
+ } else {
515
+ const langTimer = setInterval(() => {
516
+ if (api.kv.ready) { clearInterval(langTimer); restoreLang() }
517
+ }, 10)
518
+ api.lifecycle.onDispose(() => clearInterval(langTimer))
519
+ }
520
+
521
+ const pollBalance = async () => {
522
+ const provider = getBalanceProvider(balanceProviderId())
523
+ // 手动配置的 key 优先;缺失时自动复用 OpenCode 已认证的 key(auth.json / config)
524
+ const key = api.kv.get<string>(`${KV_PREFIX}.balance.${provider.id}.key`, "")
525
+ || findOpencodeKey(api, provider)
526
+ if (balanceUnsupported()) { setBalanceState({ status: "idle", data: null, lastFetch: 0, error: undefined, key: undefined }); return }
527
+ if (!key) { setBalanceState({ status: "idle", data: null, lastFetch: 0, error: undefined, key: undefined }); return }
528
+ const now = Date.now()
529
+ const prev = balanceState()
530
+ // key 已更换(重新输入)→ 强制重新查询,绕过缓存
531
+ if (prev.status === "ok" && prev.key === key && now - prev.lastFetch < BALANCE_POLL_MS) return // cache still fresh
532
+ const seq = ++balanceSeq
533
+ setBalanceState({ ...prev, status: "loading", error: undefined, key })
534
+ const controller = new AbortController()
535
+ let timedOut = false
536
+ const timer = setTimeout(() => { timedOut = true; controller.abort() }, 10_000)
537
+ try {
538
+ const data = await provider.fetchBalance(key, controller.signal)
539
+ clearTimeout(timer)
540
+ if (seq !== balanceSeq) return // 已被更新的请求取代,丢弃过期结果
541
+ setBalanceState({ status: "ok", data, lastFetch: Date.now(), error: undefined, key })
542
+ } catch (err) {
543
+ clearTimeout(timer)
544
+ if (seq !== balanceSeq) return
545
+ const code = timedOut ? "TIMEOUT" : (err instanceof Error ? err.message : "")
546
+ // 失败时清空旧数据,避免显示过期余额
547
+ setBalanceState({ status: "error", data: null, lastFetch: 0, error: code, key })
548
+ }
549
+ }
550
+
551
+ // Re-fetch when the API key is (re)configured via /cache-balance-key.
552
+ // 注意:pollBalance 内部读写 balanceState 信号,若不做 untrack 包裹,
553
+ // effect 会追踪 balanceState 的变化并与 pollBalance 的 setBalanceState
554
+ // 形成无限循环(每次重跑都发起新的 fetch 请求)。
555
+ createEffect(() => {
556
+ void balanceRefresh()
557
+ untrack(() => { void pollBalance() })
558
+ })
559
+
560
+ // 定时轮询(5 分钟);随插件生命周期清理
561
+ const balanceTimer = setInterval(pollBalance, BALANCE_POLL_MS)
562
+ api.lifecycle.onDispose(() => clearInterval(balanceTimer))
563
+
564
+ /** 菜单中 provider 选项标题:标注 key 来源(手动配置 / OpenCode 自动复用 / 未配置)。 */
565
+ const providerOptionTitle = (p: BalanceProvider, current?: string) => {
566
+ const t = createT(() => langCode())
567
+ const hasManual = !!api.kv.get<string>(`${KV_PREFIX}.balance.${p.id}.key`, "")
568
+ const hasAuto = !hasManual && !!findOpencodeKey(api, p)
569
+ const mark = hasManual
570
+ ? t("keyUser")
571
+ : hasAuto
572
+ ? t("keyOpenCode")
573
+ : t("keyNotSet")
574
+ return p.name + mark + (current && p.id === current ? " *" : "")
575
+ }
576
+
577
+ /** 弹出指定 provider 的 API Key 输入框(脱敏预填;空清除 / 含 * 保留原 key / 新 key 实时刷新)。 */
578
+ const promptBalanceKey = (dialog: TuiDialogStack | undefined, provider: BalanceProvider) => {
579
+ const t = createT(() => langCode())
580
+ const current = api.kv.get<string>(`${KV_PREFIX}.balance.${provider.id}.key`, "")
581
+ const masked = maskKey(current)
582
+ dialog?.replace(() => (
583
+ <api.ui.DialogPrompt
584
+ title={provider.name}
585
+ description={() => <text>{t("balKeyPrompt", { p: provider.name })}</text>}
586
+ placeholder={provider.keyPlaceholder ?? "sk-..."}
587
+ value={masked}
588
+ onConfirm={(val) => {
589
+ const input = val.trim()
590
+ let key: string
591
+ if (input === "") {
592
+ key = ""
593
+ } else if (input.includes("*")) {
594
+ key = current
595
+ } else {
596
+ key = input
597
+ }
598
+ api.kv.set(`${KV_PREFIX}.balance.${provider.id}.key`, key)
599
+ setBalanceRefresh(v => v + 1)
600
+ if (key) {
601
+ api.ui.toast({ message: t("keySaved") })
602
+ } else {
603
+ api.ui.toast({ message: t("keyCleared") })
604
+ }
605
+ dialog?.clear()
606
+ }}
607
+ onCancel={() => dialog?.clear()}
608
+ />
609
+ ))
610
+ }
611
+
612
+ api.command?.register(() => [
613
+ {
614
+ title: "Cache: Set Currency",
615
+ value: "cache.currency",
616
+ description: "Change the currency unit for cost display",
617
+ slash: { name: "cache-currency" },
618
+ onSelect: (dialog) => {
619
+ dialog?.replace(() => (
620
+ <api.ui.DialogSelect
621
+ title="Select Currency"
622
+ options={Object.entries(CURRENCIES).map(([code, sym]) => ({
623
+ title: `${code} (${sym})`,
624
+ value: code,
625
+ }))}
626
+ onSelect={(opt) => {
627
+ const t = createT(() => langCode())
628
+ const sym = CURRENCIES[opt.value] ?? "$"
629
+ const defRate = DEFAULT_RATES[opt.value] ?? 1
630
+ api.kv.set(`${KV_PREFIX}.currency`, sym)
631
+ api.kv.set(`${KV_PREFIX}.rate`, defRate)
632
+ // 同步余额显示币种偏好:CNY/USD 原生直显,其余币种按汇率换算
633
+ api.kv.set(`${KV_PREFIX}.balance_currency`, opt.value)
634
+ signals.setBalanceCurrency(opt.value)
635
+ signals.setCurrencySymbol(sym)
636
+ signals.setExchangeRate(defRate)
637
+ api.ui.toast({ message: t("currencySet", { v: opt.value, s: sym, r: defRate }) })
638
+ dialog?.clear()
639
+ }}
640
+ />
641
+ ))
642
+ },
643
+ },
644
+ {
645
+ title: "Cache: Set Exchange Rate",
646
+ value: "cache.rate",
647
+ description: "Set the exchange rate multiplier for the selected currency",
648
+ slash: { name: "cache-rate" },
649
+ onSelect: (dialog) => {
650
+ dialog?.replace(() => (
651
+ <api.ui.DialogPrompt
652
+ title="Exchange Rate"
653
+ description={() => <text>Enter the exchange rate from USD to your currency (e.g. 7.2 for CNY)</text>}
654
+ placeholder="1.0"
655
+ value={String(api.kv.get<number>(`${KV_PREFIX}.rate`, 1))}
656
+ onConfirm={(val) => {
657
+ const t = createT(() => langCode())
658
+ const n = parseFloat(val)
659
+ if (n > 0) {
660
+ api.kv.set(`${KV_PREFIX}.rate`, n)
661
+ signals.setExchangeRate(n)
662
+ api.ui.toast({ message: t("rateSet", { r: n }) })
663
+ }
664
+ dialog?.clear()
665
+ }}
666
+ />
667
+ ))
668
+ },
669
+ },
670
+ {
671
+ title: "Cache: Toggle Section",
672
+ value: "cache.section",
673
+ description: "Show or hide a sidebar section",
674
+ slash: { name: "cache-section" },
675
+ onSelect: (dialog) => {
676
+ const t = createT(() => langCode())
677
+ const detailOn = Boolean(api.kv.get(`${KV_PREFIX}.section.detail`, true))
678
+ const modelOn = Boolean(api.kv.get(`${KV_PREFIX}.section.model`, true))
679
+ const distOn = Boolean(api.kv.get(`${KV_PREFIX}.section.dist`, true))
680
+ const skillsOn = Boolean(api.kv.get(`${KV_PREFIX}.section.skills`, true))
681
+ const balanceOn = Boolean(api.kv.get(`${KV_PREFIX}.section.balance`, true))
682
+ const bottomOn = Boolean(api.kv.get(`${KV_PREFIX}.section.bottom`, true))
683
+ const borderOn = Boolean(api.kv.get(`${KV_PREFIX}.border`, true))
684
+ const labels: Record<string, string> = {
685
+ detail: t("secDetail"),
686
+ model: t("secModel"),
687
+ dist: t("distTitle"),
688
+ skills: t("secSkills"),
689
+ balance: t("secBalance"),
690
+ bottom: t("secBottom"),
691
+ border: t("secBorder"),
692
+ }
693
+ const optTitle = (label: string, on: boolean) => `${visualPadEnd(label, 15)}[${on ? "ON" : "OFF"}]`
694
+ dialog?.replace(() => (
695
+ <api.ui.DialogSelect
696
+ title={t("secToggle")}
697
+ options={[
698
+ { title: optTitle(labels.detail, detailOn), value: "detail" },
699
+ { title: optTitle(labels.model, modelOn), value: "model" },
700
+ { title: optTitle(labels.dist, distOn), value: "dist" },
701
+ { title: optTitle(labels.skills, skillsOn), value: "skills" },
702
+ { title: optTitle(labels.balance, balanceOn), value: "balance" },
703
+ { title: optTitle(labels.bottom, bottomOn), value: "bottom" },
704
+ { title: optTitle(labels.border, borderOn), value: "border" },
705
+ ]}
706
+ onSelect={(opt) => {
707
+ if (opt.value === "border") {
708
+ const cur = Boolean(api.kv.get(`${KV_PREFIX}.border`, true))
709
+ api.kv.set(`${KV_PREFIX}.border`, !cur)
710
+ signals.setBorderVisible(!cur)
711
+ api.ui.toast({ message: !cur ? t("borderShown") : t("borderHidden") })
712
+ } else {
713
+ const key = `${KV_PREFIX}.section.${opt.value}`
714
+ const cur = Boolean(api.kv.get(key, true))
715
+ api.kv.set(key, !cur)
716
+ if (opt.value === "detail") signals.setSectionDetail(!cur)
717
+ if (opt.value === "model") signals.setSectionModel(!cur)
718
+ if (opt.value === "dist") signals.setSectionDist(!cur)
719
+ if (opt.value === "skills") signals.setSectionSkills(!cur)
720
+ if (opt.value === "balance") signals.setSectionBalance(!cur)
721
+ if (opt.value === "bottom") signals.setSectionBottom(!cur)
722
+ const name = labels[opt.value] ?? opt.value
723
+ api.ui.toast({ message: t(!cur ? "sectionShown" : "sectionHidden", { s: name }) })
724
+ }
725
+ dialog?.clear()
726
+ }}
727
+ />
728
+ ))
729
+ },
730
+ },
731
+ {
732
+ title: "Cache: Show Config",
733
+ value: "cache.config",
734
+ description: "Display the current plugin configuration",
735
+ slash: { name: "cache-config" },
736
+ onSelect: (dialog) => {
737
+ const t = createT(() => langCode())
738
+ const sym = api.kv.get<string>(`${KV_PREFIX}.currency`) ?? "$"
739
+ const rate = api.kv.get<number>(`${KV_PREFIX}.rate`) ?? 1
740
+ const detail = Boolean(api.kv.get(`${KV_PREFIX}.section.detail`, true))
741
+ const model = Boolean(api.kv.get(`${KV_PREFIX}.section.model`, true))
742
+ const dist = Boolean(api.kv.get(`${KV_PREFIX}.section.dist`, true))
743
+ const skills = Boolean(api.kv.get(`${KV_PREFIX}.section.skills`, true))
744
+ const balance = Boolean(api.kv.get(`${KV_PREFIX}.section.balance`, true))
745
+ const bottom = Boolean(api.kv.get(`${KV_PREFIX}.section.bottom`, true))
746
+ const on = (v: boolean) => v ? "ON" : "OFF"
747
+ api.ui.toast({
748
+ title: t("panelConfigTitle"),
749
+ message: t("panelConfigMsg", {
750
+ c: sym, r: rate,
751
+ d: on(detail), m: on(model),
752
+ t: on(dist), k: on(skills),
753
+ b: on(balance), f: on(bottom),
754
+ }),
755
+ duration: 8000,
756
+ })
757
+ dialog?.clear()
758
+ },
759
+ },
760
+ {
761
+ title: "Cache: Switch Language",
762
+ value: "cache.lang",
763
+ description: "Switch between Chinese and English display",
764
+ slash: { name: "cache-lang" },
765
+ onSelect: (dialog) => {
766
+ const t = createT(() => langCode())
767
+ const cur = langCode()
768
+ dialog?.replace(() => (
769
+ <api.ui.DialogSelect
770
+ title={t("langTitle")}
771
+ options={LANG_META.map((m) => ({
772
+ title: `${visualPadEnd(m.label, 9)}${cur === m.code ? "\u2713" : ""}`,
773
+ value: m.code,
774
+ }))}
775
+ onSelect={(opt) => {
776
+ const code = opt.value as LangCode
777
+ api.kv.set(`${KV_PREFIX}.lang`, code)
778
+ setLangCode(code)
779
+ api.ui.toast({ message: t("langSwitched") })
780
+ dialog?.clear()
781
+ }}
782
+ />
783
+ ))
784
+ },
785
+ },
786
+ {
787
+ title: "Cache: Switch Balance Provider",
788
+ value: "cache.balance",
789
+ description: "切换余额提供商 / 自动切换当前会话提供商 | Switch balance provider / auto-switch session provider",
790
+ slash: { name: "cache-balance" },
791
+ onSelect: (dialog) => {
792
+ const t = createT(() => langCode())
793
+ const current = signals.balanceProviderId()
794
+ const auto = signals.autoBalance()
795
+ const autoLabel = `${t("autoSwitchOpt")} [${auto ? "ON" : "OFF"}]`
796
+ dialog?.replace(() => (
797
+ <api.ui.DialogSelect
798
+ title={t("balProvTitle")}
799
+ options={[
800
+ {
801
+ title: autoLabel,
802
+ value: "__auto__",
803
+ },
804
+ ...balanceProviders.map((p) => ({
805
+ title: providerOptionTitle(p, current),
806
+ value: p.id,
807
+ })),
808
+ ]}
809
+ onSelect={(opt) => {
810
+ if (opt.value === "__auto__") {
811
+ const next = !auto
812
+ api.kv.set(`${KV_PREFIX}.balance.auto`, next)
813
+ signals.setAutoBalance(next)
814
+ api.ui.toast({ message: next ? t("autoSwitchOn") : t("autoSwitchOff") })
815
+ dialog?.clear()
816
+ } else {
817
+ const provider = getBalanceProvider(opt.value)
818
+ // 手动切换会关闭自动切换
819
+ api.kv.set(`${KV_PREFIX}.balance.provider`, provider.id)
820
+ api.kv.set(`${KV_PREFIX}.balance.auto`, false)
821
+ signals.setBalanceProviderId(provider.id)
822
+ signals.setAutoBalance(false)
823
+ signals.setBalanceUnsupported(false)
824
+ // 切换后立即按新 provider 刷新显示(无 key 时显示 idle,避免残留上一 provider 余额)
825
+ signals.setBalanceRefresh(signals.balanceRefresh() + 1)
826
+ const hasKey = !!api.kv.get<string>(`${KV_PREFIX}.balance.${provider.id}.key`, "")
827
+ if (!hasKey) {
828
+ // 未配置 key 进入设置流程(对话框保持打开等待输入)
829
+ promptBalanceKey(dialog, provider)
830
+ } else {
831
+ api.ui.toast({ message: t("providerManual", { p: provider.name }) })
832
+ dialog?.clear()
833
+ }
834
+ }
835
+ }}
836
+ />
837
+ ))
838
+ },
839
+ },
840
+ {
841
+ title: "Cache: Set Balance API Key",
842
+ value: "cache.balance.key",
843
+ description: "Select a provider and set its API key for balance display",
844
+ slash: { name: "cache-balance-key" },
845
+ onSelect: (dialog) => {
846
+ const t = createT(() => langCode())
847
+ // 步骤 1:选择 provider
848
+ dialog?.replace(() => (
849
+ <api.ui.DialogSelect
850
+ title={t("balSelectTitle")}
851
+ options={balanceProviders.map((p) => ({
852
+ title: providerOptionTitle(p),
853
+ value: p.id,
854
+ }))}
855
+ onSelect={(opt) => {
856
+ const provider = getBalanceProvider(opt.value)
857
+ // 手动指定 provider 会关闭自动切换
858
+ api.kv.set(`${KV_PREFIX}.balance.provider`, provider.id)
859
+ api.kv.set(`${KV_PREFIX}.balance.auto`, false)
860
+ signals.setBalanceProviderId(provider.id)
861
+ signals.setAutoBalance(false)
862
+ // 切换后立即刷新显示(防止取消输入时残留上一 provider 的余额)
863
+ signals.setBalanceRefresh(signals.balanceRefresh() + 1)
864
+ // 步骤 2:输入 key
865
+ promptBalanceKey(dialog, provider)
866
+ }}
867
+ />
868
+ ))
869
+ },
870
+ },
871
+ {
872
+ title: "Cache: Debug Skills Detection",
873
+ value: "cache.debug-skills",
874
+ description: "Dump all tool parts found in the current session for skill detection debugging",
875
+ slash: { name: "cache-debug-skills" },
876
+ onSelect: () => {
877
+ const t = createT(() => langCode())
878
+ const rt = api.route.current
879
+ if (rt.name !== "session" || !rt.params) {
880
+ api.ui.toast({ message: t("runInSession"), variant: "warning" })
881
+ return
882
+ }
883
+ const sid = String(rt.params.sessionID)
884
+ const msgs = api.state.session.messages(sid)
885
+ const byTool: Record<string, number> = {}
886
+ const skillParts: string[] = []
887
+ for (const msg of msgs) {
888
+ if (msg.role !== "assistant") continue
889
+ let parts: readonly any[] = []
890
+ try { parts = api.state.part(msg.id) } catch {}
891
+ for (const p of parts) {
892
+ if (p.type === "tool") {
893
+ const t = String(p.tool ?? "?")
894
+ byTool[t] = (byTool[t] ?? 0) + 1
895
+ if (t === "skill") {
896
+ const meta = p.state?.metadata
897
+ const rootMeta = p.metadata
898
+ skillParts.push(`state.metadata=${JSON.stringify(meta)} | root.metadata=${JSON.stringify(rootMeta)} | state.title="${p.state?.title}" | state.output[:80]="${String(p.state?.output ?? "").slice(0, 80)}"`)
899
+ }
900
+ }
901
+ }
902
+ }
903
+ const summary = Object.entries(byTool).map(([k, v]) => `${k}: ${v}`).join(" | ")
904
+ const extra = skillParts.length > 0 ? "\n\nSkill parts:\n" + skillParts.join("\n") : "\n\n⚠ No skill tool parts found — AI may be reading SKILL.md instead. Try: 'Use the skill tool to load karpathy-guidelines'"
905
+ api.ui.toast({
906
+ title: `Tool Summary (${Object.keys(byTool).length} types)`,
907
+ message: summary + extra,
908
+ duration: 15000,
909
+ })
910
+ },
911
+ },
912
+ {
913
+ title: "Cache: Sub-Agent Stats",
914
+ value: "cache.session",
915
+ description: "View token cache statistics for a sub-agent by session ID",
916
+ slash: { name: "cache-session" },
917
+ onSelect: (dialog) => {
918
+ // ── 扫描当前主 session 的子代理 session ID 列表 ──
919
+ const rt = api.route.current
920
+ const parentSid = rt.name === "session" && rt.params ? String(rt.params.sessionID) : ""
921
+ const SUBAGENT_TOOLS = new Set(["task", "delegate", "call_omo_agent"])
922
+
923
+ interface ChildEntry { title: string; value: string; description: string }
924
+ const children: ChildEntry[] = []
925
+ if (parentSid) {
926
+ try {
927
+ const msgs = api.state.session.messages(parentSid)
928
+ for (const msg of msgs) {
929
+ if (msg.role !== "assistant") continue
930
+ let parts: readonly Part[] = []
931
+ try { parts = api.state.part(msg.id) } catch {}
932
+ for (const p of parts) {
933
+ if (p.type !== "tool") continue
934
+ const tool = String((p as ToolPart).tool ?? "")
935
+ if (!SUBAGENT_TOOLS.has(tool)) continue
936
+ const st = (p as any).state as Record<string, unknown> | undefined
937
+ const stMeta = st?.metadata as Record<string, unknown> | undefined
938
+ const subSid = stMeta?.session_id ?? stMeta?.sessionId
939
+ if (!subSid) continue
940
+ const sidStr = String(subSid)
941
+ const input = st?.input as Record<string, unknown> | undefined
942
+ const agent = String((p as any).subagent_type ?? input?.subagent_type ?? input?.category ?? tool)
943
+ const prompt = String(input?.prompt ?? "")
944
+ const desc = input?.description ? String(input.description) : ""
945
+ const title = desc || prompt.replace(/\n/g, " ").replace(/\s+/g, " ").trim().slice(0, 40) || agent
946
+ children.push({ title, value: sidStr, description: `${agent} · ${sidStr.slice(0, 24)}…` })
947
+ }
948
+ }
949
+ } catch {}
950
+ }
951
+
952
+ // 去重
953
+ const seen = new Set<string>()
954
+ const unique = children.filter(c => { if (seen.has(c.value)) return false; seen.add(c.value); return true })
955
+
956
+ if (unique.length > 0) {
957
+ // ── 有子代理 → DialogSelect 列表选择 ──
958
+ const t = createT(() => langCode())
959
+ const currentSid = signals.overrideSessionId() ?? api.kv.get<string>(`${KV_PREFIX}.session`, "")
960
+ const options = unique.map((c, i) => ({
961
+ title: `${i + 1}. ${c.title}`,
962
+ value: c.value,
963
+ description: c.description,
964
+ }))
965
+ // 首尾各放一个"回到主会话",长列表时顶部底部均可直达
966
+ const backValue = "__main__"
967
+ const backTitle = `\u2500 ${t("backToMainTitle")}`
968
+ options.unshift({ title: backTitle, value: backValue, description: "" })
969
+ options.push({ title: backTitle, value: backValue, description: "" })
970
+ const currentIdx = currentSid ? options.findIndex(o => o.value === currentSid) : -1
971
+ dialog?.replace(() => (
972
+ <api.ui.DialogSelect
973
+ title={t("subSelectTitle")}
974
+ options={options}
975
+ current={currentIdx >= 0 ? options[currentIdx].value : undefined}
976
+ onSelect={(opt) => {
977
+ if (opt.value === backValue) {
978
+ signals.setOverrideSessionId(undefined)
979
+ api.kv.set(`${KV_PREFIX}.session`, "")
980
+ api.ui.toast({ message: t("backToMain") })
981
+ } else {
982
+ signals.setOverrideSessionId(opt.value)
983
+ api.kv.set(`${KV_PREFIX}.session`, opt.value)
984
+ api.ui.toast({ message: t("subAgentSwitched", { s: opt.value.slice(0, 24) + "\u2026" }) })
985
+ }
986
+ dialog?.clear()
987
+ }}
988
+ />
989
+ ))
990
+ } else {
991
+ // ── 无子代理 → DialogPrompt 手动粘贴 ──
992
+ const t = createT(() => langCode())
993
+ dialog?.replace(() => (
994
+ <api.ui.DialogPrompt
995
+ title={signals.overrideSessionId() ? t("subSwitchTitle") : t("subViewTitle")}
996
+ description={() => <text>{t("subNoFound")}</text>}
997
+ placeholder="ses_..."
998
+ value={signals.overrideSessionId() ?? api.kv.get<string>(`${KV_PREFIX}.session`, "") ?? ""}
999
+ onConfirm={(val) => {
1000
+ const sid = val.trim()
1001
+ if (sid) {
1002
+ signals.setOverrideSessionId(sid)
1003
+ api.kv.set(`${KV_PREFIX}.session`, sid)
1004
+ api.ui.toast({ message: t("subAgentSwitched", { s: sid.slice(0, 24) + "\u2026" }) })
1005
+ }
1006
+ dialog?.clear()
1007
+ }}
1008
+ onCancel={() => dialog?.clear()}
1009
+ />
1010
+ ))
1011
+ }
1012
+ },
1013
+ },
1014
+ {
1015
+ title: "Cache: Back to Main",
1016
+ value: "cache.session.back",
1017
+ description: "Return to main session stats",
1018
+ slash: { name: "cache-session-back" },
1019
+ onSelect: (dialog) => {
1020
+ const t = createT(() => langCode())
1021
+ signals.setOverrideSessionId(undefined)
1022
+ api.kv.set(`${KV_PREFIX}.session`, "")
1023
+ api.ui.toast({ message: t("backToMain") })
1024
+ dialog?.clear()
1025
+ },
1026
+ },
1027
+ ])
1028
+ }
1029
+
1030
+ const mod: TuiPluginModule & { id: string } = {
1031
+ id: "opencode-visual-cache",
1032
+ tui,
1033
+ }
1034
+
1035
+ export default mod