@pothos/plugin-prisma 0.18.0 → 3.1.0

package/src/types.ts

@@ -10,6 +10,7 @@ import {
   InterfaceParam,
   ListResolveValue,
   MaybePromise,
+  Normalize,
   ObjectRef,
   OutputShape,
   OutputType,
@@ -26,52 +27,17 @@ export interface PrismaDelegate {
   findUnique: (...args: any[]) => Promise<unknown>;
 }
 
-type RelationKeys<T> = {
-  [K in keyof T]: T[K] extends (args: {}) => {
-    then: (cb: (result: unknown) => unknown) => unknown;
-  }
-    ? K
-    : never;
-}[keyof T];
-
-export type ModelTypes<Model extends {}> = Model extends {
-  findUnique: (
-    options: infer UniqueOptions & {
-      where?: infer Where | null | undefined;
-      select?: infer Select | null | undefined;
-    } & (
-        | {
-            include?: infer Include | null | undefined;
-          }
-        | {}
-      ),
-  ) => infer Chain & {
-    then: (cb: (result: infer Shape | null) => unknown) => unknown;
-  };
-}
-  ? PrismaModelTypes & {
-      Shape: Shape;
-      Include: unknown extends Include ? never : Include;
-      Where: Where;
-      Fields: keyof Select;
-      ListRelation: ListRelationFields<Include> & string;
-      Relations: {
-        [RelationName in RelationKeys<Chain>]: Chain[RelationName] extends (args: {}) => {
-          then: (cb: (result: infer Relation) => unknown) => unknown;
-        }
-          ? { Shape: Relation; Types: PrismaModelTypes }
-          : never;
-      };
-    }
-  : never;
+export const prismaModelName = Symbol.for('Pothos.prismaModelName');
 
 export interface PrismaModelTypes {
   Name: string;
   Shape: {};
   Include: unknown;
+  Select: unknown;
   Where: {};
   Fields: string;
   ListRelations: string;
+  RelationName: string;
   Relations: Record<
     string,
     {
@@ -81,52 +47,97 @@ export interface PrismaModelTypes {
   >;
 }
 
-export type ListRelationFields<T> = {
-  [K in keyof T]: T[K] extends infer Option
-    ? Option extends { orderBy?: unknown }
-      ? K
+type ExtractModel<Types extends SchemaTypes, ParentShape> = ParentShape extends {
+  [prismaModelName]?: infer Name;
+}
+  ? Types['PrismaTypes'][Name & keyof Types['PrismaTypes']] extends infer Model
+    ? Model extends PrismaModelTypes
+      ? Model
       : never
-    : never;
-}[keyof T];
+    : never
+  : never;
+
+export type PrismaObjectFieldOptions<
+  Types extends SchemaTypes,
+  ParentShape,
+  Type extends TypeParam<Types>,
+  Nullable extends FieldNullability<Type>,
+  Args extends InputFieldMap,
+  Select,
+  ResolveReturnShape,
+> = PothosSchemaTypes.ObjectFieldOptions<
+  Types,
+  unknown extends Select
+    ? ParentShape
+    : ParentShape &
+        ShapeFromSelection<
+          ExtractModel<Types, ParentShape>,
+          { select: Select extends (...args: any[]) => infer S ? S : Select }
+        >,
+  Type,
+  Nullable,
+  Args,
+  ResolveReturnShape
+> & {
+  select?: Select &
+    (
+      | ExtractModel<Types, ParentShape>['Select']
+      | ((
+          args: InputShapeFromFields<Args>,
+          ctx: Types['Context'],
+        ) => ExtractModel<Types, ParentShape>['Select'])
+    );
+};
 
-export type PrismaObjectFieldsShape<
+type PrismaObjectFieldsShape<
   Types extends SchemaTypes,
   Model extends PrismaModelTypes,
   NeedsResolve extends boolean,
   Shape extends object,
-> = (t: PrismaObjectFieldBuilder<Types, Model, NeedsResolve, Shape>) => FieldMap;
-
-export type ShapeFromInclude<Model extends PrismaModelTypes, Include> = {} extends Include
-  ? Model['Shape']
-  : {
-      [K in keyof Include &
-        keyof Model['Relations']]: Model['Relations'][K]['Shape'] extends infer RelationShape
-        ? RelationShape extends (infer ItemShape)[]
-          ? (ItemShape &
-              (Include[K] extends { include?: infer NestedInclude & object }
-                ? ShapeFromInclude<Model['Relations'][K]['Types'], NestedInclude>
-                : {}))[]
-          : RelationShape &
-              (
-                | (Include[K] extends { include?: infer NestedInclude & object }
-                    ? ShapeFromInclude<Model['Relations'][K]['Types'], NestedInclude>
-                    : {})
-                | null
-              )
-        : never;
-    };
+  Select,
+> = Model['Select'] extends Select
+  ? (t: PrismaObjectFieldBuilder<Types, Model, NeedsResolve, Shape>) => FieldMap
+  : (t: PrismaSelectionFieldBuilder<Types, Model, Shape>) => FieldMap;
 
-export type ShapeWithInclude<
+type PrismaSelectionFieldBuilder<
+  Types extends SchemaTypes,
   Model extends PrismaModelTypes,
-  Include extends Model['Include'],
-> = Model['Shape'] & ([Include] extends [never] ? {} : ShapeFromInclude<Model, Include>);
+  Shape extends object,
+> = PrismaObjectFieldBuilder<Types, Model, false, Shape>;
+
+interface BaseSelection {
+  include?: unknown;
+  select?: unknown;
+}
+
+export type SelectedKeys<T> = { [K in keyof T]: T[K] extends false ? never : K }[keyof T];
+
+export type ShapeFromSelection<Model extends PrismaModelTypes, Selection> = Normalize<
+  Selection extends BaseSelection
+    ? unknown extends Selection['select']
+      ? Model['Shape'] & RelationShapeFromInclude<Model, Selection['include']>
+      : Pick<Model['Shape'], SelectedKeys<Selection['select']>> &
+          RelationShapeFromInclude<Model, Selection['select']>
+    : Model['Shape']
+>;
+
+type RelationShapeFromInclude<Model extends PrismaModelTypes, Include> = Normalize<{
+  [K in SelectedKeys<Include> as K extends Model['RelationName']
+    ? K
+    : never]: K extends keyof Model['Relations']
+    ? Model['Relations'][K]['Shape'] extends unknown[]
+      ? ShapeFromSelection<Model['Relations'][K]['Types'], Include[K]>[]
+      : ShapeFromSelection<Model['Relations'][K]['Types'], Include[K]>
+    : unknown;
+}>;
 
 export type PrismaObjectTypeOptions<
   Types extends SchemaTypes,
   Model extends PrismaModelTypes,
   Interfaces extends InterfaceParam<Types>[],
   FindUnique,
-  Include extends Model['Include'],
+  Include,
+  Select,
   Shape extends object,
 > = NameOrVariant &
   Omit<
@@ -134,13 +145,28 @@ export type PrismaObjectTypeOptions<
     | PothosSchemaTypes.ObjectTypeWithInterfaceOptions<Types, Shape, Interfaces>,
     'fields'
   > & {
-    include?: Include;
-    fields?: PrismaObjectFieldsShape<Types, Model, FindUnique extends null ? true : false, Shape>;
-    findUnique: FindUnique &
-      (((parent: Shape, context: Types['Context']) => Model['Where']) | null);
-  };
+    fields?: PrismaObjectFieldsShape<
+      Types,
+      Model,
+      FindUnique extends null ? true : false,
+      Shape & (FindUnique extends null ? {} : { [prismaModelName]?: Model['Name'] }),
+      Select
+    >;
+  } & (
+    | {
+        include?: Include & Model['Include'];
+        select?: never;
+        findUnique: FindUnique &
+          (((parent: Shape, context: Types['Context']) => Model['Where']) | null);
+      }
+    | {
+        select: Model['Select'] & Select;
+        include?: never;
+        findUnique: (parent: Shape, context: Types['Context']) => Model['Where'];
+      }
+  );
 
-export type NameOrVariant =
+type NameOrVariant =
   | {
       name?: never;
       variant?: string;
@@ -154,7 +180,8 @@ export type PrismaNodeOptions<
   Types extends SchemaTypes,
   Model extends PrismaModelTypes,
   Interfaces extends InterfaceParam<Types>[],
-  Include extends Model['Include'],
+  Include,
+  Select,
   Shape extends object,
 > = NameOrVariant &
   Omit<
@@ -162,7 +189,6 @@ export type PrismaNodeOptions<
     | PothosSchemaTypes.ObjectTypeWithInterfaceOptions<Types, Shape, Interfaces>,
     'fields' | 'isTypeOf'
   > & {
-    include?: Include;
     id: Omit<
       FieldOptionsFromKind<
         Types,
@@ -178,11 +204,26 @@ export type PrismaNodeOptions<
     > & {
       resolve: (parent: Shape, context: Types['Context']) => MaybePromise<OutputShape<Types, 'ID'>>;
     };
-    fields?: PrismaObjectFieldsShape<Types, Model, false, Shape>;
+    fields?: PrismaObjectFieldsShape<
+      Types,
+      Model,
+      false,
+      Shape & { [prismaModelName]?: Model['Name'] },
+      Select
+    >;
     findUnique: (id: string, context: Types['Context']) => Model['Where'];
-  };
+  } & (
+    | {
+        include?: Include & Model['Include'];
+        select?: never;
+      }
+    | {
+        select: Model['Select'] & Select;
+        include?: never;
+      }
+  );
 
-export type QueryForField<
+type QueryForField<
   Types extends SchemaTypes,
   Args extends InputFieldMap,
   Include,
@@ -195,18 +236,19 @@ export type QueryForField<
         ) => Omit<Include, 'include' | 'select'>)
   : never;
 
