token-stats-timer 1.0.14 → 1.0.16

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 分支 + 其他扩展状态。
 
@@ -101,6 +101,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.14",
+  "version": "1.0.16",
   "description": "pi extension to display token usage and time spend.",
   "keywords": [
     "pi-package",

package/token-stats.ts

@@ -13,8 +13,9 @@ import type {
   ExtensionContext,
   Theme,
 } from "@earendil-works/pi-coding-agent";
+import { getMarkdownTheme } from "@earendil-works/pi-coding-agent";
 import type { AssistantMessage } from "@earendil-works/pi-ai";
-import { Text } from "@earendil-works/pi-tui";
+import { Markdown } from "@earendil-works/pi-tui";
 import {
   appendFile,
   mkdir,
@@ -150,6 +151,7 @@ export type DisplayKey =
   | "context"     // 容量（ctx%）
   | "quota5h"     // 5h 额度
   | "quotaWeek"   // 周额度
+  | "quotaMonth"  // 月额度（OpenCode Go 等）
   | "quotaClock"  // 刷新时间（⏱）
 
 export interface DisplayConfig {
@@ -212,6 +214,41 @@ function formatTokenSpeed(tokensPerSecond: number): string {
   return `${Math.round(tokensPerSecond / 1000000)}M`;
 }
 
+/**
+ * 生成 GitHub 风格 markdown 表格源码（由 Markdown 组件负责渲染对齐/换行）。
+ *
+ * @param headers 表头
+ * @param rows    数据行
+ * @param opts.aligns   每列对齐方式，缺省左对齐
+ * @param opts.totalRow 底部合计行
+ */
+function renderTable(
+  headers: string[],
+  rows: string[][],
+  opts?: {
+    aligns?: Array<"left" | "right" | "center">;
+    totalRow?: string[];
+  },
+): string[] {
+  const aligns = opts?.aligns ?? [];
+  // 转义单元格内的 | 与换行，防止破坏表格结构
+  const esc = (s: string) => s.replace(/\|/g, "\\|").replace(/\n/g, " ");
+  const alignMark = (i: number) => {
+    const a = aligns[i] ?? "left";
+    return a === "right" ? "---:" : a === "center" ? ":---:" : "---";
+  };
+
+  const lines = [
+    `| ${headers.map(esc).join(" | ")} |`,
+    `| ${headers.map((_, i) => alignMark(i)).join(" | ")} |`,
+    ...rows.map((r) => `| ${r.map(esc).join(" | ")} |`),
+  ];
+  if (opts?.totalRow) {
+    lines.push(`| ${opts.totalRow.map(esc).join(" | ")} |`);
+  }
+  return lines;
+}
+
 function isReasonableTokenSpeed(tokensPerSecond: number): boolean {
   return Number.isFinite(tokensPerSecond) && tokensPerSecond > 0 && tokensPerSecond <= MAX_REASONABLE_TOKEN_SPEED;
 }
@@ -507,6 +544,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 };
@@ -521,6 +625,7 @@ const DEFAULT_DISPLAY_CONFIG: DisplayConfig = {
     context: true,
     quota5h: true,
     quotaWeek: true,
+    quotaMonth: true,
     quotaClock: true,
   },
   contextStyle: "pct-window",
@@ -750,6 +855,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]);
@@ -1317,23 +1426,25 @@ export function createTokenStats(
     return total > 0 ? (d.sumCacheRead / total) * 100 : 0;
   }
 
