let Nokogiri parse only the necessary markup (#74)

Merge pull request 74
jekyll · May 2, 2018 · f5e2e08 · f5e2e08
1 parent 017a162
commit f5e2e08
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/lib/jemoji.rb b/lib/jemoji.rb
@@ -9,6 +9,7 @@ class Emoji
     GITHUB_DOT_COM_ASSET_HOST_URL = "https://assets-cdn.github.com"
     ASSET_PATH = "/images/icons/"
     BODY_START_TAG = "<body"
+    OPENING_BODY_TAG_REGEX = %r!<body(.*)>\s*!
 
     class << self
       def emojify(doc)
@@ -79,11 +80,11 @@ def default_asset_root
       end
 
       def replace_document_body(doc)
-        src           = emoji_src(doc.site.config)
-        parsed_doc    = Nokogiri::HTML::Document.parse(doc.output)
-        body          = parsed_doc.at_css("body")
-        body.children = filter_with_emoji(src).call(body.inner_html)[:output].to_s
-        parsed_doc.to_html
+        src                 = emoji_src(doc.site.config)
+        head, opener, tail  = doc.output.partition(OPENING_BODY_TAG_REGEX)
+        body_content, *rest = tail.partition("</body>")
+        processed_markup    = filter_with_emoji(src).call(body_content)[:output].to_s
+        String.new(head) << opener << processed_markup << rest.join
       end
     end
   end

diff --git a/spec/fixtures/index.html b/spec/fixtures/index.html
@@ -1,7 +1,6 @@
 <!DOCTYPE HTML>
 <html lang="en-US">
   <head>
-<meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
     <meta charset="UTF-8">
     <title>Jemoji</title>
     <meta name="viewport" content="width=device-width,initial-scale=1">