@pothos/plugin-prisma 0.17.2 → 0.18.0

package/src/prisma-field-builder.ts

@@ -14,7 +14,14 @@ import { prismaCursorConnectionQuery, wrapConnectionResult } from './cursors';
 import { getLoaderMapping, setLoaderMappings } from './loader-map';
 import { ModelLoader } from './model-loader';
 import { PrismaObjectRef } from './object-ref';
-import { getDelegateFromModel, getFindUniqueForRef, getRefFromModel, getRelation } from './refs';
+import {
+  getCursorFormatter,
+  getCursorParser,
+  getDelegateFromModel,
+  getFindUniqueForRef,
+  getRefFromModel,
+  getRelation,
+} from './refs';
 import {
   PrismaDelegate,
   PrismaModelTypes,
@@ -96,10 +103,13 @@ export class PrismaObjectFieldBuilder<
     const loaderCache = ModelLoader.forModel(this.model, this.builder);
     let typeName: string | undefined;
 
+    const formatCursor = getCursorFormatter(relationField.type, this.builder, cursor);
+    const parseCursor = getCursorParser(relationField.type, this.builder, cursor);
+
     const getQuery = (args: PothosSchemaTypes.DefaultConnectionArguments, ctx: {}) => ({
       ...((typeof query === 'function' ? query(args, ctx) : query) as {}),
       ...prismaCursorConnectionQuery({
-        column: cursor,
+        parseCursor,
         maxSize,
         defaultSize,
         args,
@@ -160,7 +170,7 @@ export class PrismaObjectFieldBuilder<
             await getResult(),
             args,
             connectionQuery.take,
-            cursor,
+            formatCursor,
             (parent as { _count?: Record<string, number> })._count?.[name],
           );
         },

package/src/refs.ts

@@ -1,5 +1,6 @@
 import { ObjectRef, SchemaTypes } from '@pothos/core';
 import { Prisma } from '../tests/client';
+import { formatCursor, parseCompositeCursor, parseRawCursor } from './cursors';
 import { PrismaObjectRef } from './object-ref';
 import { PrismaDelegate } from './types';
 import { PrismaModelTypes } from '.';
@@ -65,6 +66,25 @@ export function getRelation<Types extends SchemaTypes>(
   name: string,
   builder: PothosSchemaTypes.SchemaBuilder<Types>,
   relation: string,
+) {
+  const modelData = getModel(name, builder);
+
+  const fieldData = modelData.fields.find((field) => field.name === relation);
+
+  if (!fieldData) {
+    throw new Error(`Field '${relation}' not found in model '${name}'`);
+  }
+
+  if (fieldData.kind !== 'object') {
+    throw new Error(`Field ${relation} of model '${name}' is not a relation (${fieldData.kind})`);
+  }
+
+  return fieldData;
+}
+
+export function getModel<Types extends SchemaTypes>(
+  name: string,
+  builder: PothosSchemaTypes.SchemaBuilder<Types>,
 ) {
   const { client } = builder.options.prisma;
   // eslint-disable-next-line no-underscore-dangle
@@ -76,17 +96,34 @@ export function getRelation<Types extends SchemaTypes>(
     throw new Error(`Model '${name}' not found in DMMF`);
   }
 
-  const fieldData = modelData.fields.find((field) => field.name === relation);
+  return modelData;
+}
 
-  if (!fieldData) {
-    throw new Error(`Field '${relation}' not found in model '${name}'`);
-  }
+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 (fieldData.kind !== 'object') {
-    throw new Error(`Field ${relation} of model '${name}' is not a relation (${fieldData.kind})`);
-  }
+  return formatCursor(cursor === primaryKey ? modelData.primaryKey!.fields : cursor);
+}
 
-  return fieldData;
+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('_');
+
+  const parser =
+    cursor === primaryKey ? parseCompositeCursor(modelData.primaryKey!.fields) : parseRawCursor;
+
+  return (rawCursor: string) => ({
+    [cursor]: parser(rawCursor),
+  });
 }
 
 export function getRelatedDelegate<Types extends SchemaTypes>(

package/src/schema-builder.ts

@@ -37,6 +37,7 @@ schemaBuilderProto.prismaObject = function prismaObject(type, { fields, findUniq
     extensions: {
       ...options.extensions,
       pothosPrismaInclude: options.include,
+      pothosPrismaModel: type,
     },
     name,
     fields: fields ? () => fields(new PrismaObjectFieldBuilder(name, this, type)) : undefined,