From d3d87368442bf96cfaa23240bb25806a297bb62f Mon Sep 17 00:00:00 2001
From: meehawk <80167324+meehawk@users.noreply.github.com>
Date: Thu, 4 Jan 2024 14:24:32 -0500
Subject: [PATCH] feat: add generator flag support to specify generator

---
 packages/nx-prisma/src/executors/generate/executor.ts | 5 +++++
 packages/nx-prisma/src/executors/generate/schema.d.ts | 4 ++++
 packages/nx-prisma/src/executors/generate/schema.json | 4 ++++
 3 files changed, 13 insertions(+)

diff --git a/packages/nx-prisma/src/executors/generate/executor.ts b/packages/nx-prisma/src/executors/generate/executor.ts
index 1137e9fb..7e0a6e92 100644
--- a/packages/nx-prisma/src/executors/generate/executor.ts
+++ b/packages/nx-prisma/src/executors/generate/executor.ts
@@ -14,6 +14,7 @@ export default async function run(options: GenerateExecutorSchema, ctx: Executor
 const getArgs = (options: GenerateExecutorSchema, ctx: ExecutorContext): string[] => {
   const args = [];
   const schema = options?.schema ?? getDefaultScheme(ctx);
+  const generator = options?.generator;
 
   args.push(`--schema=${schema}`);
 
@@ -21,6 +22,10 @@ const getArgs = (options: GenerateExecutorSchema, ctx: ExecutorContext): string[
     args.push('--data-proxy');
   }
 
+  if (generator) {
+    args.push('--generator=${generator}');
+  }
+
   if (options?.watch) {
     args.push('--watch');
   }
diff --git a/packages/nx-prisma/src/executors/generate/schema.d.ts b/packages/nx-prisma/src/executors/generate/schema.d.ts
index 22fb803d..5fcdb886 100644
--- a/packages/nx-prisma/src/executors/generate/schema.d.ts
+++ b/packages/nx-prisma/src/executors/generate/schema.d.ts
@@ -14,6 +14,10 @@ export interface GenerateExecutorSchema extends PrismaBase {
    * The generate command will generate Prisma Client for use with the Data Proxy.
    */
   'data-proxy'?: boolean;
+  /**
+   * Specifies the generator to use
+   */
+  generator?: string;
   /**
    * The generate command will continue to watch the schema.prisma file and re-generate Prisma Client on file changes.
    */
diff --git a/packages/nx-prisma/src/executors/generate/schema.json b/packages/nx-prisma/src/executors/generate/schema.json
index 7f6c289e..0b180866 100644
--- a/packages/nx-prisma/src/executors/generate/schema.json
+++ b/packages/nx-prisma/src/executors/generate/schema.json
@@ -5,6 +5,10 @@
   "description": "The generate command generates assets like Prisma Client based on the generator and data model blocks defined in your schema.prisma file.",
   "type": "object",
   "properties": {
+    "generator": {
+      "type": "string",
+      "description": "Specifies the generator to use."
+    },
     "schema": {
       "type": "string",
       "description": "Specifies the path to the desired schema.prisma file to be processed instead of the default path. Both absolute and relative paths are supported."