1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/lib/bundler/gem_remote_fetcher.rb
hsbt be7b592912 Update bundled bundler to 1.16.0.
* lib/bundler, spec/bundler: Merge bundler-1.16.0.
  * common.mk: rspec examples of bundler-1.16.0 needs require option.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@60603 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2017-11-01 23:29:38 +00:00

43 lines
1.5 KiB
Ruby

# frozen_string_literal: true
require "rubygems/remote_fetcher"
module Bundler
# Adds support for setting custom HTTP headers when fetching gems from the
# server.
#
# TODO: Get rid of this when and if gemstash only supports RubyGems versions
# that contain https://github.com/rubygems/rubygems/commit/3db265cc20b2f813.
class GemRemoteFetcher < Gem::RemoteFetcher
attr_accessor :headers
# Extracted from RubyGems 2.4.
def fetch_http(uri, last_modified = nil, head = false, depth = 0)
fetch_type = head ? Net::HTTP::Head : Net::HTTP::Get
# beginning of change
response = request uri, fetch_type, last_modified do |req|
headers.each {|k, v| req.add_field(k, v) } if headers
end
# end of change
case response
when Net::HTTPOK, Net::HTTPNotModified then
response.uri = uri if response.respond_to? :uri
head ? response : response.body
when Net::HTTPMovedPermanently, Net::HTTPFound, Net::HTTPSeeOther,
Net::HTTPTemporaryRedirect then
raise FetchError.new("too many redirects", uri) if depth > 10
location = URI.parse response["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
end
end