token-stats-timer 1.0.15 → 1.0.17

package/README.md

@@ -16,7 +16,7 @@ Footer 上行（左对齐）：
 - `↑ ↓ Σ CH` —— 累计输入 / 输出 / 总量 / 缓存命中率
 - `⚡` —— 实时速率（2s rolling window，无流时回落到平均速率）
 - 上下文占用（样式可配）
-- `5h: W: ⏱` —— 套餐剩余（MiniMax / GLM / Kimi / DeepSeek 内置套餐，需在 /stats 里为当前 provider 启用）
+- `5h: W: ⏱` —— 套餐剩余（MiniMax / GLM / Kimi / DeepSeek / OpenCode Go 内置套餐，需在 /stats 里为当前 provider 启用）
 
 Footer 下行：cwd + git 分支 + 其他扩展状态。
 
@@ -76,8 +76,9 @@ Footer 下行：cwd + git 分支 + 其他扩展状态。
 
 ## 命令
 
-- `/stats` —— 无参进入套餐配置（为当前 provider 选择/关闭配额套餐；选 GLM 后会继续询问是否配置团队套餐凭证）
+- `/stats` —— 无参默认显示**当天 token 统计**（等价 `/stats day`）
 - `/stats day [YYYY-MM-DD]` / `hour` / `week` / `month [YYYY-MM]` —— 统计查询
+- `/stats limit` —— 套餐配置（为当前 provider 选择/关闭配额套餐；选 GLM 后会继续询问是否配置团队套餐凭证）
 - `/stats config` —— 显示样式 / 显示内容 / 配额刷新时间 / GLM 团队凭证
 - `/notify [on|off|test]` —— 通知开关 / 测试（无参查看状态）
 
@@ -101,6 +102,12 @@ Footer 下行：cwd + git 分支 + 其他扩展状态。
 
 > 组织 ID / 项目 ID 在 GLM Coding Plan 团队版后台「团队编程套餐」页面获取；如果你同时使用 Claude Code / Cursor 等工具并已在环境变量里配置，也可以在 config.json 里直接填上同名值。
 
