import { extractApplyPatchRequests } from "./apply-patch.js";
import { getString, isRecord } from "./record.js";
import type { CheckerEdit, CommentCheckRequest, TextContent, ToolResultContent, ToolResultLike } from "./types.js";

export function extractCommentCheckRequests(event: ToolResultLike): CommentCheckRequest[] {
	if (event.isError) return [];
	if (isToolFailureOutput(getContentText(event.content))) return [];

	const toolName = event.toolName.toLowerCase();
	if (toolName === "write") return extractWriteRequest(event);
	if (toolName === "edit") return extractEditRequest(event);
	if (toolName === "multiedit" || toolName === "multi_edit") return extractMultiEditRequest(event);
	if (toolName === "apply_patch") return extractApplyPatchRequests(event);
	return [];
}

export function isToolFailureOutput(text: string): boolean {
	const lower = text.trim().toLowerCase();
	return (
		lower.startsWith("error") ||
		lower.includes("error:") ||
		lower.includes("failed to") ||
		lower.includes("could not")
	);
}

function extractWriteRequest(event: ToolResultLike): CommentCheckRequest[] {
	const filePath = getString(event.input, ["filePath", "file_path", "path"]);
	const content = getString(event.input, ["content"]);
	if (!filePath || content === undefined) return [];
	return [
		{
			sourceToolName: event.toolName,
			toolName: "Write",
			filePath,
			toolInput: {
				file_path: filePath,
				content,
			},
		},
	];
}

function extractEditRequest(event: ToolResultLike): CommentCheckRequest[] {
	const filePath = getString(event.input, ["filePath", "file_path", "path"]);
	const oldString = getString(event.input, ["oldString", "old_string"]);
	const newString = getString(event.input, ["newString", "new_string"]);
	if (!filePath || oldString === undefined || newString === undefined) return [];
	return [
		{
			sourceToolName: event.toolName,
			toolName: "Edit",
			filePath,
			toolInput: {
				file_path: filePath,
				old_string: oldString,
				new_string: newString,
			},
		},
	];
}

function extractMultiEditRequest(event: ToolResultLike): CommentCheckRequest[] {
	const filePath = getString(event.input, ["filePath", "file_path", "path"]);
	const edits = getEdits(event.input["edits"]);
	if (!filePath || edits.length === 0) return [];
	return [
		{
			sourceToolName: event.toolName,
			toolName: "MultiEdit",
			filePath,
			toolInput: {
				file_path: filePath,
				edits,
			},
		},
	];
}

function getEdits(value: unknown): CheckerEdit[] {
	if (!Array.isArray(value)) return [];
	const edits: CheckerEdit[] = [];
	for (const item of value) {
		if (!isRecord(item)) continue;
		const oldString = getString(item, ["oldString", "old_string"]);
		const newString = getString(item, ["newString", "new_string"]);
		if (oldString === undefined || newString === undefined) continue;
		edits.push({
			old_string: oldString,
			new_string: newString,
		});
	}
	return edits;
}

function getContentText(content: readonly ToolResultContent[] | undefined): string {
	if (!content) return "";
	return content
		.filter((block): block is TextContent => block.type === "text")
		.map((block) => block.text)
		.join("\n");
}