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

Inherits:
Object
  • Object
show all
Extended by:
ConditionalAssignmentHelper
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::EQUAL

Class Method Summary collapse

Methods included from ConditionalAssignmentHelper

correct_branches, expand_elses, expand_when_branches, lhs, tail

Class Method Details

.correct(node) ⇒ Object



299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 299

def correct(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)
  end
end