Wrap the dot output in a div for styling, and only remove the first p…

…olygon (the background) but not the rest.
plexus · Jun 7, 2013 · 44d48f6 · 44d48f6
1 parent a237234
commit 44d48f6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/impress_renderer.rb b/lib/impress_renderer.rb
@@ -55,7 +55,9 @@ def block_code code, lang
       file = Tempfile.new(['mdpress','.dot'])
       file << code
       file.close
-      (Nokogiri(`dot #{file.path} -Tsvg`)/'svg').tap{|svg| svg.search('polygon').remove}.to_html
+      "<div class='dot-wrap'>" +
+        (Nokogiri(`dot #{file.path} -Tsvg`)/'svg').tap{|svg| svg.search('polygon').first.remove}.to_html +
+        "</div>"
     elsif lang == 'notes'
       "<div class='notes' style='display:none;'>#{code}</div>"
     else