Class: Gem::Installer

# File 'lib/rubygems/installer.rb', line 100

attr_writer :exec_format

# File 'lib/rubygems/installer.rb', line 91

attr_accessor :path_warning

# File 'lib/rubygems/installer.rb', line 149

def self.for_spec(spec, options = {})
  # FIXME: we should have a real Package class for this
  new FakePackage.new(spec), options
end

# File 'lib/rubygems/installer.rb', line 80

def inherited(klass)
  if klass.name == "Bundler::RubyGemsGemInstaller"
    require "rubygems/command"
  end

  super(klass)
end

# File 'lib/rubygems/installer.rb', line 51

attr_reader :build_root # :nodoc:

# File 'lib/rubygems/installer.rb', line 56

attr_reader :gem_home

# File 'lib/rubygems/installer.rb', line 981

def load_relative_enabled?
  rb_config["LIBRUBY_RELATIVE"] == "yes"
end

# File 'lib/rubygems/installer.rb', line 61

attr_reader :options

# File 'lib/rubygems/installer.rb', line 66

attr_reader :package

# File 'lib/rubygems/installer.rb', line 985

def bash_prolog_script
  if load_relative_enabled?
    script = +<<~EOS
      bindir="${0%/*}"
    EOS

    script << %Q(exec "$bindir/#{ruby_install_name}" "-x" "$0" "$@"\n)

    <<~EOS
      #!/bin/sh
      # -*- ruby -*-
      _=_\\
      =begin
      #{script.chomp}
      =end
    EOS
  else
    ""
  end
end

# File 'lib/rubygems/installer.rb', line 966

def build_args
  @build_args ||= begin
                    require_relative "command"
                    Gem::Command.build_args
                  end
end

# File 'lib/rubygems/installer.rb', line 840

def build_extensions
  builder = Gem::Ext::Builder.new spec, build_args

  builder.build_extensions
end

# File 'lib/rubygems/installer.rb', line 212

