Enable filtering on linked nodes (gatsbyjs#3600)

Add support to filter on linked nodes. Close gatsbyjs#3190
jastack · Jan 19, 2018 · 7120e5a · 7120e5a
1 parent 96a418a
commit 7120e5a
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 2 deletions.
diff --git a/packages/gatsby/src/schema/__tests__/infer-graphql-input-type-test.js b/packages/gatsby/src/schema/__tests__/infer-graphql-input-type-test.js
@@ -482,4 +482,33 @@ describe(`GraphQL Input args`, () => {
 
     expect(result).toMatchSnapshot()
   })
+
+  it(`filters on linked fields`, async () => {
+    const { store, getNodes } = require(`../../redux`)
+    const linkedNodeType = new GraphQLObjectType({
+      name: `Bar`,
+      fields: {
+        id: { type: GraphQLString },
+      },
+    })
+    let types = [{ name: `Bar`, nodeObjectType: linkedNodeType }]
+
+    store.dispatch({
+      type: `CREATE_NODE`,
+      payload: { id: `baz`, internal: { type: `Bar` } },
+    })
+
+    let result = await queryResult(
+      [...getNodes(), { linked___NODE: `baz`, foo: `bar` }],
+      `
+        {
+          allNode(filter: { linked: { id: { eq: "baz" } } }) {
+            edges { node { linked { id } } }
+          }
+        }
+      `,
+      { types }
+    )
+    expect(result.data.allNode.edges[0].node.linked.id).toEqual(`baz`)
+  })
 })
diff --git a/packages/gatsby/src/schema/infer-graphql-input-fields.js b/packages/gatsby/src/schema/infer-graphql-input-fields.js
@@ -19,6 +19,8 @@ const {
   isEmptyObjectOrArray,
 } = require(`./data-tree-utils`)
 
+const { findLinkedNode } = require(`./infer-graphql-type`)
+
 import type {
   GraphQLInputFieldConfig,
   GraphQLInputFieldConfigMap,
@@ -202,7 +204,10 @@ export function inferInputObjectStructureFromNodes({
     if (isRoot && EXCLUDE_KEYS[key]) return
 
     // Input arguments on linked fields aren't currently supported
-    if (_.includes(key, `___NODE`)) return
+    if (_.includes(key, `___NODE`)) {
+      value = findLinkedNode(value)
+      ;[key] = key.split(`___`)
+    }
 
     let field = inferGraphQLInputFields({
       nodes,

diff --git a/packages/gatsby/src/schema/infer-graphql-type.js b/packages/gatsby/src/schema/infer-graphql-type.js
@@ -252,7 +252,7 @@ function inferFromMapping(
   }
 }
 
-function findLinkedNode(value, linkedField, path) {
+export function findLinkedNode(value, linkedField, path) {
   let linkedNode
   // If the field doesn't link to the id, use that for searching.
   if (linkedField) {