class RuboCop::Cop::Style::MultilineMemoization

Checks expressions wrapping styles for multiline memoization.

@example EnforcedStyle: keyword (default)

# bad
foo ||= (
  bar
  baz
)

# good
foo ||= begin
  bar
  baz
end

@example EnforcedStyle: braces

# bad
foo ||= begin
  bar
  baz
end

# good
foo ||= (
  bar
  baz
)

Constants

BRACES_MSG
KEYWORD_MSG

Public Instance Methods

message(_node) click to toggle source
# File lib/rubocop/cop/style/multiline_memoization.rb, line 56
def message(_node)
  style == :braces ? BRACES_MSG : KEYWORD_MSG
end
on_or_asgn(node) click to toggle source
# File lib/rubocop/cop/style/multiline_memoization.rb, line 41
def on_or_asgn(node)
  _lhs, rhs = *node

  return unless bad_rhs?(rhs)

  add_offense(node) do |corrector|
    if style == :keyword
      keyword_autocorrect(rhs, corrector)
    else
      corrector.replace(rhs.loc.begin, '(')
      corrector.replace(rhs.loc.end, ')')
    end
  end
end

Private Instance Methods

bad_rhs?(rhs) click to toggle source
# File lib/rubocop/cop/style/multiline_memoization.rb, line 62
def bad_rhs?(rhs)
  return false unless rhs.multiline?

  if style == :keyword
    rhs.begin_type?
  else
    rhs.kwbegin_type?
  end
end
keyword_autocorrect(node, corrector) click to toggle source
# File lib/rubocop/cop/style/multiline_memoization.rb, line 72
def keyword_autocorrect(node, corrector)
  node_buf = node.source_range.source_buffer
  corrector.replace(node.loc.begin, keyword_begin_str(node, node_buf))
  corrector.replace(node.loc.end, keyword_end_str(node, node_buf))
end
keyword_begin_str(node, node_buf) click to toggle source
# File lib/rubocop/cop/style/multiline_memoization.rb, line 78
def keyword_begin_str(node, node_buf)
  if node_buf.source[node.loc.begin.end_pos] == "\n"
    'begin'
  else
    "begin\n#{' ' * (node.loc.column + configured_indentation_width)}"
  end
end
keyword_end_str(node, node_buf) click to toggle source
# File lib/rubocop/cop/style/multiline_memoization.rb, line 86
def keyword_end_str(node, node_buf)
  if /[^\s)]/.match?(node_buf.source_line(node.loc.end.line))
    "\n#{' ' * node.loc.column}end"
  else
    'end'
  end
end