array_alignment_extended.rb

# frozen_string_literal: true

module Datarockets
  module Style
    module Cop
      module Layout
        # Here we check if the elements of a multi-line array literal are
        # aligned.
        #
        # @example EnforcedStyle: with_first_argument (default)
        #   # good
        #
        #   array = [1, 2, 3,
        #            4, 5, 6]
        #   array = ['run',
        #            'forrest',
        #            'run']
        #
        #   # bad
        #
        #   array = [1, 2, 3,
        #     4, 5, 6]
        #   array = ['run',
        #        'forrest',
        #        'run']
        #
        # @example EnforcedStyle: with_fixed_indentation
        #   # good
        #
        #   array = [1, 2, 3,
        #     4, 5, 6]
        #
        #   # bad
        #
        #   array = [1, 2, 3,
        #            4, 5, 6]
        class ArrayAlignmentExtended < RuboCop::Cop::Cop
          include RuboCop::Cop::Alignment

          ALIGN_PARAMS_MSG = "Align the elements of an array literal if they span more than one line."

          FIXED_INDENT_MSG = "Use one level of indentation for elements " \
            "following the first line of a multi-line array."

          def on_array(node)
            return if node.children.size < 2

            check_alignment(node.children, base_column(node, node.children))
          end

          def autocorrect(node)
            RuboCop::Cop::AlignmentCorrector.correct(processed_source, node, column_delta)
          end

          private

          def message(_node)
            fixed_indentation? ? FIXED_INDENT_MSG : ALIGN_PARAMS_MSG
          end

          def fixed_indentation?
            cop_config["EnforcedStyle"] == "with_fixed_indentation"
          end

          def base_column(node, args)
            fixed_indentation? ? line_indentation(node) : display_column(args.first.source_range)
          end

          def line_indentation(node)
            lineno = target_method_lineno(node)
            line = node.source_range.source_buffer.source_line(lineno)
            line_indentation = /\S.*/.match(line).begin(0)
            line_indentation + configured_indentation_width
          end

          def target_method_lineno(node)
            node.loc.line
          end
        end
      end
    end
  end
end