module Cucumber::Formatter::Summary

def dump_count(count, what, state=nil)
  [count, state, "#{what}#{count == 1 ? '' : 's'}"].compact.join(" ")
end

def dump_status_counts(find_elements_proc)
  counts = [:failed, :skipped, :undefined, :pending, :passed].map do |status|
    elements = find_elements_proc.call(status)
    elements.any? ? yield("#{elements.length} #{status.to_s}", status) : nil
  end.compact
  if counts.any?
    " (#{counts.join(', ')})"
  else
    ""
  end
end

def scenario_summary(step_mother, &block)
  scenarios_proc = lambda{|status| step_mother.scenarios(status)}
  dump_count(step_mother.scenarios.length, "scenario") + dump_status_counts(scenarios_proc, &block)
end

def step_summary(step_mother, &block)
  steps_proc = lambda{|status| step_mother.steps(status)}
  dump_count(step_mother.steps.length, "step") + dump_status_counts(steps_proc, &block)
end