class RSpec::Core::RakeTask

def blank
  lambda {|s| s.nil? || s == ""}
end

def define(args, &task_block)
  desc "Run RSpec code examples" unless ::Rake.application.last_comment
  task name, *args do |_, task_args|
    RakeFileUtils.__send__(:verbose, verbose) do
      task_block.call(*[self, task_args].slice(0, task_block.arity)) if task_block
      run_task verbose
    end
  end
end

def files_to_run
  if ENV['SPEC']
    FileList[ ENV['SPEC'] ].sort
  else
    FileList[ pattern ].sort.map(&:shellescape)
  end
end

def initialize(*args, &task_block)
  @name          = args.shift || :spec
  @ruby_opts     = nil
  @rspec_opts    = nil
  @verbose       = true
  @fail_on_error = true
  @rspec_path    = DEFAULT_RSPEC_PATH
  @pattern       = DEFAULT_PATTERN
  define(args, &task_block)
end

def rspec_load_path
  @rspec_load_path ||= begin
    core_and_support = $LOAD_PATH.grep \
      %r{#{File::SEPARATOR}rspec-(core|support)[^#{File::SEPARATOR}]*#{File::SEPARATOR}lib}
    "-I#{core_and_support.map(&:shellescape).join(File::PATH_SEPARATOR)}"
  end
end

def run_task(verbose)
  command = spec_command
  begin
    puts command if verbose
    success = system(command)
  rescue
    puts failure_message if failure_message
  end
  if fail_on_error && !success
    $stderr.puts "#{command} failed"
    exit $?.exitstatus
  end
end

def spec_command
  cmd_parts = []
  cmd_parts << RUBY
  cmd_parts << ruby_opts
  cmd_parts << rspec_load_path
  cmd_parts << "-S" << rspec_path
  cmd_parts << files_to_run
  cmd_parts << rspec_opts
  cmd_parts.flatten.reject(&blank).join(" ")
end