CodeGen: Try to generate prisma client if no models are found (redwoo…

…djs#6308)
mcmx · Aug 29, 2022 · 0705da1 · 0705da1
1 parent 541ae47
commit 0705da1
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 18 deletions.
diff --git a/packages/internal/src/__tests__/__snapshots__/graphqlCodeGen.test.ts.snap b/packages/internal/src/__tests__/__snapshots__/graphqlCodeGen.test.ts.snap
@@ -3,6 +3,7 @@
 exports[`Generate gql typedefs api 1`] = `
 "import { Prisma } from "@prisma/client"
 import { GraphQLResolveInfo, GraphQLScalarType, GraphQLScalarTypeConfig } from 'graphql';
+import { PrismaModelOne as PrismaPrismaModelOne, PrismaModelTwo as PrismaPrismaModelTwo, Post as PrismaPost, Todo as PrismaTodo } from '@prisma/client';
 import { RedwoodGraphQLContext } from '@redwoodjs/graphql-server/dist/functions/types';
 export type Maybe<T> = T | null;
 export type InputMaybe<T> = Maybe<T>;
@@ -155,7 +156,7 @@ export type ResolversTypes = {
   Redwood: ResolverTypeWrapper<Redwood>;
   String: ResolverTypeWrapper<Scalars['String']>;
   Time: ResolverTypeWrapper<Scalars['Time']>;
-  Todo: ResolverTypeWrapper<Todo>;
+  Todo: ResolverTypeWrapper<PrismaTodo>;
 };
 
 /** Mapping between all available schema types and the resolvers parents */
@@ -172,7 +173,7 @@ export type ResolversParentTypes = {
   Redwood: Redwood;
   String: Scalars['String'];
   Time: Scalars['Time'];
-  Todo: Todo;
+  Todo: PrismaTodo;
 };
 
 export type requireAuthDirectiveArgs = {

diff --git a/packages/internal/src/__tests__/graphqlCodeGen.test.ts b/packages/internal/src/__tests__/graphqlCodeGen.test.ts
@@ -24,6 +24,17 @@ afterEach(() => {
   jest.restoreAllMocks()
 })
 
+jest.mock('@prisma/client', () => {
+  return {
+    ModelName: {
+      PrismaModelOne: 'PrismaModelOne',
+      PrismaModelTwo: 'PrismaModelTwo',
+      Post: 'Post',
+      Todo: 'Todo',
+    },
+  }
+})
+
 test('Generate gql typedefs web', async () => {
   await generateGraphQLSchema()
 

diff --git a/packages/internal/src/generate/graphqlCodeGen.ts b/packages/internal/src/generate/graphqlCodeGen.ts
@@ -15,6 +15,7 @@ import { CodeFileLoader } from '@graphql-tools/code-file-loader'
 import { GraphQLFileLoader } from '@graphql-tools/graphql-file-loader'
 import { loadDocuments, loadSchemaSync } from '@graphql-tools/load'
 import type { LoadTypedefsOptions } from '@graphql-tools/load'
+import execa from 'execa'
 import { DocumentNode } from 'graphql'
 
 import { getPaths } from '../paths'
@@ -133,23 +134,49 @@ function getLoadDocumentsOptions(filename: string) {
   return loadTypedefsConfig
 }
 
-function getPluginConfig() {
-  let prismaModels: Record<string, string> = {}
-  try {
-    // Extract the models from the prisma client and use those to
-    // set up internal redirects for the return values in resolvers.
-    const localPrisma = require('@prisma/client')
-    prismaModels = localPrisma.ModelName
-    Object.keys(prismaModels).forEach((key) => {
-      prismaModels[key] = `@prisma/client#${key} as Prisma${key}`
-    })
-    // This isn't really something you'd put in the GraphQL API, so
-    // we can skip the model.
-    if (prismaModels.RW_DataMigration) {
-      delete prismaModels.RW_DataMigration
+function getPrismaClient(hasGenerated = false): {
+  ModelName: Record<string, string>
+} {
+  const localPrisma = require('@prisma/client')
+
+  if (!localPrisma.ModelName) {
+    if (hasGenerated) {
+      return { ModelName: {} }
+    } else {
+      execa.sync('yarn rw prisma generate', { shell: true })
+
+      // Purge Prisma Client from node's require cache, so that the newly
+      // generated client gets picked up by any script that uses it
+      Object.keys(require.cache).forEach((key) => {
+        if (
+          key.includes('/node_modules/@prisma/client/') ||
+          key.includes('/node_modules/.prisma/client/')
+        ) {
+          delete require.cache[key]
+        }
+      })
+
+      return getPrismaClient(true)
     }
-  } catch (error) {
-    // This means they've not set up prisma types yet
+  }
+
+  return localPrisma
+}
+
+function getPluginConfig() {
+  // Extract the models from the prisma client and use those to
+  // set up internal redirects for the return values in resolvers.
+  const localPrisma = getPrismaClient()
+  const prismaModels = localPrisma.ModelName
+
+  Object.keys(prismaModels).forEach((key) => {
+    prismaModels[key] = `@prisma/client#${key} as Prisma${key}`
+  })
+
+  // This isn't really something you'd put in the GraphQL API, so
+  // we can skip the model.
+  if (prismaModels.RW_DataMigration) {
+    delete prismaModels.RW_DataMigration
   }
 
   const pluginConfig: CodegenTypes.PluginConfig &