class Bundler::Graph

def _groups
  relations = Hash.new {|h, k| h[k] = Set.new }
  @env.current_dependencies.each do |dependency|
    dependency.groups.each do |group|
      next if @without_groups.include?(group)
      relations[group.to_s].add(dependency)
      @relations[group.to_s].add(dependency.name)
      @node_options[group.to_s] ||= _make_label(group, :node)
      @edge_options["#{group}_#{dependency.name}"] = _make_label(dependency, :edge)
    end
  end
  @groups = relations.keys
  relations
end

def _make_label(symbol_or_string_or_dependency, element_type)
  case element_type.to_sym
  when :node
    if symbol_or_string_or_dependency.is_a?(Gem::Dependency)
      label = symbol_or_string_or_dependency.name.dup
      label << "\n#{spec_for_dependency(symbol_or_string_or_dependency).version}" if @show_version
    else
      label = symbol_or_string_or_dependency.to_s
    end
  when :edge
    label = nil
    if symbol_or_string_or_dependency.respond_to?(:requirements_list) && @show_requirements
      tmp = symbol_or_string_or_dependency.requirements_list.join(", ")
      label = tmp if tmp != ">= 0"
    end
  else
    raise ArgumentError, "2nd argument is invalid"
  end
  label.nil? ? {} : { label: label }
end

def _populate_relations
  parent_dependencies = _groups.values.to_set.flatten
  loop do
    break if parent_dependencies.empty?
    tmp = Set.new
    parent_dependencies.each do |dependency|
      child_dependencies = spec_for_dependency(dependency).runtime_dependencies.to_set
      @relations[dependency.name] += child_dependencies.map(&:name).to_set
      tmp += child_dependencies
      @node_options[dependency.name] = _make_label(dependency, :node)
      child_dependencies.each do |c_dependency|
        @edge_options["#{dependency.name}_#{c_dependency.name}"] = _make_label(c_dependency, :edge)
      end
    end
    parent_dependencies = tmp
  end
end

def initialize(env, output_file, show_version = false, show_requirements = false, output_format = "png", without = [])
  @env               = env
  @output_file       = output_file
  @show_version      = show_version
  @show_requirements = show_requirements
  @output_format     = output_format
  @without_groups    = without.map(&:to_sym)
  @groups            = []
  @relations         = Hash.new {|h, k| h[k] = Set.new }
  @node_options      = {}
  @edge_options      = {}
  _populate_relations
end

def spec_for_dependency(dependency)
  @env.requested_specs.find {|s| s.name == dependency.name }
end

def viz
  GraphVizClient.new(self).run
end