From 8eb0feaa8a2861a4f00c2246e041d37d33f528ca Mon Sep 17 00:00:00 2001
From: Brian Beck <exogen@gmail.com>
Date: Tue, 29 Nov 2016 23:53:01 -0800
Subject: [PATCH] Improve support for fragments

---
 docs/types.md           |  2 +-
 scripts/render-types.js |  2 +-
 src/resolvers.js        | 20 ++++++++++----------
 src/util.js             | 18 ++++++++++++++----
 4 files changed, 26 insertions(+), 16 deletions(-)

diff --git a/docs/types.md b/docs/types.md
index 89c7d3c..745d412 100644
--- a/docs/types.md
+++ b/docs/types.md
@@ -1,6 +1,6 @@
 # Schema Types
 
-You may also be interested in the [schema in GraphQL syntax](schema.md).
+You may also be interested in reading the [schema in GraphQL syntax](schema.md).
 
 <details><summary>**Table of Contents**</summary><p><ul>
   <li>[Query](#query)</li>
diff --git a/scripts/render-types.js b/scripts/render-types.js
index a5bd893..d9964f6 100644
--- a/scripts/render-types.js
+++ b/scripts/render-types.js
@@ -83,7 +83,7 @@ sortBy(interfaces, 'name')
 
 console.log('# Schema Types\n')
 
-console.log('You may also be interested in the [schema in GraphQL syntax](schema.md).\n')
+console.log('You may also be interested in reading the [schema in GraphQL syntax](schema.md).\n')
 
 console.log('<details><summary>**Table of Contents**</summary><p><ul>')
 console.log('  <li>[Query](#query)</li>')
diff --git a/src/resolvers.js b/src/resolvers.js
index a395ee4..2303d7c 100644
--- a/src/resolvers.js
+++ b/src/resolvers.js
@@ -6,16 +6,16 @@ import {
 } from 'graphql-relay'
 import { getFields, extendIncludes } from './util'
 
-export function includeRelationships (params, info) {
-  let fields = getFields(info)
+export function includeRelationships (params, info, fragments = info.fragments) {
+  let fields = getFields(info, fragments)
   if (info.fieldName !== 'relationships') {
     if (fields.relationships) {
-      fields = getFields(fields.relationships)
+      fields = getFields(fields.relationships, fragments)
     } else {
       if (fields.edges) {
-        fields = getFields(fields.edges)
+        fields = getFields(fields.edges, fragments)
         if (fields.node) {
-          return includeRelationships(params, fields.node)
+          return includeRelationships(params, fields.node, fragments)
         }
       }
       return params
@@ -36,13 +36,13 @@ export function includeRelationships (params, info) {
   return params
 }
 
-export function includeSubqueries (params, info) {
+export function includeSubqueries (params, info, fragments = info.fragments) {
   const subqueryIncludes = {
     aliases: 'aliases',
     artistCredit: 'artist-credits',
     tags: 'tags'
   }
-  let fields = getFields(info)
+  let fields = getFields(info, fragments)
   const include = []
   for (const key in subqueryIncludes) {
     if (fields[key]) {
@@ -55,9 +55,9 @@ export function includeSubqueries (params, info) {
     inc: extendIncludes(params.inc, include)
   }
   if (fields.edges) {
-    fields = getFields(fields.edges)
+    fields = getFields(fields.edges, fragments)
     if (fields.node) {
-      params = includeSubqueries(params, fields.node)
+      params = includeSubqueries(params, fields.node, fragments)
     }
   }
   return params
@@ -84,7 +84,7 @@ export function browseResolver () {
       offset: getOffsetWithDefault(after, -1) + 1
     }
     params = includeSubqueries(params, info)
-    params = includeRelationships(params, info)
+    params = includeRelationships(params, info, info.fragments)
     const formatValue = value => value.toLowerCase().replace(/ /g, '')
     params.type = params.type.map(formatValue)
     params.status = params.status.map(formatValue)
diff --git a/src/util.js b/src/util.js
index bfcc46e..1155a57 100644
--- a/src/util.js
+++ b/src/util.js
@@ -1,14 +1,24 @@
 import util from 'util'
 
-export function getFields (info) {
+export function getFields (info, fragments = info.fragments) {
   if (info.kind !== 'Field') {
     info = info.fieldNodes[0]
   }
   const selections = info.selectionSet.selections
-  return selections.reduce((fields, selection) => {
-    fields[selection.name.value] = selection
+  const reducer = (fields, selection) => {
+    if (selection.kind === 'FragmentSpread') {
+      const name = selection.name.value
+      const fragment = fragments[name]
+      if (!fragment) {
+        throw new Error(`Fragment '${name}' was not passed to getFields()`)
+      }
+      fragment.selectionSet.selections.reduce(reducer, fields)
+    } else {
+      fields[selection.name.value] = selection
+    }
     return fields
-  }, {})
+  }
+  return selections.reduce(reducer, {})
 }
 
 export function prettyPrint (obj, { depth = 5,