dsh-any-agent 0.6.1 → 0.6.3

package/src/conversation.mjs

@@ -40,7 +40,7 @@ import { rssMb } from './procstat.mjs';
 import { readConversationMeta } from './titles.mjs';
 import { gitStatus, sameWorkspaceSnapshot } from './git-snapshot.mjs';
 import { writeTombstone } from './hub-store.mjs';
-import { emitLive, livePreview, removeLive, summarizeToolInput } from './live.mjs';
+import { emitLive, livePreview, liveSettings, removeLive, summarizeToolInput } from './live.mjs';
 import {
   classifyHolder,
   describeLeaseConflict,
@@ -158,6 +158,61 @@ export class Conversation {
     emitLive(this.id, type, { run: this.#liveRunId(), turn: this.currentTurn, ...fields }, detail);
   }
 
+  /**
+   * live 事件流的 token 跟踪：`{runId, base, at, last}`。
+   * DSH 在投影缓存里维护**会话累计**的 tokenUsage.totals，并在回合中每个模型步之后更新
+   * （真机实测）；回合开始时记下基线，之后发的是"本回合增量"。
+   */
+  #liveUsage = null;
+
+  /** 读 DSH 投影缓存里的 token 累计与上下文压力；读不到返回 null。 */
+  #readTokenUsage() {
+    try {
+      const meta = readConversationMeta(this.id);
+      const t = meta?.usage?.totals;
+      const p = meta?.pressure;
+      if (!t && !p) return null;
+      const n = (v) => (Number.isFinite(v) && v > 0 ? v : 0);
+      return {
+        totals: t
+          ? { input: n(t.uncachedInputTokens), output: n(t.outputTokens), cache_read: n(t.cacheReadTokens), cache_write: n(t.cacheWriteTokens) }
+          : null,
+        context: p && Number.isFinite(p.contextWindow) ? { used: n(p.pressureTokens ?? p.surfaceTokens), window: p.contextWindow } : null,
+      };
+    } catch {
+      return null;
+    }
+  }
+
+  /**
+   * 发一条 `usage` 事件（本回合 token 增量 + 会话累计 + 上下文占用）。
+   * 回合中节流到 2 秒一次、且只在数字变化时发；回合结束后 force 再发一次最终值。
+   */
+  #liveUsageTick({ force = false } = {}) {
+    const u = this.#liveUsage;
+    if (!u) return;
+    const t = now();
+    if (!force && t - u.at < 2000) return;
+    u.at = t;
+    const cur = this.#readTokenUsage();
+    if (!cur?.totals && !cur?.context) return;
+    const base = u.base ?? { input: 0, output: 0, cache_read: 0, cache_write: 0 };
+    const tokens = cur.totals
+      ? Object.fromEntries(Object.keys(base).map((k) => [k, Math.max(0, cur.totals[k] - base[k])]))
+      : null;
+    const sig = JSON.stringify([tokens, cur.context]);
+    if (!force && sig === u.last) return;
+    u.last = sig;
+    emitLive(this.id, 'usage', {
+      run: u.runId,
+      turn: u.turn,
+      tokens,
+      session_tokens: cur.totals,
+      context: cur.context,
+      final: force || undefined,
+    });
+  }
+
   /** 审批裁决事件（decide / 超时 / 回合结束 / 关闭 统一走这里）。 */
   #liveApprovalDecided(approvalId, by, decision = null, effective = null) {
     this.#live('approval.decided', { approval_id: approvalId, by, decision, effective });
@@ -598,6 +653,7 @@ export class Conversation {
       }
       case 'usage_update':
         c.usage = u;
+        this.#liveUsageTick();
         break;
       case 'plan':
       case 'plan_update':
@@ -635,6 +691,7 @@ export class Conversation {
       },
       { title: livePreview(merged.title, 200), input_preview: summarizeToolInput(merged.input) },
     );
