Parse and stringify comments

Author: pconerly

lib/parse-tag.js

@@ -38,6 +38,18 @@ module.exports = function (tag) {
             res.voidElement = true;
         }
 
+        if (res.name === '!--') { // comment tag
+            res = {
+                type: 'comment',
+            };
+            res.comment = '';
+            var end = tag.indexOf('-->');
+            if (end !== -1) {
+                var comment = tag.slice(4, end);
+                res.comment = comment;
+            }
+            return res;
+        }
     }
 
     var reg = new RegExp(attrRE);

lib/stringify.js

@@ -19,6 +19,9 @@ function stringify(buff, doc) {
             return buff;
         }
         return buff + doc.children.reduce(stringify, '') + '</' + doc.name + '>';
+    case 'comment':
+        buff += '<!--' + doc.comment + '-->';
+        return buff;
     }
 }
 

test/parse.js

@@ -52,6 +52,39 @@ test('parse', function (t) {
     }]);
     t.equal(html, HTML.stringify(parsed));
 
+    html = '<!-- just a comment node -->';
+    parsed = HTML.parse(html);
+    t.deepEqual(parsed, [{
+      type: 'comment',
+      comment: ' just a comment node '
+    }]);
+    t.equal(html, HTML.stringify(parsed));
+
+    html = '<div><h2>Comment below this header</h2><!-- just a comment node --></div>';
+    parsed = HTML.parse(html);
+    t.deepEqual(parsed, [{
+        name: 'div',
+        type: 'tag',
+        attrs: {},
+        voidElement: false,
+        children: [{
+            attrs: {},
+            name: 'h2',
+            type: 'tag',
+            voidElement: false,
+            children: [{
+                content: 'Comment below this header',
+                type: 'text'
+            }]
+        },
+        {
+            type: 'comment' ,
+            comment: ' just a comment node ',
+        }]
+    }]);
+    t.equal(html, HTML.stringify(parsed));
+
+
     html = '<div>oh <strong>hello</strong> there! How are <span>you</span>?</div>';
     parsed = HTML.parse(html);