+## OpenCode Go 余额
+
+OpenCode Go 订阅（`opencode-go` provider，baseUrl `https://opencode.ai/zen/go/v1`）使用官方配额接口 `GET /zen/go/v1/usage`（`Authorization: Bearer <key>`，即 auth.json 里 `opencode-go` 的 `key` 或环境变量 `OPENCODE_API_KEY`），返回三个滚动窗口的已用百分比：5 小时 / 周 / 月。
+
+footer 显示 `5h: X% W: Y% M: Z% ⏱ ...`（剩余比例 = 100 - 已用），后三个子项可分别用 `/stats config` → 显示内容 的「5h额度 / 周额度 / 月额度 / 刷新时间」开关控制。
+
 ## 与原包的兼容性
 
 - 显示/套餐配置沿用 `~/.pi/agent/extensions/token-stats/`（原 token-stats 的配置直接生效）

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "token-stats-timer",
-  "version": "1.0.15",
+  "version": "1.0.17",
   "description": "pi extension to display token usage and time spend.",
   "keywords": [
     "pi-package",

package/token-stats.ts

@@ -20,6 +20,7 @@ import {
   appendFile,
   mkdir,
   readFile,
+  readdir,
   writeFile,
 } from "node:fs/promises";
 import { join } from "node:path";
@@ -151,6 +152,7 @@ export type DisplayKey =
   | "context"     // 容量（ctx%）
   | "quota5h"     // 5h 额度
   | "quotaWeek"   // 周额度
+  | "quotaMonth"  // 月额度（OpenCode Go 等）
   | "quotaClock"  // 刷新时间（⏱）
 
 export interface DisplayConfig {
@@ -248,6 +250,134 @@ function renderTable(
   return lines;
 }
 
+// ── 按模型统计（读取 raw 明细，按 model 聚合）────────────────────
+
+/** 读取若干日期的原始逐条记录 */
+async function readRawRecordsForDates(dates: string[]): Promise<RawRecord[]> {
+  const out: RawRecord[] = [];
+  for (const d of dates) {
+    try {
+      const content = await readFile(join(RAW_DIR, `${d}.jsonl`), "utf-8");
+      for (const line of content.trim().split("\n")) {
+        if (line) out.push(JSON.parse(line));
+      }
+    } catch {
+      // 该日无原始数据
+    }
+  }
+  return out;
+}
+
+/** 读取 [startDate, endDate] 闭区间内所有日期的原始记录 */
+async function readRawRecordsInRange(
+  startDate: string,
+  endDate: string,
+): Promise<RawRecord[]> {
+  const dates: string[] = [];
+  try {
+    const files = await readdir(RAW_DIR);
+    for (const f of files) {
+      if (!f.endsWith(".jsonl")) continue;
+      const d = f.slice(0, 10);
+      if (/^\d{4}-\d{2}-\d{2}$/.test(d) && d >= startDate && d <= endDate) {
+        dates.push(d);
+      }
+    }
+  } catch {
+    // RAW_DIR 不存在
+  }
+  return readRawRecordsForDates(dates);
+}
+
+interface ModelAgg {
+  count: number;
+  input: number;
+  cacheRead: number;
+  cacheWrite: number;
+  output: number;
+  tokensPerSecSum: number;
+  hitRateSum: number;
+}
+
+/** 按模型生成 markdown 用量表（模型行按总token降序，末尾合计） */
+function renderModelBreakdown(records: RawRecord[]): string[] {
+  if (records.length === 0) return ["", "> 按模型：该范围暂无明细数据"];
+
+  const byModel = new Map<string, ModelAgg>();
+  for (const r of records) {
+    const key = r.model || "unknown";
+    const agg = byModel.get(key) ?? {
+      count: 0,
+      input: 0,
+      cacheRead: 0,
+      cacheWrite: 0,
+      output: 0,
+      tokensPerSecSum: 0,
+      hitRateSum: 0,
+    };
+    agg.count++;
+    agg.input += r.input;
+    agg.cacheRead += r.cacheRead;
+    agg.cacheWrite += r.cacheWrite;
+    agg.output += r.output;
+    agg.tokensPerSecSum += r.tokensPerSec;
+    agg.hitRateSum += r.cacheHitRate;
+    byModel.set(key, agg);
+  }
+
+  // 与汇总口径一致：总token = 新增输入 + 缓存输入
+  const totalTokensOf = (a: ModelAgg) => a.input + a.cacheRead + a.cacheWrite;
+  const rows = [...byModel.entries()].sort(
+    (a, b) => totalTokensOf(b[1]) - totalTokensOf(a[1]),
+  );
+
+  const total: ModelAgg = {
+    count: 0, input: 0, cacheRead: 0, cacheWrite: 0,
+    output: 0, tokensPerSecSum: 0, hitRateSum: 0,
+  };
+  const body: string[][] = rows.map(([model, agg]) => {
+    total.count += agg.count;
+    total.input += agg.input;
+    total.cacheRead += agg.cacheRead;
+    total.cacheWrite += agg.cacheWrite;
+    total.output += agg.output;
+    total.tokensPerSecSum += agg.tokensPerSecSum;
+    total.hitRateSum += agg.hitRateSum;
+    return [
+      model,
+      String(agg.count),
+      formatTokens(agg.input),
+      formatTokens(agg.cacheRead),
+      formatTokens(agg.output),
+      formatTokens(totalTokensOf(agg)),
+      `${(agg.count > 0 ? agg.hitRateSum / agg.count : 0).toFixed(1)}%`,
+      `${(agg.count > 0 ? agg.tokensPerSecSum / agg.count : 0).toFixed(1)}`,
+    ];
+  });
+
+  return [
+    "",
+    "**按模型**",
+    ...renderTable(
+      ["模型", "次数", "新增输入", "缓存输入", "输出", "总token", "命中率", "速率"],
+      body,
+      {
+        aligns: ["left", "right", "right", "right", "right", "right", "right", "right"],
+        totalRow: [
+          "合计",
+          String(total.count),
+          formatTokens(total.input),
+          formatTokens(total.cacheRead),
+          formatTokens(total.output),
+          formatTokens(total.input + total.cacheRead + total.cacheWrite),
+          `${(total.count > 0 ? total.hitRateSum / total.count : 0).toFixed(1)}%`,
+          `${(total.count > 0 ? total.tokensPerSecSum / total.count : 0).toFixed(1)}`,
+        ],
+      },
+    ),
+  ];
+}
+
 function isReasonableTokenSpeed(tokensPerSecond: number): boolean {
   return Number.isFinite(tokensPerSecond) && tokensPerSecond > 0 && tokensPerSecond <= MAX_REASONABLE_TOKEN_SPEED;
 }
@@ -543,6 +673,73 @@ const BUILTIN_PLANS: TokenPlan[] = [
       };
     },
   },
