@pothos/plugin-prisma 3.21.1 → 3.23.0

package/src/types.ts

@@ -495,30 +495,34 @@ export type PrismaConnectionFieldOptions<
       ResolveReturnShape
     >,
     'args' | 'resolve' | 'type'
-  > & {
-    type: Type;
-    cursor: string & keyof Model['Where'];
-    defaultSize?: number;
-    maxSize?: number;
-    resolve: (
-      query: {
-        include?: Model['Include'];
-        cursor?: {};
-        take: number;
-        skip: number;
-      },
-      parent: ParentShape,
-      args: InputShapeFromFields<Args> & PothosSchemaTypes.DefaultConnectionArguments,
-      context: Types['Context'],
-      info: GraphQLResolveInfo,
-    ) => MaybePromise<Model['Shape'][]>;
-    totalCount?: (
-      parent: ParentShape,
-      args: InputShapeFromFields<Args> & PothosSchemaTypes.DefaultConnectionArguments,
-      context: Types['Context'],
-      info: GraphQLResolveInfo,
-    ) => MaybePromise<number>;
-  };
+  > &
+  (InputShapeFromFields<Args> &
+    PothosSchemaTypes.DefaultConnectionArguments extends infer ConnectionArgs
+    ? {
+        type: Type;
+        cursor: string & keyof Model['Where'];
+        defaultSize?: number | ((args: ConnectionArgs, ctx: Types['Context']) => number);
+        maxSize?: number | ((args: ConnectionArgs, ctx: Types['Context']) => number);
+        resolve: (
+          query: {
+            include?: Model['Include'];
+            cursor?: {};
+            take: number;
+            skip: number;
+          },
+          parent: ParentShape,
+          args: ConnectionArgs,
+          context: Types['Context'],
+          info: GraphQLResolveInfo,
+        ) => MaybePromise<Model['Shape'][]>;
+        totalCount?: (
+          parent: ParentShape,
+          args: ConnectionArgs,
+          context: Types['Context'],
+          info: GraphQLResolveInfo,
+        ) => MaybePromise<number>;
+      }
+    : never);
 
 export type RelatedConnectionOptions<
   Types extends SchemaTypes,
@@ -551,43 +555,47 @@ export type RelatedConnectionOptions<
       unknown
     >,
     'resolve' | 'type'
-  > & {
-    description?: string | false;
-    query?: QueryForField<Types, Args, Model['Include'][Field & keyof Model['Include']]>;
-    type?: PrismaObjectRef<Model['Relations'][Field]['Types']>;
-    cursor: CursorFromRelation<Model, Field>;
-    defaultSize?: number;
-    maxSize?: number;
-    totalCount?: NeedsResolve extends false ? boolean : false;
-  } & (NeedsResolve extends false
+  > &
+  (InputShapeFromFields<Args> &
+    PothosSchemaTypes.DefaultConnectionArguments extends infer ConnectionArgs
     ? {
-        resolve?: (
-          query: {
-            include?: Model['Include'];
-            cursor?: {};
-            take: number;
-            skip: number;
-          },
-          parent: Model['Shape'],
-          args: InputShapeFromFields<Args>,
-          context: Types['Context'],
-          info: GraphQLResolveInfo,
-        ) => MaybePromise<Model['Relations'][Field & keyof Model['Relations']]['Shape']>;
-      }
-    : {
-        resolve: (
-          query: {
-            include?: Model['Include'];
-            cursor?: {};
-            take: number;
-            skip: number;
-          },
-          parent: Model['Shape'],
-          args: InputShapeFromFields<Args>,
-          context: Types['Context'],
-          info: GraphQLResolveInfo,
-        ) => MaybePromise<Model['Relations'][Field & keyof Model['Relations']]['Shape']>;
-      });
+        description?: string | false;
+        query?: QueryForField<Types, Args, Model['Include'][Field & keyof Model['Include']]>;
+        type?: PrismaObjectRef<Model['Relations'][Field]['Types']>;
+        cursor: CursorFromRelation<Model, Field>;
+        defaultSize?: number | ((args: ConnectionArgs, ctx: Types['Context']) => number);
+        maxSize?: number | ((args: ConnectionArgs, ctx: Types['Context']) => number);
+        totalCount?: NeedsResolve extends false ? boolean : false;
+      } & (NeedsResolve extends false
+        ? {
+            resolve?: (
+              query: {
+                include?: Model['Include'];
+                cursor?: {};
+                take: number;
+                skip: number;
+              },
+              parent: Model['Shape'],
+              args: ConnectionArgs,
+              context: Types['Context'],
+              info: GraphQLResolveInfo,
+            ) => MaybePromise<Model['Relations'][Field & keyof Model['Relations']]['Shape']>;
+          }
+        : {
+            resolve: (
+              query: {
+                include?: Model['Include'];
+                cursor?: {};
+                take: number;
+                skip: number;
+              },
+              parent: Model['Shape'],
+              args: ConnectionArgs,
+              context: Types['Context'],
+              info: GraphQLResolveInfo,
+            ) => MaybePromise<Model['Relations'][Field & keyof Model['Relations']]['Shape']>;
+          })
+    : never);
 
 export type WithBrand<T> = T & { [typeBrandKey]: string };
 

