class Bundler::Index

def self.from_installed_gems
  Source::SystemGems.new.specs
end

def <<(spec)
  arr = @specs[spec.name]
  arr.delete_if do |s|
    s.version == spec.version && s.platform == spec.platform
  end
  arr << spec
  spec
end

def each(&blk)
  @specs.values.each do |specs|
    specs.each(&blk)
  end
end

def empty?
  each { return false }
  true
end

def freeze
  @specs.each do |k,v|
    v.freeze
  end
  super
end

def initialize
  @cache = {}
  @specs = Hash.new { |h,k| h[k] = [] }
end

def initialize_copy(o)
  super
  @cache = {}
  @specs = Hash.new { |h,k| h[k] = [] }
  merge!(o)
end

def merge(other)
  dup.merge!(other)
end

def merge!(other)
  other.each do |spec|
    self << spec
  end
  self
end

def search(query)
  case query
  when Gem::Specification, RemoteSpecification then search_by_spec(query)
  when String then @specs[query]
  else search_by_dependency(query)
  end
end

def search_by_dependency(dependency)
  @cache[dependency.hash] ||= begin
    specs = @specs[dependency.name]
    wants_prerelease = dependency.version_requirements.prerelease?
    only_prerelease  = specs.all? {|spec| spec.version.prerelease? }
    found = specs.select { |spec| dependency =~ spec }
    unless wants_prerelease || only_prerelease
      found.reject! { |spec| spec.version.prerelease? }
    end
    found.sort_by {|s| [s.version, s.platform.to_s == 'ruby' ? "\0" : s.platform.to_s] }
  end
end

def search_by_spec(spec)
  @specs[spec.name].select { |s| s.version == spec.version }
end