pi-web-ui 0.58.0 → 0.59.0

package/dist/server/dsh/dsh-serialize.js

@@ -0,0 +1,253 @@
+/**
+ * dsh-serialize.ts — DSH 会话事件/消息 → pi-web-ui 协议的 UiMessage[]。
+ *
+ * 事件面 ground truth（dsh 0.1.1-rc.2，实测 dump，见 docs/dsh-engine.md §2.1）：
+ *   session.event 通知 params = { sessionId, event: { type, seq, time, data } }
+ *     持久事件类型（全量 append-only，按 seq 排序）：
+ *       user/message        data: { content[], source, role, id }
+ *       assistant/message   data: { turn, step, message: { role, content[], id, time } }
+ *       tool/result         data: { turn, step, message: { role:"user",
+ *                             content:[{type:"tool-result", toolCallId, content[], isError}] },
+ *                             meta, sourceEventSeqs, surfaceOp }
+ *       assistant/chunk     data: { turn, step, chunk: {…} }
+ *         chunk: block-start {index, blockType: reasoning|text|tool-call}
+ *                reasoning-delta {index, text}      text-delta {index, text}
+ *                tool-call-delta {index, id, name, argumentsDelta}
+ *                block-end {index, block}           usage {usage}  finish {reason}
+ *       turn/start step/start step/end turn/end session/title …
+ *   session.status 通知    params = { sessionId, status: running|idle }
+ *
+ * 序列化策略（对齐 pi 引擎 serialize.ts 的 UiMessage 形状）：
+ *   - 持久消息：user/message、assistant/message、tool/result 三个事件各产一条；
+ *     id 稳定：u-<msgId> / a-<msgId> / t-<toolCallId>（前端 React key 稳定）。
+ *   - streamingMessage：由 assistant/chunk 增量累积（reasoning→thinking、
+ *     text→text、tool-call→toolCall），id 用 stream-<会话seq> 稳定跨快照。
+ */
+const TEXT_CAP = 200_000;
+const TOOL_OUTPUT_CAP = 100_000;
+const ARGS_CAP = 20_000;
+function truncate(s, cap) {
+    const str = typeof s === "string" ? s : String(s ?? "");
+    if (str.length <= cap)
+        return { text: str, truncated: false };
+    return { text: `${str.slice(0, cap)}\n\n… [truncated]`, truncated: true };
+}
+/** DSH user-role content blocks → UiContentBlock[]。 */
+export function serializeUserBlocks(blocks) {
+    if (!Array.isArray(blocks) || blocks.length === 0) {
+        return [{ type: "text", text: "" }];
+    }
+    return blocks.map((b) => {
+        if (b?.type === "image") {
+            const att = b.attachment;
+            return { type: "image", dataUrl: att?.url };
+        }
+        return { type: "text", text: String(b?.text ?? "") };
+    });
+}
+/** DSH assistant-role content blocks → UiContentBlock[]。 */
+export function serializeAssistantBlocks(blocks) {
+    if (!Array.isArray(blocks))
+        return [];
+    return blocks.map((b) => {
+        switch (b?.type) {
+            case "text": {
+                const { text, truncated } = truncate(b.text ?? "", TEXT_CAP);
+                return { type: "text", text, truncated };
+            }
+            case "reasoning": {
+                return { type: "thinking", thinking: b.text ?? "" };
+            }
+            case "tool-call": {
+                const args = typeof b.arguments === "string" ? b.arguments : "";
+                if (!args) {
+                    return { type: "toolCall", id: b.id ?? "", name: b.name ?? "tool" };
+                }
+                const { text, truncated } = truncate(args, ARGS_CAP);
+                return {
+                    type: "toolCall",
+                    id: b.id ?? "",
+                    name: b.name ?? "tool",
+                    argumentsText: text,
+                    argumentsTruncated: truncated,
+                };
+            }
+            default:
+                return { type: "unknown", ...b };
+        }
+    });
+}
+// ---------------------------------------------------------------------------
+// 事件 → UiMessage
+// ---------------------------------------------------------------------------
+/** tool/result 事件 data → UiMessage（role toolResult）。 */
+export function toolResultEventToUiMessage(data) {
+    const blocks = Array.isArray(data.message?.content) ? data.message.content : [];
+    const tr = blocks.find((b) => b?.type === "tool-result");
+    if (!tr)
+        return null;
+    const raw = (Array.isArray(tr.content) ? tr.content : [])
+        .map((c) => (c?.type === "text" ? String(c.text ?? "") : "[image result]"))
+        .filter((t) => t !== undefined && t !== null)
+        .join("\n");
+    const { text, truncated } = truncate(raw, TOOL_OUTPUT_CAP);
+    return {
+        id: `t-${tr.toolCallId}`,
+        role: "toolResult",
+        content: [{ type: "text", text, truncated }],
+        toolCallId: tr.toolCallId,
+        isError: tr.isError === true,
+        timestamp: data.message?.time,
+    };
+}
+/** user/message 事件 data → UiMessage。 */
+export function userMessageEventToUiMessage(data) {
+    return {
+        id: `u-${data.id ?? "?"}`,
+        role: "user",
+        content: serializeUserBlocks(data.content ?? []),
+        timestamp: data.time,
+    };
+}
+/** assistant/message 事件 data → UiMessage。 */
+export function assistantMessageEventToUiMessage(data) {
+    const m = data.message;
+    if (!m || !m.id)
+        return null;
+    return {
+        id: `a-${m.id}`,
+        role: "assistant",
+        content: serializeAssistantBlocks(m.content ?? []),
+        timestamp: m.time,
+        model: m.model,
+        provider: m.provider,
+        stopReason: m.finishReason,
+        errorMessage: m.errorMessage,
+    };
+}
+// ---------------------------------------------------------------------------
+// streaming 累积
+// ---------------------------------------------------------------------------
+/**
+ * assistant/chunk 增量累积器：reasoning/text/tool-call 三轨按 chunk.index
+ * 归位，产出与持久 assistant 消息同构的 content[]。
+ * id 必须跨快照稳定 —— 用会话级单调 seq 作时间戳锚（chunk 事件按 seq 到达，
+ * 首个 chunk 的 seq 恒定）。
+ */
+export class DshStreamAccumulator {
+    turn;
+    /** 稳定 id（stream-<首个 chunk seq>）。 */
+    id;
+    blocks = new Map();
+    started = false;
+    constructor(firstSeq, turn) {
+        this.turn = turn;
+        this.id = `stream-${firstSeq}-t${turn}`;
+    }
+    get isEmpty() {
+        return this.blocks.size === 0;
+    }
+    /** 处理一个 chunk，返回是否产生了内容变化。 */
+    apply(chunk) {
+        switch (chunk.type) {
+            case "block-start": {
+                const index = chunk.index;
+                const blockType = chunk.blockType;
+                if (blockType === "reasoning") {
+                    this.blocks.set(index, { type: "thinking", thinking: "" });
+                }
+                else if (blockType === "tool-call") {
+                    this.blocks.set(index, { type: "toolCall", id: "", name: "tool", argumentsText: "" });
+                }
+                else {
+                    this.blocks.set(index, { type: "text", text: "" });
+                }
+                return true;
+            }
+            case "reasoning-delta": {
+                const index = chunk.index;
+                const b = this.blocks.get(index);
+                if (b?.type === "thinking") {
+                    b.thinking += String(chunk.text ?? "");
+                    return true;
+                }
+                return false;
+            }
+            case "text-delta": {
+                const index = chunk.index;
+                const b = this.blocks.get(index);
+                if (b?.type === "text") {
+                    b.text += String(chunk.text ?? "");
+                    return true;
+                }
+                return false;
+            }
+            case "tool-call-delta": {
+                const index = chunk.index;
+                const b = this.blocks.get(index);
+                if (b?.type === "toolCall") {
+                    if (chunk.id !== undefined)
+                        b.id = String(chunk.id);
+                    if (chunk.name !== undefined)
+                        b.name = String(chunk.name);
+                    b.argumentsText = (b.argumentsText ?? "") + String(chunk.argumentsDelta ?? "");
+                    return true;
+                }
+                return false;
+            }
+            case "block-end": {
+                // 完整 block 到达：直接替换（内容与 delta 拼接一致，且带截断语义）。
+                const block = chunk.block;
+                if (block) {
+                    const index = chunk.index;
+                    if (block.type === "reasoning") {
+                        this.blocks.set(index, { type: "thinking", thinking: block.text ?? "" });
+                    }
+                    else if (block.type === "tool-call") {
+                        const { text, truncated } = truncate(typeof block.arguments === "string" ? block.arguments : "", ARGS_CAP);
+                        this.blocks.set(index, {
+                            type: "toolCall",
+                            id: block.id ?? "",
+                            name: block.name ?? "tool",
+                            argumentsText: text,
+                            argumentsTruncated: truncated,
+                        });
+                    }
+                    else if (block.type === "text") {
+                        const { text, truncated } = truncate(block.text ?? "", TEXT_CAP);
+                        this.blocks.set(index, { type: "text", text, truncated });
+                    }
+                    return true;
+                }
+                return false;
+            }
+            default:
+                // usage / finish 等不改变内容。
+                return false;
+        }
+    }
+    /** 按 index 升序输出 content[]（toolCall 空参数时省略 argumentsText）。 */
+    content() {
+        const out = [];
+        for (const index of [...this.blocks.keys()].sort((a, b) => a - b)) {
+            const b = this.blocks.get(index);
+            if (b.type === "toolCall" && (b.argumentsText ?? "") === "") {
+                out.push({ type: "toolCall", id: b.id, name: b.name });
+            }
+            else {
+                out.push(b);
+            }
+        }
+        return out;
+    }
+    toUiMessage(timestamp, model, provider) {
+        return {
+            id: this.id,
+            role: "assistant",
+            content: this.content(),
+            timestamp,
+            model,
+            provider,
+        };
+    }
+}

