fix(client): aggregate types

Closes #2959

src/packages/client/fixtures/blog/prisma/schema.prisma

@@ -6,7 +6,7 @@ datasource db {
 generator client {
   provider             = "prisma-client-js"
   binaryTargets        = ["native"]
-  experimentalFeatures = ["connectOrCreate", "aggregations"]
+  experimentalFeatures = ["connectOrCreate", "aggregateApi"]
 }
 
 model User {

src/packages/client/src/__tests__/types/aggregate/index.ts

@@ -13,6 +13,17 @@ async function main() {
   })
 
   const { count, avg, max, min, sum } = await prisma.user.aggregate({
+    cursor: {
+      email: 'a@a.de',
+    },
+    orderBy: {
+      age: 'asc',
+    },
+    skip: 12,
+    take: 10,
+    where: {
+      age: { gt: 500 },
+    },
     count: true,
     avg: {
       age: true,

src/packages/client/src/generation/TSClient.ts

@@ -740,7 +740,7 @@ export class Model implements Generatable {
     return argsTypes
   }
   private getAggregationTypes() {
-    const { model } = this
+    const { model, mapping } = this
     const aggregateType = this.dmmf.outputTypeMap[getAggregateName(model.name)]
     const aggregateTypes = [aggregateType]
 
@@ -762,6 +762,15 @@ export class Model implements Generatable {
       aggregateTypes.push(maxType)
     }
 
+    const aggregateRootField = this.dmmf.queryType.fields.find(
+      (f) => f.name === mapping?.aggregate,
+    )
+    if (!aggregateRootField) {
+      throw new Error(
+        `Could not find aggregate root field for model ${model.name}. Mapping: ${mapping?.aggregate}`,
+      )
+    }
+
     return `${aggregateTypes
       .map((type) => new SchemaOutputType(type).toTS())
       .join('\n')}
@@ -795,15 +804,18 @@ ${
 
 export type ${getAggregateArgsName(model.name)} = {
 ${indent(
-  aggregateType.fields
-    .map((f) => {
-      if (f.name === 'count') {
-        return `${f.name}?: true`
-      }
-      return `${f.name}?: ${getAggregateInputType(
-        (f.outputType.type as SchemaOutputType).name,
-      )}`
-    })
+  aggregateRootField.args
+    .map((arg) => new InputField(arg).toTS())
+    .concat(
+      aggregateType.fields.map((f) => {
+        if (f.name === 'count') {
+          return `${f.name}?: true`
+        }
+        return `${f.name}?: ${getAggregateInputType(
+          (f.outputType.type as SchemaOutputType).name,
+        )}`
+      }),
+    )
     .join('\n'),
   tab,
 )}