added unique constraint assertion tools

This commit is contained in:
2024-04-09 08:56:28 -05:00
parent 2b2ff8e4ab
commit 1b47be09de
4 changed files with 90 additions and 3 deletions

22
pkg/unique.ts Normal file
View File

@@ -0,0 +1,22 @@
import { z } from "zod";
export type AssertUniqueType<T extends Record<string, unknown>> = z.ZodEffects<z.ZodArray<z.ZodType<T>>, T[], unknown>;
export function generateUniqueValidator<T extends Record<string, unknown>>(validator: z.ZodType<T>, ...keys: (keyof T)[]) {
return validator.array().refine((val) => {
for (const key of keys ?? []) {
const values = val.map((item) => item[key as keyof T]);
return values.length === new Set(values).size;
}
}, {
message: "Encountered duplicate values in unique field"
}) satisfies AssertUniqueType<T>;
}
export function assertUniqueKeys<T extends Record<string, unknown>>(data: T[], validator: z.ZodType<T>, ...keys: (keyof T)[]) {
return generateUniqueValidator(validator, ...keys).parse(data) satisfies T[];
}
export async function assertUniqueKeysAsync<T extends Record<string, unknown>>(data: T[], validator: z.ZodType<T>, ...keys: (keyof T)[]) {
return generateUniqueValidator(validator, ...keys).parseAsync(data) satisfies Promise<T[]>;
}