import type { CacheResultCounts, JobSnapshot, PopulateSymbolGraphResult, WorkspaceCacheStatus } from "@danypops/lector";
import type { LectorTheme } from "../lector-tui-theme.ts";

type WorkspaceCacheAction = "status" | "populate" | "wait" | "job_status";

export function formatWorkspaceCacheCall(
	action: WorkspaceCacheAction,
	args: { directory?: unknown; maxFiles?: unknown; maxSymbolsPerFile?: unknown; jobId?: unknown },
	theme: LectorTheme,
): string {
	const label = theme.fg("toolTitle", theme.bold("workspace_cache"));
	if (action === "job_status" || action === "wait") {
		const jobId = typeof args.jobId === "string" ? args.jobId : "";
		return `${label} ${theme.fg("accent", action)} ${theme.fg("dim", jobId)}`;
	}
	const directory = typeof args.directory === "string" ? args.directory : "";
	const maxFiles = typeof args.maxFiles === "number" ? String(args.maxFiles) : "default";
	const maxSymbolsPerFile = typeof args.maxSymbolsPerFile === "number" ? String(args.maxSymbolsPerFile) : "default";
	const bounds =
		action === "populate" && (typeof args.maxFiles === "number" || typeof args.maxSymbolsPerFile === "number")
			? theme.fg("dim", ` (maxFiles=${maxFiles}, maxSymbolsPerFile=${maxSymbolsPerFile})`)
			: "";
	return `${label} ${theme.fg("accent", action)} ${theme.fg("dim", directory)}${bounds}`;
}

function formatResultCounts(result: CacheResultCounts): string {
	const failed = result.filesFailed > 0 ? `, ${result.filesFailed} failed` : "";
	return `${result.filesProcessed}/${result.filesAttempted} files${failed}, ${result.symbolsProcessed} symbols, ${result.nodesAdded} nodes, ${result.edgesAdded} edges`;
}

export function formatWorkspaceCacheStatusResult(status: WorkspaceCacheStatus | undefined, theme: LectorTheme): string {
	if (!status) return theme.fg("dim", "No result.");
	if (status.status === "not-cached") return theme.fg("warning", `not cached (${status.reason})`);
	if (status.status === "caching") return theme.fg("accent", `caching (job ${status.jobId})`);
	if (status.status === "waiting-for-resources") return theme.fg("accent", `waiting for resources (job ${status.jobId})`);
	if (status.status === "partial") return theme.fg("warning", `partial -- ${formatResultCounts(status.generation.result)}`);
	return theme.fg("success", `cached -- ${formatResultCounts(status.generation.result)}`);
}

export function formatJobSnapshotResult(job: JobSnapshot<PopulateSymbolGraphResult> | undefined, theme: LectorTheme): string {
	if (!job) return theme.fg("dim", "No result.");
	if (job.status === "queued") return theme.fg("dim", `queued (job ${job.id})`);
	if (job.status === "running") return theme.fg("accent", `running (job ${job.id})`);
	if (job.status === "failed") return theme.fg("error", `failed (job ${job.id}): ${job.error.code}: ${job.error.message}`);
	return theme.fg("success", `succeeded (job ${job.id}) -- ${formatResultCounts(job.result)}`);
}