import {
	type KeybindingHints,
	truncateToVisualLines,
} from "@earendil-works/pi-coding-agent";
import {
	Container,
	Image,
	Spacer,
	Text,
} from "@earendil-works/pi-tui";
import type { CodeModeRenderTheme } from "./types.js";

export interface RenderedToolContent {
	type: string;
	text?: string | undefined;
	data?: string | undefined;
	mimeType?: string | undefined;
}

export interface RenderedImage {
	data: string;
	mimeType: string;
	component: Image;
}

export function imagesByMimeType(
	contents: Array<RenderedToolContent & { data: string; mimeType: string }>,
): Map<string, Set<string>> {
	const images = new Map<string, Set<string>>();
	for (const item of contents) {
		const data = images.get(item.mimeType) ?? new Set<string>();
		data.add(item.data);
		images.set(item.mimeType, data);
	}
	return images;
}

export function previewText(text: string, theme: CodeModeRenderTheme, hideBody = false, hints?: Partial<KeybindingHints>): string {
	if (!text) return "";
	const preview = truncateToVisualLines(text, 5, 100, 0);
	if (hideBody) {
		const lines = preview.skippedCount + preview.visualLines.length;
		return theme.fg("muted", `... (${lines} ${lines === 1 ? "line" : "lines"}, ${expandHint(hints)})`);
	}
	if (preview.skippedCount <= 0) return preview.visualLines.join("\n");
	return `${theme.fg("muted", `... (${preview.skippedCount} more lines, ${expandHint(hints)})`)}\n${preview.visualLines.join("\n")}`;
}

export function expandHint(hints?: Partial<KeybindingHints>): string {
	return hints?.keyHint?.("app.tools.expand", "to expand") ?? "expand tool details";
}

export function renderTextAndImages(
	text: string,
	images: Array<RenderedToolContent & { data: string; mimeType: string }>,
	theme: CodeModeRenderTheme,
	imageCache: RenderedImage[] = [],
): Text | Container {
	const previousImages = imageCache.splice(0);
	if (images.length === 0) return new Text(text, text ? 4 : 0, 0);
	const container = new Container();
	if (text) container.addChild(new Text(text, 4, 0));
	for (const [index, image] of images.entries()) {
		if (text || index > 0) container.addChild(new Spacer(1));
		// Consume matches once so repeated images keep distinct placements, in result order.
		const previousIndex = previousImages.findIndex((previous) =>
			previous.data === image.data && previous.mimeType === image.mimeType);
		const retained = previousIndex < 0 ? undefined : previousImages.splice(previousIndex, 1)[0];
		const component = retained?.component ?? new Image(
			image.data,
			image.mimeType,
			{ fallbackColor: (value) => theme.fg("dim", value) },
			{ maxWidthCells: 60, maxHeightCells: 20 },
		);
		imageCache.push({ data: image.data, mimeType: image.mimeType, component });
		container.addChild(component);
	}
	return container;
}