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)
  _condition, if_branch, else_branch = *node
  if_branch = tail(if_branch)
  _variable, *_operator, if_assignment = *if_branch
  elsif_branches, else_branch = expand_elses(else_branch)
  elsif_branches.map! { |branch| tail(branch) }
  else_branch = tail(else_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 485
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