name : gem_remote_fetcher.rb # 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