@xyagent/cli 0.0.1

package/src-ext/runtime/_shared/sessionPicker.mjs

@@ -0,0 +1,418 @@
+/**
+ * sessionPicker.mjs — 多 runtime 历史会话枚举
+ *
+ * 导出 listSessionsForRuntime(runtimeKind) → Promise<SessionEntry[]>
+ *
+ * SessionEntry:
+ *   {
+ *     session_id: string,   // 唯一标识，用于 --resume 参数
+ *     title: string,        // 展示名
+ *     updated_at: string,   // ISO 8601
+ *     cwd?: string,         // 会话关联工作目录（hermes 无此字段）
+ *     runtime: string,      // "claude" | "codex" | "hermes"
+ *   }
+ *
+ * options 项 label 格式：`<title> · <cwd>（<relative updated_at>）`
+ * 由 sessionPicker 统一生成，不在各 handleRequest 各自渲染。
+ */
+
+import { promises as fsp } from "node:fs";
+import path from "node:path";
+import os from "node:os";
+
+// 最多返回 50 条（按 updated_at 降序取最近的）
+export const MAX_SESSIONS = 50;
+
+// ---------------------------------------------------------------------------
+// 公共工具
+// ---------------------------------------------------------------------------
+
+/** 相对时间描述（简短） */
+function relativeTime(isoString) {
+  try {
+    const diff = Date.now() - new Date(isoString).getTime();
+    const sec = Math.floor(diff / 1000);
+    if (sec < 60) return "刚刚";
+    const min = Math.floor(sec / 60);
+    if (min < 60) return `${min} 分钟前`;
+    const hr = Math.floor(min / 60);
+    if (hr < 24) return `${hr} 小时前`;
+    const day = Math.floor(hr / 24);
+    if (day < 30) return `${day} 天前`;
+    const mon = Math.floor(day / 30);
+    if (mon < 12) return `${mon} 个月前`;
+    return `${Math.floor(mon / 12)} 年前`;
+  } catch {
+    return "";
+  }
+}
+
+/**
+ * 选项主标题 = 会话名字（不含路径、不含时间）。时间走 buildSessionSubtitle。
+ * @param {SessionEntry} entry
+ * @returns {string}
+ */
+export function buildSessionLabel(entry) {
+  return (entry.title || "").trim() || "未命名会话";
+}
+
+/**
+ * 选项副标题 = 相对时间（"3 分钟前" 等），App 端在标题下方另起一行展示。
+ * @param {SessionEntry} entry
+ * @returns {string}
+ */
+export function buildSessionSubtitle(entry) {
+  const rel = relativeTime(entry.updated_at) || "";
+  const branch = (entry.gitBranch || "").trim();
+  if (rel && branch) return `${rel} · ${branch}`;
+  return rel || branch;
+}
+
+/** 归一化工作目录：解析为绝对路径并去掉尾部分隔符，便于精确比较。 */
+function normalizeCwd(p) {
+  let r = path.resolve(p);
+  if (r.length > 1 && r.endsWith(path.sep)) r = r.slice(0, -1);
+  return r;
+}
+
+/**
+ * 按当前工作区过滤会话。规则：
+ *   - filterCwd 为空 / 未给 → 不过滤，原样返回（兼容旧的「全量」行为）。
+ *   - 否则仅丢弃「有 cwd 且与 filterCwd 不匹配」的项。
+ *   - 无 cwd 的项（如 hermes，session 文件不含工作目录）一律保留——
+ *     它们没有工作区概念，按工作区过滤对其无意义。
+ * 暴露为 _testFilterByCwd 供单元测试注入假数据。
+ * @param {SessionEntry[]} entries
+ * @param {string|null|undefined} filterCwd  当前工作区绝对路径
+ * @returns {SessionEntry[]}
+ */
+export function _testFilterByCwd(entries, filterCwd) {
+  if (!filterCwd || typeof filterCwd !== "string" || !filterCwd.trim()) {
+    return entries;
+  }
+  const target = normalizeCwd(filterCwd);
+  return entries.filter((e) => !e.cwd || normalizeCwd(e.cwd) === target);
+}
+
+/**
+ * 排序 + 截断。暴露为 _testListWithEntries 供单元测试注入假数据。
+ * @param {SessionEntry[]} entries
+ * @returns {SessionEntry[]}
+ */
+export function _testListWithEntries(entries) {
+  return entries
+    .slice()
+    .sort((a, b) => {
+      const ta = new Date(a.updated_at).getTime();
+      const tb = new Date(b.updated_at).getTime();
+      return tb - ta; // 降序
+    })
+    .slice(0, MAX_SESSIONS);
+}
+
+// ---------------------------------------------------------------------------
+// claude mapper
+// ---------------------------------------------------------------------------
+
+/**
+ * 还原 claude 的项目目录编码：把 cwd 里所有非字母数字字符替换成 '-'。
+ * 例 /Users/cece/company/agentlink → -Users-cece-company-agentlink
+ *    /Users/cece/.hermes          → -Users-cece--hermes（'.' 也变 '-'，故双横线）
+ * 与 claude CLI 内部编码一致，用于「只扫当前工作区」时直接定位目录。
+ */
+function encodeClaudeProjectDir(cwd) {
+  return path.resolve(cwd).replace(/[^a-zA-Z0-9]/g, "-");
+}
+
+/**
+ * 扫描 ~/.claude/projects/<proj>/*.jsonl
+ * session_id = 文件名（不含 .jsonl）
+ * updated_at = mtime
+ * cwd = 首行 JSON 里的 cwd 字段 or 目录名（人类可读）
+ * title = 首行 JSON 里 summary/title or 目录名的末节
+ *
+ * @param {string|null} [filterCwd] 给定时只扫该工作区对应的单个项目目录
+ *   （claude 按 cwd 编码成目录名存放 session），不再 readdir 整个 projects/。
+ */
+async function listClaudeSessions(filterCwd) {
+  const projectsDir = path.join(os.homedir(), ".claude", "projects");
+  const entries = [];
+
+  // 只扫当前工作区：直接由 cwd 算出项目目录名，省去遍历全部项目 + 逐文件解析 cwd。
+  let dirNames;
+  if (filterCwd && typeof filterCwd === "string" && filterCwd.trim()) {
+    dirNames = [encodeClaudeProjectDir(filterCwd)];
+  } else {
+    try {
+      const projDirs = await fsp.readdir(projectsDir, { withFileTypes: true });
+      dirNames = projDirs.filter((d) => d.isDirectory()).map((d) => d.name);
+    } catch {
+      return [];
+    }
+  }
+
+  for (const dirName of dirNames) {
+    const projPath = path.join(projectsDir, dirName);
+    let files;
+    try {
+      files = await fsp.readdir(projPath, { withFileTypes: true });
+    } catch {
+      continue; // 目录不存在（如该工作区尚无历史会话）→ 跳过
+    }
+    for (const f of files) {
+      if (!f.isFile() || !f.name.endsWith(".jsonl")) continue;
+      const fullPath = path.join(projPath, f.name);
+      const sessionId = f.name.slice(0, -".jsonl".length);
+      let mtime;
+      let cwd = dirName; // fallback: decoded dir name
+      let title = dirName;
+      let gitBranch = "";
+
+      try {
+        const stat = await fsp.stat(fullPath);
+        mtime = stat.mtime.toISOString();
+      } catch {
+        mtime = new Date(0).toISOString();
+      }
+
+      // 扫描前若干行 JSON 取真实 cwd / summary。
+      // 注意：claude jsonl 首行常是 summary 行（无 cwd），cwd 在后续 message 行里，
+      // 只读首行会取不到真实 cwd → 回退成编码目录名（如 "-Users-cece-xxx"），
+      // 导致 `claude --resume` 用错 cwd 失败。故读较大 buffer 并逐行找 cwd。
+      try {
+        const fd = await fsp.open(fullPath, "r");
+        const buf = Buffer.alloc(65536);
+        const { bytesRead } = await fd.read(buf, 0, 65536, 0);
+        await fd.close();
+        const lines = buf.slice(0, bytesRead).toString("utf8").split("\n");
+        let foundCwd = false;
+        let foundTitle = false;
+        let firstUserMsg = null;
+        for (const line of lines) {
+          const trimmed = line.trim();
+          if (!trimmed) continue;
+          let obj;
+          try {
+            obj = JSON.parse(trimmed);
+          } catch {
+            continue; // 半行/非 JSON，跳过
+          }
+          if (!foundCwd && typeof obj.cwd === "string" && obj.cwd) {
+            cwd = obj.cwd;
+            foundCwd = true;
+          }
+          // git 分支（与 cwd 同在 message 行）：副标题里展示在时间后。
+          if (!gitBranch && typeof obj.gitBranch === "string" && obj.gitBranch) {
+            gitBranch = obj.gitBranch;
+          }
+          if (!foundTitle) {
+            if (typeof obj.summary === "string" && obj.summary) {
+              title = obj.summary;
+              foundTitle = true;
+            } else if (typeof obj.title === "string" && obj.title) {
+              title = obj.title;
+              foundTitle = true;
+            }
+          }
+          // 捕获首条 user 消息文本作为会话名字兜底（claude 通常无 summary）。
+          if (!firstUserMsg && obj.type === "user") {
+            const c = obj.message && obj.message.content;
+            if (typeof c === "string" && c.trim()) {
+              firstUserMsg = c.trim();
+            } else if (Array.isArray(c)) {
+              const ti = c.find(
+                (it) => it && it.type === "text" && typeof it.text === "string",
+              );
+              if (ti && ti.text.trim()) firstUserMsg = ti.text.trim();
+            }
+          }
+          if (foundCwd && (foundTitle || firstUserMsg)) break;
+        }
+        // title 优先级：summary/title > 首条 user 消息(截断) > 项目目录名
+        if (!foundTitle) {
+          if (firstUserMsg) {
+            title = firstUserMsg.length > 40
+              ? `${firstUserMsg.slice(0, 40)}…`
+              : firstUserMsg;
+          } else if (foundCwd) {
+            title = path.basename(cwd);
+          }
+        }
+      } catch {
+        // 读取失败，用 dirEntry.name 作 title/cwd fallback
+      }
+
+      entries.push({
+        session_id: sessionId,
+        title,
+        updated_at: mtime,
+        cwd,
+        gitBranch,
+        runtime: "claude",
+      });
+    }
+  }
+  return entries;
+}
+
+// ---------------------------------------------------------------------------
+// hermes mapper
+// ---------------------------------------------------------------------------
+
+/**
+ * 调用 hermes/sessionStore.mjs::listSessions()，派生 title/cwd。
+ * hermes session 文件只含 {messages: [...]}，无 title/cwd 字段。
+ *   - title: 首条 user message 的前 40 字
+ *   - updated_at: 文件 mtime
+ *   - cwd: 不可得，省略
+ */
+async function listHermesSessions() {
+  const { listSessions } = await import("../hermes/sessionStore.mjs");
+  let ids;
+  try {
+    ids = await listSessions();
+  } catch {
+    return [];
+  }
+
+  const hermesRoot = path.join(os.homedir(), ".agentlink", "hermes", "sessions");
+  const entries = [];
+
+  for (const sessionId of ids) {
+    const filePath = path.join(hermesRoot, `${sessionId}.json`);
+    let mtime = new Date(0).toISOString();
+    let title = sessionId;
+
+    try {
+      const stat = await fsp.stat(filePath);
+      mtime = stat.mtime.toISOString();
+    } catch { /* skip */ }
+
+    try {
+      const buf = await fsp.readFile(filePath, "utf8");
+      const data = JSON.parse(buf);
+      if (Array.isArray(data?.messages)) {
+        // 找首条 user message
+        const firstUser = data.messages.find((m) => m.role === "user");
+        if (firstUser) {
+          const content = typeof firstUser.content === "string"
+            ? firstUser.content
+            : (Array.isArray(firstUser.content)
+              ? firstUser.content.map((c) => (typeof c === "string" ? c : c?.text || "")).join("")
+              : "");
+          title = content.slice(0, 40).trim() || sessionId;
+        }
+      }
+    } catch { /* skip */ }
+
+    entries.push({
+      session_id: sessionId,
+      title,
+      updated_at: mtime,
+      runtime: "hermes",
+    });
+  }
+  return entries;
+}
+
+// ---------------------------------------------------------------------------
+// codex mapper
+// ---------------------------------------------------------------------------
+
+/**
+ * 扫描 ~/.codex/sessions/*.jsonl（递归，因 codex 按日期分级目录）。
+ * session_id = 文件名（不含 .jsonl），cwd 取首行 JSON 的 cwd 字段。
+ */
+async function listCodexSessions() {
+  const sessionsDir = path.join(os.homedir(), ".codex", "sessions");
+  const entries = [];
+
+  async function walk(dir) {
+    let items;
+    try {
+      items = await fsp.readdir(dir, { withFileTypes: true });
+    } catch {
+      return;
+    }
+    for (const item of items) {
+      const fullPath = path.join(dir, item.name);
+      if (item.isDirectory()) {
+        await walk(fullPath);
+      } else if (item.isFile() && item.name.endsWith(".jsonl")) {
+        const sessionId = item.name.slice(0, -".jsonl".length);
+        let mtime = new Date(0).toISOString();
+        let cwd = "";
+        let title = sessionId;
+
+        try {
+          const stat = await fsp.stat(fullPath);
+          mtime = stat.mtime.toISOString();
+        } catch { /* skip */ }
+
+        // 取首行 JSON
+        try {
+          const fd = await fsp.open(fullPath, "r");
+          const buf = Buffer.alloc(4096);
+          const { bytesRead } = await fd.read(buf, 0, 4096, 0);
+          await fd.close();
+          const firstLine = buf.slice(0, bytesRead).toString("utf8").split("\n")[0].trim();
+          if (firstLine) {
+            const obj = JSON.parse(firstLine);
+            if (typeof obj.cwd === "string" && obj.cwd) cwd = obj.cwd;
+            // codex sessions 的 title 可能在 payload 里
+            const payload = obj.payload || obj;
+            if (typeof payload.title === "string" && payload.title) title = payload.title;
+            else if (cwd) title = path.basename(cwd);
+          }
+        } catch { /* skip */ }
+
+        entries.push({
+          session_id: sessionId,
+          title,
+          updated_at: mtime,
+          cwd,
+          runtime: "codex",
+        });
+      }
+    }
+  }
+
+  await walk(sessionsDir);
+  return entries;
+}
+
+// ---------------------------------------------------------------------------
+// Public API
+// ---------------------------------------------------------------------------
+
+/**
+ * @param {"claude"|"codex"|"hermes"} runtimeKind
+ * @param {{ filterCwd?: string|null }} [opts]
+ *   filterCwd: 当前工作区绝对路径。给定时只返回该工作区下的会话（claude/codex
+ *   按 session 的 cwd 匹配；hermes 无 cwd，过滤对其无效，原样保留）。
+ *   不给则返回全量（旧行为）。
+ * @returns {Promise<SessionEntry[]>}  filtered by cwd, sorted by updated_at desc, capped at MAX_SESSIONS
+ */
+export async function listSessionsForRuntime(runtimeKind, { filterCwd = null } = {}) {
+  let rawEntries;
+  try {
+    if (runtimeKind === "claude") {
+      // claude 按 cwd 编码目录存放 session → 直接只扫当前工作区那一个目录。
+      rawEntries = await listClaudeSessions(filterCwd);
+    } else if (runtimeKind === "hermes") {
+      rawEntries = await listHermesSessions();
+    } else if (runtimeKind === "codex") {
+      // codex 的 session 按日期目录存放、cwd 只在文件内容里，无法靠目录定位，
+      // 仍需全量扫描后由下方 _testFilterByCwd 按 cwd 过滤。
+      rawEntries = await listCodexSessions();
+    } else {
+      return [];
+    }
+  } catch (err) {
+    // any unexpected error → warn + empty array
+    // (per-runtime internal try/catch should handle fs errors, but belt+suspenders)
+    console.warn(`[sessionPicker] ${runtimeKind} enumeration failed:`, err?.message || err);
+    return [];
+  }
+  return _testListWithEntries(_testFilterByCwd(rawEntries, filterCwd));
+}