lib/rubocop/cop/internal_affairs/method_name_equal.rb


# frozen_string_literal: true

module RuboCop
  module Cop
    module InternalAffairs
      # Checks that method names are checked using `method?` method.
      #
      # @example
      #   # bad
      #   node.method_name == :do_something
      #
      #   # good
      #   node.method?(:do_something)
      #
      class MethodNameEqual < Base
        include RangeHelp
        extend AutoCorrector

        MSG = 'Use `method?(%<method_name>s)` instead of `method_name == %<method_name>s`.'
        RESTRICT_ON_SEND = %i[==].freeze

        # @!method method_name?(node)
        def_node_matcher :method_name?, <<~PATTERN
          (send
            $(send
              (...) :method_name) :==
            $...)
        PATTERN

        def on_send(node)
          method_name?(node) do |method_name_node, method_name_arg|
            message = format(MSG, method_name: method_name_arg.first.source)

            range = range(method_name_node, node)

            add_offense(range, message: message) do |corrector|
              corrector.replace(range, "method?(#{method_name_arg.first.source})")
            end
          end
        end

        private

        def range(method_name_node, node)
          range_between(method_name_node.loc.selector.begin_pos, node.source_range.end_pos)
        end
      end
    end
  end
end