53 lines
1.6 KiB
TypeScript
53 lines
1.6 KiB
TypeScript
import type { TranscriptEntry, StdoutLineParser } from "./types";
|
|
|
|
type RunLogChunk = { ts: string; stream: "stdout" | "stderr" | "system"; chunk: string };
|
|
|
|
function appendTranscriptEntry(entries: TranscriptEntry[], entry: TranscriptEntry) {
|
|
if (entry.kind === "thinking" && entry.delta) {
|
|
const last = entries[entries.length - 1];
|
|
if (last && last.kind === "thinking" && last.delta) {
|
|
last.text += entry.text;
|
|
last.ts = entry.ts;
|
|
return;
|
|
}
|
|
}
|
|
entries.push(entry);
|
|
}
|
|
|
|
export function buildTranscript(chunks: RunLogChunk[], parser: StdoutLineParser): TranscriptEntry[] {
|
|
const entries: TranscriptEntry[] = [];
|
|
let stdoutBuffer = "";
|
|
|
|
for (const chunk of chunks) {
|
|
if (chunk.stream === "stderr") {
|
|
entries.push({ kind: "stderr", ts: chunk.ts, text: chunk.chunk });
|
|
continue;
|
|
}
|
|
if (chunk.stream === "system") {
|
|
entries.push({ kind: "system", ts: chunk.ts, text: chunk.chunk });
|
|
continue;
|
|
}
|
|
|
|
const combined = stdoutBuffer + chunk.chunk;
|
|
const lines = combined.split(/\r?\n/);
|
|
stdoutBuffer = lines.pop() ?? "";
|
|
for (const line of lines) {
|
|
const trimmed = line.trim();
|
|
if (!trimmed) continue;
|
|
for (const entry of parser(trimmed, chunk.ts)) {
|
|
appendTranscriptEntry(entries, entry);
|
|
}
|
|
}
|
|
}
|
|
|
|
const trailing = stdoutBuffer.trim();
|
|
if (trailing) {
|
|
const ts = chunks.length > 0 ? chunks[chunks.length - 1]!.ts : new Date().toISOString();
|
|
for (const entry of parser(trailing, ts)) {
|
|
appendTranscriptEntry(entries, entry);
|
|
}
|
|
}
|
|
|
|
return entries;
|
|
}
|