gatsby-node.js

const path = require('path');
const { prop } = require('./scripts/utils/functional.js');
const externalRedirects = require('./src/data/external-redirects.json');
const { createFilePath } = require('gatsby-source-filesystem');
const createSingleNav = require('./scripts/createSingleNav');
const generateTOC = require('mdast-util-toc');
// are needed for our tableOfContents override
const genMDX = require('gatsby-plugin-mdx/utils/gen-mdx.js');
const defaultOptions = require('gatsby-plugin-mdx/utils/default-options.js');
const getTableOfContents = require('gatsby-plugin-mdx/utils/get-table-of-content.js');

const TEMPLATE_DIR = 'src/templates/';
const TRAILING_SLASH = /\/$/;
const releaseNotesPerAgent = {};

const hasOwnProperty = (obj, key) =>
  Object.prototype.hasOwnProperty.call(obj, key);

const hasTrailingSlash = (pathname) =>
  pathname === '/' ? false : TRAILING_SLASH.test(pathname);

const appendTrailingSlash = (pathname) =>
  pathname.endsWith('/') ? pathname : `${pathname}/`;

exports.onPreBootstrap = () => {
  createSingleNav();
};

exports.onCreateWebpackConfig = ({ actions }) => {
  actions.setWebpackConfig({
    resolve: {
      fallback: {
        http: false,
        https: false,
        zlib: false,
      },
      alias: {
        images: path.resolve(__dirname, 'src/images/'),
      },
    },
  });
};

exports.onCreateNode = ({ node, getNode, actions }) => {
  const { createNodeField } = actions;

  if (
    node.internal.type === 'Mdx' ||
    (node.internal.type === 'MarkdownRemark' &&
      node.fileAbsolutePath.includes('src/content'))
  ) {
    createNodeField({
      node,
      name: 'slug',
      value: createFilePath({ node, getNode, trailingSlash: false }),
    });
  }
};

