class RuboCop::Cop::Style::CaseCorrector
Corrector
to correct conditional assignment in `case` statements.
Public Class Methods
correct(corrector, cop, node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 614 def correct(corrector, cop, node) when_branches, else_branch = extract_tail_branches(node) corrector.insert_before(node, lhs(else_branch)) correct_branches(corrector, when_branches) replace_branch_assignment(corrector, else_branch) corrector.insert_before(node.loc.end, indent(cop, lhs(else_branch))) end
move_assignment_inside_condition(corrector, node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 624 def move_assignment_inside_condition(corrector, node) column = node.loc.expression.column *_var, condition = *node assignment = assignment(node) corrector.remove(assignment) extract_branches(condition).flatten.each do |branch| move_branch_inside_condition(corrector, branch, condition, assignment, column) end end
Private Class Methods
extract_branches(case_node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 644 def extract_branches(case_node) when_branches = if case_node.case_type? expand_when_branches(case_node.when_branches) else expand_when_branches(case_node.in_pattern_branches) end [when_branches, case_node.else_branch] end
extract_tail_branches(node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 638 def extract_tail_branches(node) when_branches, else_branch = extract_branches(node) when_branches.map! { |branch| tail(branch) } [when_branches, tail(else_branch)] end
move_branch_inside_condition(corrector, branch, condition, assignment, column)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 654 def move_branch_inside_condition(corrector, branch, condition, assignment, column) branch_assignment = tail(branch) corrector.insert_before(branch_assignment, assignment.source) remove_whitespace_in_branches(corrector, branch, condition, column) parent_keyword = branch.parent.loc.keyword corrector.remove_preceding(parent_keyword, parent_keyword.column - column) end