Use index.js

Author: chrisg86

index.js

@@ -0,0 +1,30 @@
+const visit = require('unist-util-visit');
+
+const applyClassesToNode = require('./src/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]);
+    }
+  });
+};

package.json

@@ -1,9 +1,12 @@
 {
   "name": "gatsby-remark-classes",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Automatically add class attributes to rendered markdown elements",
-  "main": "src/index.js",
-  "keywords": ["gatsby", "gatsby-plugin"],
+  "main": "index.js",
+  "keywords": [
+    "gatsby",
+    "gatsby-plugin"
+  ],
   "author": "Chris Greindl <[email protected]>",
   "license": "MIT",
   "dependencies": {
@@ -16,4 +19,4 @@
   "bugs": {
     "url": "https://github.com/chrisg86/gatsby-remark-classes/issues"
   }
-}
+}