import * as R from "remeda";
import { z } from "zod";
import {
	date,
	falsyToNull,
	id,
	safeNullableStringSchema,
	safeStringSchema,
	stageId,
	timeString,
	weaponSplId,
} from "~/utils/zod";
import type {
	BadgeOption,
	FieldWithOptions,
	FormField,
	FormFieldArray,
	FormFieldDatetime,
	FormFieldDualSelect,
	FormFieldFieldset,
	FormFieldInputGroup,
	FormFieldItems,
	FormFieldSelect,
	FormsTranslationKey,
	SelectOption,
} from "./types";

export const formRegistry = z.registry<FormField>();

export type RequiresDefault<T extends z.ZodType> = T & {
	_requiresDefault: true;
};

type WithTypedTranslationKeys<T> = Omit<T, "label" | "bottomText"> & {
	label?: FormsTranslationKey;
	bottomText?: FormsTranslationKey;
};

type WithTypedItemLabels<T, V extends string> = Omit<T, "items"> & {
	items: Array<{ label: FormsTranslationKey | (() => string); value: V }>;
};

type WithTypedDualSelectFields<T, V extends string> = Omit<
	T,
	"fields" | "validate"
> & {
	fields: [
		{
			label?: FormsTranslationKey;
			items: Array<{ label: FormsTranslationKey | (() => string); value: V }>;
		},
		{
			label?: FormsTranslationKey;
			items: Array<{ label: FormsTranslationKey | (() => string); value: V }>;
		},
	];
	validate?: {
		func: (value: [V | null, V | null]) => boolean;
		message: FormsTranslationKey;
	};
};

function prefixKey(key: FormsTranslationKey | undefined): string | undefined {
	return key ? `forms:${key}` : undefined;
}

function prefixItems<V extends string>(
	items: Array<{ label: FormsTranslationKey | (() => string); value: V }>,
) {
	return items.map((item) => ({
		...item,
		label: typeof item.label === "string" ? `forms:${item.label}` : item.label,
	}));
}

export function customField<T extends z.ZodType>(
	args: Omit<Extract<FormField, { type: "custom" }>, "type">,
	schema: T,
) {
	// @ts-expect-error Complex generic type with registry
	return schema.register(formRegistry, {
		...args,
		type: "custom",
	});
}

export function textFieldOptional(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "text-field" }>,
			"type" | "initialValue" | "required"
		>
	>,
) {
	const schema =
		args.validate === "url"
			? z.url()
			: safeNullableStringSchema({ min: args.minLength, max: args.maxLength });

	return textFieldRefined(schema, args).register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		required: false,
		type: "text-field",
		initialValue: "",
	});
}

export function textFieldRequired(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "text-field" }>,
			"type" | "initialValue" | "required"
		>
	>,
) {
	const schema =
		args.validate === "url"
			? z.string().url()
			: safeStringSchema({ min: args.minLength, max: args.maxLength });

	return textFieldRefined(schema, args).register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		required: true,
		type: "text-field",
		initialValue: "",
	});
}

function textFieldRefined<T extends z.ZodType<string | null>>(
	schema: T,
	args: Omit<
		Extract<FormField, { type: "text-field" }>,
		"type" | "initialValue" | "required"
	>,
): T {
	let result = schema as z.ZodType<string | null>;

	if (args.regExp) {
		result = result.refine(
			(val) => {
				if (val === null) return true;
				return args.regExp!.pattern.test(val);
			},
			{ message: args.regExp!.message },
		);
	}

	if (args.validate && typeof args.validate !== "string") {
		result = result.refine(
			(val) => {
				if (val === null) return true;
				return (args.validate as { func: (value: string) => boolean }).func(
					val,
				);
			},
			{ message: args.validate!.message },
		);
	}

	if (args.toLowerCase) {
		result = result.transform(
			(val) => val?.toLowerCase() ?? null,
		) as unknown as typeof result;
	}

	return result as T;
}

export function numberField(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "text-field" }>,
			| "type"
			| "initialValue"
			| "required"
			| "validate"
			| "inputType"
			| "maxLength"
		>
	> & { maxLength?: number },
) {
	return z.coerce
		.number()
		.int()
		.nonnegative()
		.register(formRegistry, {
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			required: true,
			type: "text-field",
			inputType: "number",
			initialValue: "",
			maxLength: args.maxLength ?? 10,
		});
}

