pi-crew 0.9.51 → 0.9.53

package/src/ui/transcript-viewer.ts

@@ -1,424 +0,0 @@
-import * as fs from "node:fs";
-import { agentOutputPath, readCrewAgents } from "../runtime/crew-agent-records.ts";
-import type { TeamRunManifest } from "../state/types.ts";
-import { resolveRealContainedPath } from "../utils/safe-paths.ts";
-import { pad, truncate, truncateToVisualLines } from "../utils/visual.ts";
-import { renderDiff } from "./render-diff.ts";
-import { colorForStatus, iconForStatus, type RunStatus } from "./status-colors.ts";
-import { highlightCode, highlightJson } from "./syntax-highlight.ts";
-import type { CrewTheme } from "./theme-adapter.ts";
-import { asCrewTheme, subscribeThemeChange } from "./theme-adapter.ts";
-import { DEFAULT_TRANSCRIPT_TAIL_BYTES, getTranscriptCacheEntry, readTranscriptLinesCached } from "./transcript-cache.ts";
-
-type Component = {
-	invalidate(): void;
-	render(width: number): string[];
-	handleInput(data: string): void;
-};
-
-type TranscriptTheme = CrewTheme;
-
-interface ReadRunTranscriptResult {
-	title: string;
-	path: string;
-	lines: string[];
-	bytesRead: number;
-	size: number;
-	truncated: boolean;
-}
-
-// FIND-06: short-TTL cache keyed by (runId, taskId, readOptions) so the ~500ms
-// render tick of `DurableTranscriptViewer` does not re-stat / re-read / re-parse
-// the underlying transcript on every call. The deeper `transcriptCache` only
-// reuses parsed lines when mtime/size are stable; during an actively-written
-// run, the file's mtime changes on every tick and the inner cache is
-// invalidated. This outer cache caps the per-tick re-parse cost at one parse
-// per ~500ms even while the file is growing.
-const READ_RUN_TRANSCRIPT_TTL_MS = 500;
-const READ_RUN_TRANSCRIPT_CACHE_MAX = 32;
-interface ReadRunTranscriptCacheEntry {
-	parsedAt: number;
-	result: ReadRunTranscriptResult;
-}
-const readRunTranscriptCache = new Map<string, ReadRunTranscriptCacheEntry>();
-
-function readRunTranscriptCacheKey(
-	manifest: TeamRunManifest,
-	taskId: string | undefined,
-	readOptions: { full: boolean; maxTailBytes: number },
-): string {
-	return `${manifest.runId}|${taskId ?? ""}|${readOptions.full ? "full" : "tail"}|${readOptions.maxTailBytes}`;
-}
-
-function asRecord(value: unknown): Record<string, unknown> | undefined {
-	return value && typeof value === "object" && !Array.isArray(value) ? (value as Record<string, unknown>) : undefined;
-}
-
-function textFromContent(content: unknown): string {
-	if (typeof content === "string") return content;
-	if (!Array.isArray(content)) return "";
-	return content
-		.map((part) => {
-			const obj = asRecord(part);
-			if (!obj) return "";
-			if (typeof obj.text === "string") return obj.text;
-			if (typeof obj.content === "string") return obj.content;
-			if (typeof obj.name === "string") return `[tool:${obj.name}]`;
-			return "";
-		})
-		.filter(Boolean)
-		.join("\n");
-}
-
-function isLikelyDiff(text: string): boolean {
-	const lines = text.split(/\r?\n/);
-	const matched = lines.filter((line) => /^[-+\s]\d+\s/.test(line)).length;
-	return matched >= 2 && (text.includes("-") || text.includes("+"));
-}
-
-function highlightCodeBlocks(input: string, theme: TranscriptTheme): string[] {
-	const codeBlockRegex = /```(\S+)?\n([\s\S]*?)```/g;
-	const lines: string[] = [];
-	let index = 0;
-	let match: RegExpExecArray | null;
-	while ((match = codeBlockRegex.exec(input)) !== null) {
-		if (match.index > index) lines.push(...input.slice(index, match.index).split(/\r?\n/));
-		const lang = match[1]?.trim();
-		const block = match[2] ?? "";
-		const highlighted = highlightCode(block, lang, theme);
-		if (highlighted) {
-			lines.push(...highlighted.split(/\r?\n/));
-		}
-		index = match.index + match[0].length;
-	}
-	if (index < input.length) lines.push(...input.slice(index).split(/\r?\n/));
-	return lines.filter((line) => line.length > 0);
-}
-
-export function formatTranscriptEvent(event: unknown, themeLike: unknown = undefined): string[] {
-	const theme = asCrewTheme(themeLike);
-	const obj = asRecord(event);
-	if (!obj) return [String(event)];
-	const type = typeof obj.type === "string" ? obj.type : undefined;
-	const toolName = typeof obj.toolName === "string" ? obj.toolName : typeof obj.name === "string" ? obj.name : undefined;
-	const content = textFromContent(obj.content);
-	if (type && /tool/i.test(type)) {
-		const result = asRecord(obj.result);
-		const isError = obj.isError === true || result?.isError === true;
-		const isPartial = obj.isPartial === true;
-		const status: RunStatus = isError ? "failed" : isPartial ? "running" : "completed";
-		const header = theme.fg(
-			colorForStatus(status),
-			`${iconForStatus(status, { runningGlyph: "⋯" })} [Tool${toolName ? `: ${toolName}` : ""}] ${type}`,
-		);
-		const text = (content || (typeof obj.text === "string" ? obj.text : typeof obj.result === "string" ? obj.result : "")).trim();
-		if (!text) return [header, "(no output)"];
-		if (isLikelyDiff(text)) {
-			return [header, renderDiff(text, { theme })];
-		}
-		if (text.startsWith("{") && text.endsWith("}")) {
-			return [header, ...highlightJson(text, theme).split(/\r?\n/).filter(Boolean)];
-		}
-		if (text.includes("```") && text.includes("```")) {
-			return [header, ...highlightCodeBlocks(text, theme)];
-		}
-		return [
-			header,
-			...text
-				.split(/\r?\n/)
-				.filter(Boolean)
-				.map((line) => theme.fg("muted", line)),
-		];
-	}
-	const message = asRecord(obj.message);
-	if (message) {
-		const role = typeof message.role === "string" ? message.role : "message";
-		const text = textFromContent(message.content);
-		if (text.trim()) {
-			const label = role === "assistant" ? "Assistant" : role === "user" ? "User" : role;
-			const header = `[${label}]:`;
-			const lines = text.split(/\r?\n/);
-			if (text.includes("```") && text.includes("```")) {
-				return [theme.fg("accent", header), ...highlightCodeBlocks(text, theme)];
-			}
-			if (lines.length > 1) {
-				const block = lines.map((line) => (role === "assistant" ? theme.bold(line) : line)).join("\n");
-				return [theme.fg("accent", header), ...block.split(/\r?\n/).filter(Boolean)];
-			}
-			return [theme.fg("accent", header), ...lines.filter(Boolean)];
-		}
-	}
-	if (type) {
-		const text = content || (typeof obj.text === "string" ? obj.text : "");
-		return text.trim() ? [theme.fg("muted", `[${type}]: ${text.trim()}`)] : [`[${type}]`];
-	}
-	return [JSON.stringify(event)];
-}
-
-export function formatTranscriptText(text: string, themeLike: unknown = undefined): string[] {
-	const lines: string[] = [];
-	for (const raw of text.split(/\r?\n/).filter(Boolean)) {
-		try {
-			const parsed = JSON.parse(raw);
-			lines.push(...formatTranscriptEvent(parsed, themeLike));
-		} catch {
-			lines.push(raw);
-		}
-	}
-	return lines.length ? lines : ["(no transcript content)"];
-}
-
-export function readRunTranscript(
-	manifest: TeamRunManifest,
-	taskId?: string,
-	options: { full?: boolean; maxTailBytes?: number } = {},
-): ReadRunTranscriptResult {
-	const now = Date.now();
-	const readOptions = {
-		full: options.full === true,
-		maxTailBytes: options.maxTailBytes ?? DEFAULT_TRANSCRIPT_TAIL_BYTES,
-	};
-	const cacheKey = readRunTranscriptCacheKey(manifest, taskId, readOptions);
-	const cached = readRunTranscriptCache.get(cacheKey);
-	if (cached && now - cached.parsedAt < READ_RUN_TRANSCRIPT_TTL_MS) {
-		return cached.result;
-	}
-	const agents = readCrewAgents(manifest);
-	const agent = taskId
-		? agents.find((item) => item.taskId === taskId || item.id === taskId)
-		: (agents.find((item) => item.transcriptPath) ?? agents[0]);
-	const selectedTaskId = agent?.taskId ?? taskId ?? "unknown";
-	let transcriptPath = "";
-	try {
-		transcriptPath = agentOutputPath(manifest, selectedTaskId);
-	} catch {
-		try {
-			transcriptPath = agentOutputPath(manifest, "unknown");
-		} catch {
-			// Both fallbacks failed — transcript will be empty.
-			transcriptPath = "";
-		}
-	}
-	if (agent?.transcriptPath) {
-		try {
-			const safeTranscriptPath = resolveRealContainedPath(manifest.artifactsRoot, agent.transcriptPath);
-			if (fs.existsSync(safeTranscriptPath)) transcriptPath = safeTranscriptPath;
-		} catch {
-			// Ignore untrusted transcript paths from mutable agent state and fall back to durable agent output.
-		}
-	}
-	const lines = readTranscriptLinesCached(transcriptPath, (text) => formatTranscriptText(text), now, readOptions);
-	const entry = getTranscriptCacheEntry(transcriptPath, readOptions);
-	const result: ReadRunTranscriptResult = {
-		title: `${manifest.runId}:${selectedTaskId}`,
-		path: transcriptPath,
-		lines: lines.length ? lines : ["(no transcript content)"],
-		bytesRead: entry?.bytesRead ?? 0,
-		size: entry?.size ?? 0,
-		truncated: entry?.truncated ?? false,
-	};
-	// Map preserves insertion order; the first key is the oldest entry. Evict
-	// before insert so the cache stays at most READ_RUN_TRANSCRIPT_CACHE_MAX.
-	while (readRunTranscriptCache.size >= READ_RUN_TRANSCRIPT_CACHE_MAX) {
-		const oldest = readRunTranscriptCache.keys().next().value;
-		if (oldest === undefined) break;
-		readRunTranscriptCache.delete(oldest);
-	}
-	readRunTranscriptCache.set(cacheKey, { parsedAt: now, result });
-	return result;
-}
-
-interface ViewerState {
-	theme: TranscriptTheme;
-	autoScroll: boolean;
-	lastHeight: number;
-	scroll: number;
-}
-
-function renderViewerBase(state: ViewerState, width: number, lines: string[], title: string, subtitle: string): string[] {
-	const inner = Math.max(20, width - 4);
-	const bodyText = lines.join("\n");
-	const { visualLines, skippedCount } = truncateToVisualLines(bodyText, state.lastHeight, inner);
-	const maxScroll = Math.max(0, visualLines.length - state.lastHeight);
-	if (state.autoScroll) state.scroll = maxScroll;
-	state.scroll = Math.min(state.scroll, maxScroll);
-	const visible = visualLines.slice(state.scroll, state.scroll + state.lastHeight);
-	const statusLine = `${visualLines.length} lines · ${visualLines.length ? Math.round(((state.scroll + visible.length) / visualLines.length) * 100) : 100}% · auto-scroll ${state.autoScroll ? "on" : "off"}`;
-	const fg = (color: Parameters<TranscriptTheme["fg"]>[0], text: string) => state.theme.fg(color, text);
-	const row = (text: string) => `${fg("border", "│")} ${pad(truncate(text, inner), inner)} ${fg("border", "│")}`;
-	const linesOut: string[] = [
-		fg("border", `╭${"─".repeat(inner + 2)}╮`),
-		row(`${fg("accent", title)} ${fg("dim", subtitle)}`),
-		row(fg("dim", "j/k scroll · PgUp/PgDn · g/G top/bottom · a auto · f full/tail · q close")),
-		fg("border", `├${"─".repeat(inner + 2)}┤`),
-		...visible.map(row),
-		fg("border", `├${"─".repeat(inner + 2)}┤`),
-		row(fg("dim", statusLine)),
-		fg("border", `╰${"─".repeat(inner + 2)}╯`),
-	];
-	if (skippedCount > 0) {
-		linesOut.splice(linesOut.length - 1, 0, row(fg("muted", `… (${skippedCount} lines truncated above`)));
-	}
-	return linesOut.map((line) => truncate(line, width));
-}
-
-export class DurableTextViewer implements Component {
-	private scroll = 0;
-	private lastHeight = 16;
-	private autoScroll = true;
-	private title: string;
-	private subtitle: string;
-	private lines: string[];
-	private theme: TranscriptTheme;
-	private done: (result: undefined) => void;
-	private readonly unsubscribeTheme: () => void;
-
-	constructor(title: string, subtitle: string, lines: string[], theme: unknown, done: (result: undefined) => void) {
-		this.title = title;
-		this.subtitle = subtitle;
-		this.lines = lines.length ? lines : ["(empty)"];
-		this.theme = asCrewTheme(theme);
-		this.done = done;
-		this.unsubscribeTheme = subscribeThemeChange(theme, () => this.invalidate());
-	}
-
-	invalidate(): void {}
-
-	dispose(): void {
-		this.unsubscribeTheme();
-	}
-
-	handleInput(data: string): void {
-		if (data === "q" || data === "\u001b") {
-			this.done(undefined);
-			return;
-		}
-		const maxScroll = Math.max(0, this.lines.length - this.lastHeight);
-		if (data === "k" || data === "\u001b[A") {
-			this.scroll = Math.max(0, this.scroll - 1);
-			this.autoScroll = false;
-		} else if (data === "j" || data === "\u001b[B") {
-			this.scroll = Math.min(maxScroll, this.scroll + 1);
-			this.autoScroll = this.scroll >= maxScroll;
-		} else if (data === "\u001b[5~") {
-			this.scroll = Math.max(0, this.scroll - this.lastHeight);
-			this.autoScroll = false;
-		} else if (data === "\u001b[6~") {
-			this.scroll = Math.min(maxScroll, this.scroll + this.lastHeight);
-			this.autoScroll = this.scroll >= maxScroll;
-		} else if (data === "g" || data === "\u001b[H") {
-			this.scroll = 0;
-			this.autoScroll = false;
-		} else if (data === "G" || data === "\u001b[F") {
-			this.scroll = maxScroll;
-			this.autoScroll = true;
-		} else if (data === "a") {
-			this.autoScroll = !this.autoScroll;
-		}
-	}
-
-	render(width: number): string[] {
-		return renderViewerBase(
-			{
-				theme: this.theme,
-				autoScroll: this.autoScroll,
-				lastHeight: this.lastHeight,
-				scroll: this.scroll,
-			},
-			width,
-			this.lines,
-			this.title,
-			this.subtitle,
-		);
-	}
-}
-
-export class DurableTranscriptViewer implements Component {
-	private scroll = 0;
-	private lastHeight = 16;
-	private autoScroll = true;
-	private manifest: TeamRunManifest;
-	private theme: TranscriptTheme;
-	private done: (result: undefined) => void;
-	private taskId?: string;
-	private fullTranscript = false;
-	private maxTailBytes: number;
-	private readonly unsubscribeTheme: () => void;
-
-	constructor(
-		manifest: TeamRunManifest,
-		theme: unknown,
-		done: (result: undefined) => void,
-		taskId?: string,
-		options: { maxTailBytes?: number } = {},
-	) {
-		this.manifest = manifest;
-		this.theme = asCrewTheme(theme);
-		this.done = done;
-		this.taskId = taskId;
-		this.maxTailBytes = options.maxTailBytes ?? DEFAULT_TRANSCRIPT_TAIL_BYTES;
-		this.unsubscribeTheme = subscribeThemeChange(theme, () => this.invalidate());
-	}
-
-	invalidate(): void {}
-
-	dispose(): void {
-		this.unsubscribeTheme();
-	}
-
-	handleInput(data: string): void {
-		if (data === "q" || data === "\u001b") {
-			this.done(undefined);
-			return;
-		}
-		const content = readRunTranscript(this.manifest, this.taskId, {
-			full: this.fullTranscript,
-			maxTailBytes: this.maxTailBytes,
-		}).lines;
-		const maxScroll = Math.max(0, content.length - this.lastHeight);
-		if (data === "k" || data === "\u001b[A") {
-			this.scroll = Math.max(0, this.scroll - 1);
-			this.autoScroll = false;
-		} else if (data === "j" || data === "\u001b[B") {
-			this.scroll = Math.min(maxScroll, this.scroll + 1);
-			this.autoScroll = this.scroll >= maxScroll;
-		} else if (data === "\u001b[5~") {
-			this.scroll = Math.max(0, this.scroll - this.lastHeight);
-			this.autoScroll = false;
-		} else if (data === "\u001b[6~") {
-			this.scroll = Math.min(maxScroll, this.scroll + this.lastHeight);
-			this.autoScroll = this.scroll >= maxScroll;
-		} else if (data === "g" || data === "\u001b[H") {
-			this.scroll = 0;
-			this.autoScroll = false;
-		} else if (data === "G" || data === "\u001b[F") {
-			this.scroll = maxScroll;
-			this.autoScroll = true;
-		} else if (data === "a") {
-			this.autoScroll = !this.autoScroll;
-		} else if (data === "f") {
-			this.fullTranscript = !this.fullTranscript;
-			this.scroll = 0;
-			this.autoScroll = !this.fullTranscript;
-		}
-	}
-
-	render(width: number): string[] {
-		const data = readRunTranscript(this.manifest, this.taskId, {
-			full: this.fullTranscript,
-			maxTailBytes: this.maxTailBytes,
-		});
-		return renderViewerBase(
-			{
-				theme: this.theme,
-				autoScroll: this.autoScroll,
-				lastHeight: this.lastHeight,
-				scroll: this.scroll,
-			},
-			width,
-			data.lines,
-			"pi-crew transcript",
-			`${data.title} · ${data.truncated ? `tail ${Math.round(data.bytesRead / 1024)}KB/${Math.round(data.size / 1024)}KB` : `full ${Math.round(data.size / 1024)}KB`} · f ${this.fullTranscript ? "tail" : "full"}`,
-		);
-	}
-}