class Bundler::Source::Rubygems

def self.from_lock(options)
  s = new(options)
  Array(options["remote"]).each { |r| s.add_remote(r) }
  s
end

def add_remote(source)
  @remotes << normalize_uri(source)
end

def cache(spec)
  cached_path = cached_gem(spec)
  raise GemNotFound, "Missing gem file '#{spec.full_name}.gem'." unless cached_path
  return if File.dirname(cached_path) == Bundler.app_cache.to_s
  Bundler.ui.info "  * #{File.basename(cached_path)}"
  FileUtils.cp(cached_path, Bundler.app_cache)
end

def cached!
  @allow_cached = true
end

def cached_gem(spec)
  possibilities = @caches.map { |p| "#{p}/#{spec.file_name}" }
  cached_gem = possibilities.find { |p| File.exist?(p) }
  unless cached_gem
    raise Bundler::GemNotFound, "Could not find #{spec.file_name} for installation"
  end
  cached_gem
end

def cached_specs
  @cached_specs ||= begin
    idx = installed_specs.dup
    path = Bundler.app_cache
    Dir["#{path}/*.gem"].each do |gemfile|
      next if gemfile =~ /^bundler\-[\d\.]+?\.gem/
      begin
        s ||= Bundler.rubygems.spec_from_gem(gemfile)
      rescue Gem::Package::FormatError
        raise GemspecError, "Could not read gem at #{gemfile}. It may be corrupted."
      end
      s.source = self
      idx << s
    end
  end
  idx
end

def eql?(o)
  Rubygems === o
end

def fetch_specs
  # remote_specs usually generates a way larger Index than the other
  # sources, and large_idx.use small_idx is way faster than
  # small_idx.use large_idx.
  if @allow_remote
    idx = remote_specs.dup
  else
    idx = Index.new
  end
  idx.use(cached_specs, :override_dupes) if @allow_cached || @allow_remote
  idx.use(installed_specs, :override_dupes)
  idx
end

def hash
  Rubygems.hash
end

def initialize(options = {})
  @options = options
  @remotes = (options["remotes"] || []).map { |r| normalize_uri(r) }
  @fetchers = {}
  @allow_remote = false
  @allow_cached = false
  @caches = [ Bundler.app_cache ] +
    Bundler.rubygems.gem_path.map{|p| File.expand_path("#{p}/cache") }
end

def install(spec)
  if installed_specs[spec].any?
    Bundler.ui.info "Using #{spec.name} (#{spec.version}) "
    return
  end
  Bundler.ui.info "Installing #{spec.name} (#{spec.version}) "
  path = cached_gem(spec)
  if Bundler.requires_sudo?
    install_path = Bundler.tmp
    bin_path     = install_path.join("bin")
  else
    install_path = Bundler.rubygems.gem_dir
    bin_path     = Bundler.system_bindir
  end
  Bundler.rubygems.preserve_paths do
    Bundler::GemInstaller.new(path,
      :install_dir         => install_path.to_s,
      :bin_dir             => bin_path.to_s,
      :ignore_dependencies => true,
      :wrappers            => true,
      :env_shebang         => true
    ).install
  end
  if spec.post_install_message
    Installer.post_install_messages[spec.name] = spec.post_install_message
  end
  # SUDO HAX
  if Bundler.requires_sudo?
    Bundler.mkdir_p "#{Bundler.rubygems.gem_dir}/gems"
    Bundler.mkdir_p "#{Bundler.rubygems.gem_dir}/specifications"
    Bundler.sudo "cp -R #{Bundler.tmp}/gems/#{spec.full_name} #{Bundler.rubygems.gem_dir}/gems/"
    Bundler.sudo "cp -R #{Bundler.tmp}/specifications/#{spec.full_name}.gemspec #{Bundler.rubygems.gem_dir}/specifications/"
    spec.executables.each do |exe|
      Bundler.mkdir_p Bundler.system_bindir
      Bundler.sudo "cp -R #{Bundler.tmp}/bin/#{exe} #{Bundler.system_bindir}"
    end
  end
  spec.loaded_from = "#{Bundler.rubygems.gem_dir}/specifications/#{spec.full_name}.gemspec"
end

def installed_specs
  @installed_specs ||= begin
    idx = Index.new
    have_bundler = false
    Bundler.rubygems.all_specs.reverse.each do |spec|
      next if spec.name == 'bundler' && spec.version.to_s != VERSION
      have_bundler = true if spec.name == 'bundler'
      spec.source = self
      idx << spec
    end
    # Always have bundler locally
    unless have_bundler
     # We're running bundler directly from the source
     # so, let's create a fake gemspec for it (it's a path)
     # gemspec
     bundler = Gem::Specification.new do |s|
       s.name     = 'bundler'
       s.version  = VERSION
       s.platform = Gem::Platform::RUBY
       s.source   = self
       s.authors  = ["bundler team"]
       s.loaded_from = File.expand_path("..", __FILE__)
     end
     idx << bundler
    end
    idx
  end
end

def normalize_uri(uri)
  uri = uri.to_s
  uri = "#{uri}/" unless uri =~ %r'/$'
  uri = URI(uri)
  raise ArgumentError, "The source must be an absolute URI" unless uri.absolute?
  uri
end

def options
  { "remotes" => @remotes.map { |r| r.to_s } }
end

def remote!
  @allow_remote = true
end

def remote_specs
  @remote_specs ||= begin
    idx     = Index.new
    old     = Bundler.rubygems.sources
    sources = {}
    remotes.each do |uri|
      fetcher          = Bundler::Fetcher.new(uri)
      specs            = fetcher.specs(dependency_names, self)
      sources[fetcher] = specs.size
      idx.use specs
    end
    # don't need to fetch all specifications for every gem/version on
    # the rubygems repo if there's no api endpoints to search over
    # or it has too many specs to fetch
    fetchers              = sources.keys
    api_fetchers          = fetchers.select {|fetcher| fetcher.has_api }
    modern_index_fetchers = fetchers - api_fetchers
    if api_fetchers.any? && modern_index_fetchers.all? {|fetcher| sources[fetcher] < FORCE_MODERN_INDEX_LIMIT }
      # this will fetch all the specifications on the rubygems repo
      unmet_dependency_names = idx.unmet_dependency_names
      unmet_dependency_names -= ['bundler'] # bundler will always be unmet
      Bundler.ui.debug "Unmet Dependencies: #{unmet_dependency_names}"
      if unmet_dependency_names.any?
        api_fetchers.each do |fetcher|
          idx.use fetcher.specs(unmet_dependency_names, self)
        end
      end
    else
      Bundler::Fetcher.disable_endpoint = true
      api_fetchers.each {|fetcher| idx.use fetcher.specs([], self) }
    end
    idx
  ensure
    Bundler.rubygems.sources = old
  end
end

def replace_remotes(source)
  return false if source.remotes == @remotes
  @remotes = []
  source.remotes.each do |r|
    add_remote r.to_s
  end
  true
end

def specs
  @specs ||= fetch_specs
end

def to_lock
  out = "GEM\n"
  out << remotes.map {|r| "  remote: #{r}\n" }.join
  out << "  specs:\n"
end

def to_s
  remote_names = self.remotes.map { |r| r.to_s }.join(', ')
  "rubygems repository #{remote_names}"
end