def check_executable_overwrite(filename) # :nodoc:
  return if @force

  generated_bin = File.join @bin_dir, formatted_program_filename(filename)

  return unless File.exist? generated_bin

  ruby_executable = false
  existing = nil

  File.open generated_bin, "rb" do |io|
    line = io.gets
    shebang = /^#!.*ruby/

    if load_relative_enabled?
      until line.nil? || line =~ shebang do
        line = io.gets
      end
    end

    next unless line =~ shebang

    io.gets # blankline

    # TODO detect a specially formatted comment instead of trying
    # to run a regexp against Ruby code.
    next unless io.gets =~ /This file was generated by RubyGems/

    ruby_executable = true
    existing = io.read.slice(%r{
        ^\s*(
          gem \s |
          load \s Gem\.bin_path\( |
          load \s Gem\.activate_bin_path\(
        )
        (['"])(.*?)(\2),
      }x, 3)
  end

  return if spec.name == existing

  # somebody has written to RubyGems' directory, overwrite, too bad
  return if Gem.default_bindir != @bin_dir && !ruby_executable

  question = "#{spec.name}'s executable \"#{filename}\" conflicts with ".dup

  if ruby_executable
    question << (existing || "an unknown executable")

    return if ask_yes_no "#{question}\nOverwrite the executable?", false

    conflict = "installed executable from #{existing}"
  else
    question << generated_bin

    return if ask_yes_no "#{question}\nOverwrite the executable?", false

    conflict = generated_bin
  end

  raise Gem::InstallError,
    "\"#{filename}\" from #{spec.name} conflicts with #{conflict}"
end

# File 'lib/rubygems/installer.rb', line 686

def check_that_user_bin_dir_is_in_path # :nodoc:
  return if self.class.path_warning

  user_bin_dir = @bin_dir || Gem.bindir(gem_home)
  user_bin_dir = user_bin_dir.tr(File::ALT_SEPARATOR, File::SEPARATOR) if File::ALT_SEPARATOR

  path = ENV["PATH"]
  path = path.tr(File::ALT_SEPARATOR, File::SEPARATOR) if File::ALT_SEPARATOR

  if Gem.win_platform?
    path = path.downcase
    user_bin_dir = user_bin_dir.downcase
  end

  path = path.split(File::PATH_SEPARATOR)

  unless path.include? user_bin_dir
    unless !Gem.win_platform? && (path.include? user_bin_dir.sub(ENV["HOME"], "~"))
      alert_warning "You don't have #{user_bin_dir} in your PATH,\n\t  gem executables will not run."
      self.class.path_warning = true
    end
  end
end

# File 'lib/rubygems/installer.rb', line 448

def default_spec_file
  File.join gem_home, "specifications", "default", "#{spec.full_name}.gemspec"
end

# File 'lib/rubygems/installer.rb', line 881

def dir
  gem_dir.to_s
end

# File 'lib/rubygems/installer.rb', line 641

def ensure_dependencies_met # :nodoc:
  deps = spec.runtime_dependencies
  deps |= spec.development_dependencies if @development

  deps.each do |dep_gem|
    ensure_dependency spec, dep_gem
  end
end

# File 'lib/rubygems/installer.rb', line 410

def ensure_dependency(spec, dependency)
  unless installation_satisfies_dependency? dependency
    raise Gem::InstallError, "#{spec.name} requires #{dependency}"
  end
  true
end

# File 'lib/rubygems/installer.rb', line 629

def ensure_loadable_spec
  ruby = spec.to_ruby_for_cache
  ruby.tap(&Gem::UNTAINT)

  begin
    eval ruby
  rescue StandardError, SyntaxError => e
    raise Gem::InstallError,
          "The specification for #{spec.full_name} is corrupt (#{e.class})"
  end
end

# File 'lib/rubygems/installer.rb', line 954

def ensure_writable_dir(dir) # :nodoc:
  begin
    Dir.mkdir dir, *[options[:dir_mode] && 0755].compact
  rescue SystemCallError
    raise unless File.directory? dir
  end

  raise Gem::FilePermissionError.new(dir) unless File.writable? dir
end

# File 'lib/rubygems/installer.rb', line 791

def explicit_version_requirement(name)
  code = "version = str"
  return code unless name == "bundler"

  code += <<-TEXT

  ENV['BUNDLER_VERSION'] = str
TEXT
end

# File 'lib/rubygems/installer.rb', line 860

def extract_bin
  @package.extract_files gem_dir, "#{spec.bindir}/*"
end

# File 'lib/rubygems/installer.rb', line 851

def extract_files
  @package.extract_files gem_dir
end

# File 'lib/rubygems/installer.rb', line 867

def formatted_program_filename(filename)
  if @format_executable
    self.class.exec_format % File.basename(filename)
  else
    filename
  end
end

# File 'lib/rubygems/installer.rb', line 888

def gem
  @package.gem.path
end

# File 'lib/rubygems/installer.rb', line 279

def gem_dir
  @gem_dir ||= File.join(gem_home, "gems", spec.full_name)
end

# File 'lib/rubygems/installer.rb', line 782

def gemdeps_load(name)
  return "" if name == "bundler"

  <<-TEXT

Gem.use_gemdeps
TEXT
end

# File 'lib/rubygems/installer.rb', line 485

def generate_bin # :nodoc:
  return if spec.executables.nil? || spec.executables.empty?

  ensure_writable_dir @bin_dir

  spec.executables.each do |filename|
    filename.tap(&Gem::UNTAINT)
    bin_path = File.join gem_dir, spec.bindir, filename
    next unless File.exist? bin_path

    mode = File.stat(bin_path).mode
    dir_mode = options[:prog_mode] || (mode | 0111)

    unless dir_mode == mode
      require "fileutils"
      FileUtils.chmod dir_mode, bin_path
    end

    check_executable_overwrite filename

    if @wrappers
      generate_bin_script filename, @bin_dir
    else
      generate_bin_symlink filename, @bin_dir
    end
  end
end

# File 'lib/rubygems/installer.rb', line 533

def generate_bin_script(filename, bindir)
  bin_script_path = File.join bindir, formatted_program_filename(filename)

  require "fileutils"
  FileUtils.rm_f bin_script_path # prior install may have been --no-wrappers

  File.open bin_script_path, "wb", 0755 do |file|
    file.print app_script_text(filename)
    file.chmod(options[:prog_mode] || 0755)
  end

  verbose bin_script_path

  generate_windows_script filename, bindir
end

# File 'lib/rubygems/installer.rb', line 553

def generate_bin_symlink(filename, bindir)
  src = File.join gem_dir, spec.bindir, filename
  dst = File.join bindir, formatted_program_filename(filename)

  if File.exist? dst
    if File.symlink? dst
      link = File.readlink(dst).split File::SEPARATOR
      cur_version = Gem::Version.create(link[-3].sub(/^.*-/, ""))
      return if spec.version < cur_version
    end
    File.unlink dst
  end

  FileUtils.symlink src, dst, :verbose => Gem.configuration.really_verbose
rescue NotImplementedError, SystemCallError
  alert_warning "Unable to use symlinks, installing wrapper"
  generate_bin_script filename, bindir
end

# File 'lib/rubygems/installer.rb', line 513

def generate_plugins # :nodoc:
  latest = Gem::Specification.latest_spec_for(spec.name)
  return if latest && latest.version > spec.version

  ensure_writable_dir @plugins_dir

  if spec.plugins.empty?
    remove_plugins_for(spec, @plugins_dir)
  else
    regenerate_plugins_for(spec, @plugins_dir)
  end
end

# File 'lib/rubygems/installer.rb', line 473

def generate_windows_script(filename, bindir)
  if Gem.win_platform?
    script_name = formatted_program_filename(filename) + ".bat"
    script_path = File.join bindir, File.basename(script_name)
    File.open script_path, "w" do |file|
      file.puts windows_stub_script(bindir, filename)
    end

    verbose script_path
  end
end

# File 'lib/rubygems/installer.rb', line 301

def install
  pre_install_checks

  run_pre_install_hooks

  # Set loaded_from to ensure extension_dir is correct
  if @options[:install_as_default]
    spec.loaded_from = default_spec_file
  else
    spec.loaded_from = spec_file
  end

  # Completely remove any previous gem files
  FileUtils.rm_rf gem_dir
  FileUtils.rm_rf spec.extension_dir

  dir_mode = options[:dir_mode]
  FileUtils.mkdir_p gem_dir, :mode => dir_mode && 0755

  if @options[:install_as_default]
    extract_bin
    write_default_spec
  else
    extract_files

    build_extensions
    write_build_info_file
    run_post_build_hooks
  end

  generate_bin
  generate_plugins

  unless @options[:install_as_default]
    write_spec
    write_cache_file
  end

  File.chmod(dir_mode, gem_dir) if dir_mode

  say spec.post_install_message if options[:post_install_message] && !spec.post_install_message.nil?

  Gem::Specification.add_spec(spec)

  run_post_install_hooks

  spec

# TODO This rescue is in the wrong place. What is raising this exception?
# move this rescue to around the code that actually might raise it.
rescue Zlib::GzipFile::Error
  raise Gem::InstallError, "gzip error installing #{gem}"
end

# File 'lib/rubygems/installer.rb', line 420

def installation_satisfies_dependency?(dependency)
  return true if @options[:development] && dependency.type == :development
  return true if installed_specs.detect {|s| dependency.matches_spec? s }
  return false if @only_install_dir
  !dependency.matching_specs.empty?
end

# File 'lib/rubygems/installer.rb', line 390

def installed_specs
  @specs ||= begin
    specs = []

    Gem::Util.glob_files_in_dir("*.gemspec", File.join(gem_home, "specifications")).each do |path|
      spec = Gem::Specification.load path.tap(&Gem::UNTAINT)
      specs << spec if spec
    end

    specs
  end
end

# File 'lib/rubygems/installer.rb', line 901

def pre_install_checks
  verify_gem_home

  # The name and require_paths must be verified first, since it could contain
  # ruby code that would be eval'ed in #ensure_loadable_spec
  verify_spec

  ensure_loadable_spec

  if options[:install_as_default]
    Gem.ensure_default_gem_subdirectories gem_home
  else
    Gem.ensure_gem_subdirectories gem_home
  end

  return true if @force

  ensure_dependencies_met unless @ignore_dependencies

  true
end

# File 'lib/rubygems/installer.rb', line 650

def process_options # :nodoc:
  @options = {
    :bin_dir      => nil,
    :env_shebang  => false,
    :force        => false,
    :only_install_dir => false,
    :post_install_message => true,
  }.merge options

  @env_shebang         = options[:env_shebang]
  @force               = options[:force]
  @install_dir         = options[:install_dir]
  @gem_home            = options[:install_dir] || Gem.dir
  @plugins_dir         = Gem.plugindir(@gem_home)
  @ignore_dependencies = options[:ignore_dependencies]
  @format_executable   = options[:format_executable]
  @wrappers            = options[:wrappers]
  @only_install_dir    = options[:only_install_dir]

  # If the user has asked for the gem to be installed in a directory that is
  # the system gem directory, then use the system bin directory, else create
  # (or use) a new bin dir under the gem_home.
  @bin_dir             = options[:bin_dir] || Gem.bindir(gem_home)
  @development         = options[:development]
  @build_root          = options[:build_root]

  @build_args = options[:build_args]

  unless @build_root.nil?
    @bin_dir = File.join(@build_root, @bin_dir.gsub(/^[a-zA-Z]:/, ""))
    @gem_home = File.join(@build_root, @gem_home.gsub(/^[a-zA-Z]:/, ""))
    @plugins_dir = File.join(@build_root, @plugins_dir.gsub(/^[a-zA-Z]:/, ""))
    alert_warning "You build with buildroot.\n  Build root: #{@build_root}\n  Bin dir: #{@bin_dir}\n  Gem home: #{@gem_home}\n  Plugins dir: #{@plugins_dir}"
  end
end

# File 'lib/rubygems/installer.rb', line 973

def rb_config
  RbConfig::CONFIG
end

# File 'lib/rubygems/installer.rb', line 977

def ruby_install_name
  rb_config["ruby_install_name"]
end

# File 'lib/rubygems/installer.rb', line 366

def run_post_build_hooks # :nodoc:
  Gem.post_build_hooks.each do |hook|
    if hook.call(self) == false
      FileUtils.rm_rf gem_dir

      location = " at #{$1}" if hook.inspect =~ /[ @](.*:\d+)/

      message = "post-build hook#{location} failed for #{spec.full_name}"
      raise Gem::InstallError, message
    end
  end
end

# File 'lib/rubygems/installer.rb', line 379

def run_post_install_hooks # :nodoc:
  Gem.post_install_hooks.each do |hook|
    hook.call self
  end
end

# File 'lib/rubygems/installer.rb', line 355

def run_pre_install_hooks # :nodoc:
  Gem.pre_install_hooks.each do |hook|
    if hook.call(self) == false
      location = " at #{$1}" if hook.inspect =~ /[ @](.*:\d+)/

      message = "pre-install hook#{location} failed for #{spec.full_name}"
      raise Gem::InstallError, message
    end
  end
end

# File 'lib/rubygems/installer.rb', line 587

def shebang(bin_file_name)
  path = File.join gem_dir, spec.bindir, bin_file_name
  first_line = File.open(path, "rb") {|file| file.gets } || ""

  if first_line.start_with?("#!")
    # Preserve extra words on shebang line, like "-w".  Thanks RPA.
    shebang = first_line.sub(/\A\#!.*?ruby\S*((\s\S)+)/, "#!#{Gem.ruby}")
    opts = $1
    shebang.strip! # Avoid nasty ^M issues.
  end

  if which = Gem.configuration[:custom_shebang]
    # replace bin_file_name with "ruby" to avoid endless loops
    which = which.gsub(/ #{bin_file_name}$/," #{ruby_install_name}")

    which = which.gsub(/\$(\w+)/) do
      case $1
      when "env"
        @env_path ||= ENV_PATHS.find {|env_path| File.executable? env_path }
      when "ruby"
        "#{Gem.ruby}#{opts}"
      when "exec"
        bin_file_name
      when "name"
        spec.name
      end
    end

    "#!#{which}"
  elsif @env_shebang
    # Create a plain shebang line.
    @env_path ||= ENV_PATHS.find {|env_path| File.executable? env_path }
    "#!#{@env_path} #{ruby_install_name}"
  else
    "#{bash_prolog_script}#!#{Gem.ruby}#{opts}"
  end
end

# File 'lib/rubygems/installer.rb', line 286

def spec
  @package.spec
end

# File 'lib/rubygems/installer.rb', line 440

def spec_file
  File.join gem_home, "specifications", "#{spec.full_name}.gemspec"
end

# File 'lib/rubygems/installer.rb', line 430

def unpack(directory)
  @gem_dir = directory
  extract_files
end

# File 'lib/rubygems/installer.rb', line 710

def verify_gem_home # :nodoc:
  FileUtils.mkdir_p gem_home, :mode => options[:dir_mode] && 0755
  raise Gem::FilePermissionError, gem_home unless File.writable?(gem_home)
end

# File 'lib/rubygems/installer.rb', line 715

def verify_spec
  unless spec.name =~ Gem::Specification::VALID_NAME_PATTERN
    raise Gem::InstallError, "#{spec} has an invalid name"
  end

  if spec.raw_require_paths.any? {|path| path =~ /\R/ }
    raise Gem::InstallError, "#{spec} has an invalid require_paths"
  end

  if spec.extensions.any? {|ext| ext =~ /\R/ }
    raise Gem::InstallError, "#{spec} has an invalid extensions"
  end

  if spec.platform.to_s =~ /\R/
    raise Gem::InstallError, "#{spec.platform} is an invalid platform"
  end

  unless spec.specification_version.to_s =~ /\A\d+\z/
    raise Gem::InstallError, "#{spec} has an invalid specification_version"
  end

  if spec.dependencies.any? {|dep| dep.type != :runtime && dep.type != :development }
    raise Gem::InstallError, "#{spec} has an invalid dependencies"
  end

  if spec.dependencies.any? {|dep| dep.name =~ /(?:\R|[<>])/ }
    raise Gem::InstallError, "#{spec} has an invalid dependencies"
  end
end

# File 'lib/rubygems/installer.rb', line 804

def windows_stub_script(bindir, bin_file_name)
  rb_topdir = RbConfig::TOPDIR || File.dirname(rb_config["bindir"])

  # get ruby executable file name from RbConfig
  ruby_exe = "#{rb_config['RUBY_INSTALL_NAME']}#{rb_config['EXEEXT']}"
  ruby_exe = "ruby.exe" if ruby_exe.empty?

  if File.exist?(File.join bindir, ruby_exe)
    # stub & ruby.exe within same folder.  Portable
    <<-TEXT
@ECHO OFF
@"%~dp0#{ruby_exe}" "%~dpn0" %*
    TEXT
  elsif bindir.downcase.start_with? rb_topdir.downcase
    # stub within ruby folder, but not standard bin.  Portable
    require "pathname"
    from = Pathname.new bindir
    to   = Pathname.new "#{rb_topdir}/bin"
    rel  = to.relative_path_from from
    <<-TEXT
@ECHO OFF
@"%~dp0#{rel}/#{ruby_exe}" "%~dpn0" %*
    TEXT
  else
    # outside ruby folder, maybe -user-install or bundler.  Portable, but ruby
    # is dependent on PATH
    <<-TEXT
@ECHO OFF
@#{ruby_exe} "%~dpn0" %*
    TEXT
  end
end

# File 'lib/rubygems/installer.rb', line 927

def write_build_info_file
  return if build_args.empty?

  build_info_dir = File.join gem_home, "build_info"

  dir_mode = options[:dir_mode]
  FileUtils.mkdir_p build_info_dir, :mode => dir_mode && 0755

  build_info_file = File.join build_info_dir, "#{spec.full_name}.info"

  File.open build_info_file, "w" do |io|
    build_args.each do |arg|
      io.puts arg
    end
  end

  File.chmod(dir_mode, build_info_dir) if dir_mode
end

# File 'lib/rubygems/installer.rb', line 949

def write_cache_file
  cache_file = File.join gem_home, "cache", spec.file_name
  @package.copy_to cache_file
end

# File 'lib/rubygems/installer.rb', line 466

def write_default_spec
  Gem.write_binary(default_spec_file, spec.to_ruby)
end

# File 'lib/rubygems/installer.rb', line 456

def write_spec
  spec.installed_by_version = Gem.rubygems_version

  Gem.write_binary(spec_file, spec.to_ruby_for_cache)
end