class Cucumber::Query

# File lib/cucumber/query.rb, line 19
def initialize(repository)
  @repository = repository
end

# File lib/cucumber/query.rb, line 40
def count_test_cases_started
  find_all_test_case_started.length
end

# File lib/cucumber/query.rb, line 48
def find_all_pickle_steps
  repository.pickle_step_by_id.values
end

# File lib/cucumber/query.rb, line 44
def find_all_pickles
  repository.pickle_by_id.values
end

# File lib/cucumber/query.rb, line 52
def find_all_step_definitions
  repository.step_definition_by_id.values
end

# File lib/cucumber/query.rb, line 67
def find_all_test_case_finished
  repository.test_case_finished_by_test_case_started_id.values.reject(&:will_be_retried)
end

# File lib/cucumber/query.rb, line 59
def find_all_test_case_started
  repository.test_case_started_by_id.values.select do |test_case_started|
    test_case_finished = find_test_case_finished_by(test_case_started)
    test_case_finished.nil? || !test_case_finished.will_be_retried
  end
end

# File lib/cucumber/query.rb, line 71
def find_all_test_cases
  repository.test_case_by_id.values
end

# File lib/cucumber/query.rb, line 79
def find_all_test_run_hook_finished
  repository.test_run_hook_finished_by_test_run_hook_started_id.values
end

# File lib/cucumber/query.rb, line 75
def find_all_test_run_hook_started
  repository.test_run_hook_started_by_id.values
end

# File lib/cucumber/query.rb, line 87
def find_all_test_step_finished
  repository.test_steps_finished_by_test_case_started_id.values.flatten
end

# File lib/cucumber/query.rb, line 83
def find_all_test_step_started
  repository.test_steps_started_by_test_case_started_id.values.flatten
end

# File lib/cucumber/query.rb, line 91
def find_all_test_steps
  repository.test_step_by_id.values
end

# File lib/cucumber/query.rb, line 97
def find_hook_by(message)
  ensure_only_message_types!(message, %i[test_step test_run_hook_started test_run_hook_finished], '#find_hook_by')

  if message.is_a?(Cucumber::Messages::TestRunHookFinished)
    test_run_hook_started_message = find_test_run_hook_started_by(message)
    test_run_hook_started_message ? find_hook_by(test_run_hook_started_message) : nil
  else
    repository.hook_by_id[message.hook_id]
  end
end

# File lib/cucumber/query.rb, line 108
def find_meta
  repository.meta
end

# File lib/cucumber/query.rb, line 114
def find_most_severe_test_step_result_by(message)
  ensure_only_message_types!(message, %i[test_case_started test_case_finished], '#find_most_severe_test_step_result_by')

  if message.is_a?(Cucumber::Messages::TestCaseStarted)
    find_test_steps_finished_by(message)
      .map(&:test_step_result)
      .max_by { |test_step_result| test_step_result_rankings[test_step_result.status] }
    # Java code: "PREVIOUS".max(comparing(TestStepResult::getStatus, new TestStepResultStatusComparator()));
  else
    test_case_started_message = find_test_case_started_by(message)
    test_case_started_message && find_most_severe_test_step_result_by(test_case_started_message)
    # Java code: return findTestCaseStartedBy(testCaseFinished).flatMap(this::findMostSevereTestStepResultBy);
  end
end

# File lib/cucumber/query.rb, line 131
def find_pickle_by(message)
  ensure_only_message_types!(message, %i[test_case test_case_started test_case_finished test_step_started test_step_finished], '#find_pickle_by')

  test_case_message = message.is_a?(Cucumber::Messages::TestCase) ? message : find_test_case_by(message)
  repository.pickle_by_id[test_case_message.pickle_id]
end

# File lib/cucumber/query.rb, line 140
def find_pickle_step_by(test_step)
  ensure_only_message_types!(test_step, %i[test_step], '#find_pickle_step_by')

  repository.pickle_step_by_id[test_step.pickle_step_id]
end

# File lib/cucumber/query.rb, line 148
def find_step_by(pickle_step)
  ensure_only_message_types!(pickle_step, %i[pickle_step], '#find_step_by')

  repository.step_by_id[pickle_step.ast_node_ids.first]
end

# File lib/cucumber/query.rb, line 156
def find_step_definitions_by(test_step)
  ensure_only_message_types!(test_step, %i[test_step], '#find_step_definitions_by')

  ids = test_step.step_definition_ids.nil? ? [] : test_step.step_definition_ids
  ids.map { |id| repository.step_definition_by_id[id] }.compact
