class RuboCop::Cop::RSpec::ScatteredSetup

def autocorrect(corrector, first_occurrence, occurrence)
  return if first_occurrence == occurrence || !first_occurrence.body
  corrector.insert_after(first_occurrence.body,
                         "\n#{occurrence.body.source}")
  corrector.remove(range_by_whole_lines(occurrence.source_range,
                                        include_final_newline: true))
end

def lines_msg(numbers)
  if numbers.size == 1
    "line #{numbers.first}"
  else
    "lines #{numbers.join(', ')}"
  end
end

def message(occurrences, occurrence)
  lines = occurrences.map(&:first_line)
  lines_except_current = lines - [occurrence.first_line]
  format(MSG, hook_name: occurrences.first.method_name,
              lines: lines_msg(lines_except_current))
end

def on_block(node) # rubocop:disable InternalAffairs/NumblockHandler
  return unless example_group?(node)
  repeated_hooks(node).each do |occurrences|
    occurrences.each do |occurrence|
      message = message(occurrences, occurrence)
      add_offense(occurrence, message: message) do |corrector|
        autocorrect(corrector, occurrences.first, occurrence)
      end
    end
  end
end

def repeated_hooks(node)
  hooks = RuboCop::RSpec::ExampleGroup.new(node)
    .hooks
    .select(&:knowable_scope?)
    .group_by { |hook| [hook.name, hook.scope, hook.metadata] }
    .values
    .reject(&:one?)
  hooks.map do |hook|
    hook.map(&:to_node)
  end
end