Allow arguments for _count fields

paljs · Mar 8, 2024 · 2f43742 · 2f43742
1 parent 8f6ecbb
commit 2f43742
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/packages/generator/src/Generators.ts b/packages/generator/src/Generators.ts
@@ -222,7 +222,9 @@ export class Generators {
             .filter((field) => !this.options?.excludeInputFields?.includes(field.name))
             .forEach((field) => {
               fileContent.push(
-                `${field.name}: ${field.outputType.isList ? `[${field.outputType.type}!]` : field.outputType.type}${
+                `${field.name}${
+                  type.name.endsWith('CountOutputType') && field.args.length ? `(${this.getSDLArgs(field.args)})` : ''
+                }: ${field.outputType.isList ? `[${field.outputType.type}!]` : field.outputType.type}${
                   !field.isNullable ? '!' : ''
                 }`,
               );

diff --git a/packages/generator/tests/SDL/__snapshots__/generateSDLInputs.test.ts.snap b/packages/generator/tests/SDL/__snapshots__/generateSDLInputs.test.ts.snap
@@ -1022,7 +1022,7 @@ export default gql\`
   }
 
   type UserCountOutputType {
-    posts: Int!
+    posts(where: PostWhereInput): Int!
   }
 
   type UserCountAggregateOutputType {

diff --git a/packages/generator/tests/graphql-modules/__snapshots__/generateModulesInputs.test.ts.snap b/packages/generator/tests/graphql-modules/__snapshots__/generateModulesInputs.test.ts.snap
@@ -1022,7 +1022,7 @@ type AggregatePost {
 }
 
 type UserCountOutputType {
-  posts: Int!
+  posts(where: PostWhereInput): Int!
 }
 
 type UserCountAggregateOutputType {

diff --git a/packages/plugins/src/select.ts b/packages/plugins/src/select.ts
@@ -314,7 +314,11 @@ export class PrismaSelect<
             if (this.isAggregate) {
               selectObject[fieldName] = true;
             } else {
-              selectObject.select[fieldName] = true;
+              if (fields.name === '_count' && Object.keys(fieldsByTypeName[key].args).length) {
+                selectObject.select[fieldName] = fieldsByTypeName[key].args;
+              } else {
+                selectObject.select[fieldName] = true;
+              }
             }
           } else {
             if (this.isAggregate) {