Move code to index file

Author: chrisg86

index.js

@@ -1,6 +1,14 @@
 const visit = require('unist-util-visit');
 
-const applyClassesToNode = require('./src/applyClassesToNode');
+const applyClassesToNode = (node, classes) => {
+  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;
+};
 
 module.exports = ({ markdownAST }, { classMap = {} }) => {
   // @see: https://github.com/syntax-tree/mdast#nodes

package.json

@@ -1,6 +1,6 @@
 {
   "name": "gatsby-remark-classes",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Automatically add class attributes to rendered markdown elements",
   "main": "index.js",
   "keywords": [