+  {
+    id: "opencode-go",
+    name: "OpenCode Go",
+    matchProviders: ["opencode-go"],
+    apiKeyEnv: "OPENCODE_API_KEY",
+    baseUrl: "https://opencode.ai",
+    quotaPath: "/zen/go/v1/usage",
+    authHeader: (key) => ({ Authorization: "Bearer " + key }),
+    fetchQuota: async (plan: TokenPlan, key: string) => {
+      const r = await fetch(plan.baseUrl + plan.quotaPath, {
+        method: "GET",
+        headers: {
+          Authorization: "Bearer " + key,
+          "Content-Type": "application/json",
+        },
+        signal: AbortSignal.timeout(5000),
+      });
+      if (!r.ok) throw new Error("OpenCode Go 配额查询 HTTP " + r.status);
+      return await r.json();
+    },
+    format: (data: any) => {
+      // 官方 /v1/usage：percent 为已用比例，remaining = 100 - percent
+      const u = (data && typeof data === "object" ? (data.usage ?? data) : {}) as any;
+      const win = (k: string): any => {
+        const w = u?.[k];
+        if (w && w.status === "ok" && typeof w.percent === "number") return w;
+        return null;
+      };
+      const rolling = win("rolling");
+      const weekly = win("weekly");
+      const monthly = win("monthly");
+      if (!rolling && !weekly && !monthly) {
+        return { modelPrefix: "", display: "无数据", color: "err" as const };
+      }
+      const now = Date.now();
+      const resets = [rolling, weekly, monthly]
+        .filter((w): w is any => !!w)
+        .map((w) => {
+          const t = typeof w.resetsAt === "string" ? Date.parse(w.resetsAt) : NaN;
+          return Number.isFinite(t) && t > now ? t : null;
+        })
+        .filter((t): t is number => t !== null);
+      const nearestReset = resets.length > 0 ? Math.min(...resets) : null;
+
+      const rem = (w: any) => (w ? 100 - w.percent : null);
+      const r = rem(rolling);
+      const wk = rem(weekly);
+      const mo = rem(monthly);
+      const parts: string[] = [];
+      if (r !== null) parts.push(`5h: ${Math.round(r)}%`);
+      if (wk !== null) parts.push(`W: ${Math.round(wk)}%`);
+      if (mo !== null) parts.push(`M: ${Math.round(mo)}%`);
+      let display = parts.join(" ");
+      if (nearestReset) {
+        const diff = nearestReset - now;
+        if (diff > 0 && diff < 30 * 24 * 60 * 60 * 1000) {
+          display += ` ⏱ ${formatDuration(diff)}`;
+        }
+      }
+      const low = (v: number | null) => v !== null && v < 20;
+      const mid = (v: number | null) => v !== null && v < 50;
+      const color = low(r) || low(wk) || low(mo) ? "err" as const
+        : mid(r) || mid(wk) || mid(mo) ? "warn" as const
+          : "ok" as const;
+      return { modelPrefix: "", display, color };
+    },
+  },
 ];
 
 const DEFAULT_TOKEN_CONFIG: TokenConfig = { providerPlans: {}, ttl: 60 };
