class Autotest::Rspec2

def bundle_exec
  using_bundler? ? "bundle exec " : ""
end

def consolidate_failures(failed)
  filters = new_hash_of_arrays
  failed.each do |spec, trace|
    if trace =~ /(.*spec\.rb)/
      filters[$1] << spec
    end
  end
  return filters
end

def initialize
  super
  clear_mappings
  setup_rspec_project_mappings
  # Example for Ruby 1.8: http://rubular.com/r/AOXNVDrZpx
  # Example for Ruby 1.9: http://rubular.com/r/85ag5AZ2jP
  self.failed_results_re = /^\s*\d+\).*\n\s+Failure.*(\n\s+#\s(.*)?:\d+(?::.*)?)+$/m
  self.completed_re = /\n(?:\e\[\d*m)?\d* examples?/m
end

def make_test_cmd(files_to_test)
  files_to_test.empty? ? '' :
    "#{bundle_exec}#{ruby} #{require_rubygems}-S #{SPEC_PROGRAM} --autotest #{normalize(files_to_test).keys.flatten.map { |f| "'#{f}'"}.join(' ')}"
end

def normalize(files_to_test)
  files_to_test.keys.inject({}) do |result, filename|
    result[File.expand_path(filename)] = []
    result
  end
end

def require_rubygems
  using_bundler? ? "" : defined?(:Gem) ? "-rrubygems " : " "
end

def setup_rspec_project_mappings
  add_mapping(%r%^spec/.*_spec\.rb$%) { |filename, _|
    filename
  }
  add_mapping(%r%^lib/(.*)\.rb$%) { |_, m|
    ["spec/#{m[1]}_spec.rb"]
  }
  add_mapping(%r%^spec/(spec_helper|shared/.*)\.rb$%) {
    files_matching %r%^spec/.*_spec\.rb$%
  }
end

def using_bundler?
  File.exists?('./Gemfile')
end