opencode-tokenwatch 0.1.0 → 0.3.0

package/dist/queries.js

@@ -20,6 +20,10 @@ async function queryDb(sql) {
 function escapeSql(value) {
     return value.replace(/'/g, "''");
 }
+/** 校验日期格式必须为 YYYY-MM-DD，防止格式异常字符串进入 SQL */
+function isValidDate(s) {
+    return /^\d{4}-\d{2}-\d{2}$/.test(s);
+}
 function messageWhere(filters) {
     const where = [
         "json_extract(m.data, '$.role') = 'assistant'",
@@ -31,10 +35,13 @@ function messageWhere(filters) {
         where.push(`coalesce(json_extract(m.data, '$.providerID'), '') = '${escapeSql(filters.provider)}'`);
     if (filters.model)
         where.push(`coalesce(json_extract(m.data, '$.modelID'), '') = '${escapeSql(filters.model)}'`);
-    if (filters.startDate)
-        where.push(`date(m.time_created / 1000, 'unixepoch', 'localtime') >= '${escapeSql(filters.startDate)}'`);
-    if (filters.endDate)
-        where.push(`date(m.time_created / 1000, 'unixepoch', 'localtime') <= '${escapeSql(filters.endDate)}'`);
+    // Risk fix: 日期参数先验证格式（YYYY-MM-DD），格式不符则忽略该过滤条件
+    if (filters.startDate && isValidDate(filters.startDate)) {
+        where.push(`date(m.time_created / 1000, 'unixepoch', 'localtime') >= '${filters.startDate}'`);
+    }
+    if (filters.endDate && isValidDate(filters.endDate)) {
+        where.push(`date(m.time_created / 1000, 'unixepoch', 'localtime') <= '${filters.endDate}'`);
+    }
     return where.join(" AND ");
 }
 function parseList(value) {
@@ -83,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) => ({
@@ -134,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) => ({
@@ -165,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) => ({
@@ -197,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) => ({
@@ -235,34 +242,86 @@ 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");
 }
+/** 失败请求计数 SQL。1次运行获取成功数＋失败数＋按模型细分 */
+export async function getErrorStats(filters = {}) {
+    // 构建日期／Session/Provider/Model 过滤条件（不包含 tokens.total > 0 过滤）
+    const baseConds = [
+        "json_extract(m.data, '$.role') = 'assistant'",
+    ];
+    if (filters.sessionId)
+        baseConds.push(`m.session_id = '${escapeSql(filters.sessionId)}'`);
+    if (filters.provider)
+        baseConds.push(`coalesce(json_extract(m.data, '$.providerID'), '') = '${escapeSql(filters.provider)}'`);
+    if (filters.model)
+        baseConds.push(`coalesce(json_extract(m.data, '$.modelID'), '') = '${escapeSql(filters.model)}'`);
+    if (filters.startDate && isValidDate(filters.startDate)) {
+        baseConds.push(`date(m.time_created / 1000, 'unixepoch', 'localtime') >= '${filters.startDate}'`);
+    }
+    if (filters.endDate && isValidDate(filters.endDate)) {
+        baseConds.push(`date(m.time_created / 1000, 'unixepoch', 'localtime') <= '${filters.endDate}'`);
+    }
+    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
+  `.trim();
+    try {
+        const rows = await queryDb(sql);
+        let successCount = 0, failedCount = 0;
+        const byModel = rows.map(r => {
+            const total = r.total ?? 0;
+            const failed = r.failed ?? 0;
+            const success = total - failed;
+            successCount += success;
+            failedCount += failed;
+            return { provider: r.provider ?? 'unknown', model: r.model ?? 'unknown', failed, total };
+        });
+        const errorRate = (successCount + failedCount) > 0
+            ? failedCount / (successCount + failedCount)
+            : 0;
+        return { successCount, failedCount, errorRate, byModel };
+    }
+    catch {
+        return { successCount: 0, failedCount: 0, errorRate: 0, byModel: [] };
+    }
+}
 export async function getUsageReport(filters = {}) {
-    const [summary, models, providers, daily, sessions] = await Promise.all([
+    const [summary, models, providers, daily, sessions, errors] = await Promise.all([
         getSummary(filters),
         getModelBreakdown(filters),
         getProviderBreakdown(filters),
         getDailyBreakdown(filters),
         getSessionBreakdown(filters),
+        getErrorStats(filters),
     ]);
-    return { filters, summary, models, providers, daily, sessions };
+    return { filters, summary, models, providers, daily, sessions, errors };
 }
 function csvEscape(value) {
     const text = String(value);

package/dist/sidebar.d.ts

@@ -0,0 +1,22 @@
+import type { TuiPluginApi, TuiTheme } from "@opencode-ai/plugin/tui";
+import type { PerfTracker } from "./perf-tracker.js";
+import type { TokenMessage } from "./tui.jsx";
+export interface SidebarConfig {
+    sidebar: {
+        showPerformance: boolean;
+        showPricing: boolean;
+        showTokenDistribution: boolean;
+        showTrend: boolean;
+    };
+    language: "zh" | "en" | "auto";
+}
+export declare function loadConfig(api: TuiPluginApi): SidebarConfig;
+interface TokenWatchPanelProps {
+    api: TuiPluginApi;
+    theme: TuiTheme;
+    perfTracker: PerfTracker;
+    messages: readonly any[];
+    allTokenMessages: TokenMessage[];
+}
+export declare function TokenWatchPanel(props: TokenWatchPanelProps): any;
+export {};