Class: RuboCop::Cop::Style::IfCorrector

Inherits:
Object
  • Object
show all
Extended by:
ConditionalAssignmentHelper, ConditionalCorrectorHelper
Defined in:
lib/rubocop/cop/style/conditional_assignment.rb

Overview

Corrector to correct conditional assignment in ‘if` statements.

Constant Summary

Constants included from ConditionalAssignmentHelper

ConditionalAssignmentHelper::ALIGN_WITH, ConditionalAssignmentHelper::END_ALIGNMENT, ConditionalAssignmentHelper::EQUAL, ConditionalAssignmentHelper::KEYWORD

Class Method Summary collapse

Methods included from ConditionalCorrectorHelper

assignment, correct_branches, remove_whitespace_in_branches, white_space_range

Methods included from ConditionalAssignmentHelper

expand_elses, expand_when_branches, indent, lhs, tail

Class Method Details

.correct(cop, node) ⇒ Object 



497
498
499
500
501
502
503
504
505
506
507
508
509
510
 
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 497

def correct(cop, node)
  if_branch, elsif_branches, else_branch = extract_tail_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) ⇒ Object 



512
513
514
515
516
517
518
519
520
521
522
523
524
525
 
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 512

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