import type { ActionFunction, ActionFunctionArgs } from "react-router";

/**
 * Wraps an action for API use, converting its redirect-based responses to JSON: `successToast`
 * returns `redirect("?__success=message")`, `errorToastIfFalsy/errorToastIfErr` throw
 * `redirect("?__error=message")`, and validation failures return `{ fieldErrors }`.
 */
export async function wrapActionForApi(
	actionFn: ActionFunction,
	args: ActionFunctionArgs,
): Promise<Response> {
	try {
		const response = await actionFn(args);

		if (response instanceof Response && response.status === 302) {
			return new Response(null, { status: 200 });
		}

		if (response && typeof response === "object" && "fieldErrors" in response) {
			return new Response(
				JSON.stringify({
					error: "Validation failed",
					fieldErrors: response.fieldErrors,
				}),
				{
					status: 400,
					headers: { "Content-Type": "application/json" },
				},
			);
		}

		return response as Response;
	} catch (e) {
		if (e instanceof Response && e.status === 302) {
			const location = e.headers.get("Location") ?? "";
			const search = location.slice(location.indexOf("?") + 1);
			const errorMsg = new URLSearchParams(search).get("__error");
			if (errorMsg !== null) {
				return new Response(JSON.stringify({ error: errorMsg }), {
					status: 400,
					headers: { "Content-Type": "application/json" },
				});
			}
		}

		throw e;
	}
}