lib/rubocop/cop/style/nil_comparison.rb


# encoding: utf-8

module RuboCop
  module Cop
    module Style
      # This cop checks for comparison of something with nil using ==.
      #
      # @example
      #
      #  # bad
      #  if x == nil
      #
      #  # good
      #  if x.nil?
      class NilComparison < Cop
        MSG = 'Prefer the use of the `nil?` predicate.'

        OPS = [:==, :===]

        NIL_NODE = s(:nil)

        def on_send(node)
          _receiver, method, args = *node
          return unless OPS.include?(method)

          add_offense(node, :selector) if args == NIL_NODE
        end

        private

        def autocorrect(node)
          @corrections << lambda do |corrector|
            expr = node.loc.expression
            new_code = expr.source.sub(/\s*={2,3}\s*nil/, '.nil?')
            corrector.replace(expr, new_code)
          end
        end
      end
    end
  end
end