refactor: Extract helper functions to generate doc info for component…

…s/apis
newrelic · Jun 14, 2020 · 51558e5 · 51558e5
1 parent 535963a
commit 51558e5
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 19 deletions.
diff --git a/plugins/gatsby-source-newrelic-sdk/gatsby-node.js b/plugins/gatsby-source-newrelic-sdk/gatsby-node.js
@@ -1,4 +1,5 @@
-const loadSdk = require('./loadSdk');
+const loadSdk = require('./src/loadSdk');
+const { getComponentDoc, getApiDoc } = require('./src/docInfo');
 
 exports.sourceNodes = async (
   { actions, createNodeId, createContentDigest },
@@ -7,15 +8,10 @@ exports.sourceNodes = async (
   const { createNode } = actions;
   const sdk = await loadSdk(release);
 
-  components
-    .filter((name) => Boolean(sdk[name]))
-    .forEach((name) => {
-      const component = sdk[name];
-      const data = {
-        name,
-        description: component.__docs__.text,
-      };
+  components.forEach((name) => {
+    const data = getComponentDoc(name, sdk);
 
+    if (data) {
       createNode({
         ...data,
         id: createNodeId(`NewRelicSdkComponent-${name}`),
@@ -26,17 +22,13 @@ exports.sourceNodes = async (
           contentDigest: createContentDigest(data),
         },
       });
-    });
+    }
+  });
 
-  apis
-    .filter((name) => Boolean(sdk[name]))
-    .forEach((name) => {
-      const api = sdk[name];
-      const data = {
-        name,
-        description: api.__docs__.text,
-      };
+  apis.forEach((name) => {
+    const data = getApiDoc(name, sdk);
 
+    if (data) {
       createNode({
         ...data,
         id: createNodeId(`NewRelicSdkApi-${name}`),
@@ -47,5 +39,6 @@ exports.sourceNodes = async (
           contentDigest: createContentDigest(data),
         },
       });
-    });
+    }
+  });
 };
diff --git a/plugins/gatsby-source-newrelic-sdk/src/docInfo.js b/plugins/gatsby-source-newrelic-sdk/src/docInfo.js
@@ -0,0 +1,27 @@
+module.exports.getComponentDoc = (name, sdk) => {
+  const component = sdk[name];
+
+  if (!component) {
+    return null;
+  }
+
+  return {
+    name,
+    usage: `import { ${name} } from 'nr1'`,
+    description: component.__docs__.text,
+  };
+};
+
+module.exports.getApiDoc = (name, sdk) => {
+  const api = sdk[name];
+
+  if (!api) {
+    return null;
+  }
+
+  return {
+    name,
+    usage: `import { ${name} } from 'nr1'`,
+    description: api.__docs__.text,
+  };
+};