Sha256: 5364e4dc391f367d03456f0fd9183b5f7fc332013975c1c6f386bf79da76883a
Contents?: true
Size: 1.68 KB
Versions: 9
Compression:
Stored size: 1.68 KB
Contents
# frozen_string_literal: true
module RuboCop
module Cop
module Lint
# This cop checks for assignments in the conditions of
# if/while/until.
class AssignmentInCondition < Cop
include SafeAssignment
MSG = 'Assignment in condition - you probably meant to use `==`.'.freeze
ASGN_TYPES = [:begin, *EQUALS_ASGN_NODES, :send].freeze
def on_if(node)
check(node)
end
def on_while(node)
check(node)
end
def on_until(node)
check(node)
end
private
def check(node)
condition, = *node
return if condition.block_type?
traverse_node(condition, ASGN_TYPES) do |asgn_node|
next :skip_children if skip_children?(asgn_node)
next if allowed_construct?(asgn_node)
add_offense(asgn_node, :operator)
end
end
def allowed_construct?(asgn_node)
asgn_node.begin_type? || conditional_assignment?(asgn_node)
end
def conditional_assignment?(asgn_node)
!asgn_node.loc.operator
end
def skip_children?(asgn_node)
(asgn_node.send_type? && asgn_node.method_name !~ /=\z/) ||
empty_condition?(asgn_node) ||
(safe_assignment_allowed? && safe_assignment?(asgn_node))
end
# each_node/visit_descendants_with_types with :skip_children
def traverse_node(node, types, &block)
result = yield node if types.include?(node.type)
return if result == :skip_children
node.each_child_node { |child| traverse_node(child, types, &block) }
end
end
end
end
end
Version data entries
9 entries across 9 versions & 2 rubygems