token-stats-timer 1.0.10 → 1.0.12

package/README.md

@@ -9,10 +9,9 @@
 Footer 上行（左对齐）：
 
 ```
-● run 01:23 · prev 00:40 · max 03:12 (Review README…) | ↑12k ↓3.4k CH87% ⚡77.7 t/s 5.3%/1.0M | 5h: 87% W: 92% ⏱ 2h 15m
+↑12k ↓3.4k CH87% ⚡77.7 t/s 5.3%/1.0M | 5h: 87% W: 92% ⏱ 2h 15m
 ```
 
-- `● run 01:23` —— 当前 run 已耗时（从空闲后的第一个 `agent_start` 到 `agent_settled`，含重试/压缩恢复/排队提示）
 - `prev` / `max` —— 上一次 run 耗时 / 本会话分支内最长 run 耗时（带 15 字 prompt 预览）
 - `↑ ↓ Σ CH` —— 累计输入 / 输出 / 总量 / 缓存命中率
 - `⚡` —— 实时速率（2s rolling window，无流时回落到平均速率）
@@ -28,7 +27,20 @@ Footer 下行：cwd + git 分支 + 其他扩展状态。
 - ✅ pi 执行完成 —— 正常结束（含耗时）
 - ⏹ pi 已中止 —— 用户 Esc 中止
 
-会话关闭时另有 👋 提示（可关）。通知经 `osascript` 发送，需在系统设置 → 通知 中允许终端 App 通知。
+会话关闭时另有 👋 提示（可关）。
+
+### 投递通道（自动降级，优先用能送达的）
+
+1. **OSC 终端协议** —— 向终端写转义序列，由**终端应用自己**弹系统通知，归属明确，无需 osascript：
+   - **iTerm2 → OSC 9**（`ESC]9;内容`，iTerm2 官方文档的通知序列，**实测有效**）
+   - Ghostty / WezTerm / Hyper / rxvt-unicode → OSC 777（`ESC]777;notify;标题;内容`）
+   > ⚠️ 注意：OSC 777 在 iTerm2 3.5.x 上会静默忽略（3.6.9 才支持），官方 pi 示例（notify.ts）只发 OSC 777，在 iTerm2 上不生效——本插件已按终端选择序列。
+2. **terminal-notifier** —— 装了 `brew install terminal-notifier` 后自动使用，系统级最可靠（`-sender` 指定归属、`-group` 去重）。
+3. **osascript** —— 保底（无法用终端协议时的最后手段）。
+
+> 为什么不用纯 osascript：macOS Sequoia 起通知按“调用进程”归属，pi 是 node 进程、无法在系统设置里授权，通知会被静默丢弃（脚本 exit 0 但没弹窗）。OSC 序列把归属改到终端 App，授权后即可正常弹。
+
+每次投递结果追加到 `~/.pi/agent/extensions/token-stats-logs/notify.log`，没弹通知时先看它判断是触发问题还是投递问题。
 
 ### 配置
 
@@ -48,10 +60,10 @@ Footer 下行：cwd + git 分支 + 其他扩展状态。
 
 - `enabled` —— 总开关（也可用 `/notify on|off` 切换）
 - `minDurationSec` —— 耗时低于该秒数的 run 不通知（设为 30 可避免秒回打扰）
-- `sound` —— 通知声音（macOS 内置：Glass/Ping/Sosumi/Hero/Funk 等，`""` 静音）
+- `sound` —— 通知声音（macOS 内置：Glass/Ping/Sosumi/Hero/Funk 等，`""` 静音；注意仅 osascript/terminal-notifier 通道有声，OSC 777 无声音参数）
 - `onSuccess/onFailure/onAbort/onSessionEnd` —— 各类通知开关
 
-`/notify` —— 无参查看当前状态；`on`/`off` 开启/关闭。
+`/notify` —— 无参查看当前状态；`on`/`off` 开关；`test` 立即发一条测试通知验证通道是否可达（先看系统通知栏有没有，没有再看 notify.log）。
 
 ## 任务计时（step-timer）
 
@@ -60,16 +72,16 @@ Footer 下行：cwd + git 分支 + 其他扩展状态。
 
   `总耗时 01:23`
 
-  经 `appendEntry` 持久化、不进入 LLM 上下文，`/resume` 后仍在；无主题色样式。
+  经 `appendEntry` 持久化、不进入 LLM 上下文，`/resume` 后仍在。
 