exports.createPages = async ({ actions, graphql, reporter }) => {
  const { createPage, createRedirect } = actions;

  const { data, errors } = await graphql(`
    query {
      allMarkdownRemark(
        filter: { fileAbsolutePath: { regex: "/src/content/" } }
      ) {
        edges {
          node {
            frontmatter {
              type
            }
            fields {
              fileRelativePath
              slug
            }
          }
        }
      }

      whatsNewPosts: allMarkdownRemark(
        filter: { fileAbsolutePath: { regex: "/src/content/whats-new/" } }
      ) {
        nodes {
          fields {
            slug
          }
        }
      }

      allMdx(filter: { fileAbsolutePath: { regex: "/src/content/" } }) {
        edges {
          node {
            fields {
              fileRelativePath
              slug
            }
            frontmatter {
              type
              subject
              redirects
              hideNavs
            }
          }
        }
      }

      allI18nMdx: allMdx(
        filter: { fileAbsolutePath: { regex: "/src/i18n/content/" } }
      ) {
        edges {
          node {
            fields {
              fileRelativePath
              slug
            }
            frontmatter {
              type
              subject
              translationType
            }
          }
        }
      }

      releaseNotes: allMdx(
        filter: {
          fileAbsolutePath: {
            regex: "/src/content/docs/release-notes/.*(?<!index).mdx/"
          }
        }
        sort: { fields: frontmatter___releaseDate, order: DESC }
      ) {
        group(limit: 1, field: frontmatter___subject) {
          fieldValue
          nodes {
            frontmatter {
              releaseDate
            }
            fields {
              slug
            }
          }
          totalCount
        }
      }

      landingPagesReleaseNotes: allMdx(
        filter: {
          fileAbsolutePath: { regex: "/docs/release-notes/.*/index.mdx$/" }
        }
      ) {
        nodes {
          fields {
            slug
          }
          frontmatter {
            subject
            redirects
          }
        }
      }

      allLocale {
        nodes {
          locale
          isDefault
        }
      }

      allInstallConfig {
        edges {
          node {
            redirects
            agentName
          }
        }
      }
    }
  `);

  if (errors) {
    reporter.panicOnBuild(`Error while running GraphQL query.`);
    return;
  }

  const {
    allI18nMdx,
    allMarkdownRemark,
    allMdx,
    releaseNotes,
    landingPagesReleaseNotes,
    allLocale,
    allInstallConfig,
    whatsNewPosts,
  } = data;

  const locales = allLocale.nodes
    .filter((locale) => !locale.isDefault)
    .map(prop('locale'));

  externalRedirects.forEach(({ url, paths }) => {
    paths.forEach((path) => {
      createRedirect({
        fromPath: path,
        toPath: url,
        isPermanent: true,
        redirectInBrowser: true,
      });
    });
  });

  allInstallConfig.edges.forEach(({ node: { redirects, agentName } }) => {
    redirects?.length &&
      redirects.forEach((redirect) =>
        createLocalizedRedirect({
          locales,
          fromPath: redirect,
          toPath: `/install/${agentName}/`,
          createRedirect,
        })
      );
  });

  releaseNotes.group.forEach((el) => {
    const { fieldValue, nodes, totalCount } = el;

    const landingPage = landingPagesReleaseNotes.nodes.find(
      (node) => node.frontmatter.subject === fieldValue
    );

    if (landingPage) {
      releaseNotesPerAgent[landingPage.frontmatter.subject] = totalCount;
      const { redirects } = landingPage.frontmatter;

      createLocalizedRedirect({
        locales,
        fromPath: path.join(landingPage.fields.slug, 'current'),
        toPath: nodes[0].fields.slug,
        isPermanent: false,
        createRedirect,
      });

      if (redirects) {
        redirects.forEach((fromPath) => {
          createLocalizedRedirect({
            locales,
            fromPath,
            toPath: landingPage.fields.slug,
            isPermanent: false,
            createRedirect,
          });
        });
      }
    }
  });

  const translatedContentNodes = allI18nMdx.edges.map(({ node }) => node);

  allMdx.edges.concat(allMarkdownRemark.edges).forEach(({ node }) => {
    const {
      fields: { slug },
      frontmatter: { redirects },
    } = node;

    if (redirects) {
      redirects.forEach((fromPath) => {
        createLocalizedRedirect({
          locales,
          fromPath,
          toPath: slug,
          createRedirect,
        });
      });
    }

    createPageFromNode(node, { createPage });

    locales.forEach((locale) => {
      const i18nNode = translatedContentNodes.find(
        (i18nNode) =>
          i18nNode.fields.slug.replace(`/${locale}`, '') === node.fields.slug
      );

      createPageFromNode(
        i18nNode || node,
        {
          prefix: i18nNode ? '' : locale,
          createPage,
          disableSwiftype: !i18nNode,
        },
        true // enable DSG
      );
    });
  });

  whatsNewPosts.nodes.forEach((node) => {
    const {
      fields: { slug },
    } = node;

    createLocalizedRedirect({
      locales,
      fromPath: slug.replace(/\/\d{4}\/\d{2}/, ''),
      toPath: slug,
      createRedirect,
    });
  });

  // Redirect for VSU page to new Introduction to APM doc
  createRedirect({
    fromPath: '/docs/apm/new-relic-apm/getting-started/introduction-apm/',
    toPath: '/introduction-apm',
    isPermanent: false,
    redirectInBrowser: true,
  });
};

