Merge pull request #93 from yzhang-gh/patch-1

Fix #63 for vuepress plugin
shikijs · Sep 22, 2020 · 8b247a1 · 8b247a1
2 parents 586b741 + 6a4f696
commit 8b247a1
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/packages/vuepress-plugin/index.js b/packages/vuepress-plugin/index.js
@@ -14,10 +14,22 @@ module.exports = (options, ctx) => {
     chainMarkdown(config) {
       config.options.highlight((code, lang) => {
         if (!lang) {
-          return `<pre><code>${code}</code></pre>`
+          return `<pre><code>${escapeHtml(code)}</code></pre>`
         }
         return h.codeToHtml(code, lang)
       })
     }
   }
 }
+
+const htmlEscapes = {
+  '&': '&amp;',
+  '<': '&lt;',
+  '>': '&gt;',
+  '"': '&quot;',
+  "'": '&#39;'
+}
+
+function escapeHtml(html) {
+  return html.replace(/[&<>"']/g, chr => htmlEscapes[chr])
+}