fix in docs + refinement on whereUniqueInput

chrishoermann · Feb 3, 2023 · 6c58b32 · 6c58b32
1 parent a2f23d7
commit 6c58b32
Show file tree

Hide file tree

Showing 4 changed files with 46 additions and 2 deletions.
diff --git a/packages/generator/package.json b/packages/generator/package.json
@@ -1,6 +1,6 @@
 {
   "name": "zod-prisma-types",
-  "version": "2.0.6",
+  "version": "2.0.7",
   "description": "Generates zod schemas from Prisma models with advanced validation",
   "author": "Chris Hörmann",
   "license": "MIT",

diff --git a/packages/generator/src/classes/extendedDMMFInputType.ts b/packages/generator/src/classes/extendedDMMFInputType.ts
@@ -29,6 +29,7 @@ export class ExtendedDMMFInputType
   readonly isDecimalField: boolean;
   readonly omitFields: string[] = [];
   readonly imports: Set<string>;
+  readonly isWhereUniqueInput?: boolean;
 
   constructor(
     readonly generatorConfig: GeneratorConfig,
@@ -48,6 +49,7 @@ export class ExtendedDMMFInputType
     this.isDecimalField = this._setIsDecimalField();
     this.omitFields = this._setOmitFields();
     this.imports = this._setImports();
+    this.isWhereUniqueInput = this._setIsWhereUniqueInput();
   }
 
   /**
@@ -90,6 +92,10 @@ export class ExtendedDMMFInputType
     });
   }
 
+  private _setIsWhereUniqueInput() {
+    return /WhereUniqueInput$/.test(this.name);
+  }
+
   private _fieldIsPrismaFunctionType() {
     return this.name.match(PRISMA_FUNCTION_TYPES_WITH_VALIDATORS);
   }

diff --git a/packages/generator/src/functions/contentWriters/writeInputObjectType.ts b/packages/generator/src/functions/contentWriters/writeInputObjectType.ts
@@ -107,7 +107,23 @@ export const writeInputObjectType = (
         writer.newLine();
       });
     })
-    .write(`).strict();`);
+    .write(`).strict()`);
+
+  if (inputType.isWhereUniqueInput && inputType.fields.length > 1) {
+    writer.write(`.refine((data) => `);
+    inputType.fields.forEach((field, idx) => {
+      const writeComma = idx !== inputType.fields.length - 1;
+      writer.write(`!!data.${field?.name}`);
+      if (writeComma) {
+        writer.write(` || `);
+      }
+    });
+    writer.write(
+      `, { message: 'At least one field must be provided @ ${inputType.name}' })`,
+    );
+  }
+
+  writer.write(`;`);
 
   if (useMultipleFiles && !getSingleFileContent) {
     writer.blankLine().writeLine(`export default ${inputType.name}Schema;`);

diff --git a/packages/usage/tests/wereUniqueInput.test.ts b/packages/usage/tests/wereUniqueInput.test.ts
@@ -0,0 +1,22 @@
+import { ZodError } from 'zod';
+import { UserWhereUniqueInputSchema } from '../prisma/generated/zod';
+
+it('should accept the first of the two input values', () => {
+  const paresd = UserWhereUniqueInputSchema.parse({ id: 'idstring' });
+  expect(paresd).toEqual({ id: 'idstring' });
+});
+
+it('should accept the second of the two input values', () => {
+  const paresd = UserWhereUniqueInputSchema.parse({ email: 'email@mail.com' });
+  expect(paresd).toEqual({ email: 'email@mail.com' });
+});
+
+it('should throw an error when no input value is provided', () => {
+  try {
+    UserWhereUniqueInputSchema.parse({
+      email: undefined,
+    });
+  } catch (error) {
+    expect(error).toBeInstanceOf(ZodError);
+  }
+});