Support Prisma's official multi-file schema structure

Author: jackall3n

.changeset/unlucky-ghosts-do.md

@@ -0,0 +1,5 @@
+---
+"@trigger.dev/build": minor
+---
+
+Support Prisma's official multi-file schema structure

packages/build/package.json

@@ -78,6 +78,7 @@
   },
   "dependencies": {
     "@trigger.dev/core": "workspace:4.0.0",
+    "fast-glob": "^3.3.3",
     "pkg-types": "^1.1.3",
     "tinyglobby": "^0.2.2",
     "tsconfck": "3.1.3"

packages/build/src/extensions/prisma.ts

@@ -1,6 +1,7 @@
 import { BuildManifest, BuildTarget } from "@trigger.dev/core/v3";
 import { binaryForRuntime, BuildContext, BuildExtension } from "@trigger.dev/core/v3/build";
 import assert from "node:assert";
+import { glob } from 'fast-glob';
 import { existsSync } from "node:fs";
 import { cp, readdir } from "node:fs/promises";
 import { dirname, join, resolve } from "node:path";
@@ -112,11 +113,18 @@ export class PrismaExtension implements BuildExtension {
 
     context.logger.debug(`PrismaExtension is generating the Prisma client for version ${version}`);
 
-    const usingSchemaFolder = dirname(this._resolvedSchemaPath).endsWith("schema");
+    // Multi-file schemas can be used by specifying a directory instead of a file. https://www.prisma.io/docs/orm/prisma-schema/overview/location#multi-file-prisma-schema
+    const usingSchemaFolder = !this._resolvedSchemaPath.endsWith(".prisma");
+
+    context.logger.debug(`Using schema folder: ${usingSchemaFolder}`);
 
     const commands: string[] = [];
 
-    let prismaDir: string | undefined;
+    const prismaSourceDir = usingSchemaFolder
+      ? this._resolvedSchemaPath
+      : dirname(this._resolvedSchemaPath);
+
+    const prismaDestinationDir = join(manifest.outputPath, "prisma");
 
     const generatorFlags: string[] = [];
 
@@ -127,26 +135,22 @@ export class PrismaExtension implements BuildExtension {
     if (this.options.typedSql) {
       generatorFlags.push(`--sql`);
 
-      const prismaDir = usingSchemaFolder
-        ? dirname(dirname(this._resolvedSchemaPath))
-        : dirname(this._resolvedSchemaPath);
-
       context.logger.debug(`Using typedSql`);
 
       // Find all the files prisma/sql/*.sql
-      const sqlFiles = await readdir(join(prismaDir, "sql")).then((files) =>
+      const sqlFiles = await readdir(join(prismaSourceDir, "sql")).then((files) =>
         files.filter((file) => file.endsWith(".sql"))
       );
 
       context.logger.debug(`Found sql files`, {
         sqlFiles,
       });
 
-      const sqlDestinationPath = join(manifest.outputPath, "prisma", "sql");
+      const sqlDestinationPath = join(prismaDestinationDir, "sql");
 
       for (const file of sqlFiles) {
         const destination = join(sqlDestinationPath, file);
-        const source = join(prismaDir, "sql", file);
+        const source = join(prismaSourceDir, "sql", file);
 
         context.logger.debug(`Copying the sql from ${source} to ${destination}`);
 
@@ -155,28 +159,20 @@ export class PrismaExtension implements BuildExtension {
     }
 
     if (usingSchemaFolder) {
-      const schemaDir = dirname(this._resolvedSchemaPath);
-
-      prismaDir = dirname(schemaDir);
-
-      context.logger.debug(`Using the schema folder: ${schemaDir}`);
+      context.logger.debug(`Using the schema folder: ${this._resolvedSchemaPath}`);
 
       // Find all the files in schemaDir that end with .prisma (excluding the schema.prisma file)
-      const prismaFiles = await readdir(schemaDir).then((files) =>
-        files.filter((file) => file.endsWith(".prisma"))
-      );
+      const prismaFiles = await glob("**/*.prisma", {
+        cwd: this._resolvedSchemaPath
+      })
 
       context.logger.debug(`Found prisma files in the schema folder`, {
         prismaFiles,
       });
 
-      const schemaDestinationPath = join(manifest.outputPath, "prisma", "schema");
-
-      const allPrismaFiles = [...prismaFiles];
-
-      for (const file of allPrismaFiles) {
-        const destination = join(schemaDestinationPath, file);
-        const source = join(schemaDir, file);
+      for (const file of prismaFiles) {
+        const destination = join(prismaDestinationDir, file);
+        const source = join(this._resolvedSchemaPath, file);
 
         context.logger.debug(`Copying the prisma schema from ${source} to ${destination}`);
 
@@ -186,15 +182,14 @@ export class PrismaExtension implements BuildExtension {
       commands.push(
         `${binaryForRuntime(
           manifest.runtime
-        )} node_modules/prisma/build/index.js generate ${generatorFlags.join(" ")}` // Don't add the --schema flag or this will fail
+        )} node_modules/prisma/build/index.js generate --schema ./prisma ${generatorFlags.join(" ")}` // Don't add the --schema flag or this will fail
       );
     } else {
-      prismaDir = dirname(this._resolvedSchemaPath);
       // Now we need to add a layer that:
       // Copies the prisma schema to the build outputPath
       // Adds the `prisma` CLI dependency to the dependencies
       // Adds the `prisma generate` command, which generates the Prisma client
-      const schemaDestinationPath = join(manifest.outputPath, "prisma", "schema.prisma");
+      const schemaDestinationPath = join(prismaDestinationDir, "schema.prisma");
       // Copy the prisma schema to the build output path
       context.logger.debug(
         `Copying the prisma schema from ${this._resolvedSchemaPath} to ${schemaDestinationPath}`
@@ -215,8 +210,8 @@ export class PrismaExtension implements BuildExtension {
 
     if (this.options.migrate) {
       // Copy the migrations directory to the build output path
-      const migrationsDir = join(prismaDir, "migrations");
-      const migrationsDestinationPath = join(manifest.outputPath, "prisma", "migrations");
+      const migrationsDir = join(prismaSourceDir, "migrations");
+      const migrationsDestinationPath = join(prismaDestinationDir, "migrations");
 
       context.logger.debug(
         `Copying the prisma migrations from ${migrationsDir} to ${migrationsDestinationPath}`

pnpm-lock.yaml

@@ -42,7 +42,7 @@ export default defineConfig({
       emitDecoratorMetadata(),
       audioWaveform(),
       prismaExtension({
-        schema: "prisma/schema/schema.prisma",
+        schema: "prisma/schema.prisma",
         migrate: true,
         directUrlEnvVarName: "DATABASE_URL_UNPOOLED",
         clientGenerator: "client",