Fixed a bug where totalCount was not selected correctly when selected…

… through a fragment
hayes · May 16, 2023 · e8139e7 · e8139e7 · vercel · May 16, 2023
1 parent 9d968a6
commit e8139e7
Show file tree

Hide file tree

Showing 3 changed files with 226 additions and 94 deletions.
diff --git a/.changeset/lemon-oranges-relate.md b/.changeset/lemon-oranges-relate.md
@@ -0,0 +1,5 @@
+---
+'@pothos/plugin-prisma': patch
+---
+
+Fixed a bug where totalCount was not selected correctly when selected through a fragment
diff --git a/packages/plugin-prisma/src/util/map-query.ts b/packages/plugin-prisma/src/util/map-query.ts
@@ -134,9 +134,12 @@ function resolveIndirectInclude(
         }
         continue;
       case Kind.FRAGMENT_SPREAD:
-        if (info.fragments[sel.name.value].typeCondition.name.value === include.type) {
+        if (
+          !include.type ||
+          info.fragments[sel.name.value].typeCondition.name.value === include.type
+        ) {
           resolveIndirectInclude(
-            info.schema.getType(include.type)!,
+            include.type ? info.schema.getType(include.type)! : type,
             info,
             info.fragments[sel.name.value],
             includePath,
@@ -148,7 +151,7 @@ function resolveIndirectInclude(
         continue;
 
       case Kind.INLINE_FRAGMENT:
-        if (!sel.typeCondition || sel.typeCondition.name.value === include.type) {
+        if (!sel.typeCondition || !include.type || sel.typeCondition.name.value === include.type) {
           resolveIndirectInclude(
             sel.typeCondition ? info.schema.getType(sel.typeCondition.name.value)! : type,
             info,