From 78ae3fcac4cc5e6794a5d2224ea61452ba201346 Mon Sep 17 00:00:00 2001
From: Andrew Porritt <aporritt@gogo-robot.com>
Date: Mon, 1 Jul 2024 16:37:45 +0100
Subject: [PATCH] Fix CreateManyAndReturn type

---
 packages/generator/src/sdl/GenerateTypes.ts   | 31 +++++++++----
 .../generateSDLTypescript.test.ts.snap        | 46 +++++++++++++------
 2 files changed, 53 insertions(+), 24 deletions(-)

diff --git a/packages/generator/src/sdl/GenerateTypes.ts b/packages/generator/src/sdl/GenerateTypes.ts
index bbb0b660..28bc50c8 100644
--- a/packages/generator/src/sdl/GenerateTypes.ts
+++ b/packages/generator/src/sdl/GenerateTypes.ts
@@ -43,6 +43,23 @@ export class GenerateTypes {
     );
   }
 
+  getParentType(name: string) {
+    if (['Query', 'Mutation'].includes(name)) {
+      return '{}';
+    }
+    if (name === 'AffectedRowsOutput') {
+      return 'Client.Prisma.BatchPayload';
+    }
+    if (this.isModel(name)) {
+      return `Client.${name}`;
+    }
+    if (name.startsWith('CreateMany') && name.endsWith('AndReturnOutputType')) {
+      const innerType = name.replace(/^CreateMany|AndReturnOutputType$/g, '');
+      return `ReturnType<Client.Prisma.${innerType}Delegate["createManyAndReturn"]>`;
+    }
+    return `Client.Prisma.${name}`;
+  }
+
   capital(name: string) {
     return name.charAt(0).toUpperCase() + name.slice(1);
   }
