pi-session-peek 0.5.0 → 0.5.2

package/src/preview.ts

@@ -0,0 +1,139 @@
+import { Markdown, truncateToWidth, visibleWidth, type MarkdownTheme } from "@earendil-works/pi-tui";
+import { msg } from "./i18n.ts";
+import { anyMatch, highlight, lineHasMatch, type RoleTerms } from "./query.ts";
+import type { PeekMsg, PeekSession } from "./sessions.ts";
+import { padEndVisible } from "./text.ts";
+import type { PeekTheme } from "./theme.ts";
+
+// 右栏：把一个会话渲染成预览行，全是纯函数。滚动偏移和"要不要重建"的缓存键由组件持有
+
+const MAX_MSGS = 500; // 超长会话只画最后这么多条，更早的命中仍计数并提示
+
+/** 每条消息渲染好的行，按宽度缓存。键是消息对象本身，会话从缓存里掉了它也跟着掉 */
+export type RenderCache = WeakMap<PeekMsg, { w: number; lines: string[] }>;
+
+// 用 pi 自己的 Markdown 组件渲染一条消息，和主界面里的对话长得一样；参数照抄 pi 的
+// user-message / assistant-message 组件。渲染比较贵，按消息和宽度缓存，换关键词时不用重来
+export function renderMsg(m: PeekMsg, rw: number, t: PeekTheme, mdTheme: MarkdownTheme, cache: RenderCache): string[] {
+  const hit = cache.get(m);
+  if (hit && hit.w === rw) return hit.lines;
+  const md =
+    m.role === "user"
+      ? new Markdown(
+          m.text,
+          1,
+          0,
+          mdTheme,
+          { color: (s) => t.fg("userMessageText", s), bgColor: (s) => t.bg("userMessageBg", s) },
+          { preserveOrderedListMarkers: true, preserveBackslashEscapes: true },
+        )
+      : new Markdown(m.text, 1, 0, mdTheme);
+  const lines = md.render(rw);
+  cache.set(m, { w: rw, lines });
+  return lines;
+}
+
+export interface Preview {
+  lines: string[];
+  matchLines: number[]; // 含关键词的行（升序），Ctrl+N / Ctrl+P 用
+}
+
+// 右栏内容。没有会话时只有一行提示
+export function buildPreview(
+  s: PeekSession | undefined,
+  rt: RoleTerms,
+  rw: number,
+  showTools: boolean,
+  t: PeekTheme,
+  mdTheme: MarkdownTheme,
+  cache: RenderCache,
+): Preview {
+  if (!s) return { lines: [t.fg("dim", msg("noMatch"))], matchLines: [] };
+
+  const lines: string[] = [];
+  const matchLines: number[] = [];
+  // 消息数只算有文字的，纯工具轮次不计
+  const textCount = s.msgs.reduce((n, m) => n + (m.text ? 1 : 0), 0);
+  lines.push(
+    truncateToWidth(
+      t.fg("dim", `${s.cwd} • ${msg("msgCount", { n: textCount })}${s.name ? " • " + s.name : ""}`),
+      rw,
+    ),
+  );
+  lines.push("");
+
+  const hasBody = rt.user.length > 0 || rt.assistant.length > 0;
+  const truncated = s.msgs.length > MAX_MSGS;
+  const msgs = truncated ? s.msgs.slice(-MAX_MSGS) : s.msgs;
+  if (truncated) {
+    lines.push(t.fg("dim", msg("truncated", { n: MAX_MSGS })));
+    lines.push("");
+  }
+  if (hasBody && !msgs.some((m) => anyMatch(m.text, rt[m.role]))) {
+    const inTruncated = truncated && s.msgs.some((m) => anyMatch(m.text, rt[m.role]));
+    lines.push(
+      t.fg(
+        "warning",
+        inTruncated ? msg("hitInTruncated") : msg("hitOnlyMeta"),
+      ),
+    );
+    lines.push("");
+  }
+
+  // 用户消息整块背景色，AI 消息只有一条细线标签
+  const fillLabel = (label: string): string => {
+    const w = visibleWidth(label);
+    return truncateToWidth(label + " " + "─".repeat(Math.max(2, rw - w - 1)), rw);
+  };
+
+  let prevRole: PeekMsg["role"] | undefined; // 上一条画出来的消息的角色
+  for (const m of msgs) {
+    const tools = showTools ? (m.tools ?? []) : [];
+    if (!m.text && !tools.length) continue; // 工具隐藏时，只有工具调用的 AI 轮次整条跳过
+    const terms = rt[m.role]; // user: 的词只在用户消息里高亮，ai: 的只在 AI 回复里
+    const isMatch = terms.length > 0 && anyMatch(m.text, terms);
+    // 紧跟在 AI 文字后面、只有工具调用的轮次不重复画标签，接在上一条正文下面，看起来是同一轮
+    const continues = m.role === "assistant" && prevRole === "assistant" && !m.text;
+    if (continues) {
+      lines.pop();
+    } else if (m.role === "user") {
+      lines.push(
+        t.bg("userMessageBg", padEndVisible(t.bold(t.fg("accent", fillLabel(msg("you")))), rw)),
+      );
+    } else {
+      lines.push(t.fg("muted", fillLabel(msg("ai"))));
+    }
+    const head = lines.length - 1;
+    // 先渲染再高亮：往 markdown 源码里插转义码会把链接、代码块的语法弄坏
+    const before = matchLines.length;
+    if (m.text) {
+      for (const l of renderMsg(m, rw, t, mdTheme, cache)) {
+        if (isMatch && lineHasMatch(l, terms)) matchLines.push(lines.length);
+        lines.push(isMatch ? highlight(l, terms, t) : l);
+      }
+    }
+    // 关键词被折行拆开时哪一行都找不到，退回到消息标题行，别把这条消息漏掉
+    if (isMatch && matchLines.length === before) matchLines.push(head);
+    for (const tool of tools) {
+      lines.push(t.fg("dim", truncateToWidth(` ⚙ ${tool.name}${tool.summary ? "  " + tool.summary : ""}`, rw)));
+    }
+    lines.push("");
+    prevRole = m.role;
+  }
+
+  return { lines, matchLines };
+}
+
+// 刚建好的预览该停在哪：有关键词时滚到第一个命中处（上面留两行），没有时滚到底部看最新消息
+export function initialOffset(p: Preview, H: number): number {
+  if (p.matchLines.length) return Math.max(0, p.matchLines[0] - 2);
+  return Math.max(0, p.lines.length - H);
+}
+
+// Ctrl+N / Ctrl+P：当前视口 [top, bottom] 之外的下一个 / 上一个命中行，到头了回绕。同一屏里的多个命中算一处，
+// 否则一段里连着几行都命中要按好几次才过得去。没有命中返回 undefined
+export function nextMatchTarget(matchLines: number[], top: number, bottom: number, dir: 1 | -1): number | undefined {
+  if (!matchLines.length) return undefined;
+  if (dir === 1) return matchLines.find((l) => l > bottom) ?? matchLines[0];
+  return [...matchLines].reverse().find((l) => l < top) ?? matchLines[matchLines.length - 1];
+}

