class Buildkite::TestCollector::RSpecPlugin::Reporter

def blank?(string)
  string.nil? || string.strip.empty?
end

def dump_summary(_notification)
  Buildkite::TestCollector.session.send_remaining_data
  Buildkite::TestCollector.session.close
end

def failure_info(notification)
  failure_expanded = []
  if Buildkite::TestCollector::RSpecPlugin::Reporter::MULTIPLE_ERRORS.include?(notification.exception.class)
    failure_reason = notification.exception.summary
    notification.exception.all_exceptions.each do |exception|
      # an example with multiple failures doesn't give us a
      # separate message lines and backtrace object to send, so
      # I've reached into RSpec internals and duplicated the
      # construction of these
      message_lines = RSpec::Core::Formatters::ExceptionPresenter.new(exception, notification.example).colorized_message_lines
      failure_expanded << {
        expanded: format_message_lines(message_lines),
        backtrace:  RSpec.configuration.backtrace_formatter.format_backtrace(exception.backtrace)
      }
    end
  else
    message_lines = notification.colorized_message_lines
    failure_reason = strip_diff_colors(message_lines.shift)
    failure_expanded << {
      expanded:  format_message_lines(message_lines),
      backtrace: notification.formatted_backtrace
    }
  end
  return failure_reason, failure_expanded
end

def format_message_lines(message_lines)
  message_lines.map! { |l| strip_diff_colors(l) }
  # the first line is sometimes blank, depending on the error reported
  message_lines.shift if blank?(message_lines.first)
  # the last line is sometimes blank, depending on the error reported
  message_lines.pop if blank?(message_lines.last)
  message_lines
end

def handle_example(notification)
  example = notification.example
  trace = Buildkite::TestCollector.uploader.traces[example.id]
  if trace
    trace.example = example
    if example.execution_result.status == :failed
      trace.failure_reason, trace.failure_expanded = failure_info(notification)
    end
    Buildkite::TestCollector.session.add_example_to_send_queue(example.id)
  end
end

def initialize(output)
  Buildkite::TestCollector.session = Buildkite::TestCollector::Session.new
  @output = output
end

def strip_diff_colors(string)
  string.gsub(/\e\[([;\d]+)?m/, '')
end