diff options
author | Agis Anastasopoulos <agis.anast@gmail.com> | 2015-10-19 00:34:13 +0300 |
---|---|---|
committer | Agis Anastasopoulos <agis.anast@gmail.com> | 2015-10-20 19:52:49 +0300 |
commit | c95d283702bea53a4fee2193b4832a1ce0d0bacd (patch) | |
tree | edeb81d9f49857bedc065cc60f1e56b4ad81c120 | |
parent | 25b53401c7912c4b64e3e08832d539ce6ffc38e3 (diff) | |
download | bundler-c95d283702bea53a4fee2193b4832a1ce0d0bacd.tar.gz |
Introduce GemRemoteFetcher
This is essentially Gem::RemoteFetcher with added support for setting
extra HTTP headers to be sent when fetching the gems.
-rw-r--r-- | lib/bundler.rb | 1 | ||||
-rw-r--r-- | lib/bundler/gem_remote_fetcher.rb | 41 |
2 files changed, 42 insertions, 0 deletions
diff --git a/lib/bundler.rb b/lib/bundler.rb index d222befc98..f4373662ba 100644 --- a/lib/bundler.rb +++ b/lib/bundler.rb @@ -3,6 +3,7 @@ require "pathname" require "rbconfig" require "thread" require "bundler/gem_path_manipulation" +require "bundler/gem_remote_fetcher" require "bundler/rubygems_ext" require "bundler/rubygems_integration" require "bundler/version" diff --git a/lib/bundler/gem_remote_fetcher.rb b/lib/bundler/gem_remote_fetcher.rb new file mode 100644 index 0000000000..77cab0c4ad --- /dev/null +++ b/lib/bundler/gem_remote_fetcher.rb @@ -0,0 +1,41 @@ +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 |