Sha256: 2aad21c354a8ccbf9d69a5725bb767bdb7b81a26f5edf273bf732f1d365f38c2
Contents?: true
Size: 1.9 KB
Versions: 7
Compression:
Stored size: 1.9 KB
Contents
# frozen_string_literal: true
module RuboCop
module Cop
module Style
# Enforces the use of `Object#instance_of?` instead of class comparison
# for equality.
#
# @example
# # bad
# var.class == Date
# var.class.equal?(Date)
# var.class.eql?(Date)
# var.class.name == 'Date'
#
# # good
# var.instance_of?(Date)
#
class ClassEqualityComparison < Base
include RangeHelp
include IgnoredMethods
extend AutoCorrector
MSG = 'Use `instance_of?(%<class_name>s)` instead of comparing classes.'
RESTRICT_ON_SEND = %i[== equal? eql?].freeze
# @!method class_comparison_candidate?(node)
def_node_matcher :class_comparison_candidate?, <<~PATTERN
(send
{$(send _ :class) (send $(send _ :class) :name)}
{:== :equal? :eql?} $_)
PATTERN
def on_send(node)
def_node = node.each_ancestor(:def, :defs).first
return if def_node && ignored_method?(def_node.method_name)
class_comparison_candidate?(node) do |receiver_node, class_node|
range = offense_range(receiver_node, node)
class_name = class_name(class_node, node)
add_offense(range, message: format(MSG, class_name: class_name)) do |corrector|
corrector.replace(range, "instance_of?(#{class_name})")
end
end
end
private
def class_name(class_node, node)
if node.children.first.method?(:name)
return class_node.receiver.source if class_node.receiver
class_node.source.delete('"').delete("'")
else
class_node.source
end
end
def offense_range(receiver_node, node)
range_between(receiver_node.loc.selector.begin_pos, node.source_range.end_pos)
end
end
end
end
end
Version data entries
7 entries across 7 versions & 2 rubygems