class Minitest::TestProf::EventProfReporter

def before_test(test)
  prerecord(test.class, test.name)
end

def change_current_group(group, example)
  @current_group = {
    name: group.name,
    location: location_without_line_number(group, example)
  }
  @profiler.group_started(@current_group)
end

def configure_profiler(options)
  ::TestProf::EventProf.configure do |config|
    config.event = options[:event]
    config.rank_by = options[:rank_by] if options[:rank_by]
    config.top_count = options[:top_count] if options[:top_count]
    config.per_example = options[:per_example] if options[:per_example]
    ::TestProf::EventProf::CustomEvents.activate_all config.event
  end
  ::TestProf::EventProf.build
end

def initialize(io = $stdout, options = {})
  super
  @profiler = configure_profiler(options)
  log :info, "EventProf enabled (#{@profiler.events.join(", ")})"
  @formatter = EventProfFormatter.new(@profiler)
  @current_group = nil
  @current_example = nil
end

def prerecord(group, example)
  change_current_group(group, example) unless @current_group
  track_current_example(group, example)
end

def record(*)
  @profiler.example_finished(@current_example)
end

def report
  @profiler.group_finished(@current_group)
  result = @formatter.prepare_results
  log :info, result
end

def track_current_example(group, example)
  unless @current_group[:name] == group.name
    @profiler.group_finished(@current_group)
    change_current_group(group, example)
  end
  @current_example = {
    name: example.gsub(/^test_(?:\d+_)?/, ""),
    location: location_with_line_number(group, example)
  }
  @profiler.example_started(@current_example)
end