class Cucumber::Formatter::Json

def add_failed_around_hook(result)
  @step_or_hook_hash = {}
  around_hooks << @step_or_hook_hash
  @step_or_hook_hash[:match] = { location: 'unknown_hook_location:1' }
  @step_or_hook_hash[:result] = create_result_hash(result)
end

def add_match_and_result(test_step, result)
  @step_or_hook_hash[:match] = create_match_hash(test_step, result)
  @step_or_hook_hash[:result] = create_result_hash(result)
end

def after_hooks
  @element_hash[:after] ||= []
end

def after_step_hooks
  @step_hash[:after] ||= []
end

def around_hooks
  @element_hash[:around] ||= []
end

def before_hooks
  @element_hash[:before] ||= []
end

def create_data_table_value(data_table)
  data_table[:rows].map do |row|
    { cells: row[:cells].map { |cell| cell[:value] } }
  end
end

def create_doc_string_hash(doc_string)
  content_type = doc_string[:content_type] || ''
  {
    value: doc_string[:content],
    content_type: content_type,
    line: doc_string[:location][:line]
  }
end

def create_error_message(result)
  message_element = result.failed? ? result.exception : result
  message = "#{message_element.message} (#{message_element.class})"
  ([message] + message_element.backtrace).join("\n")
end

def create_match_hash(test_step, _result)
  { location: test_step.action_location.to_s }
end

def create_result_hash(result)
  result_hash = {
    status: result.to_sym
  }
  result_hash[:error_message] = create_error_message(result) if result.failed? || result.pending?
  result.duration.tap { |duration| result_hash[:duration] = duration.nanoseconds }
  result_hash
end

def create_step_hash(test_step)
  step_source = @ast_lookup.step_source(test_step).step
  step_hash = {
    keyword: step_source[:keyword],
    name: test_step.text,
    line: test_step.location.lines.min
  }
  step_hash[:doc_string] = create_doc_string_hash(step_source[:doc_string]) unless step_source[:doc_string].nil?
  step_hash[:rows] = create_data_table_value(step_source[:data_table]) unless step_source[:data_table].nil?
  step_hash
end

def current_feature
  @feature_hash ||= {} # rubocop:disable Naming/MemoizedInstanceVariableName
end

def embed(src, mime_type, _label)
  if File.file?(src)
    content = File.open(src, 'rb', &:read)
    data = encode64(content)
  elsif mime_type =~ /;base64$/
    mime_type = mime_type[0..-8]
    data = src
  else
    data = encode64(src)
  end
  test_step_embeddings << { mime_type: mime_type, data: data }
end

def encode64(data)
  # strip newlines from the encoded data
  Base64.encode64(data).delete("\n")
end

def feature_elements
  @feature_hash[:elements] ||= []
end

def first_step_after_background?(test_step)
  @in_background && test_step.location.lines.max >= @test_case_hash[:line]
end

def hooks_of_type(hook_step)
  case hook_step.text
  when 'Before hook'
    before_hooks
  when 'After hook'
    after_hooks
  when 'AfterStep hook'
    after_step_hooks
  else
    raise 'Unknown hook type ' + hook_step.to_s
  end
end

def initialize(config)
  @io = ensure_io(config.out_stream)
  @ast_lookup = AstLookup.new(config)
  @feature_hashes = []
  @step_or_hook_hash = {}
  config.on_event :test_case_started, &method(:on_test_case_started)
  config.on_event :test_case_finished, &method(:on_test_case_finished)
  config.on_event :test_step_started, &method(:on_test_step_started)
  config.on_event :test_step_finished, &method(:on_test_step_finished)
  config.on_event :test_run_finished, &method(:on_test_run_finished)
end

def internal_hook?(test_step)
  test_step.location.file.include?('lib/cucumber/')
end

def on_test_case_finished(event)
  _test_case, result = *event.attributes
  result = result.with_filtered_backtrace(Cucumber::Formatter::BacktraceFilter)
  add_failed_around_hook(result) if result.failed? && !@any_step_failed
end

def on_test_case_started(event)
  test_case = event.test_case
  builder = Builder.new(test_case, @ast_lookup)
  unless same_feature_as_previous_test_case?(test_case)
    @feature_hash = builder.feature_hash
    @feature_hashes << @feature_hash
  end
  @test_case_hash = builder.test_case_hash
  if builder.background?
    @in_background = true
    feature_elements << builder.background_hash
    @element_hash = builder.background_hash
  else
    @in_background = false
    feature_elements << @test_case_hash
    @element_hash = @test_case_hash
  end
  @any_step_failed = false
end

def on_test_run_finished(_event)
  @io.write(MultiJson.dump(@feature_hashes, pretty: true))
end

def on_test_step_finished(event)
  test_step, result = *event.attributes
  result = result.with_filtered_backtrace(Cucumber::Formatter::BacktraceFilter)
  return if internal_hook?(test_step)
  add_match_and_result(test_step, result)
  @any_step_failed = true if result.failed?
end

def on_test_step_started(event)
  test_step = event.test_step
  return if internal_hook?(test_step)
  if test_step.hook?
    @step_or_hook_hash = {}
    hooks_of_type(test_step) << @step_or_hook_hash
    return
  end
  if first_step_after_background?(test_step)
    @in_background = false
    feature_elements << @test_case_hash
    @element_hash = @test_case_hash
  end
  @step_or_hook_hash = create_step_hash(test_step)
  steps << @step_or_hook_hash
  @step_hash = @step_or_hook_hash
end

def puts(message)
  test_step_output << message
end

def same_feature_as_previous_test_case?(test_case)
  current_feature[:uri] == test_case.location.file
end

def steps
  @element_hash[:steps] ||= []
end

def test_step_embeddings
  @step_or_hook_hash[:embeddings] ||= []
end

def test_step_output
  @step_or_hook_hash[:output] ||= []
end