package/src/query.ts

@@ -1,8 +1,10 @@
+import { applySgr, ANSI_SEQ_RE, splitAnsi, type SgrStyle } from "./ansi.ts";
 import { msg, type MsgKey } from "./i18n.ts";
 import type { PeekMsg, PeekSession } from "./sessions.ts";
+import { stylePrefix, type PeekTheme } from "./theme.ts";
 
 // 搜索框语法。空白分隔多个词，全部满足才算（AND）。每个词可以是：
-//   foo            子串，大小写不敏感
+//   foo            子串，大小写不敏感，在对话正文和会话名里找；不看工作目录（见 termHits）
 //   "foo bar"      短语，引号里的一段空白匹配正文里任意一段空白（含换行）
 //   a|b            任一命中（OR）；要搜字面的 | 就加引号
 //   /re/           正则（JS 语法，自动带 i 和 m 标志，不能含空白，用 \s 代替）；写错了按普通文字搜
@@ -124,11 +126,13 @@ function termHits(s: PeekSession, t: Term): boolean {
     case "assistant":
       return s.msgs.some((m) => m.role === t.field && hit(t.re, m.text));
     default:
-      return hit(t.re, s.name) || hit(t.re, s.cwd) || s.msgs.some((m) => hit(t.re, m.text));
+      // 普通词不看工作目录：当前目录树范围下所有会话的路径都带同一段前缀，路径里的词会命中全部会话。
+      // 要搜路径用 dir:。pi 自带的选择器会把 cwd 拼进搜索文本，这里是有意不同
+      return hit(t.re, s.name) || s.msgs.some((m) => hit(t.re, m.text));
   }
 }
 