package/src/util/cursors.ts

@@ -1,6 +1,5 @@
 /* eslint-disable no-nested-ternary */
-import { MaybePromise, SchemaTypes } from '@pothos/core';
-// eslint-disable-next-line import/no-cycle
+import { decodeBase64, encodeBase64, MaybePromise, SchemaTypes } from '@pothos/core';
 import { getModel } from './datamodel';
 import { DMMFField } from './get-client';
 
@@ -27,12 +26,10 @@ export function formatCursorChunk(value: unknown) {
 export function formatCursor(fields: string | string[]) {
   return (value: Record<string, unknown>) => {
     if (typeof fields === 'string') {
-      return Buffer.from(`GPC:${formatCursorChunk(value[fields])}`).toString('base64');
+      return encodeBase64(`GPC:${formatCursorChunk(value[fields])}`);
     }
 
-    return Buffer.from(`GPC:J:${JSON.stringify(fields.map((name) => value[name]))}`).toString(
-      'base64',
-    );
+    return encodeBase64(`GPC:J:${JSON.stringify(fields.map((name) => value[name]))}`);
   };
 }
 
@@ -42,7 +39,7 @@ export function parseRawCursor(cursor: unknown) {
   }
 
   try {
-    const decoded = Buffer.from(cursor, 'base64').toString();
+    const decoded = decodeBase64(cursor);
     const [, type, value] = decoded.match(/^GPC:(\w):(.*)/) as [string, string, string];
 
     switch (type) {
@@ -199,8 +196,9 @@ export function parseCompositeCursor(fields: string[]) {
 
 export interface PrismaCursorConnectionQueryOptions {
   args: PothosSchemaTypes.DefaultConnectionArguments;
-  defaultSize?: number;
-  maxSize?: number;
+  ctx: {};
+  defaultSize?: number | ((args: {}, ctx: {}) => number);
+  maxSize?: number | ((args: {}, ctx: {}) => number);
   parseCursor: (cursor: string) => Record<string, unknown>;
 }
 
@@ -210,11 +208,13 @@ interface ResolvePrismaCursorConnectionOptions extends PrismaCursorConnectionQue
 }
 
 export function prismaCursorConnectionQuery({
-  args: { before, after, first, last },
+  args,
+  ctx,
   maxSize = DEFAULT_MAX_SIZE,
   defaultSize = DEFAULT_SIZE,
   parseCursor,
 }: PrismaCursorConnectionQueryOptions) {
+  const { before, after, first, last } = args;
   if (first != null && first < 0) {
     throw new TypeError('Argument "first" must be a non-negative integer');
   }
@@ -227,10 +227,6 @@ export function prismaCursorConnectionQuery({
     throw new Error('Arguments "before" and "after" are not supported at the same time');
   }
 
-  if (before != null && last == null) {
-    throw new Error('Argument "last" must be provided when using "before"');
-  }
-
   if (before != null && first != null) {
     throw new Error('Arguments "before" and "first" are not supported at the same time');
   }
@@ -241,7 +237,11 @@ export function prismaCursorConnectionQuery({
 
   const cursor = before ?? after;
 
-  let take = Math.min(first ?? last ?? defaultSize, maxSize) + 1;
+  const maxSizeForConnection = typeof maxSize === 'function' ? maxSize(args, ctx) : maxSize;
+  const defaultSizeForConnection =
+    typeof defaultSize === 'function' ? defaultSize(args, ctx) : defaultSize;
+
+  let take = Math.min(first ?? last ?? defaultSizeForConnection, maxSizeForConnection) + 1;
 
   if (before) {
     take = -take;
@@ -304,3 +304,48 @@ export async function resolvePrismaCursorConnection<T extends {}>(
 
   return wrapConnectionResult(results, options.args, query.take, cursor, options.totalCount);
 }
+
+export function getCursorFormatter<Types extends SchemaTypes>(
+  name: string,
+  builder: PothosSchemaTypes.SchemaBuilder<Types>,
+  cursor: string,
+) {
+  const modelData = getModel(name, builder);
+  const primaryKey = modelData.primaryKey?.name ?? modelData.primaryKey?.fields.join('_');
+  if (primaryKey === cursor) {
+    return formatCursor(modelData.primaryKey!.fields);
+  }
+
+  const uniqueIndex = modelData.uniqueIndexes.find(
+    (idx) => (idx.name ?? idx.fields.join('_')) === cursor,
+  );
+
+  return formatCursor(uniqueIndex?.fields ?? cursor);
+}
+
+export function getCursorParser<Types extends SchemaTypes>(
+  name: string,
+  builder: PothosSchemaTypes.SchemaBuilder<Types>,
+  cursor: string,
+) {
+  const modelData = getModel(name, builder);
+  const primaryKey = modelData.primaryKey?.name ?? modelData.primaryKey?.fields.join('_');
+
+  let parser = parseRawCursor;
+
+  if (primaryKey === cursor) {
+    parser = parseCompositeCursor(modelData.primaryKey!.fields);
+  } else {
+    const uniqueIndex = modelData.uniqueIndexes.find(
+      (idx) => (idx.name ?? idx.fields.join('_')) === cursor,
+    );
+
+    if (uniqueIndex) {
+      parser = parseCompositeCursor(uniqueIndex.fields);
+    }
+  }
+
+  return (rawCursor: string) => ({
+    [cursor]: parser(rawCursor),
+  });
+}

package/src/util/datamodel.ts

@@ -1,8 +1,6 @@
 import { ObjectRef, SchemaTypes } from '@pothos/core';
 import { PrismaObjectRef } from '../object-ref';
 import { PrismaClient, PrismaDelegate, PrismaModelTypes } from '../types';
-// eslint-disable-next-line import/no-cycle
-import { formatCursor, parseCompositeCursor, parseRawCursor } from './cursors';
 import { getDMMF } from './get-client';
 
 export const refMap = new WeakMap<object, Map<string, PrismaObjectRef<PrismaModelTypes>>>();
@@ -76,51 +74,6 @@ export function getModel<Types extends SchemaTypes>(
   return modelData;
 }
 
-export function getCursorFormatter<Types extends SchemaTypes>(
-  name: string,
-  builder: PothosSchemaTypes.SchemaBuilder<Types>,
-  cursor: string,
-) {
-  const modelData = getModel(name, builder);
-  const primaryKey = modelData.primaryKey?.name ?? modelData.primaryKey?.fields.join('_');
-  if (primaryKey === cursor) {
-    return formatCursor(modelData.primaryKey!.fields);
-  }
-
-  const uniqueIndex = modelData.uniqueIndexes.find(
-    (idx) => (idx.name ?? idx.fields.join('_')) === cursor,
-  );
-
-  return formatCursor(uniqueIndex?.fields ?? cursor);
-}
-
-export function getCursorParser<Types extends SchemaTypes>(
-  name: string,
-  builder: PothosSchemaTypes.SchemaBuilder<Types>,
-  cursor: string,
-) {
-  const modelData = getModel(name, builder);
-  const primaryKey = modelData.primaryKey?.name ?? modelData.primaryKey?.fields.join('_');
-
-  let parser = parseRawCursor;
-
-  if (primaryKey === cursor) {
-    parser = parseCompositeCursor(modelData.primaryKey!.fields);
-  } else {
-    const uniqueIndex = modelData.uniqueIndexes.find(
-      (idx) => (idx.name ?? idx.fields.join('_')) === cursor,
-    );
-
-    if (uniqueIndex) {
-      parser = parseCompositeCursor(uniqueIndex.fields);
-    }
-  }
-
-  return (rawCursor: string) => ({
-    [cursor]: parser(rawCursor),
-  });
-}
-
 export function getDelegateFromModel(client: PrismaClient, model: string) {
   const lowerCase = `${model.slice(0, 1).toLowerCase()}${model.slice(1)}`;