exports.createSchemaCustomization = (
  {
    getNode,
    getNodes,
    pathPrefix,
    reporter,
    cache,
    actions,
    schema,
    store,
    ...helpers
  },
  pluginOptions
) => {
  const { createTypes } = actions;

  const typeDefs = `
  type NavYaml implements Node @dontInfer {
    id: ID!
    title: String!
    path: String
    icon: String
    label: String
    filterable: Boolean!
    pages: [NavYaml!]!
    rootNav: Boolean!
  }
  type MarkdownRemark implements Node {
    frontmatter: Frontmatter
  }
  type Mdx implements Node {
    frontmatter: Frontmatter
  }
  type Frontmatter {
    isFeatured: Boolean
    translationType: String
    dataSource: String
    hideNavs: Boolean
    eolDate: String
    downloadLink: String
    signupBanner: SignupBanner
    features: [String]
    bugs: [String]
    security: [String]
  }
  type SignupBanner {
    cta: String
    url: String
    text: String
  }

  `;

  // this was taken from gatsby-plugin-mdx/gatsby/create-schema-customization.js
  const options = defaultOptions(pluginOptions);

  const pendingPromises = new WeakMap();
  const processMDX = ({ node }) => {
    let promise = pendingPromises.get(node);
    if (!promise) {
      promise = genMDX({
        node,
        options,
        store,
        pathPrefix,
        getNode,
        getNodes,
        cache,
        reporter,
        actions,
        schema,
        ...helpers,
      });
      pendingPromises.set(node, promise);
      promise.then(() => {
        pendingPromises.delete(node);
      });
    }
    return promise;
  };

  const tocExtension = schema.buildObjectType({
    name: `Mdx`,
    fields: {
      tableOfContents: {
        type: `JSON`,
        args: {
          maxDepth: {
            type: `Int`,
            default: 6,
          },
        },
        async resolve(mdxNode, { maxDepth }) {
          const { mdast } = await processMDX({ node: mdxNode });
          const toc = generateTOC(mdast, {
            maxDepth,

            // override gatsby-plugin-mdx tableOfContents options
            // to allow Step headers to show in PageTools
            parents: [
              (node) => node.type === 'mdxBlockElement' && node.name === 'Step',
              'root',
            ],
          });

          return getTableOfContents(toc.map, {});
        },
      },
    },
  });
  createTypes([typeDefs, tocExtension]);
};

exports.createResolvers = ({ createResolvers }) => {
  createResolvers({
    NavYaml: {
      pages: {
        resolve: (source) => {
          return source.pages || [];
        },
      },
      filterable: {
        resolve: (source) =>
          hasOwnProperty(source, 'filterable') ? source.filterable : true,
      },
      rootNav: {
        resolve: (source) =>
          hasOwnProperty(source, 'rootNav') ? source.rootNav : false,
      },
    },
    Frontmatter: {
      isFeatured: {
        resolve: (source) =>
          hasOwnProperty(source, 'isFeatured') ? source.isFeatured : false,
      },
      translationType: {
        resolve: (source) =>
          hasOwnProperty(source, 'translationType')
            ? source.translationType
            : null,
      },
      dataSource: {
        resolve: (source) =>
          hasOwnProperty(source, 'dataSource') ? source.dataSource : null,
      },
      hideNavs: {
        resolve: (source) =>
          hasOwnProperty(source, 'hideNavs') ? source.hideNavs : null,
      },
      eolDate: {
        resolve: (source) =>
          hasOwnProperty(source, 'eolDate') ? source.eolDate : null,
      },
      downloadLink: {
        resolve: (source) =>
          hasOwnProperty(source, 'downloadLink') ? source.downloadLink : null,
      },
      features: {
        resolve: (source) =>
          hasOwnProperty(source, 'features') ? source.features : null,
      },
      bugs: {
        resolve: (source) =>
          hasOwnProperty(source, 'bugs') ? source.bugs : null,
      },
      security: {
        resolve: (source) =>
          hasOwnProperty(source, 'security') ? source.security : null,
      },
    },
    SignupBanner: {
      cta: {
        resolve: (source) =>
          hasOwnProperty(source, 'cta') ? source.cta : null,
      },
      url: {
        resolve: (source) =>
          hasOwnProperty(source, 'url') ? source.url : null,
      },
      text: {
        resolve: (source) =>
          hasOwnProperty(source, 'text') ? source.text : null,
      },
    },
  });
};

