class Bundler::ParallelInstaller

def self.call(*args)
  new(*args).call
end

def call
  # Since `autoload` has the potential for threading issues on 1.8.7
  # TODO:  remove in bundler 2.0
  require "bundler/gem_remote_fetcher" if RUBY_VERSION < "1.9"
  check_for_corrupt_lockfile
  if @size > 1
    install_with_worker
  else
    install_serially
  end
  handle_error if @specs.any?(&:failed?)
  @specs
ensure
  worker_pool && worker_pool.stop
end

def check_for_corrupt_lockfile
  missing_dependencies = @specs.map do |s|
    [
      s,
      s.missing_lockfile_dependencies(@specs.map(&:name)),
    ]
  end.reject { |a| a.last.empty? }
  return if missing_dependencies.empty?
  warning = []
  warning << "Your lockfile was created by an old Bundler that left some things out."
  if @size != 1
    warning << "Because of the missing DEPENDENCIES, we can only install gems one at a time, instead of installing #{@size} at a time."
    @size = 1
  end
  warning << "You can fix this by adding the missing gems to your Gemfile, running bundle install, and then removing the gems from your Gemfile."
  warning << "The missing gems are:"
  missing_dependencies.each do |spec, missing|
    warning << "* #{missing.map(&:name).join(", ")} depended upon by #{spec.name}"
  end
  Bundler.ui.warn(warning.join("\n"))
end

def do_install(spec_install, worker_num)
  Plugin.hook(Plugin::Events::GEM_BEFORE_INSTALL, spec_install)
  gem_installer = Bundler::GemInstaller.new(
    spec_install.spec, @installer, @standalone, worker_num, @force
  )
  success, message = begin
    gem_installer.install_from_spec
  rescue RuntimeError => e
    raise e, "#{e}\n\n#{require_tree_for_spec(spec_install.spec)}"
  end
  if success
    spec_install.state = :installed
    spec_install.post_install_message = message unless message.nil?
  else
    spec_install.state = :failed
    spec_install.error = "#{message}\n\n#{require_tree_for_spec(spec_install.spec)}"
  end
  Plugin.hook(Plugin::Events::GEM_AFTER_INSTALL, spec_install)
  spec_install
end

def enqueue_specs
  @specs.select(&:ready_to_enqueue?).each do |spec|
    next if @rake && !@rake.installed? && spec.name != @rake.name
    if spec.dependencies_installed? @specs
      spec.state = :enqueued
      worker_pool.enq spec
    end
  end
end

def finished_installing?
  @specs.all? do |spec|
    return true if spec.failed?
    spec.installed?
  end
end

def handle_error
  errors = @specs.select(&:failed?).map(&:error)
  if exception = errors.find {|e| e.is_a?(Bundler::BundlerError) }
    raise exception
  end
  raise Bundler::InstallError, errors.map(&:to_s).join("\n\n")
end

def initialize(installer, all_specs, size, standalone, force)
  @installer = installer
  @size = size
  @standalone = standalone
  @force = force
  @specs = all_specs.map {|s| SpecInstallation.new(s) }
  @spec_set = all_specs
  @rake = @specs.find {|s| s.name == "rake" }
end

def install_serially
  until finished_installing?
    raise "failed to find a spec to enqueue while installing serially" unless spec_install = @specs.find(&:ready_to_enqueue?)
    spec_install.state = :enqueued
    do_install(spec_install, 0)
  end
end

def install_with_worker
  enqueue_specs
  process_specs until finished_installing?
end

def process_specs
  worker_pool.deq
  enqueue_specs
end

def require_tree_for_spec(spec)
  tree = @spec_set.what_required(spec)
  t = String.new("In #{File.basename(SharedHelpers.default_gemfile)}:\n")
  tree.each_with_index do |s, depth|
    t << "  " * depth.succ << s.name
    unless tree.last == s
      t << %( was resolved to #{s.version}, which depends on)
    end
    t << %(\n)
  end
  t
end

def worker_pool
  @worker_pool ||= Bundler::Worker.new @size, "Parallel Installer", lambda { |spec_install, worker_num|
    do_install(spec_install, worker_num)
  }
end