Initial commit

Author: chrisg86

package-lock.json

@@ -0,0 +1,12 @@
+{
+  "name": "gatsby-remark-classes",
+  "version": "0.1.0",
+  "description": "",
+  "main": "src/index.js",
+  "keywords": ["gatsby"],
+  "author": "Chris Greindl <[email protected]>",
+  "license": "MIT",
+  "dependencies": {
+    "unist-util-visit": "^1.4.0"
+  }
+}

package.json

@@ -0,0 +1,11 @@
+// Format:
+// data: { hProperties: { className: [] } }
+module.exports.applyClassesToNode = () => {
+    node.data = node.data || {};
+    node.data.hProperties = node.data.hProperties || {};
+    node.data.hProperties.className = node.data.hProperties.className || [];
+
+    node.data.hProperties.className.push(classes);
+
+    return node;
+}

src/applyClassesToNode.js

@@ -0,0 +1,33 @@
+const visit = require('unist-util-visit');
+
+const applyClassesToNode = require('./applyClassesToNode');
+
+module.exports = (
+    { markdownAST },
+    { classMap = {} }
+) => {
+    // @see: https://github.com/syntax-tree/mdast#nodes
+    visit(markdownAST, `heading`, node => {
+        const selector = `h${node.depth}`;
+
+        if (selector in classMap) {
+            node = applyClassesToNode(node, classMap[selector]);
+        }
+    });
+
+    visit(markdownAST, `table`, node => {
+        const selector = `table`;
+
+        if (selector in classMap) {
+            node = applyClassesToNode(node, classMap[selector]);
+        }
+    });
+
+    visit(markdownAST, `paragraph`, node => {
+        const selector = `paragraph`;
+
+        if (selector in classMap) {
+            node = applyClassesToNode(node, classMap[selector]);
+        }
+    });
+}