class RuboCop::Cop::Style::IfCorrector
Corrector
to correct conditional assignment in ‘if` statements.
Public Class Methods
correct(corrector, cop, node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 569 def correct(corrector, cop, node) correct_if_branches(corrector, cop, node) end
move_assignment_inside_condition(corrector, node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 573 def move_assignment_inside_condition(corrector, node) column = node.loc.expression.column *_var, condition = *node assignment = assignment(node) corrector.remove(assignment) condition.branches.flatten.each do |branch| move_branch_inside_condition(corrector, branch, condition, assignment, column) end end
Private Class Methods
extract_tail_branches(node)
click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 587 def extract_tail_branches(node) if_branch, *elsif_branches, else_branch = *node.branches elsif_branches.map! { |branch| tail(branch) } [tail(if_branch), elsif_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 594 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) return unless (branch_else = branch.parent.loc.else) corrector.remove_preceding(branch_else, branch_else.column - column) end