class Cucumber::Formatter::Json

# File lib/cucumber/formatter/json.rb, line 15
def initialize(config)
  @io = ensure_io(config.out_stream, config.error_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)
  config.on_event :attach_called, &method(:on_attach_called)
end

# File lib/cucumber/formatter/json.rb, line 91
def on_attach_called(event)
  if event.media_type == 'text/x.cucumber.log+plain'
    test_step_output << event.src
    return
  end
  if event.media_type =~ /;base64$/
    media_type = event.media_type[0..-8]
    data = event.src
  else
    data = encode64(event.src)
    media_type = event.media_type
  end
  test_step_embeddings << { mime_type: media_type, data: data }
end

# File lib/cucumber/formatter/json.rb, line 78
def on_test_case_finished(event)
  feature_elements << @test_case_hash if @in_background

  _test_case = event.test_case
  result = event.result
  result = result.with_filtered_backtrace(Cucumber::Formatter::BacktraceFilter)
  add_failed_around_hook(result) if result.failed? && !@any_step_failed
end

# File lib/cucumber/formatter/json.rb, line 28
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

  @element_hash = nil
  @element_background_hash = builder.background_hash
  @in_background = builder.background?

  @any_step_failed = false
end

# File lib/cucumber/formatter/json.rb, line 87
def on_test_run_finished(_event)
  @io.write(JSON.pretty_generate(@feature_hashes))
end

# File lib/cucumber/formatter/json.rb, line 68
def on_test_step_finished(event)
  test_step = event.test_step
  result = event.result
  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

# File lib/cucumber/formatter/json.rb, line 44
def on_test_step_started(event)
  test_step = event.test_step
  return if internal_hook?(test_step)

  if @element_hash.nil?
    @element_hash = create_element_hash(test_step)
    feature_elements << @element_hash
  end

  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

# File lib/cucumber/formatter/json.rb, line 205
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

# File lib/cucumber/formatter/json.rb, line 199
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)
  result.embeddings.each { |e| embed(e['src'], e['mime_type'], e['label']) } if result.respond_to?(:embeddings)
end

# File lib/cucumber/formatter/json.rb, line 145
def after_hooks
  @element_hash[:after] ||= []
end

# File lib/cucumber/formatter/json.rb, line 153
def after_step_hooks
  @step_hash[:after] ||= []
end

# File lib/cucumber/formatter/json.rb, line 149
def around_hooks
  @element_hash[:around] ||= []
end

# File lib/cucumber/formatter/json.rb, line 141
def before_hooks
  @element_hash[:before] ||= []
end

# File lib/cucumber/formatter/json.rb, line 193
def create_data_table_value(data_table)
  data_table.rows.map do |row|
    { cells: row.cells.map(&:value) }
  end
end

# File lib/cucumber/formatter/json.rb, line 184
def create_doc_string_hash(doc_string, doc_string_content)
  content_type = doc_string.media_type || ''
  {
    value: doc_string_content,
    content_type: content_type,
    line: doc_string.location.line
  }
end

# File lib/cucumber/formatter/json.rb, line 165
def create_element_hash(test_step)
  return @element_background_hash if @in_background && !first_step_after_background?(test_step)

  @in_background = false
  @test_case_hash
end

# File lib/cucumber/formatter/json.rb, line 226
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

# File lib/cucumber/formatter/json.rb, line 213
def create_match_hash(test_step, _result)
  { location: test_step.action_location.to_s }
end

# File lib/cucumber/formatter/json.rb, line 217
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

# File lib/cucumber/formatter/json.rb, line 172
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, test_step.multiline_arg.content) 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

# File lib/cucumber/formatter/json.rb, line 232
def encode64(data)
  # strip newlines from the encoded data
  Base64.encode64(data).delete("\n")
end

# File lib/cucumber/formatter/json.rb, line 120
def feature_elements
  @feature_hash[:elements] ||= []
end

# File lib/cucumber/formatter/json.rb, line 112
def first_step_after_background?(test_step)
  @in_background && test_step.location.file == @feature_hash[:uri] && test_step.location.lines.max >= @test_case_hash[:line]
end

# File lib/cucumber/formatter/json.rb, line 128
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}"
  end
end

# File lib/cucumber/formatter/json.rb, line 116
def internal_hook?(test_step)
  test_step.location.file.include?('lib/cucumber/')
end

# File lib/cucumber/formatter/json.rb, line 108
def same_feature_as_previous_test_case?(test_case)
  @feature_hash&.fetch(:uri, nil) == test_case.location.file
end

# File lib/cucumber/formatter/json.rb, line 124
def steps
  @element_hash[:steps] ||= []
end

# File lib/cucumber/formatter/json.rb, line 161
def test_step_embeddings
  @step_or_hook_hash[:embeddings] ||= []
end

# File lib/cucumber/formatter/json.rb, line 157
def test_step_output
  @step_or_hook_hash[:output] ||= []
end