Fix html stripping related to preview mode (tmcw#70)

* Don't strip top level html * Fix html stripping related to preview mode
rehan-dckap · Jan 28, 2018 · 4999028 · 4999028
1 parent 3f9a511
commit 4999028
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/src/components/content.js b/src/components/content.js
@@ -53,16 +53,8 @@ function chunkifyAST(ast, language) {
       } else if (node.type === 'heading' && node.depth < 4 && !title) {
         title = node.children[0].value;
         left.push(node);
-      } else if (node.type === 'html') {
-        if (node.value.indexOf('<!--') === 0) {
-          var content = node.value
-            .replace(/^<!--/, '')
-            .replace(/-->$/, '')
-            .trim();
-          if (content === 'preview') {
-            preview = true;
-          }
-        }
+      } else if (node.type === 'html' && node.value.match(/^<!--\s*preview\s*-->$/)) {
+        preview = true;
       } else {
         left.push(node);
       }