class RSpec::Core::Notifications::SummaryNotification

def colorized_rerun_commands(colorizer=::RSpec::Core::Formatters::ConsoleCodes)
  "\nFailed examples:\n\n" +
  failed_examples.map do |example|
    colorizer.wrap("rspec #{rerun_argument_for(example)}", RSpec.configuration.failure_color) + " " +
    colorizer.wrap("# #{example.full_description}",   RSpec.configuration.detail_color)
  end.join("\n")
end

def colorized_totals_line(colorizer=::RSpec::Core::Formatters::ConsoleCodes)
  if failure_count > 0
    colorizer.wrap(totals_line, RSpec.configuration.failure_color)
  elsif pending_count > 0
    colorizer.wrap(totals_line, RSpec.configuration.pending_color)
  else
    colorizer.wrap(totals_line, RSpec.configuration.success_color)
  end
end

def duplicate_rerun_locations
  @duplicate_rerun_locations ||= begin
    locations = RSpec.world.all_examples.map(&:location_rerun_argument)
    Set.new.tap do |s|
      locations.group_by { |l| l }.each do |l, ls|
        s << l if ls.count > 1
      end
    end
  end
end

def example_count
  @example_count ||= examples.size
end

def failure_count
  @failure_count ||= failed_examples.size
end

def formatted_duration
  Formatters::Helpers.format_duration(duration)
end

def formatted_load_time
  Formatters::Helpers.format_duration(load_time)
end

def fully_formatted(colorizer=::RSpec::Core::Formatters::ConsoleCodes)
  formatted = "\nFinished in #{formatted_duration} " \
              "(files took #{formatted_load_time} to load)\n" \
              "#{colorized_totals_line(colorizer)}\n"
  unless failed_examples.empty?
    formatted << colorized_rerun_commands(colorizer) << "\n"
  end
  formatted
end

def pending_count
  @pending_count ||= pending_examples.size
end

def rerun_argument_for(example)
  location = example.location_rerun_argument
  return location unless duplicate_rerun_locations.include?(location)
  conditionally_quote(example.id)
end

def totals_line
  summary = Formatters::Helpers.pluralize(example_count, "example")
  summary << ", " << Formatters::Helpers.pluralize(failure_count, "failure")
  summary << ", #{pending_count} pending" if pending_count > 0
  summary
end