opencode-tokenwatch 0.3.0 → 0.3.1

package/dist/i18n.js

@@ -14,6 +14,7 @@ const zh = {
     trendUp: "↑",
     trendDown: "↓",
     cache: "缓存",
+    lat: "延迟",
     performance: "性能",
     pricing: "Pricing",
     tokenDistribution: "Token分布",
@@ -86,6 +87,7 @@ const en = {
     trendUp: "↑",
     trendDown: "↓",
     cache: "Cache",
+    lat: "Lat",
     performance: "Performance",
     pricing: "Pricing",
     tokenDistribution: "Token Distribution",

package/dist/perf-tracker.d.ts

@@ -54,6 +54,7 @@ declare class PerfTracker {
     getSessionStats(): SessionPerfStats;
     readLogs(last?: number): LogEntry[];
     reset(): void;
+    loadSession(sessionID: string): void;
 }
 export declare function createPerfTracker(): PerfTracker;
 export type { PartEvent, PerfTracker };

package/dist/perf-tracker.js

@@ -2,6 +2,7 @@ import { appendFileSync, readFileSync, writeFileSync } from "node:fs";
 import { join } from "node:path";
 import { homedir } from "node:os";
 import { existsSync, statSync } from "node:fs";
+import { updatePersistedStats } from "./stats-store.js";
 const LOG_PATH = join(homedir(), ".opencode", "tokenwatch.jsonl");
 class PerfTracker {
     firstPartTimes = new Map();
@@ -76,6 +77,7 @@ class PerfTracker {
         try {
             // Risk fix: JSONL 日志轮转保护，防止长期使用后文件无限增长
             // 超过 5MB 时截断，保留最新 2000 行
+            // 注意：轮转前先调用 updatePersistedStats，确保被轮转行的数据已持久化
             const MAX_SIZE = 5 * 1024 * 1024; // 5 MB
             const KEEP_LINES = 2000;
             if (existsSync(LOG_PATH) && statSync(LOG_PATH).size > MAX_SIZE) {
@@ -87,6 +89,9 @@ class PerfTracker {
         catch {
             // Silently fail — logging is non-critical
         }
+        // 无论 JSONL 写入是否成功，都尝试更新持久化聚合统计
+        // 这样即使日志被轮转，历史统计数据也永不丢失
+        updatePersistedStats(entry);
     }
     handleMessageRemoved(event) {
         const mid = event.properties?.messageID ?? "";
@@ -247,6 +252,38 @@ class PerfTracker {
         this.ttftSamples.clear();
         this.latencySamples.clear();
     }
+    loadSession(sessionID) {
+        this.firstPartTimes.clear();
+        this.statsMap.clear();
+        this.ttftSamples.clear();
+        this.latencySamples.clear();
+        if (!sessionID)
+            return;
+        try {
+            if (!existsSync(LOG_PATH))
+                return;
+            const content = readFileSync(LOG_PATH, "utf-8").trim();
+            if (!content)
+                return;
+            const lines = content.split("\n");
+            for (const line of lines) {
+                if (!line)
+                    continue;
+                try {
+                    const entry = JSON.parse(line);
+                    if (entry.sessionID === sessionID) {
+                        this.updateStats(entry.model, entry);
+                    }
+                }
+                catch {
+                    // Skip malformed lines
+                }
+            }
+        }
+        catch {
+            // Non-critical loading failure
+        }
+    }
 }
 export function createPerfTracker() {
     return new PerfTracker();

package/dist/queries.js

@@ -90,41 +90,41 @@ export async function getCurrentSessionStats(sessionId) {
     return getSummary(filters);
 }
 export async function getSummary(filters = {}) {
-    const sql = `
-SELECT
-  group_concat(distinct coalesce(json_extract(m.data, '$.modelID'), 'unknown')) as models_used,
-  group_concat(distinct coalesce(json_extract(m.data, '$.providerID'), 'unknown')) as providers_used,
-  count(*) as request_count,
-  sum(coalesce(json_extract(m.data, '$.tokens.total'), 0)) as total_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.input'), 0)) as input_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.output'), 0)) as output_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.reasoning'), 0)) as reasoning_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.cache.read'), 0)) as cache_read,
-  sum(coalesce(json_extract(m.data, '$.tokens.cache.write'), 0)) as cache_write,
-  sum(coalesce(json_extract(m.data, '$.cost'), 0)) as total_cost
-FROM message m
-WHERE ${messageWhere(filters)}
+    const sql = `
+SELECT
+  group_concat(distinct coalesce(json_extract(m.data, '$.modelID'), 'unknown')) as models_used,
+  group_concat(distinct coalesce(json_extract(m.data, '$.providerID'), 'unknown')) as providers_used,
+  count(*) as request_count,
+  sum(coalesce(json_extract(m.data, '$.tokens.total'), 0)) as total_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.input'), 0)) as input_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.output'), 0)) as output_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.reasoning'), 0)) as reasoning_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.cache.read'), 0)) as cache_read,
+  sum(coalesce(json_extract(m.data, '$.tokens.cache.write'), 0)) as cache_write,
+  sum(coalesce(json_extract(m.data, '$.cost'), 0)) as total_cost
+FROM message m
+WHERE ${messageWhere(filters)}
   `.trim();
     const rows = await queryDb(sql);
     return toSessionTokenData(rows[0]);
 }
 export async function getModelBreakdown(filters = {}) {
-    const sql = `
-SELECT
-  coalesce(json_extract(m.data, '$.providerID'), 'unknown') as provider,
-  coalesce(json_extract(m.data, '$.modelID'), 'unknown') as model,
-  count(*) as requests,
-  count(distinct m.session_id) as sessions,
-  sum(coalesce(json_extract(m.data, '$.tokens.total'), 0)) as total_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.input'), 0)) as input_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.output'), 0)) as output_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.reasoning'), 0)) as reasoning_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.cache.read'), 0)) as cache_read,
-  sum(coalesce(json_extract(m.data, '$.cost'), 0)) as total_cost
-FROM message m
-WHERE ${messageWhere(filters)}
-GROUP BY provider, model
-ORDER BY total_tokens DESC
+    const sql = `
+SELECT
+  coalesce(json_extract(m.data, '$.providerID'), 'unknown') as provider,
+  coalesce(json_extract(m.data, '$.modelID'), 'unknown') as model,
+  count(*) as requests,
+  count(distinct m.session_id) as sessions,
+  sum(coalesce(json_extract(m.data, '$.tokens.total'), 0)) as total_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.input'), 0)) as input_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.output'), 0)) as output_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.reasoning'), 0)) as reasoning_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.cache.read'), 0)) as cache_read,
+  sum(coalesce(json_extract(m.data, '$.cost'), 0)) as total_cost
+FROM message m
+WHERE ${messageWhere(filters)}
+GROUP BY provider, model
+ORDER BY total_tokens DESC
   `.trim();
     const rows = await queryDb(sql);
     return rows.map((row) => ({
@@ -141,21 +141,21 @@ ORDER BY total_tokens DESC
     }));
 }
 export async function getProviderBreakdown(filters = {}) {
-    const sql = `
-SELECT
-  coalesce(json_extract(m.data, '$.providerID'), 'unknown') as provider,
-  count(*) as requests,
-  count(distinct m.session_id) as sessions,
-  sum(coalesce(json_extract(m.data, '$.tokens.total'), 0)) as total_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.input'), 0)) as input_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.output'), 0)) as output_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.reasoning'), 0)) as reasoning_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.cache.read'), 0)) as cache_read,
-  sum(coalesce(json_extract(m.data, '$.cost'), 0)) as total_cost
-FROM message m
-WHERE ${messageWhere(filters)}
-GROUP BY provider
-ORDER BY total_tokens DESC
+    const sql = `
+SELECT
+  coalesce(json_extract(m.data, '$.providerID'), 'unknown') as provider,
+  count(*) as requests,
+  count(distinct m.session_id) as sessions,
+  sum(coalesce(json_extract(m.data, '$.tokens.total'), 0)) as total_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.input'), 0)) as input_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.output'), 0)) as output_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.reasoning'), 0)) as reasoning_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.cache.read'), 0)) as cache_read,
+  sum(coalesce(json_extract(m.data, '$.cost'), 0)) as total_cost
+FROM message m
+WHERE ${messageWhere(filters)}
+GROUP BY provider
+ORDER BY total_tokens DESC
   `.trim();
     const rows = await queryDb(sql);
     return rows.map((row) => ({
@@ -172,22 +172,22 @@ ORDER BY total_tokens DESC
 }
 export async function getDailyBreakdown(filters = {}) {
     const limit = filters.limit ?? 30;
-    const sql = `
-SELECT
-  date(m.time_created / 1000, 'unixepoch', 'localtime') as day,
-  count(*) as requests,
-  count(distinct m.session_id) as sessions,
-  sum(coalesce(json_extract(m.data, '$.tokens.total'), 0)) as total_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.input'), 0)) as input_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.output'), 0)) as output_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.reasoning'), 0)) as reasoning_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.cache.read'), 0)) as cache_read,
-  sum(coalesce(json_extract(m.data, '$.cost'), 0)) as total_cost
-FROM message m
-WHERE ${messageWhere(filters)}
-GROUP BY day
-ORDER BY day DESC
-LIMIT ${Math.max(1, limit)}
+    const sql = `
+SELECT
+  date(m.time_created / 1000, 'unixepoch', 'localtime') as day,
+  count(*) as requests,
+  count(distinct m.session_id) as sessions,
+  sum(coalesce(json_extract(m.data, '$.tokens.total'), 0)) as total_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.input'), 0)) as input_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.output'), 0)) as output_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.reasoning'), 0)) as reasoning_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.cache.read'), 0)) as cache_read,
+  sum(coalesce(json_extract(m.data, '$.cost'), 0)) as total_cost
+FROM message m
+WHERE ${messageWhere(filters)}
+GROUP BY day
+ORDER BY day DESC
+LIMIT ${Math.max(1, limit)}
   `.trim();
     const rows = await queryDb(sql);
     return rows.map((row) => ({
@@ -204,26 +204,26 @@ LIMIT ${Math.max(1, limit)}
 }
 export async function getSessionBreakdown(filters = {}) {
     const limit = filters.limit ?? 15;
-    const sql = `
-SELECT
-  s.id as session_id,
-  s.title as title,
-  coalesce(json_extract(m.data, '$.providerID'), json_extract(s.model, '$.providerID'), 'unknown') as provider,
-  coalesce(json_extract(m.data, '$.modelID'), json_extract(s.model, '$.id'), 'unknown') as model,
-  count(*) as requests,
-  sum(coalesce(json_extract(m.data, '$.tokens.total'), 0)) as total_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.input'), 0)) as input_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.output'), 0)) as output_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.reasoning'), 0)) as reasoning_tokens,
-  sum(coalesce(json_extract(m.data, '$.tokens.cache.read'), 0)) as cache_read,
-  sum(coalesce(json_extract(m.data, '$.cost'), 0)) as total_cost,
-  date(max(m.time_created) / 1000, 'unixepoch', 'localtime') as day
-FROM message m
-JOIN session s ON s.id = m.session_id
-WHERE ${messageWhere(filters)}
-GROUP BY s.id, s.title, provider, model
-ORDER BY max(m.time_created) DESC
-LIMIT ${Math.max(1, limit)}
+    const sql = `
+SELECT
+  s.id as session_id,
+  s.title as title,
+  coalesce(json_extract(m.data, '$.providerID'), json_extract(s.model, '$.providerID'), 'unknown') as provider,
+  coalesce(json_extract(m.data, '$.modelID'), json_extract(s.model, '$.id'), 'unknown') as model,
+  count(*) as requests,
+  sum(coalesce(json_extract(m.data, '$.tokens.total'), 0)) as total_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.input'), 0)) as input_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.output'), 0)) as output_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.reasoning'), 0)) as reasoning_tokens,
+  sum(coalesce(json_extract(m.data, '$.tokens.cache.read'), 0)) as cache_read,
+  sum(coalesce(json_extract(m.data, '$.cost'), 0)) as total_cost,
+  date(max(m.time_created) / 1000, 'unixepoch', 'localtime') as day
+FROM message m
+JOIN session s ON s.id = m.session_id
+WHERE ${messageWhere(filters)}
+GROUP BY s.id, s.title, provider, model
+ORDER BY max(m.time_created) DESC
+LIMIT ${Math.max(1, limit)}
   `.trim();
     const rows = await queryDb(sql);
     return rows.map((row) => ({
@@ -242,21 +242,21 @@ LIMIT ${Math.max(1, limit)}
     }));
 }
 export async function getAvailableModels() {
-    const sql = `
-SELECT distinct coalesce(json_extract(m.data, '$.modelID'), 'unknown') as value
-FROM message m
-WHERE ${messageWhere({})}
-ORDER BY value ASC
+    const sql = `
+SELECT distinct coalesce(json_extract(m.data, '$.modelID'), 'unknown') as value
+FROM message m
+WHERE ${messageWhere({})}
+ORDER BY value ASC
   `.trim();
     const rows = await queryDb(sql);
     return rows.map((row) => row.value ?? "unknown");
 }
 export async function getAvailableProviders() {
-    const sql = `
-SELECT distinct coalesce(json_extract(m.data, '$.providerID'), 'unknown') as value
-FROM message m
-WHERE ${messageWhere({})}
-ORDER BY value ASC
+    const sql = `
+SELECT distinct coalesce(json_extract(m.data, '$.providerID'), 'unknown') as value
+FROM message m
+WHERE ${messageWhere({})}
+ORDER BY value ASC
   `.trim();
     const rows = await queryDb(sql);
     return rows.map((row) => row.value ?? "unknown");
@@ -281,16 +281,16 @@ export async function getErrorStats(filters = {}) {
     }
     const baseWhere = baseConds.join(" AND ");
     // 按模型细化：同时统计成功和失败请求
-    const sql = `
-SELECT
-  coalesce(json_extract(m.data, '$.providerID'), 'unknown') as provider,
-  coalesce(json_extract(m.data, '$.modelID'), 'unknown') as model,
-  count(*) as total,
-  sum(CASE WHEN coalesce(json_extract(m.data, '$.tokens.total'), 0) = 0 THEN 1 ELSE 0 END) as failed
-FROM message m
-WHERE ${baseWhere}
-GROUP BY provider, model
-ORDER BY failed DESC
+    const sql = `
+SELECT
+  coalesce(json_extract(m.data, '$.providerID'), 'unknown') as provider,
+  coalesce(json_extract(m.data, '$.modelID'), 'unknown') as model,
+  count(*) as total,
+  sum(CASE WHEN coalesce(json_extract(m.data, '$.tokens.total'), 0) = 0 THEN 1 ELSE 0 END) as failed
+FROM message m
+WHERE ${baseWhere}
+GROUP BY provider, model
+ORDER BY failed DESC
   `.trim();
     try {
         const rows = await queryDb(sql);

package/dist/sidebar.d.ts

@@ -15,8 +15,8 @@ interface TokenWatchPanelProps {
     api: TuiPluginApi;
     theme: TuiTheme;
     perfTracker: PerfTracker;
-    messages: readonly any[];
-    allTokenMessages: TokenMessage[];
+    messages: () => readonly any[];
+    allTokenMessages: () => TokenMessage[];
 }
 export declare function TokenWatchPanel(props: TokenWatchPanelProps): any;
 export {};