lib/rubocop/cop/internal_affairs/operator_keyword.rb


# frozen_string_literal: true

module RuboCop
  module Cop
    module InternalAffairs
      # Enforces the use of `node.operator_keyword?` instead of `node.and_type? || node.or_type?`.
      #
      # @example
      #   # bad
      #   node.and_type? || node.or_type?
      #   node.or_type? || node.and_type?
      #
      #   # good
      #   node.operator_keyword?
      #
      class OperatorKeyword < Base
        extend AutoCorrector

        MSG = 'Use `%<prefer>s`.'
        PREFERRED_METHOD = 'operator_keyword?'

        # @!method and_or_type(node)
        def_node_matcher :and_or_type, <<~PATTERN
          {
            (or $(send _node :and_type?) $(send _node :or_type?))
            (or $(send _node :or_type?) $(send _node :and_type?))
            (or
              (or _ $(send _node :and_type?)) $(send _node :or_type?))
            (or
              (or _ $(send _node :or_type?)) $(send _node :and_type?))
          }
        PATTERN

        def on_or(node)
          return unless (lhs, rhs = and_or_type(node))

          begin_range = lhs.receiver&.source_range || lhs.loc.selector
          offense = begin_range.join(rhs.source_range.end)
          prefer = lhs.receiver ? "#{lhs.receiver.source}.#{PREFERRED_METHOD}" : PREFERRED_METHOD

          add_offense(offense, message: format(MSG, prefer: prefer)) do |corrector|
            corrector.replace(offense, prefer)
          end
        end
      end
    end
  end
end