From 11664c827789535581a8eaa8491d9a251e2cfb7f Mon Sep 17 00:00:00 2001
From: Ilya Kreymer <ikreymer@gmail.com>
Date: Wed, 13 May 2015 19:26:03 -0700
Subject: [PATCH] html_rewriter: handle parse_comments by rewriting as html,
 not as js, should address ikreymer/pywb-webrecorder#7

---
 pywb/rewrite/html_rewriter.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/pywb/rewrite/html_rewriter.py b/pywb/rewrite/html_rewriter.py
index 9d1310b1..25f23d70 100644
--- a/pywb/rewrite/html_rewriter.py
+++ b/pywb/rewrite/html_rewriter.py
@@ -101,6 +101,7 @@ class HTMLRewriterMixin(object):
         self.head_insert = head_insert
         self.parse_comments = parse_comments
 
+        self.defmod = defmod
         self.rewrite_tags = self._init_rewrite_tags(defmod)
 
         # get opts from urlrewriter
@@ -391,7 +392,14 @@ class HTMLRewriter(HTMLRewriterMixin, HTMLParser):
     def handle_comment(self, data):
         self.out.write('<!--')
         if self.parse_comments:
-            data = self._rewrite_script(data)
+            #data = self._rewrite_script(data)
+
+            # Rewrite with seperate HTMLRewriter
+            comment_rewriter = HTMLRewriter(self.url_rewriter,
+                                            defmod=self.defmod)
+
+            data = comment_rewriter.rewrite(data)
+            data += comment_rewriter.close()
             self.out.write(data)
         else:
             self.parse_data(data)