end

# File lib/cucumber/query.rb, line 165
def find_test_case_by(message)
  ensure_only_message_types!(message, %i[test_case_started test_case_finished test_step_started test_step_finished], '#find_test_case_by')

  test_case_started_message = message.is_a?(Cucumber::Messages::TestCaseStarted) ? message : find_test_case_started_by(message)
  repository.test_case_by_id[test_case_started_message.test_case_id]
end

# File lib/cucumber/query.rb, line 182
def find_test_case_finished_by(test_case_started)
  ensure_only_message_types!(test_case_started, %i[test_case_started], '#find_test_case_finished_by')

  repository.test_case_finished_by_test_case_started_id[test_case_started.id]
end

# File lib/cucumber/query.rb, line 174
def find_test_case_started_by(message)
  ensure_only_message_types!(message, %i[test_case_finished test_step_started test_step_finished], '#find_test_case_started_by')

  repository.test_case_started_by_id[message.test_case_started_id]
end

# File lib/cucumber/query.rb, line 188
def find_test_run_duration
  if repository.test_run_started.nil? || repository.test_run_finished.nil?
    nil
  else
    timestamp_to_time(repository.test_run_finished.timestamp) - timestamp_to_time(repository.test_run_started.timestamp)
  end
end

# File lib/cucumber/query.rb, line 216
def find_test_run_finished
  repository.test_run_finished
end

# File lib/cucumber/query.rb, line 206
def find_test_run_hook_finished_by(test_run_hook_started)
  ensure_only_message_types!(test_run_hook_started, %i[test_run_hook_started], '#find_test_run_hook_finished_by')

  repository.test_run_hook_finished_by_test_run_hook_started_id[test_run_hook_started.id]
end

# File lib/cucumber/query.rb, line 198
def find_test_run_hook_started_by(test_run_hook_finished)
  ensure_only_message_types!(test_run_hook_finished, %i[test_run_hook_finished], '#find_test_run_hook_started_by')

  repository.test_run_hook_started_by_id[test_run_hook_finished.test_run_hook_started_id]
end

# File lib/cucumber/query.rb, line 212
def find_test_run_started
  repository.test_run_started
end

# File lib/cucumber/query.rb, line 222
def find_test_step_by(message)
  ensure_only_message_types!(message, %i[test_case_started test_case_finished], '#find_test_step_by')

  repository.test_step_by_id[message.test_step_id]
end

# File lib/cucumber/query.rb, line 239
def find_test_steps_finished_by(message)
  ensure_only_message_types!(message, %i[test_case_started test_case_finished], '#find_test_steps_finished_by')

  if message.is_a?(Cucumber::Messages::TestCaseStarted)
    repository.test_steps_finished_by_test_case_started_id.fetch(message.id, [])
  else
    test_case_started_message = find_test_case_started_by(message)
    test_case_started_message.nil? ? [] : find_test_steps_finished_by(test_case_started_message)
  end
end

# File lib/cucumber/query.rb, line 230
def find_test_steps_started_by(message)
  ensure_only_message_types!(message, %i[test_case_started test_case_finished], '#find_test_steps_started_by')

  key = message.is_a?(Cucumber::Messages::TestCaseStarted) ? message.id : message.test_case_started_id
  repository.test_steps_started_by_test_case_started_id.fetch(key, [])
end

# File lib/cucumber/query.rb, line 252
def ensure_only_message_types!(supplied_message, permissible_message_types, method_name)
  raise ArgumentError, "Supplied argument is not a Cucumber Message. Argument: #{supplied_message.class}" unless supplied_message.is_a?(Cucumber::Messages::Message)

  permitted_klasses = permissible_message_types.map { |message| message_types[message] }
  raise ArgumentError, "Supplied message type '#{supplied_message.class}' is not permitted to be used when calling #{method_name}" unless permitted_klasses.include?(supplied_message.class)
end

# File lib/cucumber/query.rb, line 259
def message_types
  {
    pickle_step: Cucumber::Messages::PickleStep,
    test_case: Cucumber::Messages::TestCase,
    test_case_started: Cucumber::Messages::TestCaseStarted,
    test_case_finished: Cucumber::Messages::TestCaseFinished,
    test_run_hook_started: Cucumber::Messages::TestRunHookStarted,
    test_run_hook_finished: Cucumber::Messages::TestRunHookFinished,
    test_step: Cucumber::Messages::TestStep,
    test_step_started: Cucumber::Messages::TestStepStarted,
    test_step_finished: Cucumber::Messages::TestStepFinished
  }
end