@@ -57,6 +74,10 @@ export class GenerateTypes {
             options.type
           }Args>`;
         }
+        if (options.type.startsWith('CreateMany') && options.type.endsWith('AndReturnOutputType')) {
+          const innerType = options.type.replace(/^CreateMany|AndReturnOutputType$/g, '');
+          return `ReturnType<Client.Prisma.${innerType}Delegate["createManyAndReturn"]>`;
+        }
         const type =
           options.type.toString() === 'AffectedRowsOutput'
             ? 'Prisma.BatchPayload'
@@ -105,15 +126,7 @@ export class GenerateTypes {
 
       // generate fields
       type.fields.forEach((field) => {
-        const parentType = ['Query', 'Mutation'].includes(type.name)
-          ? '{}'
-          : `Client.${
-              type.name === 'AffectedRowsOutput'
-                ? 'Prisma.BatchPayload'
-                : !this.isModel(type.name)
-                  ? 'Prisma.' + type.name
-                  : type.name
-            }`;
+        const parentType = this.getParentType(type.name);
         const argsType =
           field.args.length > 0
             ? `${['Query', 'Mutation'].includes(type.name) ? '' : type.name}${this.capital(field.name)}Args`
diff --git a/packages/generator/tests/SDL/__snapshots__/generateSDLTypescript.test.ts.snap b/packages/generator/tests/SDL/__snapshots__/generateSDLTypescript.test.ts.snap
index 4b634532..e0a9b50a 100644
--- a/packages/generator/tests/SDL/__snapshots__/generateSDLTypescript.test.ts.snap
+++ b/packages/generator/tests/SDL/__snapshots__/generateSDLTypescript.test.ts.snap
@@ -77,25 +77,33 @@ export type Post = { [key: string]: Resolver<any, any, any> } & {
 export type CreateManyUserAndReturnOutputType = {
   [key: string]: Resolver<any, any, any>
 } & {
-  id?: Resolver<Client.Prisma.CreateManyUserAndReturnOutputType, {}, number>
+  id?: Resolver<
+    ReturnType<Client.Prisma.UserDelegate['createManyAndReturn']>,
+    {},
+    number
+  >
   createdAt?: Resolver<
-    Client.Prisma.CreateManyUserAndReturnOutputType,
+    ReturnType<Client.Prisma.UserDelegate['createManyAndReturn']>,
     {},
     Date
   >
-  email?: Resolver<Client.Prisma.CreateManyUserAndReturnOutputType, {}, string>
+  email?: Resolver<
+    ReturnType<Client.Prisma.UserDelegate['createManyAndReturn']>,
+    {},
+    string
+  >
   name?: Resolver<
-    Client.Prisma.CreateManyUserAndReturnOutputType,
+    ReturnType<Client.Prisma.UserDelegate['createManyAndReturn']>,
     {},
     string | null
   >
   password?: Resolver<
-    Client.Prisma.CreateManyUserAndReturnOutputType,
+    ReturnType<Client.Prisma.UserDelegate['createManyAndReturn']>,
     {},
     string
   >
   permissions?: Resolver<
-    Client.Prisma.CreateManyUserAndReturnOutputType,
+    ReturnType<Client.Prisma.UserDelegate['createManyAndReturn']>,
     {},
     any
   >
@@ -104,30 +112,38 @@ export type CreateManyUserAndReturnOutputType = {
 export type CreateManyPostAndReturnOutputType = {
   [key: string]: Resolver<any, any, any>
 } & {
-  id?: Resolver<Client.Prisma.CreateManyPostAndReturnOutputType, {}, number>
+  id?: Resolver<
+    ReturnType<Client.Prisma.PostDelegate['createManyAndReturn']>,
+    {},
+    number
+  >
   published?: Resolver<
-    Client.Prisma.CreateManyPostAndReturnOutputType,
+    ReturnType<Client.Prisma.PostDelegate['createManyAndReturn']>,
     {},
     boolean
   >
-  title?: Resolver<Client.Prisma.CreateManyPostAndReturnOutputType, {}, string>
+  title?: Resolver<
+    ReturnType<Client.Prisma.PostDelegate['createManyAndReturn']>,
+    {},
+    string
+  >
   authorId?: Resolver<
-    Client.Prisma.CreateManyPostAndReturnOutputType,
+    ReturnType<Client.Prisma.PostDelegate['createManyAndReturn']>,
     {},
     number | null
   >
   createdAt?: Resolver<
-    Client.Prisma.CreateManyPostAndReturnOutputType,
+    ReturnType<Client.Prisma.PostDelegate['createManyAndReturn']>,
     {},
     Date
   >
   updatedAt?: Resolver<
-    Client.Prisma.CreateManyPostAndReturnOutputType,
+    ReturnType<Client.Prisma.PostDelegate['createManyAndReturn']>,
     {},
     Date
   >
   author?: Resolver<
-    Client.Prisma.CreateManyPostAndReturnOutputType,
+    ReturnType<Client.Prisma.PostDelegate['createManyAndReturn']>,
     CreateManyPostAndReturnOutputTypeAuthorArgs,
     Client.User | null
   >
@@ -191,7 +207,7 @@ export type Mutation = { [key: string]: Resolver<any, any, any> } & {
   createManyUserAndReturn?: Resolver<
     {},
     CreateManyUserAndReturnArgs,
-    Client.Prisma.CreateManyUserAndReturnOutputType[]
+    ReturnType<Client.Prisma.UserDelegate['createManyAndReturn']>
   >
   deleteOneUser?: Resolver<{}, DeleteOneUserArgs, Client.User | null>
   updateOneUser?: Resolver<{}, UpdateOneUserArgs, Client.User | null>
@@ -203,7 +219,7 @@ export type Mutation = { [key: string]: Resolver<any, any, any> } & {
   createManyPostAndReturn?: Resolver<
     {},
     CreateManyPostAndReturnArgs,
-    Client.Prisma.CreateManyPostAndReturnOutputType[]
+    ReturnType<Client.Prisma.PostDelegate['createManyAndReturn']>
   >
   deleteOnePost?: Resolver<{}, DeleteOnePostArgs, Client.Post | null>
   updateOnePost?: Resolver<{}, UpdateOnePostArgs, Client.Post | null>