class RuboCop::Cop::Style::IfCorrector
Corrector to correct conditional assignment in `if` statements.
Public Class Methods
correct(cop, node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 466 def correct(cop, node) if_branch, elsif_branches, else_branch = extract_branches(node) _variable, *_operator, if_assignment = *if_branch _else_variable, *_operator, else_assignment = *else_branch lambda do |corrector| corrector.insert_before(node.source_range, lhs(if_branch)) corrector.replace(if_branch.source_range, if_assignment.source) correct_branches(corrector, elsif_branches) corrector.replace(else_branch.source_range, else_assignment.source) corrector.insert_before(node.loc.end, indent(cop, lhs(if_branch))) end end
move_assignment_inside_condition(node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 481 def move_assignment_inside_condition(node) column = node.loc.expression.column *_var, condition = *node _condition, if_branch, else_branch = *condition elsif_branches, else_branch = expand_elses(else_branch) assignment = assignment(node) lambda do |corrector| corrector.remove(assignment) [if_branch, *elsif_branches, else_branch].each do |branch| branch_assignment = tail(branch) corrector.insert_before(branch_assignment.loc.expression, assignment.source) remove_whitespace_in_branches(corrector, branch, condition, column) corrector .remove_preceding(branch.parent.loc.else, branch.parent.loc.else.column - column) end end end
Private Class Methods
extract_branches(node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 508 def extract_branches(node) _condition, if_branch, else_branch = *node elsif_branches, else_branch = expand_elses(else_branch) elsif_branches.map! { |branch| tail(branch) } [tail(if_branch), elsif_branches, tail(else_branch)] end