class Cucumber::Formatter::Steps

def after_features(features)
  print_summary
end

def collect_steps(step_mother)
  step_mother.step_definitions.inject({}) do |step_definitions, step_definition|
    step_definitions[step_definition.file] ||= []
    step_definitions[step_definition.file] << [ step_definition.file_colon_line, step_definition.regexp_source ]
    step_definitions
  end
end

def initialize(step_mother, path_or_io, options)
  @io = ensure_io(path_or_io, "steps")
  @options = options
  @step_definition_files = collect_steps(step_mother)
end

def print_summary
  count = 0
  @step_definition_files.keys.sort.each do |step_definition_file|
    @io.puts step_definition_file
    
    sources = @step_definition_files[step_definition_file]
    source_indent = source_indent(sources)
    sources.sort.each do |file_colon_line, regexp_source|
      @io.print regexp_source.indent(2)
      @io.print " # #{file_colon_line}".indent(source_indent - regexp_source.jlength)
      @io.puts
    end
    @io.puts
    count += sources.size
  end
  @io.puts "#{count} step definition(s) in #{@step_definition_files.size} source file(s)."
end

def source_indent(sources)
  sources.map { |file_colon_line, regexp| regexp.size }.max + 1
end