Adds gatsby-source-swiftype to theme

packages/gatsby-theme-newrelic/gatsby-config.js

@@ -1,4 +1,4 @@
-module.exports = ({ newrelic, robots = {}, gaTrackingId }) => {
+module.exports = ({ newrelic, swiftype, robots = {}, gaTrackingId }) => {
   return {
     plugins: [
       'gatsby-plugin-emotion',
@@ -25,6 +25,10 @@ module.exports = ({ newrelic, robots = {}, gaTrackingId }) => {
         resolve: 'gatsby-plugin-newrelic',
         options: newrelic,
       },
+      swiftype && {
+        resolve: 'gatsby-source-swiftype',
+        options: swiftype,
+      },
       {
         resolve: 'gatsby-source-filesystem',
         options: {

packages/gatsby-theme-newrelic/plugins/gatsby-source-swiftype/gatsby-node.js

@@ -0,0 +1,93 @@
+const fs = require('fs');
+const createRelatedResourceNode = require('./src/createRelatedResourceNode');
+const getRelatedResources = require('./src/getRelatedResources');
+
+const writeableData = {};
+
+exports.onPreBootstrap = (_, pluginOptions) => {
+  const { file } = pluginOptions;
+
+  if (!fs.existsSync(file)) {
+    fs.writeFileSync(file, '{}');
+  }
+};
+
+exports.onCreateNode = async (
+  { actions, node, getNodesByType, createNodeId, createContentDigest },
+  pluginOptions
+) => {
+  const { createNode, createParentChildLink } = actions;
+  const { filterNode = () => false, getPath } = pluginOptions;
+
+  if (node.internal.type !== 'Mdx' || !filterNode({ node })) {
+    return;
+  }
+
+  const [
+    {
+      siteMetadata: { siteUrl },
+    },
+  ] = getNodesByType('Site');
+
+  const pathname = getPath({ node });
+  const resources = await getRelatedResources({ node, siteUrl }, pluginOptions);
+
+  writeableData[pathname] = resources;
+
+  resources.forEach((resource) => {
+    const child = createRelatedResourceNode({
+      parent: node.id,
+      resource,
+      createContentDigest,
+      createNode,
+      createNodeId,
+    });
+
+    createParentChildLink({ parent: node, child: child });
+  });
+};
+
+exports.createSchemaCustomization = ({ actions }) => {
+  const { createTypes } = actions;
+
+  const typeDefs = `
+  type RelatedResource implements Node {
+    id: ID!
+    title: String!
+    url: String!
+  }
+  `;
+
+  createTypes(typeDefs);
+};
+
+exports.createResolvers = ({ createResolvers }) => {
+  createResolvers({
+    Mdx: {
+      relatedResources: {
+        args: {
+          limit: {
+            type: 'Int',
+            defaultValue: 5,
+          },
+        },
+        type: ['RelatedResource!'],
+        resolve: (source, args, context) => {
+          const { limit } = args;
+
+          return context.nodeModel
+            .getNodesByIds({ ids: source.children })
+            .slice(0, Math.max(limit, 0));
+        },
+      },
+    },
+  });
+};
+
+exports.onPostBootstrap = (_, pluginOptions) => {
+  const { refetch, file } = pluginOptions;
+
+  if (refetch) {
+    fs.writeFileSync(file, JSON.stringify(writeableData, null, 2));
+  }
+};

packages/gatsby-theme-newrelic/plugins/gatsby-source-swiftype/package.json

@@ -0,0 +1 @@
+{}

packages/gatsby-theme-newrelic/plugins/gatsby-source-swiftype/src/createRelatedResourceNode.js

@@ -0,0 +1,25 @@
+module.exports = ({
+  createNode,
+  createNodeId,
+  createContentDigest,
+  resource,
+  parent,
+}) => {
+  const node = {
+    id: createNodeId(`RelatedResource-${resource.url}`),
+    title: resource.title,
+    url: resource.url,
+    parent,
+    children: [],
+    plugin: 'gatsby-source-swiftype',
+    internal: {
+      type: 'RelatedResource',
+      content: JSON.stringify(resource),
+      contentDigest: createContentDigest(resource),
+    },
+  };
+
+  createNode(node);
+
+  return node;
+};

packages/gatsby-theme-newrelic/plugins/gatsby-source-swiftype/src/getRelatedResources.js

@@ -0,0 +1,26 @@
+const fs = require('fs');
+const search = require('./search');
+
+module.exports = async ({ node, siteUrl }, pluginOptions) => {
+  const {
+    refetch,
+    engineKey,
+    limit,
+    file,
+    getParams = () => ({}),
+    getPath,
+  } = pluginOptions;
+
+  const pathname = getPath({ node });
+
+  if (refetch) {
+    return search(siteUrl + pathname, getParams({ node }), {
+      engineKey,
+      limit,
+    });
+  }
+
+  const data = JSON.parse(fs.readFileSync(file));
+
+  return data[pathname] || [];
+};

packages/gatsby-theme-newrelic/plugins/gatsby-source-swiftype/src/search.js

@@ -0,0 +1,47 @@
+const fetch = require('node-fetch');
+const { appendTrailingSlash, stripTrailingSlash } = require('./utils/url');
+
+const normalizeUrl = (url) => {
+  const prefix = url.startsWith('!') ? '!' : '';
+  const plainUrl = url.replace(/^!/, '');
+
+  return [
+    prefix + appendTrailingSlash(plainUrl),
+    prefix + stripTrailingSlash(plainUrl),
+  ];
+};
+
+const uniq = (arr) => [...new Set(arr)];
+
+module.exports = async (url, params = {}, { engineKey, limit }) => {
+  const { page: pageFilters = {} } = params.filters || {};
+
+  const res = await fetch(
+    'https://search-api.swiftype.com/api/v1/public/engines/search.json',
+    {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+      },
+      body: JSON.stringify({
+        ...params,
+        engine_key: engineKey,
+        per_page: limit,
+        filters: {
+          ...params.filters,
+          page: {
+            ...pageFilters,
+            url: uniq([
+              ...normalizeUrl(`!${url}`),
+              ...(pageFilters.url || []).flatMap(normalizeUrl),
+            ]),
+          },
+        },
+      }),
+    }
+  );
+
+  const { records } = await res.json();
+
+  return records.page;
+};

packages/gatsby-theme-newrelic/plugins/gatsby-source-swiftype/src/utils/url.js

@@ -0,0 +1,21 @@
+const hasQueryParams = (urlString) => {
+  const url = new URL(urlString);
+
+  return Boolean(url.search);
+};
+
+exports.appendTrailingSlash = (url) => {
+  if (hasQueryParams(url)) {
+    return url;
+  }
+
+  return url.endsWith('/') ? url : `${url}/`;
+};
+
+exports.stripTrailingSlash = (url) => {
+  if (hasQueryParams(url)) {
+    return url;
+  }
+
+  return url.endsWith('/') ? url.replace(/\/$/, '') : url;
+};