pi-session-peek 0.5.1 → 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

@@ -3,6 +3,7 @@ import { readdir, readFile, stat } from "node:fs/promises";
 import { randomBytes } from "node:crypto";
 import { homedir } from "node:os";
 import { dirname, join } from "node:path";
+import { sanitizeText } from "./ansi.ts";
 import { normPath } from "./text.ts";
 
 // 读 ~/.pi/agent/sessions 下的会话 JSONL，按 mtime 缓存；重命名和删除也在这里。
@@ -64,11 +65,6 @@ async function walkJsonl(dir: string, out: string[]): Promise<void> {
   }
 }
 
-// 终端转义序列（CSI / OSC / DCS / APC 等）和除 \t \n 外的 C0 控制字符。正文里带这些会原样写到屏幕上：
-// 排版错乱、宽度算错，OSC 52 之类的还能改剪贴板 / 窗口标题
-const CONTROL_RE = /\x1b(?:\[[0-?]*[ -/]*[@-~]|[\]P_^X][^\x07\x1b]*(?:\x07|\x1b\\)?|[@-Z\\-_])|[\x00-\x08\x0b-\x1f\x7f]/g;
-export const sanitizeText = (s: string) => s.replace(/\r\n?/g, "\n").replace(CONTROL_RE, "");
-
 function extractText(content: unknown): string {
   if (typeof content === "string") return sanitizeText(content);
   if (!Array.isArray(content)) return "";

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) : "";
+}