fix: Compatibility Issues with Prisma 5

fix: unlight#179

Author: unlight

src/handlers/input-type.ts

@@ -10,6 +10,8 @@ import { getPropertyType } from '../helpers/get-property-type';
 import { ImportDeclarationMap } from '../helpers/import-declaration-map';
 import { propertyStructure } from '../helpers/property-structure';
 import { EventArguments, InputType } from '../types';
+import { isWhereUniqueInputType } from '../helpers/is-where-unique-input-type';
+import { getWhereUniqueAtLeastKeys } from '../helpers/get-where-unique-at-least-keys';
 
 export function inputType(
   args: EventArguments & {
@@ -57,8 +59,6 @@ export function inputType(
   const modelFieldSettings = model && fieldSettings.get(model.name);
   const moduleSpecifier = '@nestjs/graphql';
 
-  // console.log('sourceFile.getBaseName()', sourceFile.getBaseName());
-
   importDeclarations
     .set('Field', {
       namedImports: [{ name: 'Field' }],
@@ -96,8 +96,13 @@ export function inputType(
     });
     const modelField = model?.fields.find(f => f.name === name);
     const isCustomsApplicable = typeName === modelField?.type;
+    const whereUniqueInputType =
+      isWhereUniqueInputType(typeName) &&
+      model &&
+      `Prisma.AtLeast<${typeName}, ${getWhereUniqueAtLeastKeys(model)}>`;
     const propertyType = castArray(
       propertySettings?.name ||
+        whereUniqueInputType ||
         getPropertyType({
           location,
           type: typeName,
@@ -115,7 +120,10 @@ export function inputType(
     if (propertySettings) {
       importDeclarations.create({ ...propertySettings });
     } else if (propertyType.includes('Decimal')) {
+      // TODO: Deprecated and should be removed
       importDeclarations.add('Decimal', '@prisma/client/runtime/library');
+    } else if (propertyType.some(p => p.startsWith('Prisma.'))) {
+      importDeclarations.add('Prisma', '@prisma/client');
     }
 
     // Get graphql type

src/helpers/get-where-unique-at-least-keys.ts

@@ -0,0 +1,15 @@
+import { DMMF } from '../types';
+
+export function getWhereUniqueAtLeastKeys(model: DMMF.Model) {
+  const names = model.fields
+    .filter(field => field.isUnique || field.isId)
+    .map(field => field.name);
+
+  for (const uniqueIndex of model.uniqueIndexes) {
+    const name = uniqueIndex.name || uniqueIndex.fields.join('_');
+
+    names.push(name);
+  }
+
+  return names.map(name => `'${name}'`).join(' | ');
+}

src/test/generate.spec.ts

@@ -316,6 +316,16 @@ describe('model with one id int', () => {
 
     expect(s.classFile.getName()).toEqual('UserAggregateArgs');
   });
+
+  it('where uniq must be wrapped to prisma atleast', () => {
+    const s = testSourceFile({
+      project,
+      class: 'FindManyUserArgs',
+      property: 'cursor',
+    });
+
+    expect(s.property?.type).toEqual(`Prisma.AtLeast<UserWhereUniqueInput, 'id'>`);
+  });
 });
 
 describe('duplicated', () => {
@@ -1877,7 +1887,7 @@ describe('object model options', () => {
 });
 
 describe('compound index', () => {
-  it('user unique input compound', async () => {
+  before(async () => {
     ({ project, sourceFiles } = await testGenerate({
       schema: `
                 model User {
@@ -1899,10 +1909,29 @@ describe('compound index', () => {
         `fields_Validator_input = true`,
       ],
     }));
-    setSourceFile('user-email-name-compound-unique.input.ts');
-    const minLength = classFile.getProperty('name')?.getDecorator('MinLength');
+  });
+
+  it('user unique input compound', () => {
+    const s = testSourceFile({
+      project,
+      class: 'UserEmailNameCompoundUniqueInput',
+    });
+
+    const minLength = s.classFile.getProperty('name')?.getDecorator('MinLength');
     expect(minLength?.getText()).toEqual('@Validator.MinLength(3)');
   });
+
+  it('compound uniq where must be wrapped to prisma atleast', () => {
+    const s = testSourceFile({
+      project,
+      class: 'FindManyUserArgs',
+      property: 'cursor',
+    });
+
+    expect(s.property?.type).toEqual(
+      `Prisma.AtLeast<UserWhereUniqueInput, 'id' | 'email_name'>`,
+    );
+  });
 });
 
 describe('field type', () => {