module RuboCop::Cop::Style::ConditionalAssignmentHelper
Helper module to provide common methods to classes needed for the ConditionalAssignment Cop.
Constants
- ALIGN_WITH
- END_ALIGNMENT
- EQUAL
- KEYWORD
Public Instance Methods
end_with_eq?(sym)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 63 def end_with_eq?(sym) sym.to_s.end_with?(EQUAL) end
expand_elses(branch)
click to toggle source
`elsif` branches show up in the `node` as an `else`. We need to recursively iterate over all `else` branches and consider all but the last `node` an `elsif` branch and consider the last `node` the actual `else` branch.
# File lib/rubocop/cop/style/conditional_assignment.rb, line 19 def expand_elses(branch) elsif_branches = expand_elsif(branch) else_branch = elsif_branches.any? ? elsif_branches.pop : branch [elsif_branches, else_branch] end
expand_when_branches(when_branches)
click to toggle source
`when` nodes contain the entire branch including the condition. We only need the contents of the branch, not the condition.
# File lib/rubocop/cop/style/conditional_assignment.rb, line 27 def expand_when_branches(when_branches) when_branches.map { |branch| branch.children[1] } end
indent(cop, source)
click to toggle source
rubocop:enable Metrics/AbcSize
# File lib/rubocop/cop/style/conditional_assignment.rb, line 54 def indent(cop, source) conf = cop.config.for_cop(END_ALIGNMENT) if conf[ALIGN_WITH] == KEYWORD ' ' * source.length else '' end end
lhs(node)
click to toggle source
rubocop:disable Metrics/AbcSize
# File lib/rubocop/cop/style/conditional_assignment.rb, line 36 def lhs(node) case node.type when :send lhs_for_send(node) when :op_asgn "#{node.children[0].source} #{node.children[1]}= " when :and_asgn, :or_asgn "#{node.children[0].source} #{node.loc.operator.source} " when :casgn "#{node.children[1]} = " when *ConditionalAssignment::VARIABLE_ASSIGNMENT_TYPES "#{node.children[0]} = " else node.source end end
tail(branch)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 31 def tail(branch) branch.begin_type? ? [*branch].last : branch end
Private Instance Methods
assignment_rhs_exist?(node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 99 def assignment_rhs_exist?(node) parent = node.parent return true unless parent !(parent.mlhs_type? || parent.resbody_type?) end
expand_elsif(node, elsif_branches = [])
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 69 def expand_elsif(node, elsif_branches = []) return [] if node.nil? || !node.if_type? _condition, elsif_branch, else_branch = *node elsif_branches << elsif_branch if else_branch && else_branch.if_type? expand_elsif(else_branch, elsif_branches) else elsif_branches << else_branch end elsif_branches end
lhs_for_send(node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 81 def lhs_for_send(node) receiver = node.receiver ? node.receiver.source : '' if node.method?(:[]=) indices = node.arguments[0...-1].map(&:source).join(', ') "#{receiver}[#{indices}] = " elsif node.setter_method? "#{receiver}.#{node.method_name[0...-1]} = " else "#{receiver} #{node.method_name} " end end
setter_method?(method_name)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 94 def setter_method?(method_name) method_name.to_s.end_with?(EQUAL) && !%i[!= == === >= <=].include?(method_name) end