export function numberFieldOptional(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "text-field" }>,
			| "type"
			| "initialValue"
			| "required"
			| "validate"
			| "inputType"
			| "maxLength"
		>
	> & { maxLength?: number },
) {
	return z.coerce
		.number()
		.int()
		.nonnegative()
		.optional()
		.register(formRegistry, {
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			required: false,
			type: "text-field",
			inputType: "number",
			initialValue: "",
			maxLength: args.maxLength ?? 10,
		});
}

export function textAreaOptional(
	args: WithTypedTranslationKeys<
		Omit<Extract<FormField, { type: "text-area" }>, "type" | "initialValue">
	>,
) {
	return safeNullableStringSchema({ max: args.maxLength }).register(
		formRegistry,
		{
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			type: "text-area",
			initialValue: "",
		},
	);
}

export function textAreaRequired(
	args: WithTypedTranslationKeys<
		Omit<Extract<FormField, { type: "text-area" }>, "type" | "initialValue">
	>,
) {
	return safeStringSchema({ max: args.maxLength }).register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		type: "text-area",
		initialValue: "",
	});
}

export function toggle(
	args: WithTypedTranslationKeys<
		Omit<Extract<FormField, { type: "switch" }>, "type" | "initialValue">
	>,
) {
	return z
		.boolean()
		.optional()
		.default(false)
		.register(formRegistry, {
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			type: "switch",
			initialValue: false,
		});
}

function itemsSchema<V extends string>(items: FormFieldItems<V>) {
	return z.enum(items.map((item) => item.value) as [V, ...V[]]);
}

function clearableItemsSchema<V extends string>(items: FormFieldItems<V>) {
	return z.preprocess(
		falsyToNull,
		z.enum(items.map((item) => item.value) as [V, ...V[]]).nullable(),
	);
}

export function selectOptional<V extends string>(
	args: WithTypedTranslationKeys<
		WithTypedItemLabels<
			Omit<FormFieldSelect<"select", V>, "type" | "initialValue" | "clearable">,
			V
		>
	>,
) {
	return clearableItemsSchema(args.items).register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		items: prefixItems(args.items),
		type: "select",
		initialValue: null,
		clearable: true,
	});
}

export function select<V extends string>(
	args: WithTypedTranslationKeys<
		WithTypedItemLabels<
			Omit<FormFieldSelect<"select", V>, "type" | "initialValue" | "clearable">,
			V
		>
	>,
) {
	return itemsSchema(args.items).register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		items: prefixItems(args.items),
		type: "select",
		initialValue: args.items[0].value,
		clearable: false,
	});
}

export function selectDynamic(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "select-dynamic" }>,
			"type" | "initialValue" | "clearable"
		>
	>,
) {
	return z.string().register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		type: "select-dynamic",
		initialValue: null,
		clearable: false,
	}) as unknown as z.ZodType<string> & FieldWithOptions<SelectOption[]>;
}

export function selectDynamicOptional(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "select-dynamic" }>,
			"type" | "initialValue" | "clearable"
		>
	>,
) {
	return z
		.preprocess(falsyToNull, z.string().nullable())
		.register(formRegistry, {
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			type: "select-dynamic",
			initialValue: null,
			clearable: true,
		}) as unknown as z.ZodType<string | null> &
		FieldWithOptions<SelectOption[]>;
}

export function dualSelectOptional<V extends string>(
	args: WithTypedTranslationKeys<
		WithTypedDualSelectFields<
			Omit<
				FormFieldDualSelect<"dual-select", V>,
				"type" | "initialValue" | "clearable"
			>,
			V
		>
	>,
) {
	let schema = z
		.tuple([
			clearableItemsSchema(args.fields[0].items),
			clearableItemsSchema(args.fields[1].items),
		])
		.optional();

	if (args.validate) {
		schema = schema.refine(
			(val) => {
				if (!val) return true;
				const [first, second] = val;
				return args.validate!.func([first, second]);
			},
			{ message: `forms:${args.validate!.message}` },
		);
	}

	// @ts-expect-error Complex generic type
	return schema.register(formRegistry, {
		...args,
		bottomText: prefixKey(args.bottomText),
		fields: args.fields.map((field) => ({
			...field,
			label: prefixKey(field.label),
			items: prefixItems(field.items),
		})),
		type: "dual-select",
		initialValue: [null, null],
		clearable: true,
	});
}

