[zmarkdown] Solve a bug on code visitor

zestedesavoir · Aug 31, 2020 · 04fdbd9 · 04fdbd9
1 parent 2a2ce52
commit 04fdbd9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/packages/zmarkdown/postprocessors/html-wrap-code.js b/packages/zmarkdown/postprocessors/html-wrap-code.js
@@ -20,7 +20,11 @@ module.exports = () => (tree) => {
 
       if (preElem.length === 1) {
         const classes = preElem[0].children[0].properties.className
+        if (!classes) return
+
         const langClass = classes.filter(c => c.startsWith('language'))
+        if (langClass.length === 0 || langClass[0].length < 9) return
+
         const language = langClass[0].substring(9)
 
         // rewrite class name