class Gem::Source

# File lib/rubygems/source.rb, line 28
def initialize(uri)
  require_relative "uri"
  @uri = Gem::Uri.parse!(uri)
  @update_cache = nil
end

# File lib/rubygems/source.rb, line 37
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
    unless @uri
      return 0 unless other.uri
      return 1
    end

    return -1 unless 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
  end
end

# File lib/rubygems/source.rb, line 84
def cache_dir(uri)
  # Correct for windows paths
  escaped_path = uri.path.sub(%r{^/([a-z]):/}i, '/\\1-/')

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

# File lib/rubygems/source.rb, line 73
def dependency_resolver_set(prerelease = false)
  new_dependency_resolver_set.tap {|set| set.prerelease = prerelease }
end

# File lib/rubygems/source.rb, line 184
def download(spec, dir = Dir.pwd)
  fetcher = Gem::RemoteFetcher.fetcher
  fetcher.download spec, uri.to_s, dir
end

# File lib/rubygems/source.rb, line 107
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
    Gem.load_safe_marshal
    spec = begin
             Gem::SafeMarshal.safe_load(spec)
           rescue StandardError
             nil
           end
    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

  Gem.load_safe_marshal
  # TODO: Investigate setting Gem::Specification#loaded_from to a URI
  Gem::SafeMarshal.safe_load spec
end

# File lib/rubygems/source.rb, line 161
def load_specs(type)
  load_compact_index_specs(type) || load_marshal_specs(type)
end

# File lib/rubygems/source.rb, line 204
def typo_squatting?(host, distance_threshold = 4)
  return if @uri.host.nil?
  levenshtein_distance(@uri.host, host).between? 1, distance_threshold
end

# File lib/rubygems/source.rb, line 94
def update_cache?
  return @update_cache unless @update_cache.nil?
  @update_cache =
    begin
      File.stat(Gem.user_home).uid == Process.uid
    rescue Errno::ENOENT
      false
    end
end

# File lib/rubygems/source.rb, line 293
def compact_index_cache_dir(index_uri)
  if update_cache?
    # Correct for windows paths
    escaped_path = index_uri.path.sub(%r{^/([a-z]):/}i, '/\\1-/')

    File.join Gem.spec_cache_dir, "compact_index",
      "#{index_uri.host}%#{index_uri.port}", *escaped_path.split("/").reject(&:empty?)
  else
    require "tmpdir"
    require "fileutils"
    dir = Dir.mktmpdir "gem_compact_index"
    at_exit { FileUtils.rm_rf dir }
    dir
  end
end

# File lib/rubygems/source.rb, line 283
def compact_index_uri
  if uri.host == "rubygems.org"
    index_uri = uri.dup
    index_uri.host = "index.rubygems.org"
    index_uri
  else
    uri
  end
end

# File lib/rubygems/source.rb, line 272
def compact_index_versions
  @compact_index_versions ||= compact_index_client.versions
rescue Gem::RemoteFetcher::FetchError, Gem::CompactIndexClient::Error
  @compact_index_versions = {}
  nil
end

# File lib/rubygems/source.rb, line 324
def enforce_trailing_slash(uri)
  uri.merge(uri.path.gsub(%r{/+$}, "") + "/")
end

# File lib/rubygems/source.rb, line 246
def load_compact_index_specs(type)
  return unless %w[http https].include?(uri.scheme)

  versions = compact_index_versions
  return if versions.nil? || versions.empty?

  tuples = []

  versions.each_value do |rows|
    gem_tuples = rows.filter_map do |name, version_string, platform|
      next unless Gem::Version.correct?(version_string)

      version = Gem::Version.new(version_string)
      next if version.prerelease? != (type == :prerelease)

      Gem::NameTuple.new(name, version, platform || "ruby")
    end

    gem_tuples = max_versions_by_platform(gem_tuples) if type == :latest

    tuples.concat(gem_tuples)
  end

  tuples
end

# File lib/rubygems/source.rb, line 211
def load_marshal_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?

  Gem.load_safe_marshal
  begin
    Gem::NameTuple.from_list Gem::SafeMarshal.safe_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

# File lib/rubygems/source.rb, line 279
def max_versions_by_platform(tuples)
  tuples.group_by(&:platform).map {|_, platform_tuples| platform_tuples.max_by(&:version) }
end

# File lib/rubygems/source.rb, line 309
def new_dependency_resolver_set
  return Gem::Resolver::IndexSet.new self if uri.scheme == "file"

  bundler_api_uri = enforce_trailing_slash(compact_index_uri) + "versions"

  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