export function radioGroup<V extends string>(
	args: WithTypedTranslationKeys<
		WithTypedItemLabels<
			Omit<FormFieldInputGroup<"radio-group", V>, "type" | "initialValue">,
			V
		>
	>,
) {
	return itemsSchema(args.items).register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		items: prefixItems(args.items),
		type: "radio-group",
		initialValue: args.items[0].value,
	});
}

type DateTimeArgs = WithTypedTranslationKeys<
	Omit<FormFieldDatetime<"datetime">, "type" | "initialValue" | "required">
> & {
	minMessage?: FormsTranslationKey;
	maxMessage?: FormsTranslationKey;
};

export function datetimeRequired(args: DateTimeArgs) {
	const resolveMin = args.min ?? (() => new Date(Date.UTC(2015, 4, 28)));
	const resolveMax = args.max ?? (() => new Date(Date.UTC(2030, 4, 28)));

	return z
		.preprocess(
			date,
			z
				.date({ message: "forms:errors.required" })
				.refine((d) => d >= resolveMin(), {
					message: `forms:${args.minMessage ?? "errors.dateTooEarly"}`,
				})
				.refine((d) => d <= resolveMax(), {
					message: `forms:${args.maxMessage ?? "errors.dateTooLate"}`,
				}),
		)
		.register(formRegistry, {
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			type: "datetime",
			initialValue: null,
			required: true,
		});
}

export function datetimeOptional(args: DateTimeArgs) {
	const resolveMin = args.min ?? (() => new Date(Date.UTC(2015, 4, 28)));
	const resolveMax = args.max ?? (() => new Date(Date.UTC(2030, 4, 28)));

	return z
		.preprocess(
			date,
			z
				.date()
				.refine((d) => d >= resolveMin(), {
					message: `forms:${args.minMessage ?? "errors.dateTooEarly"}`,
				})
				.refine((d) => d <= resolveMax(), {
					message: `forms:${args.maxMessage ?? "errors.dateTooLate"}`,
				})
				.nullish(),
		)
		.register(formRegistry, {
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			type: "datetime",
			initialValue: null,
			required: false,
		});
}

export function dayMonthYearRequired(args: DateTimeArgs) {
	const resolveMin = args.min ?? (() => new Date(Date.UTC(2015, 4, 28)));
	const resolveMax = args.max ?? (() => new Date(Date.UTC(2030, 4, 28)));

	return z
		.preprocess(
			date,
			z
				.date({ message: "forms:errors.required" })
				.refine((d) => d >= resolveMin(), {
					message: `forms:${args.minMessage ?? "errors.dateTooEarly"}`,
				})
				.refine((d) => d <= resolveMax(), {
					message: `forms:${args.maxMessage ?? "errors.dateTooLate"}`,
				}),
		)
		.transform((d) => ({
			day: d.getDate(),
			month: d.getMonth(),
			year: d.getFullYear(),
		}))
		.register(formRegistry, {
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			type: "date",
			initialValue: null,
			required: true,
		});
}

// xxx: should validate for duplicates too (if any duplicates -> error)
export function checkboxGroup<V extends string>(
	args: WithTypedTranslationKeys<
		WithTypedItemLabels<
			Omit<FormFieldInputGroup<"checkbox-group", V>, "type" | "initialValue">,
			V
		>
	>,
) {
	return z
		.array(itemsSchema(args.items))
		.min(args.minLength ?? 0)
		.register(formRegistry, {
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			items: prefixItems(args.items),
			type: "checkbox-group",
			initialValue: [],
		});
}

export function weaponPool(
	args: WithTypedTranslationKeys<
		Omit<Extract<FormField, { type: "weapon-pool" }>, "type" | "initialValue">
	>,
) {
	let schema = z
		.array(
			z.object({
				id: weaponSplId,
				isFavorite: z.boolean(),
			}),
		)
		.min(args.minCount ?? 0)
		.max(args.maxCount);

	if (!args.allowDuplicates) {
		schema = schema.refine(
			(val) => val.length === R.uniqueBy(val, (item) => item.id).length,
		);
	}

	return schema.register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		type: "weapon-pool",
		initialValue: [],
	});
}