package/dist/server/dsh/dsh-sessions.js

@@ -0,0 +1,207 @@
+/**
+ * dsh-sessions.ts — DSH JSONL 会话存储的只读访问（会话列表/回放/fork 素材）。
+ *
+ * 布局（实测，dsh 0.1.1-rc.2 + dsh-session-persistence-jsonl）：
+ *   <DSH_SESSION_ROOT>/--<cwd projectKey>--/<sessionId>/session.jsonl.zstd
+ * 每个写批次追加一个 zstd 帧 → 必须多帧解压（zstdDecompressSync 只解首帧）。
+ *
+ * JSONL 行：type "session"（header）→ 逐行事件。text-chunks / reasoning-chunks
+ * / tool-call-chunks 是打包记录，解包成多个 assistant/chunk 事件。
+ */
+import { existsSync, readFileSync, readdirSync, statSync } from "node:fs";
+import { join } from "node:path";
+import { zstdDecompressSync } from "node:zlib";
+/** projectKey(cwd) — 镜像 DSH 运行时的会话目录命名（--<cwd>--）。分隔符 → "-"，非法字符 → ~XXXX（大写 hex）。 */
+export function projectKey(cwd) {
+    let readable = "";
+    let separatorRun = false;
+    for (let i = 0; i < cwd.length; i++) {
+        const code = cwd.charCodeAt(i);
+        const ch = String.fromCharCode(code);
+        if (ch === "/" || ch === "\\" || ch === ":") {
+            if (!separatorRun)
+                readable += "-";
+            separatorRun = true;
+        }
+        else if (ch !== "~" && /^[A-Za-z0-9._-]$/.test(ch)) {
+            readable += ch;
+            separatorRun = false;
+        }
+        else {
+            readable += "~" + code.toString(16).toUpperCase().padStart(4, "0");
+            separatorRun = false;
+        }
+    }
+    return `--${(readable.replace(/^-+/, "") || "root").slice(0, 251)}--`;
+}
+/** 解包一条存储记录：打包的 chunk 行 → 多个 assistant/chunk 事件；否则原样返回。 */
+function decodeStorageRecord(value) {
+    const t = value.type;
+    if (t === "text-chunks" ||
+        t === "reasoning-chunks" ||
+        t === "tool-call-chunks") {
+        const { seq0, time0, data } = value;
+        if (!data || (!Array.isArray(data.texts) && !Array.isArray(data.args))) {
+            return [value];
+        }
+        const texts = Array.isArray(data.texts) ? data.texts : [];
+        const args = Array.isArray(data.args) ? data.args : [];
+        const count = t === "tool-call-chunks" ? args.length : texts.length;
+        const out = [];
+        let time = time0 ?? 0;
+        for (let k = 0; k < count; k++) {
+            if (k > 0 && Array.isArray(data.dt) && k - 1 < data.dt.length) {
+                time += data.dt[k - 1];
+            }
+            const chunk = t === "text-chunks"
+                ? { type: "text-delta", index: data.index, text: texts[k] }
+                : t === "reasoning-chunks"
+                    ? { type: "reasoning-delta", index: data.index, text: texts[k] }
+                    : {
+                        type: "tool-call-delta",
+                        index: data.index,
+                        id: data.id,
+                        name: data.name,
+                        argumentsDelta: args[k],
+                    };
+            out.push({
+                type: "assistant/chunk",
+                seq: (seq0 ?? 0) + k,
+                time,
+                data: { turn: data.turn, step: data.step, chunk },
+            });
+        }
+        return out;
+    }
+    return [value];
+}
+/** 多帧 zstd 解压（运行时每个写批次追加一帧）。失败回退原文。 */
+function zstdDecompressAll(buf) {
+    const MAGIC = Buffer.from([0x28, 0xb5, 0x2f, 0xfd]);
+    const starts = [];
+    for (let i = 0; i + 4 <= buf.length; i++) {
+        if (buf[i] === MAGIC[0] &&
+            buf[i + 1] === MAGIC[1] &&
+            buf[i + 2] === MAGIC[2] &&
+            buf[i + 3] === MAGIC[3]) {
+            starts.push(i);
+        }
+    }
+    let out = "";
+    for (let i = 0; i < starts.length; i++) {
+        const from = starts[i];
+        const to = i + 1 < starts.length ? starts[i + 1] : buf.length;
+        try {
+            out += zstdDecompressSync(buf.subarray(from, to)).toString("utf8");
+        }
+        catch {
+            /* skip unreadable frame */
+        }
+    }
+    return out || buf.toString("utf8");
+}
+/** 读一个会话 JSONL 文件 → { header, events }。 */
+export function readSessionLog(file) {
+    const raw = readFileSync(file);
+    const text = file.endsWith(".jsonl.zstd")
+        ? zstdDecompressAll(raw)
+        : raw.toString("utf8");
+    const lines = text.split("\n").filter((l) => l.trim());
+    let header = null;
+    const events = [];
+    for (const line of lines) {
+        let value;
+        try {
+            value = JSON.parse(line);
+        }
+        catch {
+            continue;
+        }
+        if (!value || typeof value !== "object")
+            continue;
+        if (value.type === "session") {
+            header = value;
+            continue;
+        }
+        for (const ev of decodeStorageRecord(value)) {
+            events.push(ev);
+        }
+    }
+    return { header, events };
+}
+/** 递归找 root 下的 *.jsonl / *.jsonl.zstd 会话文件（按 mtime 倒序）。 */
+export function findSessionFiles(root) {
+    if (!existsSync(root))
+        return [];
+    const out = [];
+    const walk = (dir) => {
+        let entries;
+        try {
+            entries = readdirSync(dir, { withFileTypes: true });
+        }
+        catch {
+            return;
+        }
+        for (const e of entries) {
+            const full = join(dir, e.name);
+            if (e.isDirectory())
+                walk(full);
+            else if (e.name.endsWith(".jsonl") || e.name.endsWith(".jsonl.zstd")) {
+                out.push(full);
+            }
+        }
+    };
+    walk(root);
+    return out.sort((a, b) => statSync(b).mtimeMs - statSync(a).mtimeMs);
+}
+/** 一个工作区的会话文件：官方布局（root/--<cwd>--）+ 旧版 per-cwd 布局。 */
+export function findSessionFilesForCwd(sessionRoot, cwd) {
+    const dirs = [
+        join(sessionRoot, projectKey(cwd)),
+        join(sessionRoot, encodeURIComponent(cwd), projectKey(cwd)),
+    ];
+    const out = [];
+    for (const d of dirs)
+        out.push(...findSessionFiles(d));
+    return out.sort((a, b) => statSync(b).mtimeMs - statSync(a).mtimeMs);
+}
+/** 第一个用户文本（会话标题素材）。 */
+export function firstUserText(events) {
+    for (const ev of events) {
+        if (ev.type === "user/message") {
+            const blocks = ev.data?.content;
+            if (Array.isArray(blocks)) {
+                for (const b of blocks) {
+                    const bb = b;
+                    if (bb?.type === "text" && typeof bb.text === "string" && bb.text.trim()) {
+                        const t = bb.text.trim().replace(/\s+/g, " ");
+                        return t.length > 30 ? `${t.slice(0, 30)}…` : t;
+                    }
+                }
+            }
+        }
+    }
+    return "新对话";
+}
+import { assistantMessageEventToUiMessage, toolResultEventToUiMessage, userMessageEventToUiMessage, } from "./dsh-serialize.js";
+export function replayEventsToMessages(events) {
+    const messages = [];
+    const seen = new Set();
+    for (const ev of events) {
+        let msg = null;
+        if (ev.type === "user/message") {
+            msg = userMessageEventToUiMessage(ev.data);
+        }
+        else if (ev.type === "assistant/message") {
+            msg = assistantMessageEventToUiMessage(ev.data);
+        }
+        else if (ev.type === "tool/result") {
+            msg = toolResultEventToUiMessage(ev.data);
+        }
+        if (msg && !seen.has(msg.id)) {
+            seen.add(msg.id);
+            messages.push(msg);
+        }
+    }
+    return messages;
+}

package/dist/server/dsh/runtime/cordis.yml

@@ -0,0 +1 @@
+[]