enum_syntax.rb

# frozen_string_literal: true

module RuboCop
  module Cop
    module Rails
      # Looks for enums written with keyword arguments syntax.
      #
      # Defining enums with keyword arguments syntax is deprecated and will be removed in Rails 8.0.
      # Positional arguments should be used instead:
      #
      # @example
      #   # bad
      #   enum status: { active: 0, archived: 1 }, _prefix: true
      #
      #   # good
      #   enum :status, { active: 0, archived: 1 }, prefix: true
      #
      class EnumSyntax < Base
        extend AutoCorrector
        extend TargetRubyVersion
        extend TargetRailsVersion

        minimum_target_ruby_version 3.0
        minimum_target_rails_version 7.0

        MSG = 'Enum defined with keyword arguments in `%<enum>s` enum declaration. Use positional arguments instead.'
        MSG_OPTIONS = 'Enum defined with deprecated options in `%<enum>s` enum declaration. Remove the `_` prefix.'
        RESTRICT_ON_SEND = %i[enum].freeze

        # From https://github.com/rails/rails/blob/v7.2.1/activerecord/lib/active_record/enum.rb#L231
        OPTION_NAMES = %w[prefix suffix scopes default instance_methods].freeze
        UNDERSCORED_OPTION_NAMES = OPTION_NAMES.map { |option| "_#{option}" }.freeze

        def_node_matcher :enum?, <<~PATTERN
          (send nil? :enum (hash $...))
        PATTERN

        def_node_matcher :enum_with_options?, <<~PATTERN
          (send nil? :enum $_ ${array hash} $_)
        PATTERN

        def on_send(node)
          check_and_correct_keyword_args(node)
          check_enum_options(node)
        end

        private

        def check_and_correct_keyword_args(node)
          enum?(node) do |pairs|
            pairs.each do |pair|
              next if option_key?(pair)

              correct_keyword_args(node, pair.key, pair.value, pairs[1..])
            end
          end
        end

        def check_enum_options(node)
          enum_with_options?(node) do |key, _, options|
            options.children.each do |option|
              next unless option_key?(option)

              add_offense(option.key, message: format(MSG_OPTIONS, enum: enum_name_value(key))) do |corrector|
                corrector.replace(option.key, option.key.source.delete_prefix('_'))
              end
            end
          end
        end

        def correct_keyword_args(node, key, values, options)
          add_offense(values, message: format(MSG, enum: enum_name_value(key))) do |corrector|
            # TODO: Multi-line autocorrect could be implemented in the future.
            next if multiple_enum_definitions?(node)

            preferred_syntax = "enum #{enum_name(key)}, #{values.source}#{correct_options(options)}"

            corrector.replace(node, preferred_syntax)
          end
        end

        def multiple_enum_definitions?(node)
          keys = node.first_argument.keys.map { |key| key.source.delete_prefix('_') }
          filterred_keys = keys.filter { |key| !OPTION_NAMES.include?(key) }
          filterred_keys.size >= 2
        end

        def enum_name_value(key)
          case key.type
          when :sym, :str
            key.value
          else
            key.source
          end
        end

        def enum_name(elem)
          case elem.type
          when :str
            elem.value.dump
          when :sym
            elem.value.inspect
          else
            elem.source
          end
        end

        def option_key?(pair)
          return false unless pair.respond_to?(:key)

          UNDERSCORED_OPTION_NAMES.include?(pair.key.source)
        end

        def correct_options(options)
          corrected_options = options.map do |pair|
            name = if pair.key.source[0] == '_'
                     pair.key.source[1..]
                   else
                     pair.key.source
                   end

            "#{name}: #{pair.value.source}"
          end.join(', ')

          ", #{corrected_options}" unless corrected_options.empty?
        end
      end
    end
  end
end