-  function renderDaySummary(daily: DailyRecord): string {
+  function renderDaySummary(daily: DailyRecord): string[] {
     const d = daily;
     const avgInput = d.count > 0 ? d.sumInput / d.count : 0;
     const avgOutput = d.count > 0 ? d.sumOutput / d.count : 0;
     const totalPrompt = d.sumInput + d.sumCacheRead + d.sumCacheWrite;
     const cacheHitRate = weightedCacheHitRate(d);
 
-    const lines = [
-      `对话次数:  ${d.count}`,
-      `新增输入:  ${formatTokens(d.sumInput)}  (平均 ${formatTokens(avgInput)}/次，未命中缓存)`,
-      `缓存输入:  ${formatTokens(d.sumCacheRead)}`,
-      `总输出:    ${formatTokens(d.sumOutput)}  (平均 ${formatTokens(avgOutput)}/次)`,
-      `总token:   ${formatTokens(totalPrompt)}  (新增 + 缓存)`,
-      `缓存命中率: ${cacheHitRate.toFixed(1)}%`,
-      `平均速率:  ${(d.sumTokensPerSec / d.count).toFixed(1)} t/s`,
-    ];
-    return lines.join("\n");
+    return renderTable(
+      ["指标", "数值"],
+      [
+        ["对话次数", String(d.count)],
+        ["新增输入", `${formatTokens(d.sumInput)}（平均 ${formatTokens(avgInput)}/次，未命中缓存）`],
+        ["缓存输入", formatTokens(d.sumCacheRead)],
+        ["总输出", `${formatTokens(d.sumOutput)}（平均 ${formatTokens(avgOutput)}/次）`],
+        ["总token", `${formatTokens(totalPrompt)}（新增 + 缓存）`],
+        ["缓存命中率", `${cacheHitRate.toFixed(1)}%`],
+        ["平均速率", `${(d.sumTokensPerSec / d.count).toFixed(1)} t/s`],
+      ],
+    );
   }
 
   async function showStats(
@@ -1341,9 +1452,7 @@ export function createTokenStats(
     title: string,
     ctx: ExtensionContext,
   ) {
-    const theme = ctx.ui.theme;
-    const text = `${theme.fg("accent", theme.bold(title))}\n${theme.fg("dim", "─".repeat(42))}\n` +
-      lines.map((l) => theme.fg("dim", l)).join("\n");
+    const text = `## ${title}\n\n${lines.join("\n")}`;
     pi.sendMessage({
       customType: "token-stats",
       content: text,
@@ -1369,7 +1478,7 @@ export function createTokenStats(
     }
 
     await showStats(
-      renderDaySummary(daily).split("\n"),
+      renderDaySummary(daily),
       `Token 统计  |  ${date}`,
       ctx,
     );
@@ -1393,18 +1502,18 @@ export function createTokenStats(
 
     records.sort((a, b) => a.hour - b.hour);
 
-    const lines = [
-      "时  次数  输入      输出      命中率  速率",
-      "─".repeat(40),
-      ...records.map((r) =>
-        `${String(r.hour).padStart(2, "0")}  ` +
-        `${String(r.count).padStart(3)}  ` +
-        `${formatTokens(r.sumInput).padStart(7)}  ` +
-        `${formatTokens(r.sumOutput).padStart(7)}  ` +
-        `${weightedCacheHitRate(r).toFixed(1).padStart(5)}%  ` +
-        `${(r.sumTokensPerSec / r.count).toFixed(1).padStart(5)}`,
-      ),
-    ];
+    const lines = renderTable(
+      ["时间", "次数", "输入", "输出", "命中率", "速率"],
+      records.map((r) => [
+        String(r.hour).padStart(2, "0"),
+        String(r.count),
+        formatTokens(r.sumInput),
+        formatTokens(r.sumOutput),
+        `${weightedCacheHitRate(r).toFixed(1)}%`,
+        `${(r.sumTokensPerSec / r.count).toFixed(1)}`,
+      ]),
+      { aligns: ["left", "right", "right", "right", "right", "right"] },
+    );
 
     await showStats(lines, `按小时分布  |  ${date}`, ctx);
   }
@@ -1433,23 +1542,25 @@ export function createTokenStats(
       return;
     }
 
-    const lines = [
-      "日期        次数  新增输入  缓存输入  输出      总token   命中率  速率",
-      "─".repeat(70),
-      ...weekRecords.map((r) => {
+    const lines = renderTable(
+      ["日期", "次数", "新增输入", "缓存输入", "输出", "总token", "命中率", "速率"],
+      weekRecords.map((r) => {
         const totalPrompt = r.sumInput + r.sumCacheRead + r.sumCacheWrite;
-        return (
-          `${r.date}  ` +
-          `${String(r.count).padStart(3)}  ` +
-          `${formatTokens(r.sumInput).padStart(7)}  ` +
-          `${formatTokens(r.sumCacheRead).padStart(7)}  ` +
-          `${formatTokens(r.sumOutput).padStart(7)}  ` +
-          `${formatTokens(totalPrompt).padStart(7)}  ` +
-          `${weightedCacheHitRate(r).toFixed(1).padStart(5)}%  ` +
-          `${(r.sumTokensPerSec / r.count).toFixed(1).padStart(5)}`
-        );
+        return [
+          r.date,
+          String(r.count),
+          formatTokens(r.sumInput),
+          formatTokens(r.sumCacheRead),
+          formatTokens(r.sumOutput),
+          formatTokens(totalPrompt),
+          `${weightedCacheHitRate(r).toFixed(1)}%`,
+          `${(r.sumTokensPerSec / r.count).toFixed(1)}`,
+        ];
       }),
-    ];
+      {
+        aligns: ["left", "right", "right", "right", "right", "right", "right", "right"],
+      },
+    );
 
     await showStats(lines, "本周每天汇总", ctx);
   }
@@ -1495,40 +1606,46 @@ export function createTokenStats(
     const totalPrompt = total.sumInput + total.sumCacheRead + total.sumCacheWrite;
     const cacheHitRate = weightedCacheHitRate(total);
 
-    const lines = [
-      "日期        次数  新增输入  缓存输入  输出      总token   命中率  速率",
-      "─".repeat(70),
-      ...monthRecords.map((r) => {
+    const lines = renderTable(
+      ["日期", "次数", "新增输入", "缓存输入", "输出", "总token", "命中率", "速率"],
+      monthRecords.map((r) => {
         const tp = r.sumInput + r.sumCacheRead + r.sumCacheWrite;
-        return (
-          `${r.date}  ` +
-          `${String(r.count).padStart(3)}  ` +
-          `${formatTokens(r.sumInput).padStart(7)}  ` +
-          `${formatTokens(r.sumCacheRead).padStart(7)}  ` +
-          `${formatTokens(r.sumOutput).padStart(7)}  ` +
-          `${formatTokens(tp).padStart(7)}  ` +
-          `${weightedCacheHitRate(r).toFixed(1).padStart(5)}%  ` +
-          `${(r.sumTokensPerSec / r.count).toFixed(1).padStart(5)}`
-        );
+        return [
+          r.date,
+          String(r.count),
+          formatTokens(r.sumInput),
+          formatTokens(r.sumCacheRead),
+          formatTokens(r.sumOutput),
+          formatTokens(tp),
+          `${weightedCacheHitRate(r).toFixed(1)}%`,
+          `${(r.sumTokensPerSec / r.count).toFixed(1)}`,
+        ];
       }),
-      "",
-      `合计      ${String(total.count).padStart(3)}  ` +
-      `${formatTokens(total.sumInput).padStart(7)}  ` +
-      `${formatTokens(total.sumCacheRead).padStart(7)}  ` +
-      `${formatTokens(total.sumOutput).padStart(7)}  ` +
-      `${formatTokens(totalPrompt).padStart(7)}  ` +
-      `${cacheHitRate.toFixed(1).padStart(5)}%  ` +
-      `${(total.sumTokensPerSec / total.count).toFixed(1).padStart(5)}`,
-    ];
+      {
+        aligns: ["left", "right", "right", "right", "right", "right", "right", "right"],
+        totalRow: [
+          "合计",
+          String(total.count),
+          formatTokens(total.sumInput),
+          formatTokens(total.sumCacheRead),
+          formatTokens(total.sumOutput),
+          formatTokens(totalPrompt),
+          `${cacheHitRate.toFixed(1)}%`,
+          `${(total.sumTokensPerSec / total.count).toFixed(1)}`,
+        ],
+      },
+    );
 
     await showStats(lines, `${month} 月度汇总`, ctx);
   }
 
   // ── 事件注册 ─────────────────────────────────────────
 
-  // ── message renderer: 渲染 /stats 发出的消息 ─────────
-  pi.registerMessageRenderer("token-stats", (message, _options, _theme) => {
-    return new Text(message.content, 0, 0);
+  // ── message renderer: 渲染 /stats 发出的消息（markdown 表格）──
+  pi.registerMessageRenderer("token-stats", (message, _options, theme) => {
+    return new Markdown(message.content, 0, 0, getMarkdownTheme(), {
+      color: (text) => theme.fg("dim", text),
+    });
   });
 
   // ── turn_start: 记录时间 + 检测供应商切换 ──────────
@@ -1925,12 +2042,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 =>