exports.onCreatePage = ({ page, actions }) => {
  const { createPage } = actions;

  if (page.path === '/') {
    page.context.quicklaunchSlug =
      'docs/new-relic-solutions/get-started/quick-launch-guide';
    page.context.layout = 'homepage';
  }
  if (page.path === '/jp/') {
    page.context.quicklaunchSlug =
      'jp/docs/new-relic-solutions/get-started/quick-launch-guide';
  }
  if (page.path === '/kr/') {
    page.context.quicklaunchSlug =
      'kr/docs/new-relic-solutions/get-started/quick-launch-guide';
  }

  if (page.path.match(/404/)) {
    page.context.layout = 'basic';
  }

  if (page.path.match(/404/) && page.path.match(/\/docs\//)) {
    page.context.layout = 'default';
  }

  if (page.path.includes('/install/')) {
    const pagePathArray = page.path.split('/');
    const lastItem = pagePathArray[pagePathArray.length - 1];
    page.context.agent =
      lastItem !== '' ? lastItem : pagePathArray[pagePathArray.length - 2];
  }

  if (hasTrailingSlash(page.context.slug)) {
    page.context.slug = page.context.slug.replace(TRAILING_SLASH, '');
  }

  createPage(page);
};

const createLocalizedRedirect = ({
  fromPath,
  toPath,
  locales,
  redirectInBrowser = true,
  isPermanent = true,
  createRedirect,
}) => {
  // Create redirects for paths with and without a trailing slash
  const pathWithTrailingSlash = hasTrailingSlash(fromPath)
    ? fromPath
    : path.join(fromPath, '/');

  createRedirect({
    fromPath: pathWithTrailingSlash,
    toPath: appendTrailingSlash(toPath),
    isPermanent,
    redirectInBrowser,
  });

  locales.forEach((locale) => {
    createRedirect({
      fromPath: path.join(`/${locale}`, pathWithTrailingSlash),
      toPath: appendTrailingSlash(path.join(`/${locale}`, toPath)),
      isPermanent,
      redirectInBrowser,
    });
  });
};

const createPageFromNode = (
  node,
  { createPage, prefix = '', disableSwiftype = false },
  defer = false
) => {
  const {
    frontmatter: { subject: agentName, hideNavs },
    fields: { fileRelativePath, slug },
  } = node;

  const { template, context = {} } = getTemplate(node);

  if (process.env.NODE_ENV === 'development' && !template) {
    createPage({
      path: path.join(prefix, slug, '/'),
      component: path.resolve(TEMPLATE_DIR, 'dev/missingTemplate.js'),
      context: {
        ...context,
        fileRelativePath,
        layout: 'basic',
      },
    });
  } else if (template === 'releaseNoteLandingPage') {
    const releaseNotes = releaseNotesPerAgent[agentName];
    const releaseNotesPerPage = 10;
    const numPages = Math.ceil(releaseNotes / releaseNotesPerPage);
    Array.from({ length: numPages }).forEach((_, i) => {
      createPage({
        path:
          i === 0
            ? path.join(prefix, slug, '/')
            : path.join(prefix, slug, `/${i + 1}/`),
        component: path.resolve(path.join(TEMPLATE_DIR, `${template}.js`)),
        context: {
          limit: releaseNotesPerPage,
          skip: i * releaseNotesPerPage,
          numPages,
          currentPage: i + 1,
          ...context,
          fileRelativePath,
          slug,
          slugRegex: `${slug}/.+/`,
          disableSwiftype,
        },
        defer,
      });
    });
  } else {
    createPage({
      path: path.join(prefix, slug, '/'),
      component: path.resolve(path.join(TEMPLATE_DIR, `${template}.js`)),
      context: {
        ...context,
        fileRelativePath,
        hideNavs,
        slug,
        slugRegex: `${slug}/.+/`,
        disableSwiftype,
      },
      defer,
    });
  }
};

const TEMPLATES_BY_TYPE = {
  landingPage: 'landingPage',
  apiDoc: 'docPage',
  releaseNote: 'releaseNote',
  troubleshooting: 'docPage',
};

const getTemplate = (node) => {
  const {
    frontmatter,
    fields: { fileRelativePath },
  } = node;

  switch (true) {
    case Boolean(frontmatter.type):
      return { template: TEMPLATES_BY_TYPE[frontmatter.type] };

    case /docs\/release-notes\/.*\/index.mdx$/.test(fileRelativePath):
      return {
        template: 'releaseNoteLandingPage',
        context: { subject: frontmatter.subject },
      };

    case fileRelativePath.includes('src/content/docs/release-notes'):
      return { template: 'releaseNote' };

    case fileRelativePath.includes('src/content/whats-new'):
      return { template: 'whatsNew' };

    default:
      return { template: 'docPage' };
  }
};