import { ValidationSchemas } from '~/domain';
import { themis } from '~/main';

const wrapSut = <S>(sut: S) => ({ sut });

const getMakeSutFn = <T extends ValidationSchemas.Type>(type: T, rules: ValidationSchemas.Rules<T>) => {
	const makeSchemaFn = themis[type] as (rules: ValidationSchemas.Rules<T>) => ValidationSchemas<T>;
	return (moreRules: ValidationSchemas.Rules<T>) => {
		return wrapSut(makeSchemaFn({ ...rules, ...moreRules }));
	};
};

export const getMakeSut = Object.assign(getMakeSutFn, {
	customFormat: <T extends ValidationSchemas.Type>(
		format: ValidationSchemas.Format<T>,
		rules: ValidationSchemas.Rules<T>
	) => {
		return (moreRules: ValidationSchemas.Rules<T>) => {
			return wrapSut(themis.custom[format]({ ...rules, ...moreRules }));
		};
	},
});