active_record_aliases.rb

# frozen_string_literal: true

module RuboCop
  module Cop
    module Rails
      # Checks that ActiveRecord aliases are not used. The direct method names
      # are more clear and easier to read.
      #
      # @example
      #   #bad
      #   Book.update_attributes!(author: 'Alice')
      #
      #   #good
      #   Book.update!(author: 'Alice')
      class ActiveRecordAliases < Cop
        MSG = 'Use `%<prefer>s` instead of `%<current>s`.'

        ALIASES = {
          update_attributes: :update,
          update_attributes!: :update!
        }.freeze

        RESTRICT_ON_SEND = ALIASES.keys.freeze

        def on_send(node)
          method_name = node.method_name

          add_offense(
            node,
            message: format(MSG, prefer: ALIASES[method_name], current: method_name),
            location: :selector,
            severity: :warning
          )
        end

        alias on_csend on_send

        def autocorrect(node)
          lambda do |corrector|
            corrector.replace(
              node.loc.selector,
              ALIASES[node.method_name].to_s
            )
          end
        end
      end
    end
  end
end