# frozen_string_literal: true

module RuboCop
  module Cop
    # Common functionality for checking for a line break before the first
    # element in a multi-line collection.
    module FirstElementLineBreak
      def autocorrect(node)
        ->(corrector) { corrector.insert_before(node.source_range, "\n") }
      end

      private

      def check_method_line_break(node, children)
        return if children.empty?

        return unless method_uses_parens?(node, children.first)

        check_children_line_break(node, children)
      end

      def method_uses_parens?(node, limit)
        source = node.source_range.source_line[0...limit.loc.column]
        source =~ /\s*\(\s*$/
      end

      def check_children_line_break(node, children, start = node)
        return if children.size < 2

        line = start.loc.line

        min = first_by_line(children)
        return if line != min.loc.first_line

        max = last_by_line(children)
        return if line == max.loc.last_line

        add_offense(min)
      end

      def first_by_line(nodes)
        nodes.min_by { |n| n.loc.first_line }
      end

      def last_by_line(nodes)
        nodes.max_by { |n| n.loc.last_line }
      end
    end
  end
end