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, correct_if_branches, remove_whitespace_in_branches, replace_branch_assignment, white_space_range

Methods included from ConditionalAssignmentHelper

end_with_eq?, expand_elses, expand_when_branches, indent, lhs, tail

Methods included from NodePattern::Macros

#def_node_matcher, #def_node_search, #node_search, #node_search_all, #node_search_body, #node_search_first

Class Method Details

.correct(cop, node) ⇒ Object


561
562
563
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 561

def correct(cop, node)
  ->(corrector) { correct_if_branches(corrector, cop, node) }
end

.move_assignment_inside_condition(node) ⇒ Object


565
566
567
568
569
570
571
572
573
574
575
576
577
578
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 565

def move_assignment_inside_condition(node)
  column = node.loc.expression.column
  *_var, condition = *node
  assignment = assignment(node)

  lambda do |corrector|
    corrector.remove(assignment)

    condition.branches.flatten.each do |branch|
      move_branch_inside_condition(corrector, branch, condition,
                                   assignment, column)
    end
  end
end