-无独立开关，随包启用；计时口径与 run-timer 一致（一次 run = 首个 agent_start → agent_settled，含重试/压缩/排队提示）。早期版本的每 turn 摘要（timing-turn）、“⏱ 思考”每条消息内联标注与 Thinking 实时标签均已移除，历史条目自动隐藏。
+无独立开关，随包启用；计时口径与 run-timer 一致（一次 run = 首个 agent_start → agent_settled，含重试/压缩/排队提示）。
 
 ## 命令
 
 - `/stats` —— 无参进入套餐配置（为当前 provider 选择/关闭配额套餐）
 - `/stats day [YYYY-MM-DD]` / `hour` / `week` / `month [YYYY-MM]` —— 统计查询
 - `/stats config` —— 显示样式 / 显示内容 / 配额刷新时间
-- `/notify [on|off]` —— macOS 完成通知开关（无参查看状态）
+- `/notify [on|off|test]` —— 通知开关 / 测试（无参查看状态）
 
 ## 与两个原包的兼容性
 
@@ -77,7 +89,7 @@ Footer 下行：cwd + git 分支 + 其他扩展状态。
 - 统计日志沿用 `~/.pi/agent/extensions/token-stats-logs/`（历史数据 `/stats` 可直接查询）
 - run 计时状态沿用 session 内 `run-timer-state` 自定义条目（`/reload` 后自动恢复，兼容旧 `turn-timer-state`）
 