export function stringConstant<T extends string>(value: T) {
	// @ts-expect-error Complex generic type with registry
	return z.literal(value).register(formRegistry, {
		type: "string-constant",
		initialValue: value,
		value,
	});
}

export function idConstant<T extends number>(value: T): z.ZodLiteral<T>;
export function idConstant(): RequiresDefault<z.ZodNumber>;
export function idConstant<T extends number>(value?: T) {
	const schema = value !== undefined ? z.literal(value) : id.clone();
	return schema.register(formRegistry, {
		type: "id-constant",
		initialValue: value,
		value: value ?? null,
	}) as never;
}

export function idConstantOptional<T extends number>(value?: T) {
	const schema = value ? z.literal(value).optional() : id.optional();
	return schema.register(formRegistry, {
		type: "id-constant",
		initialValue: value,
		value: value ?? null,
	});
}

export function array<S extends z.ZodType>(
	args: WithTypedTranslationKeys<
		Omit<FormFieldArray<"array", S>, "type" | "initialValue">
	>,
) {
	const schema = z
		.array(args.field)
		.min(args.min ?? 0)
		.max(args.max);
	// @ts-expect-error Complex generic type with registry
	return schema.register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		type: "array",
		initialValue: [],
	});
}

type TimeRangeArgs = WithTypedTranslationKeys<
	Omit<
		Extract<FormField, { type: "time-range" }>,
		"type" | "initialValue" | "startLabel" | "endLabel"
	>
> & {
	startLabel?: FormsTranslationKey;
	endLabel?: FormsTranslationKey;
};

export function timeRangeOptional(args: TimeRangeArgs) {
	return z
		.object({
			start: timeString,
			end: timeString,
		})
		.nullable()
		.register(formRegistry, {
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			startLabel: prefixKey(args.startLabel),
			endLabel: prefixKey(args.endLabel),
			type: "time-range",
			initialValue: null,
		});
}

export function fieldset<S extends z.ZodRawShape>(
	args: WithTypedTranslationKeys<
		Omit<FormFieldFieldset<"fieldset", S>, "type" | "initialValue">
	>,
) {
	// @ts-expect-error Complex generic type with registry
	return args.fields.register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		type: "fieldset",
		initialValue: {},
	});
}

export function userSearch(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "user-search" }>,
			"type" | "initialValue" | "required"
		>
	>,
) {
	return id.clone().register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		type: "user-search",
		initialValue: null,
		required: true,
	});
}

export function userSearchOptional(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "user-search" }>,
			"type" | "initialValue" | "required"
		>
	>,
) {
	return id.optional().register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		type: "user-search",
		initialValue: null,
		required: false,
	});
}

export function badges(
	args: WithTypedTranslationKeys<
		Omit<Extract<FormField, { type: "badges" }>, "type" | "initialValue">
	>,
) {
	return z
		.array(id)
		.max(args.maxCount ?? 50)
		.register(formRegistry, {
			...args,
			label: prefixKey(args.label),
			bottomText: prefixKey(args.bottomText),
			type: "badges",
			initialValue: [],
		}) as z.ZodArray<typeof id> & FieldWithOptions<BadgeOption[]>;
}

export function stageSelect(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "stage-select" }>,
			"type" | "initialValue" | "required"
		>
	>,
) {
	return stageId.register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		type: "stage-select",
		initialValue: 1,
		required: true,
	});
}

export function weaponSelect(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "weapon-select" }>,
			"type" | "initialValue" | "required"
		>
	>,
) {
	return weaponSplId.register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		type: "weapon-select",
		initialValue: null,
		required: true,
	});
}

export function weaponSelectOptional(
	args: WithTypedTranslationKeys<
		Omit<
			Extract<FormField, { type: "weapon-select" }>,
			"type" | "initialValue" | "required"
		>
	>,
) {
	return weaponSplId.optional().register(formRegistry, {
		...args,
		label: prefixKey(args.label),
		bottomText: prefixKey(args.bottomText),
		type: "weapon-select",
		initialValue: null,
		required: false,
	});
}