-export type IncludeFromRelation<
+type QueryFromRelation<
   Model extends PrismaModelTypes,
   Field extends keyof Model['Include'],
 > = Model['Include'][Field] extends infer Include
   ? Include extends {
-      include?: infer T;
+      include?: infer I;
+      select?: infer S;
     }
-    ? NonNullable<T>
+    ? { include?: NonNullable<I>; select?: NonNullable<S> }
     : never
   : never;
 
-export type CursorFromRelation<
+type CursorFromRelation<
   Model extends PrismaModelTypes,
   Field extends Model['ListRelations'],
 > = Field extends keyof Model['Include']
@@ -217,7 +259,7 @@ export type CursorFromRelation<
     : never
   : never;
 
-export type RefForRelation<
+type RefForRelation<
   Model extends PrismaModelTypes,
   Field extends keyof Model['Relations'],
 > = Model['Relations'][Field] extends unknown[]
@@ -247,7 +289,7 @@ export type RelatedFieldOptions<
   (NeedsResolve extends false
     ? {
         resolve?: (
-          query: { include?: IncludeFromRelation<Model, Field & keyof Model['Include']> },
+          query: QueryFromRelation<Model, Field & keyof Model['Include']>,
           parent: Shape,
           args: InputShapeFromFields<Args>,
           context: Types['Context'],
@@ -258,7 +300,7 @@ export type RelatedFieldOptions<
       }
     : {
         resolve: (
-          query: { include?: IncludeFromRelation<Model, Field & keyof Model['Include']> },
+          query: QueryFromRelation<Model, Field & keyof Model['Include']>,
           parent: Shape,
           args: InputShapeFromFields<Args>,
           context: Types['Context'],
@@ -401,6 +443,12 @@ export type PrismaConnectionFieldOptions<
       context: Types['Context'],
       info: GraphQLResolveInfo,
     ) => MaybePromise<Model['Shape'][]>;
+    totalCount?: (
+      parent: ParentShape,
+      args: InputShapeFromFields<Args> & PothosSchemaTypes.DefaultConnectionArguments,
+      context: Types['Context'],
+      info: GraphQLResolveInfo,
+    ) => MaybePromise<number>;
   };
 
 export type RelatedConnectionOptions<
@@ -472,31 +520,36 @@ export type RelatedConnectionOptions<
 
 export type WithBrand<T> = T & { [typeBrandKey]: string };
 
-export type IncludeMap = Record<string, Record<string, unknown> | boolean>;
+export type IncludeMap = Record<string, SelectionMap | boolean>;
 
-export interface IncludeCounts {
-  current: Record<string, boolean>;
-  parent: Record<string, boolean>;
+export interface SelectionMap {
+  select?: Record<string, SelectionMap | boolean>;
+  include?: Record<string, SelectionMap | boolean>;
+  where?: {};
 }
 
+export type FieldSelection =
+  | Record<string, SelectionMap | boolean>
+  | ((
+      args: object,
+      context: object,
+      query: (
+        selection: SelectionMap | boolean | ((args: object, context: object) => SelectionMap),
+      ) => SelectionMap | boolean,
+    ) => SelectionMap);
+
 export type LoaderMappings = Record<
   string,
   {
     field: string;
-    alias?: string;
     mappings: LoaderMappings;
     indirectPath: string[];
-  }[]
+  }
 >;
 
-export interface SubFieldInclude {
-  type?: string;
-  name: string;
-}
-
-export interface IndirectLoadMap {
-  subFields: SubFieldInclude[];
-  path: string[];
+export interface IndirectInclude {
+  getType: () => string;
+  path: { type?: string; name: string }[];
 }
 
 export type ShapeFromConnection<T> = T extends { shape: unknown } ? T['shape'] : never;

package/src/{cursors.ts → util/cursors.ts}

@@ -75,7 +75,7 @@ export function parseCompositeCursor(fields: string[]) {
   };
 }
 
-interface PrismaCursorConnectionQueryOptions {
+export interface PrismaCursorConnectionQueryOptions {
   args: PothosSchemaTypes.DefaultConnectionArguments;
   defaultSize?: number;
   maxSize?: number;
@@ -84,7 +84,7 @@ interface PrismaCursorConnectionQueryOptions {
 
 interface ResolvePrismaCursorConnectionOptions extends PrismaCursorConnectionQueryOptions {
   query: {};
-  totalCount?: number;
+  totalCount?: number | (() => MaybePromise<number>);
 }
 
 export function prismaCursorConnectionQuery({
@@ -139,7 +139,7 @@ export function wrapConnectionResult<T extends {}>(
   args: PothosSchemaTypes.DefaultConnectionArguments,
   take: number,
   cursor: (node: T) => string,
-  totalCount?: number,
+  totalCount?: number | (() => MaybePromise<number>),
 ) {
   const gotFullResults = results.length === Math.abs(take);
   const hasNextPage = args.before ? true : gotFullResults;

package/src/{refs.ts → util/datamodel.ts}

@@ -1,9 +1,8 @@
 import { ObjectRef, SchemaTypes } from '@pothos/core';
-import { Prisma } from '../tests/client';
+import { Prisma } from '../../tests/client';
+import { PrismaObjectRef } from '../object-ref';
+import { PrismaDelegate, PrismaModelTypes } from '../types';
 import { formatCursor, parseCompositeCursor, parseRawCursor } from './cursors';
-import { PrismaObjectRef } from './object-ref';
-import { PrismaDelegate } from './types';
-import { PrismaModelTypes } from '.';
 
 export const refMap = new WeakMap<object, Map<string, PrismaObjectRef<PrismaModelTypes>>>();
 export const findUniqueMap = new WeakMap<
@@ -32,36 +31,6 @@ export function getRefFromModel<Types extends SchemaTypes>(
   return cache.get(name)!;
 }
 
-export function getFindUniqueForRef<Types extends SchemaTypes>(
-  ref: ObjectRef<unknown>,
-  builder: PothosSchemaTypes.SchemaBuilder<Types>,
-) {
-  if (!findUniqueMap.has(builder)) {
-    findUniqueMap.set(builder, new Map());
-  }
-  const cache = findUniqueMap.get(builder)!;
-
-  if (!cache.has(ref)) {
-    return null;
-  }
-
-  return cache.get(ref)! as (args: unknown, context: Types['Context']) => unknown;
-}
-
-export function setFindUniqueForRef<Types extends SchemaTypes>(
-  ref: ObjectRef<unknown>,
-  builder: PothosSchemaTypes.SchemaBuilder<Types>,
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  findUnique: ((args: any, context: Types['Context']) => unknown) | null,
-) {
-  if (!findUniqueMap.has(builder)) {
-    findUniqueMap.set(builder, new Map());
-  }
-  const cache = findUniqueMap.get(builder)!;
-
-  cache.set(ref, findUnique);
-}
-
 export function getRelation<Types extends SchemaTypes>(
   name: string,
   builder: PothosSchemaTypes.SchemaBuilder<Types>,
@@ -126,16 +95,6 @@ export function getCursorParser<Types extends SchemaTypes>(
   });
 }
 
-export function getRelatedDelegate<Types extends SchemaTypes>(
-  name: string,
-  builder: PothosSchemaTypes.SchemaBuilder<Types>,
-  relation: string,
-) {
-  const fieldData = getRelation(name, builder, relation);
-
-  return fieldData.type;
-}
-
 export function getDelegateFromModel(client: Record<string, unknown>, model: string) {
   const lowerCase = `${model.slice(0, 1).toLowerCase()}${model.slice(1)}`;
 

package/src/util/deep-equal.ts

@@ -0,0 +1,51 @@
+/* eslint-disable no-continue */
+export function deepEqual(left: unknown, right: unknown, ignore?: Set<string>) {
+  if (left === right) {
+    return true;
+  }
+
+  if (left && right && typeof left === 'object' && typeof right === 'object') {
+    if (Array.isArray(left)) {
+      if (!Array.isArray(right)) {
+        return false;
+      }
+
+      const { length } = left;
+
+      if (right.length !== length) {
+        return false;
+      }
+
+      for (let i = 0; i < length; i += 1) {
+        if (!deepEqual(left[i], right[i])) {
+          return false;
+        }
+      }
+
+      return true;
+    }
+
+    const keys = Object.keys(left);
+    const keyLength = keys.length;
+
+    if (keyLength !== Object.keys(right).length) {
+      return false;
+    }
+
+    for (const key of keys) {
+      if (ignore?.has(key)) {
+        continue;
+      }
+
+      if (
+        !deepEqual((left as Record<string, unknown>)[key], (right as Record<string, unknown>)[key])
+      ) {
+        return false;
+      }
+    }
+
+    return true;
+  }
+
+  return false;
+}

package/src/{loader-map.ts → util/loader-map.ts}

@@ -1,10 +1,11 @@
-import { GraphQLResolveInfo } from 'graphql';
+import { GraphQLNamedType, GraphQLResolveInfo } from 'graphql';
 import { createContextCache } from '@pothos/core';
-import { LoaderMappings } from './types';
+import { LoaderMappings } from '../types';
+import { getIndirectType } from './map-query';
 
 const cache = createContextCache((ctx) => new Map<string, LoaderMappings>());
 
-export function cacheKey(path: GraphQLResolveInfo['path'], subPath: string[]) {
+export function cacheKey(type: string, path: GraphQLResolveInfo['path'], subPath: string[]) {
   let key = '';
   let current: GraphQLResolveInfo['path'] | undefined = path;
 
@@ -19,30 +20,29 @@ export function cacheKey(path: GraphQLResolveInfo['path'], subPath: string[]) {
     key = `${key}.${entry}`;
   }
 
-  return key;
+  return `${type}@${key}`;
 }
 
 export function setLoaderMappings(
   ctx: object,
-  path: GraphQLResolveInfo['path'],
+  info: GraphQLResolveInfo,
   value: LoaderMappings,
+  type: GraphQLNamedType,
 ) {
   Object.keys(value).forEach((field) => {
     const map = cache(ctx);
 
-    for (const mapping of value[field]) {
-      const selectionName = mapping.alias ?? mapping.field;
-      const subPath = [...mapping.indirectPath, selectionName];
-      const key = cacheKey(path, subPath);
+    const mapping = value[field];
+    const subPath = [...mapping.indirectPath, field];
+    const key = cacheKey(getIndirectType(type, info).name, info.path, subPath);
 
-      map.set(key, mapping.mappings);
-    }
+    map.set(key, mapping.mappings);
   });
 }
 
-export function getLoaderMapping(ctx: object, path: GraphQLResolveInfo['path']) {
+export function getLoaderMapping(ctx: object, path: GraphQLResolveInfo['path'], type: string) {
   const map = cache(ctx);
-  const key = cacheKey(path, []);
+  const key = cacheKey(type, path, []);
 
   return map.get(key) ?? null;
 }