Google

Edit File: source.rb

# frozen_string_literal: true

require_relative "text"
##
# A Source knows how to list and fetch gems from a RubyGems marshal index.
#
# There are other Source subclasses for installed gems, local gems, the
# bundler dependency API and so-forth.

class Gem::Source
  include Comparable
  include Gem::Text

FILES = { # :nodoc:
    :released   => 'specs',
    :latest     => 'latest_specs',
    :prerelease => 'prerelease_specs',
  }.freeze

##
  # The URI this source will fetch gems from.

attr_reader :uri

##
  # Creates a new Source which will use the index located at +uri+.

def initialize(uri)
    begin
      unless uri.kind_of? URI
        uri = URI.parse(uri.to_s)
      end
    rescue URI::InvalidURIError
      raise if Gem::Source == self.class
    end

@uri = uri
  end

##
  # Sources are ordered by installation preference.

def <=>(other)
    case other
    when Gem::Source::Installed,
         Gem::Source::Local,
         Gem::Source::Lock,
         Gem::Source::SpecificFile,
         Gem::Source::Git,
         Gem::Source::Vendor then
      -1
    when Gem::Source then
      if !@uri
        return 0 unless other.uri
        return 1
      end

return -1 if !other.uri

# Returning 1 here ensures that when sorting a list of sources, the
      # original ordering of sources supplied by the user is preserved.
      return 1 unless @uri.to_s == other.uri.to_s

0
    else
      nil
    end
  end

def ==(other) # :nodoc:
    self.class === other and @uri == other.uri
  end

alias_method :eql?, :== # :nodoc:

##
  # Returns a Set that can fetch specifications from this source.

def dependency_resolver_set # :nodoc:
    return Gem::Resolver::IndexSet.new self if 'file' == uri.scheme

fetch_uri = if uri.host == "rubygems.org"
                  index_uri = uri.dup
                  index_uri.host = "index.rubygems.org"
                  index_uri
                else
                  uri
                end

bundler_api_uri = enforce_trailing_slash(fetch_uri)

begin
      fetcher = Gem::RemoteFetcher.fetcher
      response = fetcher.fetch_path bundler_api_uri, nil, true
    rescue Gem::RemoteFetcher::FetchError
      Gem::Resolver::IndexSet.new self
    else
      Gem::Resolver::APISet.new response.uri + "./info/"
    end
  end

def hash # :nodoc:
    @uri.hash
  end

##
  # Returns the local directory to write +uri+ to.

def cache_dir(uri)
    # Correct for windows paths
    escaped_path = uri.path.sub(/^\/([a-z]):\//i, '/\\1-/')
    escaped_path.tap(&Gem::UNTAINT)

File.join Gem.spec_cache_dir, "#{uri.host}%#{uri.port}", File.dirname(escaped_path)
  end

##
  # Returns true when it is possible and safe to update the cache directory.

def update_cache?
    @update_cache ||=
      begin
        File.stat(Gem.user_home).uid == Process.uid
      rescue Errno::ENOENT
        false
      end
  end

##
  # Fetches a specification for the given +name_tuple+.

def fetch_spec(name_tuple)
    fetcher = Gem::RemoteFetcher.fetcher

spec_file_name = name_tuple.spec_name

source_uri = enforce_trailing_slash(uri) + "#{Gem::MARSHAL_SPEC_DIR}#{spec_file_name}"

cache_dir = cache_dir source_uri

local_spec = File.join cache_dir, spec_file_name

if File.exist? local_spec
      spec = Gem.read_binary local_spec
      spec = Marshal.load(spec) rescue nil
      return spec if spec
    end

source_uri.path << '.rz'

spec = fetcher.fetch_path source_uri
    spec = Gem::Util.inflate spec

if update_cache?
      require "fileutils"
      FileUtils.mkdir_p cache_dir

File.open local_spec, 'wb' do |io|
        io.write spec
      end
    end

# TODO: Investigate setting Gem::Specification#loaded_from to a URI
    Marshal.load spec
  end

##
  # Loads +type+ kind of specs fetching from +@uri+ if the on-disk cache is
  # out of date.
  #
  # +type+ is one of the following:
  #
  # :released   => Return the list of all released specs
  # :latest     => Return the list of only the highest version of each gem
  # :prerelease => Return the list of all prerelease only specs
  #

def load_specs(type)
    file       = FILES[type]
    fetcher    = Gem::RemoteFetcher.fetcher
    file_name  = "#{file}.#{Gem.marshal_version}"
    spec_path  = enforce_trailing_slash(uri) + "#{file_name}.gz"
    cache_dir  = cache_dir spec_path
    local_file = File.join(cache_dir, file_name)
    retried    = false

if update_cache?
      require "fileutils"
      FileUtils.mkdir_p cache_dir
    end

spec_dump = fetcher.cache_update_path spec_path, local_file, update_cache?

begin
      Gem::NameTuple.from_list Marshal.load(spec_dump)
    rescue ArgumentError
      if update_cache? && !retried
        FileUtils.rm local_file
        retried = true
        retry
      else
        raise Gem::Exception.new("Invalid spec cache file in #{local_file}")
      end
    end
  end

##
  # Downloads +spec+ and writes it to +dir+.  See also
  # Gem::RemoteFetcher#download.

def download(spec, dir=Dir.pwd)
    fetcher = Gem::RemoteFetcher.fetcher
    fetcher.download spec, uri.to_s, dir
  end

def pretty_print(q) # :nodoc:
    q.group 2, '[Remote:', ']' do
      q.breakable
      q.text @uri.to_s

if api = uri
        q.breakable
        q.text 'API URI: '
        q.text api.to_s
      end
    end
  end

def typo_squatting?(host, distance_threshold=4)
    return if @uri.host.nil?
    levenshtein_distance(@uri.host, host).between? 1, distance_threshold
  end

private

def enforce_trailing_slash(uri)
    uri.merge(uri.path.gsub(/\/+$/, '') + '/')
  end
end

require_relative 'source/git'
require_relative 'source/installed'
require_relative 'source/specific_file'
require_relative 'source/local'
require_relative 'source/lock'
require_relative 'source/vendor'