mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
43 lines
1.5 KiB
Ruby
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
|