Look through inline fragments to find selections

equalogic · Aug 20, 2020 · c155b22 · c155b22
1 parent 80b1592
commit c155b22
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/src/RelationMapper.ts b/src/RelationMapper.ts
@@ -157,6 +157,26 @@ export class RelationMapper {
   }
 
   private findFieldInSelection(fieldName: string, selectionSet: SelectionSetNode): SelectionNode | undefined {
-    return selectionSet.selections.find(selectionNode => this.getNameFromNode(selectionNode) === fieldName);
+    let foundNode: SelectionNode | undefined = undefined;
+
+    for (const selectionNode of selectionSet.selections) {
+      if (foundNode !== undefined) {
+        break;
+      }
+
+      if (selectionNode.kind === 'InlineFragment') {
+        foundNode = this.findFieldInSelection(fieldName, selectionNode.selectionSet);
+
+        break;
+      }
+
+      if (this.getNameFromNode(selectionNode) === fieldName) {
+        foundNode = selectionNode;
+
+        break;
+      }
+    }
+
+    return foundNode;
   }
 }