class RuboCop::Cop::Style::CaseCorrector

Corrector to correct conditional assignment in `case` statements.

Public Class Methods

correct(cop, node) click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 617
def correct(cop, node)
  when_branches, else_branch = extract_tail_branches(node)

  lambda do |corrector|
    corrector.insert_before(node.source_range, 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
end
move_assignment_inside_condition(node) click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 630
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

Private Class Methods

extract_branches(node) click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 653
def extract_branches(node)
  _condition, *when_branches, else_branch = *node
  when_branches = expand_when_branches(when_branches)
  [when_branches, else_branch]
end
extract_tail_branches(node) click to toggle source
# File lib/rubocop/cop/style/conditional_assignment.rb, line 647
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 659
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