-## 安装（替换两个原包）
+## 安装（替换原包）
 
 ```bash
 pi remove npm:@liziy/token-stats

package/notify.ts

@@ -11,16 +11,49 @@
 //   sound         通知声音（默认 "Glass"，"" 表示静音）
 //   onSuccess / onAbort / onSessionEnd  各类通知开关
 //
-// 命令：/notify on | off | status
+// 命令：/notify on | off | status | test
+//
+// ── 投递通道（按优先级自动降级）──────────────────────────────
+//   1. OSC 终端协议：向 stdout 写转义序列，由终端自己弹系统通知，归属终端应用。
+//      ⚠️ 重要：iTerm2 官方文档的通知序列是 OSC 9（`ESC]9;msg`）！
+//         OSC 777 是 rxvt 系/Ghostty/WezTerm 用的，iTerm2 3.6.9 才支持，
+//         iTerm2 3.5.x 收到 OSC 777 会静默忽略（实测确认）。
+//      - iTerm2 → OSC 9（无标题字段，格式为 `标题\n内容`）
+//      - Ghostty / WezTerm / Hyper / rxvt-unicode → OSC 777（title;body）
+//   2. terminal-notifier：brew install terminal-notifier 装了之后用它，
+//      系统级最可靠，-sender 指定归属 bundle id，-group 去重。
+//   3. osascript：保底（无法用终端协议时的最后手段）。
+//
+// 每次投递结果追加到 ~/.pi/agent/extensions/token-stats-logs/notify.log，
+// 便于排查"没弹通知"是触发问题还是投递问题。
 
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { spawn } from "node:child_process";
-import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { spawn, spawnSync } from "node:child_process";
+import {
+  appendFileSync,
+  existsSync,
+  mkdirSync,
+  readFileSync,
+  writeFileSync,
+} from "node:fs";
 import { homedir } from "node:os";
 import { join } from "node:path";
 
 const CONFIG_DIR = join(homedir(), ".pi/agent/extensions/token-stats");
 const CONFIG_FILE = join(CONFIG_DIR, "notify-config.json");
+const LOG_DIR = join(homedir(), ".pi/agent/extensions/token-stats-logs");
+const NOTIFY_LOG = join(LOG_DIR, "notify.log");
+
+/** 支持 OSC 777 终端协议的应用（按 TERM_PROGRAM 匹配） */
+const OSC777_TERMINALS = new Set([
+  "iTerm.app",
+  "iTerm2",
+  "WezTerm",
+  "WezTerm.app",
+  "ghostty",
+  "Ghostty",
+  "Hyper",
+]);
 
 export interface NotifyConfig {
   enabled: boolean;
@@ -58,6 +91,19 @@ function saveConfig(cfg: NotifyConfig): void {
 /** 连续两次通知的最短间隔（毫秒），防止并行 agent 结束时刷屏 */
 const DEBOUNCE_MS = 3000;
 
+function logLine(channel: string, title: string, ok: boolean, detail = ""): void {
+  try {
+    mkdirSync(LOG_DIR, { recursive: true });
+    appendFileSync(
+      NOTIFY_LOG,
+      `${new Date().toISOString()} [${channel}] ${ok ? "OK" : "FAIL"} ${JSON.stringify(title)}${detail ? ` ${detail}` : ""}\n`,
+      "utf-8",
+    );
+  } catch {
+    // 日志失败不影响通知本身
+  }
+}
+
 function formatDuration(ms: number): string {
   const totalSeconds = Math.max(0, Math.floor(ms / 1000));
   const hours = Math.floor(totalSeconds / 3600);
@@ -73,6 +119,79 @@ function truncate(s: string, max = 80): string {
   return s.length > max ? s.slice(0, max) + "…" : s;
 }
 
+/** 终端是否支持 OSC 777 通知协议（非 iTerm2 的现代终端：Ghostty/WezTerm/Hyper/rxvt） */
+function supportsOSC777(): boolean {
+  const tp = process.env.TERM_PROGRAM || "";
+  if (OSC777_TERMINALS.has(tp)) return true;
+  // rxvt-unicode 系列仅能从 TERM 判断
+  return /rxvt|urxvt/i.test(process.env.TERM || "");
+}
+
+/** 是否 iTerm2（iTerm2 官方文档的通知序列是 OSC 9，不是 OSC 777） */
+function isITerm2(): boolean {
+  const tp = process.env.TERM_PROGRAM || "";
+  return tp === "iTerm.app" || tp === "iTerm2";
+}
+
+/** 通道1a：OSC 9 终端通知（iTerm2 官方文档序列，无标题字段，格式 `标题\n内容`） */
+function notifyOSC9(title: string, message: string): void {
+  const safe = (s: string) => s.replace(/[\x07\x1b]/g, "").replace(/;/g, "·");
+  process.stdout.write(`\x1b]9;${safe(title)}\n${safe(message)}\x07`);
+}
+
+/** 通道1b：OSC 777 终端通知（Ghostty / WezTerm / Hyper / rxvt，title;body 结构） */
+function notifyOSC777(title: string, message: string): void {
+  // 去掉 BEL/ESC 等可能截断/注入的字符；OSC 参数以 ; 分隔，替换为 · 避免歧义
+  const safe = (s: string) => s.replace(/[\x07\x1b]/g, "").replace(/;/g, "·");
+  process.stdout.write(`\x1b]777;notify;${safe(title)};${safe(message)}\x07`);
+}
+
+/** 通道2：terminal-notifier（brew 安装后自动使用），-sender 归属终端保证权限生效 */
+function notifyTerminalNotifier(title: string, message: string, sound: string): boolean {
+  const candidates = [
+    "/opt/homebrew/bin/terminal-notifier",
+    "/usr/local/bin/terminal-notifier",
+  ];
+  const exe = candidates.find((p) => existsSync(p));
+  if (!exe) return false;
+
+  const sender =
+    process.env.TERM_PROGRAM === "iTerm.app"
+      ? "com.googlecode.iterm2"
+      : process.env.TERM_PROGRAM === "WezTerm"
+        ? "org.wezfurlong.wezterm"
+        : undefined;
+  const args = ["-title", title, "-message", message, "-group", "pi-agent"];
+  if (sound) args.push("-sound", sound);
+  if (sender) args.push("-sender", sender);
+
+  try {
+    const r = spawnSync(exe, args, { timeout: 5000 });
+    return !r.error && r.status === 0;
+  } catch {
+    return false;
+  }
+}
+
+/** 通道3：osascript 保底（不支持 OSC 777 的终端）；失败信息写日志供排查 */
+function notifyOSAScript(title: string, message: string, sound: string): void {
+  const soundPart = sound ? ` sound name ${JSON.stringify(sound)}` : "";
+  const script = `display notification ${JSON.stringify(message)} with title ${JSON.stringify(title)}${soundPart}`;
+  const child = spawn("osascript", ["-e", script], {
+    detached: true,
+    stdio: ["ignore", "ignore", "pipe"],
+  });
+  let err = "";
+  child.stderr?.on("data", (d) => {
+    err += d.toString();
+  });
+  child.on("error", (e) => logLine("osascript", title, false, e.message));
+  child.on("exit", (code) => {
+    if (code !== 0) logLine("osascript", title, false, err.trim() || `exit ${code}`);
+  });
+  child.unref();
+}
+
 export function createNotifier(pi: ExtensionAPI): void {
   let config: NotifyConfig = loadConfig();
 
@@ -83,6 +202,7 @@ export function createNotifier(pi: ExtensionAPI): void {
   // 进程内去抖计时
   let lastNotifyAt = 0;
 
+  /** 多通道投递：iTerm2 用 OSC 9 → 其他终端 OSC 777 → terminal-notifier → osascript */
   function notify(title: string, message: string, sound: string): void {
     if (process.platform !== "darwin") return; // 非 macOS 直接跳过
 
@@ -90,11 +210,26 @@ export function createNotifier(pi: ExtensionAPI): void {
     if (now - lastNotifyAt < DEBOUNCE_MS) return;
     lastNotifyAt = now;
 
-    // spawn + 参数数组传参，避免 shell 注入；detached + unref 不阻塞 pi
-    const soundPart = sound ? ` sound name ${JSON.stringify(sound)}` : "";
-    const script = `display notification ${JSON.stringify(message)} with title ${JSON.stringify(title)}${soundPart}`;
-    const child = spawn("osascript", ["-e", script], { detached: true });
-    child.unref();
+    if (isITerm2()) {
+      // iTerm2：官方文档通知序列 OSC 9（实测有效）
+      notifyOSC9(title, message);
+      logLine("osc9", title, true);
+      return;
+    }
+
+    if (supportsOSC777()) {
+      // Ghostty / WezTerm / Hyper / rxvt：OSC 777
+      notifyOSC777(title, message);
+      logLine("osc777", title, true);
+      return;
+    }
+
+    if (notifyTerminalNotifier(title, message, sound)) {
+      logLine("terminal-notifier", title, true);
+      return;
+    }
+
+    notifyOSAScript(title, message, sound);
   }
 
   pi.on("agent_start", () => {
@@ -155,7 +290,7 @@ export function createNotifier(pi: ExtensionAPI): void {
 
   // ── /notify 命令 ─────────────────────────────────────
   pi.registerCommand("notify", {
-    description: "macOS 完成通知: on | off | status（详细配置见 notify-config.json）",
+    description: "macOS 完成通知: on | off | status | test（详细配置见 notify-config.json）",
     handler: async (args, ctx) => {
       const arg = args.trim();
       if (arg === "on") {
@@ -166,6 +301,19 @@ export function createNotifier(pi: ExtensionAPI): void {
         config = { ...config, enabled: false };
         saveConfig(config);
         ctx.ui.notify("完成通知已关闭", "info");
+      } else if (arg === "test") {
+        // 立即发一条测试通知，验证当前投递通道是否可达
+        const title = "🔔 pi 通知测试";
+        const channel = isITerm2()
+          ? "osc9 终端协议"
+          : supportsOSC777()
+            ? "osc777 终端协议"
+            : existsSync("/opt/homebrew/bin/terminal-notifier") ||
+              existsSync("/usr/local/bin/terminal-notifier")
+              ? "terminal-notifier"
+              : "osascript";
+        notify(title, `投递通道：${channel}\n时间：${new Date().toLocaleTimeString()}`, config.sound);
+        ctx.ui.notify("测试通知已发送，请查看系统通知栏", "info");
       } else {
         const sound = config.sound ? `, 声音=${config.sound}` : ", 静音";
         ctx.ui.notify(

package/package.json

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

package/step-timer.ts

@@ -92,11 +92,12 @@ export function createStepTimer(pi: ExtensionAPI): void {
     lastCtx = undefined;
   });
 
-  pi.registerEntryRenderer<FinalTimingData>(FINAL_TYPE, (entry, _opts, _theme) => {
+  pi.registerEntryRenderer<FinalTimingData>(FINAL_TYPE, (entry, _opts, theme) => {
     const d = entry.data;
     if (!d) return undefined;
-    const box = new Box(1, 1);
-    box.addChild(new Text(`总耗时 ${formatDuration(d.totalMs)}`, 0, 0));
+    const title = theme.fg("accent", "总耗时");
+    const box = new Box(1, 1, (text) => theme.bg("customMessageBg", text));
+    box.addChild(new Text(`${title} ${formatDuration(d.totalMs)}`, 0, 0));
     return box;
   });
 }