lib/jekyll-diagrams/erd.rb in jekyll-diagrams-0.9.2 vs lib/jekyll-diagrams/erd.rb in jekyll-diagrams-0.9.3

@@ -3,20 +3,26 @@
 module Jekyll
   module Diagrams
     class ErdBlock < Block
       XML_REGEX = /^<\?xml(([^>]|\n)*>\n?){2}/.freeze
       CONFIGURATIONS = %w[config edge].freeze
+      SWITCHES = {
+        'dot-entity' => false
+      }.freeze
 
       def render_svg(code, config)
         command = build_command(config)
 
         svg = render_with_stdin_stdout(command, code)
         svg.sub!(XML_REGEX, '')
       end
 
       def build_command(config)
         command = +'erd --fmt=svg'
-        command << ' --dot-entity' if config.fetch('dot-entity', false) != false
+
+        SWITCHES.merge(config.slice(*SWITCHES.keys)).each do |switch, value|
+          command << " --#{switch}" if value != false
+        end
 
         CONFIGURATIONS.each do |conf|
           command << " --#{conf}=#{config[conf]}" if config.key?(conf)
         end