class Gem::InstallerTestCase

def setup
  super
  Gem::Installer.path_warning = false
end

def setup_base_gem
  @spec = setup_base_spec
  util_build_gem @spec
  @spec.cache_file
end

def setup_base_installer(force = true)
  @gem = setup_base_gem
  util_installer @spec, @gemhome, false, force
end

def setup_base_spec
  quick_gem 'a' do |spec|
    util_make_exec spec
  end
end

def setup_base_user_installer
  @user_spec = quick_gem 'b' do |spec|
    util_make_exec spec
  end
  util_build_gem @user_spec
  @user_gem = @user_spec.cache_file
  util_installer @user_spec, Gem.user_dir, :user
end

def symlink_supported?
  if @@symlink_supported.nil?
    begin
      File.symlink("", "")
    rescue Errno::ENOENT, Errno::EEXIST
      @@symlink_supported = true
    rescue NotImplementedError, SystemCallError
      @@symlink_supported = false
    end
  end
  @@symlink_supported
end

def util_inst_bindir
  File.join @gemhome, "bin"
end

def util_installer(spec, gem_home, user=false, force=true)
  Gem::Installer.at(spec.cache_file,
                     :install_dir => gem_home,
                     :user_install => user,
                     :force => force)
end

def util_make_exec(spec = @spec, shebang = "#!/usr/bin/ruby", bindir = "bin")
  spec.executables = %w[executable]
  spec.bindir = bindir
  exec_path = spec.bin_file "executable"
  write_file exec_path do |io|
    io.puts shebang
  end
  bin_path = File.join @tempdir, "bin", "executable"
  write_file bin_path do |io|
    io.puts shebang
  end
end

def util_setup_gem(ui = @ui, force = true)
  @spec.files << File.join('lib', 'code.rb')
  @spec.extensions << File.join('ext', 'a', 'mkrf_conf.rb')
  Dir.chdir @tempdir do
    FileUtils.mkdir_p 'bin'
    FileUtils.mkdir_p 'lib'
    FileUtils.mkdir_p File.join('ext', 'a')
    File.open File.join('bin', 'executable'), 'w' do |f|
      f.puts "raise 'ran executable'"
    end
    File.open File.join('lib', 'code.rb'), 'w' do |f|
      f.puts '1'
    end
    File.open File.join('ext', 'a', 'mkrf_conf.rb'), 'w' do |f|
      f << <<-EOF
        File.open 'Rakefile', 'w' do |rf| rf.puts "task :default" end
      EOF
    end
    yield @spec if block_given?
    use_ui ui do
      FileUtils.rm_f @gem
      @gem = Gem::Package.build @spec
    end
  end
  Gem::Installer.at @gem, :force => force
end

def util_setup_installer(&block)
  @gem = setup_base_gem
  util_setup_gem(&block)
end