class Bundler::Fetcher

def bundler_cert_store
  store = OpenSSL::X509::Store.new
  if Bundler.settings[:ssl_ca_cert]
    if File.directory? Bundler.settings[:ssl_ca_cert]
      store.add_path Bundler.settings[:ssl_ca_cert]
    else
      store.add_file Bundler.settings[:ssl_ca_cert]
    end
  else
    store.set_default_paths
    certs = File.expand_path("../ssl_certs/*/*.pem", __FILE__)
    Dir.glob(certs).each {|c| store.add_file c }
  end
  store
end

def cis
  env_cis = {
    "TRAVIS" => "travis",
    "CIRCLECI" => "circle",
    "SEMAPHORE" => "semaphore",
    "JENKINS_URL" => "jenkins",
    "BUILDBOX" => "buildbox",
    "GO_SERVER_URL" => "go",
    "SNAP_CI" => "snap",
    "CI_NAME" => ENV["CI_NAME"],
    "CI" => "ci"
  }
  env_cis.find_all {|env, _| ENV[env] }.map {|_, ci| ci }
end

def connection
  @connection ||= begin
    needs_ssl = remote_uri.scheme == "https" ||
      Bundler.settings[:ssl_verify_mode] ||
      Bundler.settings[:ssl_client_cert]
    raise SSLError if needs_ssl && !defined?(OpenSSL::SSL)
    con = Net::HTTP::Persistent.new "bundler", :ENV
    if gem_proxy = Bundler.rubygems.configuration[:http_proxy]
      con.proxy = URI.parse(gem_proxy) if gem_proxy != :no_proxy
    end
    if remote_uri.scheme == "https"
      con.verify_mode = (Bundler.settings[:ssl_verify_mode] ||
        OpenSSL::SSL::VERIFY_PEER)
      con.cert_store = bundler_cert_store
    end
    if Bundler.settings[:ssl_client_cert]
      pem = File.read(Bundler.settings[:ssl_client_cert])
      con.cert = OpenSSL::X509::Certificate.new(pem)
      con.key  = OpenSSL::PKey::RSA.new(pem)
    end
    con.read_timeout = Fetcher.api_timeout
    con.override_headers["User-Agent"] = user_agent
    con.override_headers["X-Gemfile-Source"] = @remote.original_uri.to_s if @remote.original_uri
    con
  end
end

def downloader
  @downloader ||= Downloader.new(connection, self.class.redirect_limit)
end

def fetch_spec(spec)
  spec -= [nil, "ruby", ""]
  spec_file_name = "#{spec.join "-"}.gemspec"
  uri = URI.parse("#{remote_uri}#{Gem::MARSHAL_SPEC_DIR}#{spec_file_name}.rz")
  if uri.scheme == "file"
    Bundler.load_marshal Gem.inflate(Gem.read_binary(uri.path))
  elsif cached_spec_path = gemspec_cached_path(spec_file_name)
    Bundler.load_gemspec(cached_spec_path)
  else
    Bundler.load_marshal Gem.inflate(downloader.fetch(uri).body)
  end
rescue MarshalError
  raise HTTPError, "Gemspec #{spec} contained invalid data.\n" \
    "Your network or your gem server is probably having issues right now."
end

def fetchers
  @fetchers ||= FETCHERS.map {|f| f.new(downloader, @remote, uri) }
end

def gemspec_cached_path(spec_file_name)
  paths = Bundler.rubygems.spec_cache_dirs.map {|dir| File.join(dir, spec_file_name) }
  paths = paths.select {|path| File.file? path }
  paths.first
end

def http_proxy
  return unless uri = connection.proxy_uri
  uri.to_s
end

def initialize(remote)
  @remote = remote
  Socket.do_not_reverse_lookup = true
  connection # create persistent connection
end

def inspect
  "#<#{self.class}:0x#{object_id} uri=#{uri}>"
end

def remote_uri
  @remote.uri
end

def specs(gem_names, source)
  old = Bundler.rubygems.sources
  index = Bundler::Index.new
  if Bundler::Fetcher.disable_endpoint
    @use_api = false
    specs = fetchers.last.specs(gem_names)
  else
    specs = []
    fetchers.shift until fetchers.first.available? || fetchers.empty?
    fetchers.dup.each do |f|
      break unless f.api_fetcher? && !gem_names || !specs = f.specs(gem_names)
      fetchers.delete(f)
    end
    @use_api = false if fetchers.none?(&:api_fetcher?)
  end
  specs.each do |name, version, platform, dependencies|
    next if name == "bundler"
    spec = if dependencies
      EndpointSpecification.new(name, version, platform, dependencies)
    else
      RemoteSpecification.new(name, version, platform, self)
    end
    spec.source = source
    spec.remote = @remote
    index << spec
  end
  index
rescue CertificateFailureError
  Bundler.ui.info "" if gem_names && use_api # newline after dots
  raise
ensure
  Bundler.rubygems.sources = old
end

def specs_with_retry(gem_names, source)
  Bundler::Retry.new("fetcher", FAIL_ERRORS).attempts do
    specs(gem_names, source)
  end
end

def uri
  @remote.anonymized_uri
end

def use_api
  return @use_api if defined?(@use_api)
  fetchers.shift until fetchers.first.available?
  @use_api = if remote_uri.scheme == "file" || Bundler::Fetcher.disable_endpoint
    false
  else
    fetchers.first.api_fetcher?
  end
end

def user_agent
  @user_agent ||= begin
    ruby = Bundler::RubyVersion.system
    agent = String.new("bundler/#{Bundler::VERSION}")
    agent << " rubygems/#{Gem::VERSION}"
    agent << " ruby/#{ruby.versions_string(ruby.versions)}"
    agent << " (#{ruby.host})"
    agent << " command/#{ARGV.first}"
    if ruby.engine != "ruby"
      # engine_version raises on unknown engines
      engine_version = begin
                         ruby.engine_versions
                       rescue
                         "???"
                       end
      agent << " #{ruby.engine}/#{ruby.versions_string(engine_version)}"
    end
    agent << " options/#{Bundler.settings.all.join(",")}"
    agent << " ci/#{cis.join(",")}" if cis.any?
    # add a random ID so we can consolidate runs server-side
    agent << " " << SecureRandom.hex(8)
    # add any user agent strings set in the config
    extra_ua = Bundler.settings[:user_agent]
    agent << " " << extra_ua if extra_ua
    agent
  end
end