class Bundler::Graph::GraphVizClient

def g
  @g ||= ::GraphViz.digraph(@graph_name, :concentrate => true, :normalize => true, :nodesep => 0.55) do |g|
    g.edge[:weight]   = 2
    g.edge[:fontname] = g.node[:fontname] = "Arial, Helvetica, SansSerif"
    g.edge[:fontsize] = 12
  end
end

def initialize(graph_instance)
  @graph_name    = graph_instance.class::GRAPH_NAME
  @groups        = graph_instance.groups
  @relations     = graph_instance.relations
  @node_options  = graph_instance.node_options
  @edge_options  = graph_instance.edge_options
  @output_file   = graph_instance.output_file
  @output_format = graph_instance.output_format
end

def run
  @groups.each do |group|
    g.add_nodes(
      group, {
        :style     => "filled",
        :fillcolor => "#B9B9D5",
        :shape     => "box3d",
        :fontsize  => 16,
      }.merge(@node_options[group])
    )
  end
  @relations.each do |parent, children|
    children.each do |child|
      if @groups.include?(parent)
        g.add_nodes(child, { :style => "filled", :fillcolor => "#B9B9D5" }.merge(@node_options[child]))
        g.add_edges(parent, child, { :constraint => false }.merge(@edge_options["#{parent}_#{child}"]))
      else
        g.add_nodes(child, @node_options[child])
        g.add_edges(parent, child, @edge_options["#{parent}_#{child}"])
      end
    end
  end
  if @output_format.to_s == "debug"
    $stdout.puts g.output :none => String
    Bundler.ui.info "debugging bundle viz..."
  else
    begin
      g.output @output_format.to_sym => "#{@output_file}.#{@output_format}"
      Bundler.ui.info "#{@output_file}.#{@output_format}"
    rescue ArgumentError => e
      warn "Unsupported output format. See Ruby-Graphviz/lib/graphviz/constants.rb"
      raise e
    end
  end
end