class Cucumber::Formatter::Rerun

def after_examples(*args)
  @in_examples = false
end

def after_feature(*)
  unless @lines.empty?
    after_first_time do
      @io.print ' '
    end
    @io.print "#{@file}:#{@lines.join(':')}"
    @io.flush
  end
end

def after_feature_element(feature_element)
  if (@rerun || feature_element.failed?) && !(Ast::ScenarioOutline === feature_element)
    @lines << feature_element.line
  end
end

def after_features(features)
  @io.close
end

def after_first_time
  yield if @not_first_time
  @not_first_time = true
end

def after_table_row(table_row)
  return unless @in_examples and Cucumber::Ast::OutlineTable::ExampleRow === table_row
  unless @header_row
    if table_row.failed?
      @rerun = true
      @lines << table_row.line
    end
  end
  @header_row = false if @header_row
end

def before_examples(*args)
  @header_row = true
  @in_examples = true
end

def before_feature(feature_element)
  @lines = []
  @file = feature_element.file
end

def before_feature_element(feature_element)
  @rerun = false
end

def before_table_row(table_row)
  return unless @in_examples
end

def initialize(runtime, path_or_io, options)
  @io = ensure_io(path_or_io, "rerun")
  @options = options
  @file_names = []
  @file_colon_lines = Hash.new{|h,k| h[k] = []}
end

def step_name(keyword, step_match, status, source_indent, background, file_colon_line)
  @rerun = true if [:failed, :pending, :undefined].index(status)
end