@pothos/plugin-prisma 3.42.0 → 3.44.0

package/src/types.ts

@@ -49,7 +49,8 @@ export interface PrismaModelTypes {
     string,
     {
       Shape: unknown;
-      Types: PrismaModelTypes;
+      Name: string;
+      // Types: PrismaModelTypes;
     }
   >;
 }
@@ -78,6 +79,7 @@ export type PrismaObjectFieldOptions<
     ? ParentShape
     : ParentShape &
         ShapeFromSelection<
+          Types,
           ExtractModel<Types, ParentShape>,
           { select: Select extends (...args: any[]) => infer S ? S : Select }
         >,
@@ -123,12 +125,16 @@ interface BaseSelection {
 
 export type SelectedKeys<T> = { [K in keyof T]: T[K] extends false ? never : K }[keyof T];
 
-export type ShapeFromSelection<Model extends PrismaModelTypes, Selection> = Normalize<
+export type ShapeFromSelection<
+  Types extends SchemaTypes,
+  Model extends PrismaModelTypes,
+  Selection,
+> = Normalize<
   Selection extends BaseSelection
     ? unknown extends Selection['select']
-      ? Model['Shape'] & RelationShapeFromInclude<Model, Selection['include']>
+      ? Model['Shape'] & RelationShapeFromInclude<Types, Model, Selection['include']>
       : Pick<Model['Shape'], SelectedKeys<Selection['select']>> &
-          RelationShapeFromInclude<Model, Selection['select']> &
+          RelationShapeFromInclude<Types, Model, Selection['select']> &
           ('_count' extends keyof Selection['select']
             ? ShapeFromCount<Selection['select']['_count']>
             : {})
@@ -141,13 +147,27 @@ export type ShapeFromCount<Selection> = Selection extends true
   ? { _count: { [K in keyof Counts]: number } }
   : never;
 
-type RelationShapeFromInclude<Model extends PrismaModelTypes, Include> = Normalize<{
+export type TypesForRelation<
+  Types extends SchemaTypes,
+  Model extends PrismaModelTypes,
+  Relation extends keyof Model['Relations'],
+> = Model['Relations'][Relation]['Name'] extends infer Name
+  ? Name extends keyof Types['PrismaTypes']
+    ? Types['PrismaTypes'][Name] & PrismaModelTypes
+    : never
+  : never;
+
+type RelationShapeFromInclude<
+  Types extends SchemaTypes,
+  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]>
+      ? ShapeFromSelection<Types, TypesForRelation<Types, Model, K>, Include[K]>[]
+      : ShapeFromSelection<Types, TypesForRelation<Types, Model, K>, Include[K]>
     : unknown;
 }>;
 
@@ -372,7 +392,7 @@ export type RelatedFieldOptions<
         >;
       }) & {
     description?: string | false;
-    type?: PrismaObjectRef<Model['Relations'][Field]['Types']>;
+    type?: PrismaObjectRef<TypesForRelation<Types, Model, Field>>;
     query?: QueryForField<Types, Args, Model['Include'][Field & keyof Model['Include']]>;
   };
 
@@ -643,7 +663,7 @@ export type RelatedConnectionOptions<
     ? {
         description?: string | false;
         query?: QueryForField<Types, Args, Model['Include'][Field & keyof Model['Include']]>;
-        type?: PrismaObjectRef<Model['Relations'][Field]['Types']>;
+        type?: PrismaObjectRef<TypesForRelation<Types, Model, Field>>;
         cursor: CursorFromRelation<Model, Field>;
         defaultSize?: number | ((args: ConnectionArgs, ctx: Types['Context']) => number);
         maxSize?: number | ((args: ConnectionArgs, ctx: Types['Context']) => number);