Module: RuboCop::Cop::Style::ConditionalAssignmentHelper

Included in:
CaseCorrector, ConditionalAssignment, IfCorrector, TernaryCorrector, UnlessCorrector
Defined in:
lib/rubocop/cop/style/conditional_assignment.rb

Overview

Helper module to provide common methods to classes needed for the ConditionalAssignment Cop.

Constant Summary collapse

EQUAL =
'='.freeze
END_ALIGNMENT =
'Lint/EndAlignment'.freeze
ALIGN_WITH =
'AlignWith'.freeze
KEYWORD =
'keyword'.freeze

Instance Method Summary collapse

Instance Method Details

#correct_branches(corrector, branches) ⇒ Object



31
32
33
34
35
36
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 31

def correct_branches(corrector, branches)
  branches.each do |branch|
    *_, assignment = *branch
    corrector.replace(branch.source_range, assignment.source)
  end
end

#expand_elses(branch) ⇒ Object

‘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.



19
20
21
22
23
# 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) ⇒ Object

‘when` nodes contain the entire branch including the condition. We only need the contents of the branch, not the condition.



27
28
29
# 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) ⇒ Object



61
62
63
64
65
66
67
68
69
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 61

def indent(cop, source)
  if cop.config[END_ALIGNMENT] &&
     cop.config[END_ALIGNMENT][ALIGN_WITH] &&
     cop.config[END_ALIGNMENT][ALIGN_WITH] == KEYWORD
    ' ' * source.length
  else
    ''
  end
end

#lhs(node) ⇒ Object



42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 42

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
    "#{node.children[0].source} &&= "
  when :or_asgn
    "#{node.children[0].source} ||= "
  when :casgn
    "#{node.children[1]} = "
  when *ConditionalAssignment::VARIABLE_ASSIGNMENT_TYPES
    "#{node.children[0]} = "
  else
    node.source
  end
end

#tail(branch) ⇒ Object



38
39
40
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 38

def tail(branch)
  branch.begin_type? ? [*branch].last : branch
end