class RuboCop::Cop::Style::CaseCorrector
Corrector to correct conditional assignment in ‘case` statements.
def correct(cop, node)
def correct(cop, node) when_branches, else_branch = extract_tail_branches(node) _variable, *_operator, else_assignment = *else_branch lambda do |corrector| corrector.insert_before(node.source_range, lhs(else_branch)) correct_branches(corrector, when_branches) corrector.replace(else_branch.source_range, else_assignment.source) corrector.insert_before(node.loc.end, indent(cop, lhs(else_branch))) end end
def extract_branches(node)
def extract_branches(node) _condition, *when_branches, else_branch = *node when_branches = expand_when_branches(when_branches) [when_branches, else_branch] end
def extract_tail_branches(node)
def extract_tail_branches(node) when_branches, else_branch = extract_branches(node) when_branches.map! { |branch| tail(branch) } [when_branches, tail(else_branch)] end
def move_assignment_inside_condition(node)
def move_assignment_inside_condition(node) column = node.loc.expression.column *_var, condition = *node assignment = assignment(node) lambda do |corrector| corrector.remove(assignment) extract_branches(condition).flatten.each do |branch| move_branch_inside_condition(corrector, branch, condition, assignment, column) end end end
def move_branch_inside_condition(corrector, branch, condition,
def move_branch_inside_condition(corrector, branch, condition, assignment, column) branch_assignment = tail(branch) corrector.insert_before(branch_assignment.loc.expression, 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