+    this.#liveUsageTick();
   }
 
   async onPermissionRequest(params) {
@@ -789,6 +846,7 @@ export class Conversation {
     runRecord.status = 'error';
     runRecord.error = error?.message ?? String(error);
     runRecord.elapsed_ms = 0;
+    if (this.#liveUsage?.runId === runRecord.run_id) this.#liveUsage = null;
     emitLive(
       this.id,
       'run.end',
@@ -912,6 +970,10 @@ export class Conversation {
     this.pushOut('user', text.length > MAX_PROMPT_ECHO ? `${text.slice(0, MAX_PROMPT_ECHO)}…（已截断）` : text);
     this.pushOut('turn', `第 ${this.currentTurn} 轮开始`, { phase: 'start' });
     this.#liveTools.clear();
+    // live 关闭时不读投影缓存（emitLive 会直接丢弃，读了也白读）。
+    this.#liveUsage = liveSettings().level === 'off'
+      ? null
+      : { runId: runRecord.run_id, turn: this.currentTurn, base: this.#readTokenUsage()?.totals ?? null, at: 0, last: '' };
     this.#live(
       'run.start',
       { via, permission: this.permission, on_approval: this.onApproval, cwd: this.cwd },
@@ -1019,6 +1081,12 @@ export class Conversation {
           this.#scheduleDrain();
           // 到这里才允许被 git 拖慢；结果/审核单/哨兵必须等快照完成后再落。
           const after = await afterPromise;
+          // 回合结束后 DSH 的 token 累计已经落盘（真机实测在哨兵之前）：发一次最终值。
+          // 只发给本回合自己的跟踪器——下一回合可能已经在 git 快照期间开始了。
+          if (this.#liveUsage?.runId === runRecord.run_id) {
+            this.#liveUsageTick({ force: true });
+            this.#liveUsage = null;
+          }
           const result = {
             conversation_id: this.id,
             run_id: runRecord.run_id,

package/src/doctor.mjs

@@ -1,13 +1,14 @@
 /**
  * dsh_doctor：只读体检（DSH 可执行文件与版本、profile、状态文件、runs/locks 目录、
- * 孤儿锁、会话数与 RSS、生效配置）。只报告，不修改任何东西。
+ * live 事件流、孤儿锁、会话数与 RSS、生效配置）。只报告，不修改任何东西
+ * （目录可写性用一个临时探测文件，写完立刻删掉）。
  *
  * 从 tools.mjs 拆出来：它是处理器表里最长的一项，且只依赖 `hub` 的两个只读字段
  * （conversations / hosts）与一批探测函数。返回值形状与原处理器一致：
  * `{ structured: { ok, verdict, checks }, text }`。
  */
 
-import { mkdirSync, readFileSync, rmSync, writeFileSync } from 'node:fs';
+import { mkdirSync, readFileSync, readdirSync, rmSync, statSync, writeFileSync } from 'node:fs';
 import { join } from 'node:path';
 import {
   DEFAULT_PERMISSION,
@@ -34,6 +35,58 @@ import {
 import { inspectProfile } from './bootstrap.mjs';
 import { listOrphans, LOCKS_DIR } from './locks.mjs';
 import { rssMb } from './procstat.mjs';
+import { liveSettings } from './live.mjs';
+
+/** live 目录占用超过这个值就提醒（TTL 正常时远到不了）。 */
+const LIVE_WARN_BYTES = 500 * 1024 * 1024;
+
+/**
+ * 体检 live 事件流：级别、目录可写、会话数、占用、最近一条事件的时间。
+ * 返回 `[status, detail, fix]`，供 add('live', …) 使用。
+ */
+export function inspectLive({ dir = liveSettings().dir, level = liveSettings().level, now = Date.now() } = {}) {
+  if (level === 'off') {
+    return ['ok', `已关闭（DSH_MCP_LIVE=off）：watch / dashboard 看不到新的派活；目录 ${dir}`, null];
+  }
+  try {
+    mkdirSync(dir, { recursive: true });
+    const probe = join(dir, `.doctor-${process.pid}`);
+    writeFileSync(probe, '');
+    rmSync(probe, { force: true });
+  } catch (e) {
+    return [
+      'warn',
+      `级别=${level}，但目录不可写：${dir}（${e.message}）—— 派活不受影响，只是 watch / dashboard 看不到事件`,
+      '检查权限，或用 DSH_MCP_LIVE_DIR 指向可写目录',
+    ];
+  }
+  let convs = 0;
+  let bytes = 0;
+  let newest = 0;
+  for (const ent of readdirSync(dir, { withFileTypes: true })) {
+    if (!ent.isDirectory()) continue;
+    convs++;
+    for (const f of readdirSync(join(dir, ent.name), { withFileTypes: true })) {
+      if (!f.isFile()) continue;
+      try {
+        const st = statSync(join(dir, ent.name, f.name));
+        bytes += st.size;
+        if (st.mtimeMs > newest) newest = st.mtimeMs;
+      } catch {
+        /* 正在滚动 / 被清理：跳过 */
+      }
+    }
+  }
+  const mb = (bytes / 1024 / 1024).toFixed(1);
+  const age = newest ? Math.round((now - newest) / 60000) : null;
+  const last = newest ? (age < 1 ? '最近一条事件：刚刚' : `最近一条事件：${age} 分钟前`) : '还没有任何事件';
+  const big = bytes > LIVE_WARN_BYTES;
+  return [
+    big ? 'warn' : 'ok',
+    `级别=${level}，${dir}，${convs} 个会话，${mb}MB；${last}`,
+    big ? '占用偏大：检查 DSH_MCP_SENTINEL_TTL_MS 是否被设成 0（关闭清理），或改用 DSH_MCP_LIVE=meta' : null,
+  ];
+}
 
 export async function runDoctor(hub) {
   const checks = [];
@@ -148,6 +201,14 @@ export async function runDoctor(hub) {
     }
   }
 
+  // live 事件流（watch / dashboard 的数据源）。可选功能、写失败不影响派活，
+  // 所以有问题只报 warn，不报 fail；但"为什么 watch 里什么都看不到"要能从这里查出来。
+  try {
+    add('live', ...inspectLive());
+  } catch (e) {
+    add('live', 'warn', e.message);
+  }
+
   try {
     const orphans = listOrphans();
     add(

package/src/live-cli.mjs

@@ -6,7 +6,7 @@
  */
 
 import { LIVE_DIR, STATE_FILE } from './config.mjs';
-import { LiveTail, renderText } from './live-view.mjs';
+import { LiveTail, TEXT, detectLang, diffFrame, layoutFrame, renderText } from './live-view.mjs';
 import { readConversationMeta } from './titles.mjs';
 
 function parseFlags(args) {
@@ -36,14 +36,17 @@ export function makeTail(dir = LIVE_DIR) {
   });
 }
 
-export const WATCH_USAGE =
-  '用法: dsh-any-agent watch [会话id前缀] [--once] [--interval 毫秒] [--runs N] [--items N] [--no-color]\n';
+export const WATCH_USAGE = {
+  zh: '用法: dsh-any-agent watch [会话id前缀] [--once] [--interval 毫秒] [--runs N] [--items N] [--no-color] [--lang zh|en]\n',
+  en: 'usage: dsh-any-agent watch [conversation id prefix] [--once] [--interval ms] [--runs N] [--items N] [--no-color] [--lang zh|en]\n',
+};
 
 /** watch：终端里实时显示调用链（Ctrl+C 退出）。`--once` 打印一次就退出。 */
 export async function runWatch(args, { out = process.stdout } = {}) {
   const { flags, positional } = parseFlags(args);
+  const lang = detectLang(typeof flags.lang === 'string' ? flags.lang : null);
   if (flags.help || flags.h) {
-    out.write(WATCH_USAGE);
+    out.write(WATCH_USAGE[lang]);
     return 0;
   }
   const tail = makeTail(typeof flags.dir === 'string' ? flags.dir : LIVE_DIR);
@@ -52,6 +55,7 @@ export async function runWatch(args, { out = process.stdout } = {}) {
     runs: intFlag(flags.runs, 3, 1, 30),
     items: intFlag(flags.items, 12, 1, 300),
     color: !flags['no-color'] && Boolean(out.isTTY),
+    lang,
   };
   tail.poll();
   if (flags.once) {
@@ -59,19 +63,42 @@ export async function runWatch(args, { out = process.stdout } = {}) {
     return 0;
   }
   const interval = intFlag(flags.interval, 500, 100, 10_000);
-  const draw = () => {
-    // 清屏 + 光标回到左上角，再整屏重画（树会变长变短，增量刷新不值得）。
-    out.write('\x1b[2J\x1b[H' + renderText(tail.view(), opts) + `\n\n${'\x1b[2m'}live 目录: ${tail.dir} · Ctrl+C 退出${'\x1b[0m'}\n`);
+  const tty = Boolean(out.isTTY);
+  const clock = () => new Date().toTimeString().slice(0, 8);
+  const frameText = () => renderText(tail.view(), opts) + `\n\n\x1b[2m${TEXT[lang].footer(clock(), tail.dir)}\x1b[0m`;
+
+  // ★ 在 TTY 上用"备用屏幕"原地刷新（同 top / htop）：每帧只重写变化的行。
+  //   旧实现每次 `ESC[2J` 清屏，Windows Terminal 会把被清掉的整屏推进回滚区，
+  //   往上一翻就是一摞重复的帧。备用屏幕没有回滚区，退出时原样恢复用户的终端。
+  let prev = null;
+  let lastText = '';
+  const draw = (force = false) => {
+    const text = frameText();
+    if (!tty) {
+      // 管道 / 重定向：没有光标控制，内容变了才追加一帧（去掉时间行再比较，免得每次都算变化）。
+      const body = text.slice(0, text.lastIndexOf('\n\n'));
+      if (body !== lastText) out.write(text.replace(/\x1b\[[0-9;?]*[A-Za-z]/g, '') + '\n\n');
+      lastText = body;
+      return;
+    }
+    const next = layoutFrame(text, { cols: out.columns ?? 120, rows: out.rows ?? 40, lang });
+    const seq = diffFrame(force ? null : prev, next);
+    if (seq) out.write(seq);
+    prev = next;
+  };
+  const restore = () => {
+    if (tty) out.write('\x1b[0m\x1b[?25h\x1b[?1049l');
   };
-  draw();
-  let lastDraw = Date.now();
+  if (tty) out.write('\x1b[?1049h\x1b[?25l');
+  process.once('exit', restore);
+  draw(true);
+  const onResize = () => draw(true);
+  if (tty) out.on?.('resize', onResize);
   await new Promise((resolve) => {
     const timer = setInterval(() => {
-      // 有新事件就重画；没有也每 5 秒重画一次（刷新"写者是否还活着"）。
-      if (tail.poll() || Date.now() - lastDraw > 5000) {
-        draw();
-        lastDraw = Date.now();
-      }
+      // 有新事件才重算；时间行每秒在变，所以每次都会写，但 diffFrame 只重写真正变了的那几行。
+      tail.poll();
+      draw();
     }, interval);
     const stop = () => {
       clearInterval(timer);
@@ -80,16 +107,55 @@ export async function runWatch(args, { out = process.stdout } = {}) {
     process.once('SIGINT', stop);
     process.once('SIGTERM', stop);
   });
+  out.off?.('resize', onResize);
+  restore();
+  process.off('exit', restore);
   return 0;
 }
 
-export const DASHBOARD_USAGE = '用法: dsh-any-agent dashboard [--port 端口] [--interval 毫秒]\n';
+export const DASHBOARD_USAGE = {
+  zh: '用法: dsh-any-agent dashboard [--port 端口] [--interval 毫秒] [--open] [--lang zh|en]\n',
+  en: 'usage: dsh-any-agent dashboard [--port port] [--interval ms] [--open] [--lang zh|en]\n',
+};
+
+const DASHBOARD_BANNER = {
+  zh: (url, dir) =>
+    `DSH live dashboard（只读）\n  打开: ${url}\n  live 目录: ${dir}\n  链接里带一次性 token，只在本机 127.0.0.1 可访问；Ctrl+C 退出。\n`,
+  en: (url, dir) =>
+    `DSH live dashboard (read-only)\n  open: ${url}\n  live dir: ${dir}\n  The link carries a one-time token and only works on 127.0.0.1; Ctrl+C to quit.\n`,
+};
+
+/**
+ * 用系统默认浏览器打开地址。失败不抛错（打印的地址照样能手动打开）。
+ * 地址里的 token 只含 base64url 字符，不会被 cmd 的 `start` 当成元字符。
+ *
+ * ★ Windows：`start` 的第一个带引号参数是窗口标题，所以要先给一个空标题。
+ *   必须传空字符串 ''（Node 会把它转成 `""`）；传字面量 '""' 会被转义成 `"\"\""`，
+ *   cmd 收到的就不是空标题了（实测）。
+ */
+export function openInBrowser(url, { platform = process.platform, spawnFn } = {}) {
+  const [cmd, args] =
+    platform === 'win32'
+      ? ['cmd', ['/c', 'start', '', url]]
+      : platform === 'darwin'
+        ? ['open', [url]]
+        : ['xdg-open', [url]];
+  return import('node:child_process')
+    .then(({ spawn }) => {
+      const child = (spawnFn ?? spawn)(cmd, args, { stdio: 'ignore', detached: true, windowsHide: true });
+      child.on?.('error', () => {});
+      child.unref?.();
+      return true;
+    })
+    .catch(() => false);
+}
 
 /** dashboard：起一个只读本地页面（127.0.0.1 + 一次性 token），Ctrl+C 退出。 */
 export async function runDashboard(args, { out = process.stdout } = {}) {
   const { flags } = parseFlags(args);
+  const lang = detectLang(typeof flags.lang === 'string' ? flags.lang : null);
   if (flags.help || flags.h) {
-    out.write(DASHBOARD_USAGE);
+    out.write(DASHBOARD_USAGE[lang]);
     return 0;
   }
   const { startDashboard } = await import('./live-dashboard.mjs');
@@ -99,12 +165,10 @@ export async function runDashboard(args, { out = process.stdout } = {}) {
     port: intFlag(flags.port, 0, 0, 65535),
     pollMs: intFlag(flags.interval, 500, 100, 10_000),
   });
-  out.write(
-    `DSH live dashboard（只读）\n` +
-      `  打开: ${dash.url}\n` +
-      `  live 目录: ${tail.dir}\n` +
-      `  链接里带一次性 token，只在本机 127.0.0.1 可访问；Ctrl+C 退出。\n`,
-  );
+  // 页面自己也会按浏览器语言选；这里把命令行的语言带过去，两边一致（页面上还能再切换）。
+  const url = `${dash.url}&lang=${lang}`;
+  out.write(DASHBOARD_BANNER[lang](url, tail.dir));
+  if (flags.open) await openInBrowser(url);
   await new Promise((resolve) => {
     const stop = () => resolve();
     process.once('SIGINT', stop);