class Bundler::Retry

def attempt(&block)
  @current_run = 0
  @failed      = false
  @error       = nil
  while keep_trying? do
    run(&block)
  end
  @result
end

def default_attempts
  return Integer(self.class.attempts) if self.class.attempts
  DEFAULT_ATTEMPTS
end

def fail(e)
  @failed = true
  raise e if last_attempt? || @exceptions.any?{|k| k === e }
  return true unless name
  Bundler.ui.warn "Retrying#{" #{name}" if name} due to error (#{current_run.next}/#{total_runs}): #{e.class} #{e.message}"
end

def initialize(name, exceptions = nil, attempts = nil)
  @name        = name
  attempts    ||= default_attempts
  @exceptions = Array(exceptions) || []
  @total_runs =  attempts.next # will run once, then upto attempts.times
end

def keep_trying?
  return true  if current_run.zero?
  return false if last_attempt?
  return true  if @failed
end

def last_attempt?
  current_run >= total_runs
end

def run(&block)
  @failed      = false
  @current_run += 1
  @result = block.call
rescue => e
  fail(e)
end