Class: Gem::RemoteFetcher

# File 'lib/rubygems/remote_fetcher.rb', line 326

def close_all
  @pools.each_value(&:close_all)
end

# File 'lib/rubygems/remote_fetcher.rb', line 112

def download(spec, source_uri, install_dir = Gem.dir)
  install_cache_dir = File.join install_dir, "cache"
  cache_dir =
    if Dir.pwd == install_dir # see fetch_command
      install_dir
    elsif File.writable?(install_cache_dir) || (File.writable?(install_dir) && !File.exist?(install_cache_dir))
      install_cache_dir
    else
      File.join Gem.user_dir, "cache"
    end

  gem_file_name = File.basename spec.cache_file
  local_gem_path = File.join cache_dir, gem_file_name

  require "fileutils"
  begin
    FileUtils.mkdir_p cache_dir
  rescue StandardError
    nil
  end unless File.exist? cache_dir

  source_uri = Gem::Uri.new(source_uri)

  scheme = source_uri.scheme

  # Gem::URI.parse gets confused by MS Windows paths with forward slashes.
  scheme = nil if /^[a-z]$/i.match?(scheme)

  # REFACTOR: split this up and dispatch on scheme (eg download_http)
  # REFACTOR: be sure to clean up fake fetcher when you do this... cleaner
  case scheme
  when "http", "https", "s3" then
    unless File.exist? local_gem_path
      begin
        verbose "Downloading gem #{gem_file_name}"

        remote_gem_path = source_uri + "gems/#{gem_file_name}"

        cache_update_path remote_gem_path, local_gem_path
      rescue FetchError
        raise if spec.original_platform == spec.platform

        alternate_name = "#{spec.original_name}.gem"

        verbose "Failed, downloading gem #{alternate_name}"

        remote_gem_path = source_uri + "gems/#{alternate_name}"

        cache_update_path remote_gem_path, local_gem_path
      end
    end
  when "file" then
    begin
      path = source_uri.path
      path = File.dirname(path) if File.extname(path) == ".gem"

      remote_gem_path = Gem::Util.correct_for_windows_path(File.join(path, "gems", gem_file_name))

      FileUtils.cp(remote_gem_path, local_gem_path)
    rescue Errno::EACCES
      local_gem_path = source_uri.to_s
    end

    verbose "Using local gem #{local_gem_path}"
  when nil then # TODO: test for local overriding cache
    source_path = if Gem.win_platform? && source_uri.scheme &&
                     !source_uri.path.include?(":")
      "#{source_uri.scheme}:#{source_uri.path}"
    else
      source_uri.path
    end

    source_path = Gem::UriFormatter.new(source_path).unescape

    begin
      FileUtils.cp source_path, local_gem_path unless
        File.identical?(source_path, local_gem_path)
    rescue Errno::EACCES
      local_gem_path = source_uri.to_s
    end

    verbose "Using local gem #{local_gem_path}"
  else
    raise ArgumentError, "unsupported URI scheme #{source_uri.scheme}"
  end

  local_gem_path
end

# File 'lib/rubygems/remote_fetcher.rb', line 97

def download_to_cache(dependency)
  found, _ = Gem::SpecFetcher.fetcher.spec_for_dependency dependency

  return if found.empty?

  spec, source = found.max_by {|(s,_)| s.version }

  download spec, source.uri
end

# File 'lib/rubygems/remote_fetcher.rb', line 204

def fetch_file(uri, *_)
  Gem.read_binary Gem::Util.correct_for_windows_path uri.path
end

# File 'lib/rubygems/remote_fetcher.rb', line 211

def fetch_http(uri, last_modified = nil, head = false, depth = 0)
  fetch_type = head ? Gem::Net::HTTP::Head : Gem::Net::HTTP::Get
  response   = request uri, fetch_type, last_modified do |req|
    headers.each {|k,v| req.add_field(k,v) }
  end

  case response
  when Gem::Net::HTTPOK, Gem::Net::HTTPNotModified then
    response.uri = uri
    head ? response : response.body
  when Gem::Net::HTTPMovedPermanently, Gem::Net::HTTPFound, Gem::Net::HTTPSeeOther,
       Gem::Net::HTTPTemporaryRedirect then
    raise FetchError.new("too many redirects", uri) if depth > 10

    unless location = response["Location"]
      raise FetchError.new("redirecting but no redirect location was given", uri)
    end
    location = Gem::Uri.new location

    if https?(uri) && !https?(location)
      raise FetchError.new("redirecting to non-https resource: #{location}", uri)
    end

    fetch_http(location, last_modified, head, depth + 1)
  else
    raise FetchError.new("bad response #{response.message} #{response.code}", uri)
  end
end

# File 'lib/rubygems/remote_fetcher.rb', line 240

alias_method :fetch_https, :fetch_http

# File 'lib/rubygems/remote_fetcher.rb', line 245

def fetch_path(uri, mtime = nil, head = false)
  uri = Gem::Uri.new uri

  unless uri.scheme
    raise ArgumentError, "uri scheme is invalid: #{uri.scheme.inspect}"
  end

  data = send "fetch_#{uri.scheme}", uri, mtime, head

  if data && !head && uri.to_s.end_with?(".gz")
    begin
      data = Gem::Util.gunzip data
    rescue Zlib::GzipFile::Error
      raise FetchError.new("server did not return a valid file", uri)
    end
  end

  data
rescue Gem::Timeout::Error, IOError, SocketError, SystemCallError,
       *(OpenSSL::SSL::SSLError if Gem::HAVE_OPENSSL) => e
  raise FetchError.new("#{e.class}: #{e}", uri)
end

# File 'lib/rubygems/remote_fetcher.rb', line 268

def fetch_s3(uri, mtime = nil, head = false)
  begin
    public_uri = s3_uri_signer(uri).sign
  rescue Gem::S3URISigner::ConfigurationError, Gem::S3URISigner::InstanceProfileError => e
    raise FetchError.new(e.message, "s3://#{uri.host}")
  end
  fetch_https public_uri, mtime, head
end

# File 'lib/rubygems/remote_fetcher.rb', line 322

def https?(uri)
  uri.scheme.casecmp("https").zero?
end

# File 'lib/rubygems/remote_fetcher.rb', line 336

def pools_for(proxy)
  @pool_lock.synchronize do
    @pools[proxy] ||= Gem::Request::ConnectionPools.new proxy, @cert_files
  end
end

# File 'lib/rubygems/remote_fetcher.rb', line 332

def proxy_for(proxy, uri)
  Gem::Request.proxy_uri(proxy || Gem::Request.get_proxy_from_env(uri.scheme))
end

# File 'lib/rubygems/remote_fetcher.rb', line 311

def request(uri, request_class, last_modified = nil)
  proxy = proxy_for @proxy, uri
  pool  = pools_for(proxy).pool_for uri

  request = Gem::Request.new uri, request_class, last_modified, pool

  request.fetch do |req|
    yield req if block_given?
  end
end

# File 'lib/rubygems/remote_fetcher.rb', line 278

def s3_uri_signer(uri)
  Gem::S3URISigner.new(uri)
end