lib/jekyll-diagrams/erd.rb in jekyll-diagrams-0.7.0 vs lib/jekyll-diagrams/erd.rb in jekyll-diagrams-0.7.1

@@ -1,21 +1,25 @@
 module Jekyll
     module Diagrams
       class ErdBlock < Block
+        CONFIGURATIONS = %w(config edge).freeze
+        
         def render_svg(code, config)
-          render_with_stdin_stdout(build_command(config), code)
+          svg = render_with_stdin_stdout(build_command(config), code)
+          svg.sub!(/^<\?xml(([^>]|\n)*>\n?){2}/, '')
         end
   
         def build_command(config)
           command = 'erd --fmt=svg'
+          command << ' --dot-entity' if config.has_key?('dot-entity')
 
-          %w(config edge).each do |conf|
+          CONFIGURATIONS.each do |conf|
             command << " --#{conf}=#{config[conf]}" if config.has_key?(conf)
           end
 
           command
         end
       end
     end
   end
   
-  Liquid::Template.register_tag(:erb, Jekyll::Diagrams::ErdBlock)
+  Liquid::Template.register_tag(:erd, Jekyll::Diagrams::ErdBlock)
\ No newline at end of file