/**
 * Edit mode wrapper for the Codex `apply_patch` envelope format.
 *
 * The mode accepts a single `input` string containing a full
 * `*** Begin Patch ... *** End Patch` block, parses it, and fans out to
 * the existing `executePatchSingle` — so all the machinery (plan mode,
 * LSP writethrough, fs-cache invalidation, diagnostics) is shared with
 * the `patch` mode.
 */

import * as z from "zod/v4";
import { parseApplyPatch, parseApplyPatchStreaming } from "../apply-patch/parser";
import { ApplyPatchError } from "../diff";
import type { PatchEditEntry } from "./patch";

export const applyPatchSchema = z.object({
	input: z.string().describe("apply_patch envelope"),
});

export type ApplyPatchParams = z.infer<typeof applyPatchSchema>;

export type ApplyPatchEntry = PatchEditEntry & { path: string };

/**
 * Parse the envelope and lower each hunk to a `PatchEditEntry` so it can
 * be routed through `executePatchSingle`.
 */
export function expandApplyPatchToEntries(params: ApplyPatchParams): ApplyPatchEntry[] {
	const hunks = parseApplyPatch(params.input);
	if (hunks.length === 0) {
		throw new ApplyPatchError("No files were modified.");
	}
	return hunks.map(
		(h): ApplyPatchEntry => ({
			path: h.path,
			op: h.op,
			rename: h.rename,
			diff: h.diff,
		}),
	);
}

export function expandApplyPatchToPreviewEntries(params: ApplyPatchParams): ApplyPatchEntry[] {
	const hunks = parseApplyPatchStreaming(params.input);
	return hunks.map(
		(h): ApplyPatchEntry => ({
			path: h.path,
			op: h.op,
			rename: h.rename,
			diff: h.diff,
		}),
	);
}