pi-web-ui 0.86.2 → 0.87.1

package/dist/server/composer-drafts.js

@@ -0,0 +1,138 @@
+/**
+ * composer-drafts.ts — 未发送输入框草稿的单中心文件存储（全局 <dataDir>/composer-drafts.json）。
+ *
+ * 设计（issue #166 单中心文件方案）：不进 session JSONL、不按会话建 sidecar，
+ * 全会话的草稿集中在一个文件里，按 **sessionId**（uuidv7，重启稳定）键入。
+ * conversationId 每次重启都变，不能做 key（见 agent-service 的 compaction 注释）。
+ *
+ * - 空白新会话也能存：sessionId 在 SessionManager.create() 时内存里就有了，
+ *   不依赖转录文件落盘（SDK 的 _persist 门控要等首轮 assistant 才写盘）。
+ * - 每会话只留最新一条（last-write-wins，按 ts，比 marker-store 的扫描模式更轻）。
+ * - 与 per-client 的 client-state.json 不同：按 sessionId 全局存，跨标签页可见。
+ * - 文件 I/O 一律 best-effort：持久化故障绝不能弄崩 server（同 subagent-templates）。
+ */
+import { mkdirSync, readFileSync, renameSync, writeFileSync } from "node:fs";
+import { basename, dirname } from "node:path";
+/** 单条草稿上限字符数：超出截断再存（快照全量下发时携带，太大浪费流量）。 */
+export const DRAFT_TEXT_MAX = 20_000;
+/** 草稿保留期：超过这么久没更新，下次加载时清扫（兜底；正常路径靠 clear/prune）。 */
+export const DRAFT_TTL_MS = 30 * 24 * 3600 * 1000;
+/** 转录文件名 `<timestamp>_<sessionId>.jsonl` → sessionId。
+ *  时间戳里的 `:`/`.` 已被 SDK 换成 `-`（无下划线），取首个 `_` 之后即 sessionId
+ *  （uuidv7 本身也不含下划线）。对不上形状返回 undefined（调用方跳过剪枝）。 */
+export function sessionIdFromTranscriptPath(p) {
+    const base = basename(p);
+    if (!base.endsWith(".jsonl"))
+        return undefined;
+    const noExt = base.slice(0, -".jsonl".length);
+    const i = noExt.indexOf("_");
+    if (i < 0 || i + 1 >= noExt.length)
+        return undefined;
+    return noExt.slice(i + 1);
+}
+/** 归一化待存文本：超长截断；纯空白 → undefined（调用方按「删除」处理，不存空条目）。 */
+export function normalizeDraftText(text) {
+    const t = (text ?? "").slice(0, DRAFT_TEXT_MAX);
+    return t.trim() ? t : undefined;
+}
+function isValidDraft(d) {
+    if (!d || typeof d !== "object")
+        return false;
+    const o = d;
+    return typeof o.text === "string" && typeof o.ts === "number" && typeof o.updatedAt === "number";
+}
+export class ComposerDraftsStore {
+    filePath;
+    cache = null;
+    constructor(filePath) {
+        this.filePath = filePath;
+    }
+    load() {
+        if (this.cache)
+            return this.cache;
+        let raw = {};
+        try {
+            raw = JSON.parse(readFileSync(this.filePath, "utf8"));
+        }
+        catch {
+            raw = {};
+        }
+        const now = Date.now();
+        const kept = {};
+        let pruned = false;
+        for (const [k, v] of Object.entries(raw)) {
+            if (!isValidDraft(v)) {
+                pruned = true;
+                continue;
+            }
+            if (now - v.updatedAt > DRAFT_TTL_MS) {
+                pruned = true;
+                continue;
+            }
+            kept[k] = v;
+        }
+        this.cache = kept;
+        // 读到过期/脏条目才回写：正常加载不碰磁盘。
+        if (pruned)
+            this.persist();
+        return this.cache;
+    }
+    persist() {
+        if (!this.cache)
+            return;
+        try {
+            mkdirSync(dirname(this.filePath), { recursive: true });
+            const tmp = `${this.filePath}.tmp-${process.pid}`;
+            writeFileSync(tmp, JSON.stringify(this.cache, null, 2) + "\n");
+            renameSync(tmp, this.filePath);
+        }
+        catch {
+            // best-effort：草稿丢了可以重打，server 绝不能因此崩。
+        }
+    }
+    /** 读一条草稿（{text, ts}；没有返回 undefined）。 */
+    get(sessionId) {
+        if (!sessionId)
+            return undefined;
+        const d = this.load()[sessionId];
+        return d ? { text: d.text, ts: d.ts } : undefined;
+    }
+    /** 存一条草稿：空文本 = 删除；同 key 上 ts 更大的才覆盖（last-write-wins，
+     *  `>=` 让同 ts 的重发幂等）。只有实际变化才写盘。 */
+    save(sessionId, text, ts) {
+        if (!sessionId)
+            return;
+        const map = this.load();
+        const norm = normalizeDraftText(text);
+        if (norm === undefined) {
+            if (map[sessionId] !== undefined) {
+                delete map[sessionId];
+                this.persist();
+            }
+            return;
+        }
+        const prev = map[sessionId];
+        if (prev && prev.ts > ts)
+            return;
+        if (prev && prev.ts === ts && prev.text === norm)
+            return;
+        map[sessionId] = { text: norm, ts, updatedAt: Date.now() };
+        this.persist();
+    }
+    /** 发送成功 / 会话删除后清掉（有 key 才写盘）。 */
+    clear(sessionId) {
+        if (!sessionId)
+            return;
+        const map = this.load();
+        if (map[sessionId] !== undefined) {
+            delete map[sessionId];
+            this.persist();
+        }
+    }
+    /** delete_session 用：按转录文件路径反解 sessionId 并清掉（形状对不上就跳过）。 */
+    pruneSessionFile(sessionPath) {
+        const id = sessionIdFromTranscriptPath(sessionPath);
+        if (id)
+            this.clear(id);
+    }
+}