class Bundler::SourceList

def add_git_source(options = {})
  add_source_to_list Source::Git.new(options), git_sources
end

def add_path_source(options = {})
  add_source_to_list Source::Path.new(options), path_sources
end

def add_rubygems_remote(uri)
  @rubygems_aggregate.add_remote(uri)
  @rubygems_aggregate
end

def add_rubygems_source(options = {})
  add_source_to_list Source::Rubygems.new(options), @rubygems_sources
end

def add_source_to_list(source, list)
  list.unshift(source).uniq!
  source
end

def all_sources
  path_sources + git_sources + rubygems_sources
end

def cached!
  all_sources.each(&:cached!)
end

def combine_rubygems_sources
  Source::Rubygems.new("remotes" => rubygems_remotes)
end

def get(source)
  source_list_for(source).find {|s| source == s }
end

def initialize
  @path_sources       = []
  @git_sources        = []
  @rubygems_aggregate = Source::Rubygems.new
  @rubygems_sources   = []
end

def lock_sources
  lock_sources = (path_sources + git_sources).sort_by(&:to_s)
  lock_sources << combine_rubygems_sources
end

def remote!
  all_sources.each(&:remote!)
end

def replace_sources!(replacement_sources)
  return true if replacement_sources.empty?
  [path_sources, git_sources].each do |source_list|
    source_list.map! do |source|
      replacement_sources.find {|s| s == source } || source
    end
  end
  replacement_rubygems =
    replacement_sources.detect {|s| s.is_a?(Source::Rubygems) }
  @rubygems_aggregate = replacement_rubygems if replacement_rubygems
  # Return true if there were changes
  lock_sources.to_set != replacement_sources.to_set ||
    rubygems_remotes.to_set != replacement_rubygems.remotes.to_set
end

def rubygems_primary_remotes
  @rubygems_aggregate.remotes
end

def rubygems_remotes
  rubygems_sources.map(&:remotes).flatten.uniq
end

def rubygems_sources
  @rubygems_sources + [@rubygems_aggregate]
end

def source_list_for(source)
  case source
  when Source::Git      then git_sources
  when Source::Path     then path_sources
  when Source::Rubygems then rubygems_sources
  else raise ArgumentError, "Invalid source: #{source.inspect}"
  end
end