@@ -557,6 +754,7 @@ const DEFAULT_DISPLAY_CONFIG: DisplayConfig = {
     context: true,
     quota5h: true,
     quotaWeek: true,
+    quotaMonth: true,
     quotaClock: true,
   },
   contextStyle: "pct-window",
@@ -786,6 +984,10 @@ export function createTokenStats(
           const m = fullDisplay.match(/\bW:\s+\d+%/);
           if (m) filteredParts.push(m[0]);
         }
+        if (cfg.quotaMonth) {
+          const m = fullDisplay.match(/\bM:\s+\d+%/);
+          if (m) filteredParts.push(m[0]);
+        }
         if (cfg.quotaClock) {
           const m = fullDisplay.match(/⏱\s*\d+[hm]/);
           if (m) filteredParts.push(m[0]);
@@ -1405,7 +1607,7 @@ export function createTokenStats(
     }
 
     await showStats(
-      renderDaySummary(daily),
+      [...renderDaySummary(daily), ...renderModelBreakdown(await readRawRecordsForDates([date]))],
       `Token 统计  |  ${date}`,
       ctx,
     );
@@ -1489,7 +1691,11 @@ export function createTokenStats(
       },
     );
 
-    await showStats(lines, "本周每天汇总", ctx);
+    await showStats(
+      [...lines, ...renderModelBreakdown(await readRawRecordsInRange(sevenDaysAgo, today))],
+      "本周每天汇总",
+      ctx,
+    );
   }
 
   function getMonthStr(date: Date = new Date()): string {
@@ -1563,7 +1769,17 @@ export function createTokenStats(
       },
     );
 
-    await showStats(lines, `${month} 月度汇总`, ctx);
+    const monthDates = monthRecords.map((r) => r.date).sort();
+    await showStats(
+      [
+        ...lines,
+        ...renderModelBreakdown(
+          await readRawRecordsInRange(monthDates[0], monthDates[monthDates.length - 1]),
+        ),
+      ],
+      `${month} 月度汇总`,
+      ctx,
+    );
   }
 
   // ── 事件注册 ─────────────────────────────────────────
@@ -1784,12 +2000,18 @@ export function createTokenStats(
   // ── /stats 命令 ─────────────────────────────────────
 
   pi.registerCommand("stats", {
-    description: "Token 统计 (day | hour | week | month | config)  无参默认进入套餐配置",
+    description: "Token 统计 (day | hour | week | month | config | limit)  无参默认显示当天统计；limit 进入套餐配置",
     handler: async (args, ctx) => {
       const arg = args.trim();
 
-      // 无参 → 套餐配置
+      // 无参 → 当天统计（等价于 /stats day）
       if (!arg) {
+        await showDay(getDateStr(), ctx);
+        return;
+      }
+
+      // limit → 套餐配置（原无参行为）
+      if (arg === "limit") {
         const provider = ctx.model?.provider;
         if (!provider) {
           ctx.ui.notify("无法获取当前供应商，请先切换对话", "warning");
@@ -1969,12 +2191,12 @@ export function createTokenStats(
         } else if (subChoice === "显示内容") {
           const itemLabels: DisplayKey[] = [
             "input", "output", "totalTokens", "cacheHit", "speed", "context",
-            "quota5h", "quotaWeek", "quotaClock",
+            "quota5h", "quotaWeek", "quotaMonth", "quotaClock",
           ];
           const itemNames: Record<DisplayKey, string> = {
             input: "输入", output: "输出", totalTokens: "总token",
             cacheHit: "缓存命中", speed: "速度", context: "容量",
-            quota5h: "5h额度", quotaWeek: "周额度", quotaClock: "刷新时间",
+            quota5h: "5h额度", quotaWeek: "周额度", quotaMonth: "月额度", quotaClock: "刷新时间",
           };
           while (true) {
             const options = itemLabels.map(k =>