Module: RuboCop::Cop::Style::ConditionalCorrectorHelper

Included in:
CaseCorrector, IfCorrector, TernaryCorrector
Defined in:
lib/rubocop/cop/style/conditional_assignment.rb

Overview

Helper module to provide common methods to ConditionalAssignment correctors

Instance Method Summary collapse

Instance Method Details

#assignment(node) ⇒ Object


451
452
453
454
455
456
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 451

def assignment(node)
  *_, condition = *node
  Parser::Source::Range.new(node.loc.expression.source_buffer,
                            node.loc.expression.begin_pos,
                            condition.loc.expression.begin_pos)
end

#correct_branches(corrector, branches) ⇒ Object


481
482
483
484
485
486
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 481

def correct_branches(corrector, branches)
  branches.each do |branch|
    *_, assignment = *branch
    corrector.replace(branch.source_range, assignment.source)
  end
end

#correct_if_branches(corrector, cop, node) ⇒ Object


458
459
460
461
462
463
464
465
466
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 458

def correct_if_branches(corrector, cop, node)
  if_branch, elsif_branches, else_branch = extract_tail_branches(node)

  corrector.insert_before(node.source_range, lhs(if_branch))
  replace_branch_assignment(corrector, if_branch)
  correct_branches(corrector, elsif_branches)
  replace_branch_assignment(corrector, else_branch)
  corrector.insert_before(node.loc.end, indent(cop, lhs(if_branch)))
end

#remove_whitespace_in_branches(corrector, branch, condition, column) ⇒ Object


431
432
433
434
435
436
437
438
439
440
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 431

def remove_whitespace_in_branches(corrector, branch, condition, column)
  branch.each_node do |child|
    white_space = white_space_range(child, column)
    corrector.remove(white_space) if white_space.source.strip.empty?
  end

  [condition.loc.else, condition.loc.end].each do |loc|
    corrector.remove_preceding(loc, loc.column - column)
  end
end

#replace_branch_assignment(corrector, branch) ⇒ Object


468
469
470
471
472
473
474
475
476
477
478
479
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 468

def replace_branch_assignment(corrector, branch)
  _variable, *_operator, assignment = *branch
  source = assignment.source

  replacement = if assignment.array_type? && !assignment.bracketed?
                  "[#{source}]"
                else
                  source
                end

  corrector.replace(branch.source_range, replacement)
end

#white_space_range(node, column) ⇒ Object


442
443
444
445
446
447
448
449
# File 'lib/rubocop/cop/style/conditional_assignment.rb', line 442

def white_space_range(node, column)
  expression = node.loc.expression
  begin_pos = expression.begin_pos - (expression.column - column - 2)

  Parser::Source::Range.new(expression.source_buffer,
                            begin_pos,
                            expression.begin_pos)
end