Cleaned up gatsby-remark-prismjs hack a little

gatsbyjs · Nov 24, 2017 · bded111 · bded111
1 parent 2f350f6
commit bded111
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/packages/gatsby-remark-embed-snippet/package.json b/packages/gatsby-remark-embed-snippet/package.json
@@ -5,6 +5,7 @@
   "author": "Brian Vaughn <brian.david.vaughn@gmail.com>",
   "dependencies": {
     "babel-runtime": "^6.26.0",
+    "gatsby-remark-prismjs": "^1.2.9",
     "normalize-path": "^2.1.1",
     "unist-util-map": "^1.0.3"
   },

diff --git a/packages/gatsby-remark-embed-snippet/src/index.js b/packages/gatsby-remark-embed-snippet/src/index.js
@@ -4,9 +4,8 @@ const fs = require(`fs`)
 const normalizePath = require(`normalize-path`)
 const visit = require(`unist-util-visit`)
 
-// TODO: This is a terrible hack!
-// Let's find a real way to share this utility code between these 2 plug-ins?
-const highlightCode = require(`../../gatsby-remark-prismjs/src/highlight-code`)
+// HACK: It would be nice to find a better way to share this utility code.
+const highlightCode = require(`gatsby-remark-prismjs/highlight-code`)
 
 // Language defaults to extension.toLowerCase();
 // This map tracks languages that don't match their extension.
@@ -88,6 +87,7 @@ module.exports = (
           return line
         })
         .join('\n')
+        .trim()
 
       // PrismJS's theme styles are targeting pre[class*="language-"]
       // to apply its styles. We do the same here so that users