token-stats-timer 1.0.15 → 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.15",
+  "version": "1.0.16",
   "description": "pi extension to display token usage and time spend.",
   "keywords": [
     "pi-package",

package/token-stats.ts

@@ -151,6 +151,7 @@ export type DisplayKey =
   | "context"     // 容量（ctx%）
   | "quota5h"     // 5h 额度
   | "quotaWeek"   // 周额度
+  | "quotaMonth"  // 月额度（OpenCode Go 等）
   | "quotaClock"  // 刷新时间（⏱）
 
 export interface DisplayConfig {
@@ -543,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 };
@@ -557,6 +625,7 @@ const DEFAULT_DISPLAY_CONFIG: DisplayConfig = {
     context: true,
     quota5h: true,
     quotaWeek: true,
+    quotaMonth: true,
     quotaClock: true,
   },
   contextStyle: "pct-window",
@@ -786,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]);
@@ -1969,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 =>