-// 每个词都要满足：普通词在会话名、cwd 或某条消息里出现，排除词则哪里都不出现；带前缀的只看对应的地方。
+// 每个词都要满足：普通词在会话名或某条消息里出现，排除词则哪里都不出现；带前缀的只看对应的地方。
 // 正文不预存小写副本，直接在原文上用不区分大小写的正则匹配
 export function matchesSession(s: PeekSession, terms: Term[]): boolean {
   return terms.every((t) => termHits(s, t) !== t.negate);
@@ -165,70 +169,20 @@ function nextMatch(text: string, terms: Term[], from: number): [number, number]
   return [best, len];
 }
 
-// pi-tui 会产生的三类转义序列：CSI（颜色等）、OSC（超链接）、APC
-const ESC_RE = /\x1b\[[0-9;?]*[A-Za-z]|\x1b[\]_][^\x07\x1b]*(?:\x07|\x1b\\)/g;
-
-// 把样式函数拆出前缀，pi 的 Markdown 组件也是这么做的
-function stylePrefix(fn: (s: string) => string): string {
-  const s = fn(" ");
-  const i = s.indexOf(" ");
-  return i >= 0 ? s.slice(0, i) : "";
-}
-
-// 只跟踪高亮会碰到的几项：前景色、粗体 / 暗淡、下划线
-interface Sgr {
-  fg: string;
-  bold: boolean;
-  dim: boolean;
-  underline: boolean;
-}
-
-function applySgr(st: Sgr, seq: string): void {
-  const m = /^\x1b\[([0-9;]*)m$/.exec(seq);
-  if (!m) return;
-  const p = m[1] === "" ? [0] : m[1].split(";").map(Number);
-  for (let i = 0; i < p.length; i++) {
-    const c = p[i];
-    if (c === 0) {
-      st.fg = "";
-      st.bold = st.dim = st.underline = false;
-    } else if (c === 1) st.bold = true;
-    else if (c === 2) st.dim = true;
-    else if (c === 22) st.bold = st.dim = false;
-    else if (c === 4) st.underline = true;
-    else if (c === 24) st.underline = false;
-    else if ((c >= 30 && c <= 37) || (c >= 90 && c <= 97)) st.fg = `\x1b[${c}m`;
-    else if (c === 39) st.fg = "";
-    else if (c === 38 || c === 48) {
-      // 扩展色 38;5;n / 38;2;r;g;b；背景色（48）只需要跳过它的参数
-      const n = p[i + 1] === 5 ? 2 : p[i + 1] === 2 ? 4 : 0;
-      if (c === 38 && n) st.fg = `\x1b[${p.slice(i, i + n + 1).join(";")}m`;
-      i += n;
-    }
-  }
-}
-
-// 渲染后的一行（可带 ANSI）里有没有命中，预览按行记命中用
+// 渲染后的一行（可带 ANSI）里有没有命中，预览按行记命中用。和 highlight 用同一个正则去掉序列，两边才对得上
 export function lineHasMatch(line: string, terms: Term[]): boolean {
-  return terms.length > 0 && anyMatch(line.replace(ESC_RE, ""), terms);
+  return terms.length > 0 && anyMatch(line.replace(ANSI_SEQ_RE, ""), terms);
 }
 
 // 在一行文字里高亮命中，返回带 ANSI 的字符串。
 // 输入可以是已经渲染过、带样式的行：转义序列不参与匹配也不会被切断；高亮结束时把原来的
 // 前景色 / 粗体 / 下划线恢复回去，所以放在标题、链接等样式里面也不会把后面的文字弄丢样式。
 // 不用主题自带的 searchMatchBg：dark/light 主题里它和 selectedBg 同色，看不出来
-export function highlight(line: string, terms: Term[], theme: any): string {
+export function highlight(line: string, terms: Term[], theme: PeekTheme): string {
   if (!terms.length || !line) return line;
 
   // 拆成可见文本和转义序列两类片段
-  const segs: { text: string; esc: boolean }[] = [];
-  let last = 0;
-  for (const m of line.matchAll(ESC_RE)) {
-    if (m.index > last) segs.push({ text: line.slice(last, m.index), esc: false });
-    segs.push({ text: m[0], esc: true });
-    last = m.index + m[0].length;
-  }
-  if (last < line.length) segs.push({ text: line.slice(last), esc: false });
+  const segs = splitAnsi(line);
 
   let visible = "";
   for (const s of segs) if (!s.esc) visible += s.text;
@@ -243,7 +197,7 @@ export function highlight(line: string, terms: Term[], theme: any): string {
   if (!ranges.length) return line;
 
   const on = "\x1b[1m\x1b[4m" + stylePrefix((s) => theme.fg("warning", s));
-  const st: Sgr = { fg: "", bold: false, dim: false, underline: false };
+  const st: SgrStyle = { fg: "", bold: false, dim: false, underline: false };
   const off = () =>
     (st.underline ? "" : "\x1b[24m") +
     (st.bold ? "" : "\x1b[22m" + (st.dim ? "\x1b[2m" : "")) +

package/src/selection.ts

@@ -1,4 +1,5 @@
 import { sliceByColumn, stripTerminalSequences, visibleWidth } from "@earendil-works/pi-tui";
+import { ANSI_TAIL_RE, isSgr, splitAnsi } from "./ansi.ts";
 
 // 鼠标拖出来的选区。只在一个栏里：行号是这一栏内容的绝对行号（左栏每个会话两行，右栏是预览行），
 // 列号是栏内列号。起止格都算在内，中间的行整行选中，所以从左栏拖到右栏也只会选左栏的字
@@ -16,9 +17,6 @@ export interface Selection {
   focus: Cell; // 现在拖到的格
 }
 
-// CSI / OSC / APC 序列，反显时要原样保留
-const SEQ = /\x1b\[[0-9;?<>=!]*[ -/]*[@-~]|\x1b\][^\x07\x1b]*(?:\x07|\x1b\\)|\x1b_[^\x07\x1b]*(?:\x07|\x1b\\)/g;
-const TAIL = new RegExp(`(?:${SEQ.source})+$`); // 行尾的样式收尾码
 const graphemes = new Intl.Segmenter(undefined, { granularity: "grapheme" });
 
 /** 起止格，按阅读顺序 */
@@ -60,16 +58,14 @@ export function rowColumns(sel: Selection, row: number, line: string, width: num
   return c1 > c0 ? [c0, c1] : undefined;
 }
 
-// 整段反显。段里原有的样式码照留，每个样式码后面补一次反显，免得被 0m 之类的重置掉
+// 整段反显。段里原有的序列照留（反显时要原样保留），每个样式码后面补一次反显，免得被 0m 之类的重置掉
 function inverse(text: string): string {
   let out = "\x1b[7m";
-  let last = 0;
-  for (const m of text.matchAll(SEQ)) {
-    out += text.slice(last, m.index) + m[0];
-    if (m[0].startsWith("\x1b[") && m[0].endsWith("m")) out += "\x1b[7m";
-    last = m.index + m[0].length;
+  for (const seg of splitAnsi(text)) {
+    out += seg.text;
+    if (seg.esc && isSgr(seg.text)) out += "\x1b[7m";
   }
-  return out + text.slice(last) + "\x1b[27m";
+  return out + "\x1b[27m";
 }
 
 /** 给一行的 [c0, c1) 列加反显，其余部分原样 */
@@ -79,7 +75,7 @@ export function highlightColumns(line: string, c0: number, c1: number): string {
   const mid = sliceByColumn(line, c0, c1 - c0, true);
   const after = sliceByColumn(line, c1, Math.max(0, w - c1), true);
   // sliceByColumn 切到行尾时会丢掉最后那些收尾码（39m、49m 之类），补回去，否则背景色会漏到右边
-  const tail = TAIL.exec(line)?.[0] ?? "";
+  const tail = ANSI_TAIL_RE.exec(line)?.[0] ?? "";
   return before + inverse(mid) + after + tail;
 }
 

package/src/sessions.ts

@@ -2,7 +2,9 @@ import { appendFileSync, existsSync, readFileSync, rmSync } from "node:fs";
 import { readdir, readFile, stat } from "node:fs/promises";
 import { randomBytes } from "node:crypto";
 import { homedir } from "node:os";
-import { join } from "node:path";
+import { dirname, join } from "node:path";
+import { sanitizeText } from "./ansi.ts";
+import { normPath } from "./text.ts";
 
 // 读 ~/.pi/agent/sessions 下的会话 JSONL，按 mtime 缓存；重命名和删除也在这里。
 // 搜索不预存小写全文：会话正文只在 msgs 里放一份，匹配时用不区分大小写的正则直接扫（见 query.ts 的 matchesSession）
@@ -26,18 +28,29 @@ export interface PeekSession {
   first: string; // 首条消息，列表里显示用
   msgs: PeekMsg[];
   mtime: number;
+  size: number; // 和 mtime 一起当缓存键：mtime 粒度粗的文件系统上连续两次写入可能同一个 mtime
 }
 
 const sessionCache = new Map<string, PeekSession>();
 const READ_CONCURRENCY = 16; // 同时读的文件数，太高会把 libuv 线程池排满
 
-// 和 pi 一样认 PI_CODING_AGENT_DIR
-function sessionsDir(): string {
-  const env = process.env.PI_CODING_AGENT_DIR;
-  const agent = env ? (env.startsWith("~") ? join(homedir(), env.slice(1)) : env) : join(homedir(), ".pi", "agent");
+const expandTilde = (p: string) => (p.startsWith("~") ? join(homedir(), p.slice(1)) : p);
+
+// 默认布局下所有项目的会话根目录：和 pi 一样认 PI_CODING_AGENT_DIR，下面每个 cwd 一个子目录
+export function sessionsDir(env: NodeJS.ProcessEnv = process.env): string {
+  const agent = env.PI_CODING_AGENT_DIR ? expandTilde(env.PI_CODING_AGENT_DIR) : join(homedir(), ".pi", "agent");
   return join(agent, "sessions");
 }
 
+/** 要扫描的根目录。pi 的 getSessionDir() 是参数、环境变量、settings.json 合并后的结果：
+ * 默认布局时它是 <root>/sessions/<按 cwd 编码的子目录>，取上一级才能看到所有项目；
+ * 用户自定义了目录（--session-dir 等）时就是那个目录本身，直接扫它 */
+export function scanRoot(currentSessionDir: string | undefined, env: NodeJS.ProcessEnv = process.env): string {
+  const root = sessionsDir(env);
+  if (!currentSessionDir) return root;
+  return normPath(dirname(currentSessionDir)) === normPath(root) ? root : currentSessionDir;
+}
+
 async function walkJsonl(dir: string, out: string[]): Promise<void> {
   let entries;
   try {
@@ -53,12 +66,14 @@ async function walkJsonl(dir: string, out: string[]): Promise<void> {
 }
 
 function extractText(content: unknown): string {
-  if (typeof content === "string") return content;
+  if (typeof content === "string") return sanitizeText(content);
   if (!Array.isArray(content)) return "";
-  return content
-    .filter((c: any) => c?.type === "text" && typeof c.text === "string")
-    .map((c: any) => c.text)
-    .join("\n");
+  return sanitizeText(
+    content
+      .filter((c: any) => c?.type === "text" && typeof c.text === "string")
+      .map((c: any) => c.text)
+      .join("\n"),
+  );
 }
 
 // 工具调用的主参数：按常见键名找第一个字符串，都没有就取第一个字符串参数。不存结果，bash 输出动辄几十 KB
@@ -71,7 +86,7 @@ export function toolSummary(args: unknown): string {
   let v = TOOL_ARG_KEYS.map((k) => o[k]).find((x) => typeof x === "string" && x.trim());
   if (typeof v !== "string") v = Object.values(o).find((x) => typeof x === "string" && (x as string).trim());
   if (typeof v !== "string") return "";
-  return v.replace(/\s+/g, " ").trim().slice(0, TOOL_SUMMARY_MAX);
+  return sanitizeText(v).replace(/\s+/g, " ").trim().slice(0, TOOL_SUMMARY_MAX);
 }
 
 function extractTools(content: unknown): PeekTool[] {
@@ -81,7 +96,7 @@ function extractTools(content: unknown): PeekTool[] {
     .map((c: any) => ({ name: c.name, summary: toolSummary(c.arguments) }));
 }
 
-function parseSession(path: string, raw: string, mtime: number): PeekSession | null {
+function parseSession(path: string, raw: string, mtime: number, size: number): PeekSession | null {
   let cwd = "";
   let time = "";
   let name = "";
@@ -98,7 +113,7 @@ function parseSession(path: string, raw: string, mtime: number): PeekSession | n
       }
       if (line.includes('"type":"session_info"')) {
         const o = JSON.parse(line);
-        if (typeof o.name === "string" && o.name) name = o.name;
+        if (typeof o.name === "string") name = sanitizeText(o.name).trim(); // 空名字是显式清掉
         continue;
       }
       // 先用子串粗筛，省掉大部分行的 JSON.parse
@@ -127,6 +142,7 @@ function parseSession(path: string, raw: string, mtime: number): PeekSession | n
     name,
     msgs,
     mtime,
+    size,
     first: firstText.text.replace(/\s+/g, " ").slice(0, 80),
   };
 }
@@ -134,29 +150,32 @@ function parseSession(path: string, raw: string, mtime: number): PeekSession | n
 // 单个文件：mtime 没变直接用缓存，否则重新读和解析
 async function loadSession(file: string): Promise<PeekSession | null> {
   let mtime: number;
+  let size: number;
   try {
-    mtime = (await stat(file)).mtimeMs;
+    const st = await stat(file);
+    mtime = st.mtimeMs;
+    size = st.size;
   } catch {
     return null;
   }
   const hit = sessionCache.get(file);
-  if (hit && hit.mtime === mtime) return hit;
+  if (hit && hit.mtime === mtime && hit.size === size) return hit;
   let raw: string;
   try {
     raw = await readFile(file, "utf8");
   } catch {
     return null;
   }
-  const parsed = parseSession(file, raw, mtime);
+  const parsed = parseSession(file, raw, mtime, size);
   if (parsed) sessionCache.set(file, parsed);
   return parsed;
 }
 
 /** 全部会话，按最后修改时间倒序；文件没变的直接用缓存。
  * 目录遍历和文件读取都走异步 I/O，几百个会话也不会把 TUI 卡住 */
-export async function scanSessions(): Promise<PeekSession[]> {
+export async function scanSessions(root: string = sessionsDir()): Promise<PeekSession[]> {
   const files: string[] = [];
-  await walkJsonl(sessionsDir(), files);
+  await walkJsonl(root, files);
 
   const out: PeekSession[] = [];
   let next = 0;
@@ -179,6 +198,7 @@ export async function scanSessions(): Promise<PeekSession[]> {
 
 // 和 pi 自带的 /name 写同一种 session_info 记录，parentId 用文件里最后一条记录的 id
 export function renameSession(path: string, name: string): void {
+  name = sanitizeText(name).replace(/\n+/g, " ").trim(); // 和 pi 的 appendSessionInfo 一样，换行不能进 JSONL 的一行
   const raw = readFileSync(path, "utf8");
   const lines = raw.split("\n");
   let parentId: string | null = null;
@@ -227,18 +247,21 @@ export function trashCommands(path: string, platform: NodeJS.Platform = process.
 }
 
 // 优先进回收站（能恢复），每个途径都不行再直接删。exec 由调用方传入，方便测试；
-// 命令退出码为 0 但文件还在也当失败，继续试下一个
+// 命令退出码为 0 但文件还在也当失败，继续试下一个。命令超时被杀但其实已经把文件移走了的，
+// 事后再看一眼文件，别把"已进回收站"报成"永久删除"
 export async function deleteSession(
   path: string,
   exec: (cmd: string, args: string[]) => Promise<{ code: number }>,
   platform: NodeJS.Platform = process.platform,
 ): Promise<DeleteResult> {
+  const wasThere = existsSync(path);
   for (const [cmd, args] of trashCommands(path, platform)) {
     try {
       const r = await exec(cmd, args);
       if (r.code === 0 && !existsSync(path)) return "trash";
     } catch {
     }
+    if (wasThere && !existsSync(path)) return "trash";
   }
   try {
     rmSync(path, { force: true });

package/src/theme.ts

@@ -0,0 +1,12 @@
+import type { Theme } from "@earendil-works/pi-coding-agent";
+
+// 组件和高亮只用到主题的这三个方法。从 pi 导出的 Theme 取子集：颜色名跟着 pi 的 ThemeColor / ThemeBg 联合走，
+// 写错名字 typecheck 就报；测试里的桩只要有这三个方法就能当主题用。类型导入编译后被擦除，运行时不依赖 pi-coding-agent
+export type PeekTheme = Pick<Theme, "fg" | "bg" | "bold">;
+
+// 把样式函数拆出前缀（fn(" ") 里空格前面那段转义码），pi 的 Markdown 组件也是这么做的
+export function stylePrefix(fn: (s: string) => string): string {
+  const s = fn(" ");
+  const i = s.indexOf(" ");
+  return i >= 0 ? s.slice(0, i) : "";
+}