Class: RuboCop::Cop::Style::Semicolon

# File 'lib/rubocop/cop/style/semicolon.rb', line 70

def each_semicolon
  tokens_for_lines.each do |line, tokens|
    semicolon_pos = semicolon_position(tokens)
    after_expr_pos = semicolon_pos == -1 ? -2 : semicolon_pos

    yield line, tokens[semicolon_pos].column, tokens[after_expr_pos] if semicolon_pos
  end
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 106

def exist_semicolon_after_left_curly_brace?(tokens)
  tokens[1]&.left_curly_brace? && tokens[2]&.semicolon?
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 110

def exist_semicolon_after_left_lambda_curly_brace?(tokens)
  tokens[2]&.type == :tLAMBEG && tokens[3]&.semicolon?
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 118

def exist_semicolon_after_left_string_interpolation_brace?(tokens)
  tokens[1]&.type == :tSTRING_DBEG && tokens[2]&.semicolon?
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 102

def exist_semicolon_before_right_curly_brace?(tokens)
  tokens[-2]&.right_curly_brace? && tokens[-3]&.semicolon?
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 114

def exist_semicolon_before_right_string_interpolation_brace?(tokens)
  tokens[-3]&.type == :tSTRING_DEND && tokens[-4]&.semicolon?
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 142

def expressions_per_line(exprs)
  # create a map matching lines to the number of expressions on them
  exprs_lines = exprs.map(&:last_line)
  exprs_lines.group_by(&:itself)
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 156

def find_range_node(token_before_semicolon)
  range_nodes.detect do |range_node|
    range_node.source_range.contains?(token_before_semicolon.pos)
  end
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 148

def find_semicolon_positions(line)
  # Scan for all the semicolons on the line
  semicolons = processed_source[line - 1].enum_for(:scan, ';')
  semicolons.each do
    yield Regexp.last_match.begin(0)
  end
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 45

def on_begin(node)
  return if cop_config['AllowAsExpressionSeparator']
  return unless node.source.include?(';')

  exprs = node.children

  return if exprs.size < 2

  expressions_per_line(exprs).each do |line, expr_on_line|
    # Every line with more than one expression on it is a
    # potential offense
    next unless expr_on_line.size > 1

    find_semicolon_positions(line) { |pos| register_semicolon(line, pos, true) }
  end
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 39

def on_new_investigation
  return if processed_source.blank? || !processed_source.raw_source.include?(';')

  check_for_line_terminator_or_opener
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 162

def range_nodes
  return @range_nodes if instance_variable_defined?(:@range_nodes)

  ast = processed_source.ast
  @range_nodes = ast.range_type? ? [ast] : []
  @range_nodes.concat(ast.each_descendant(:irange, :erange).to_a)
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 122

def register_semicolon(line, column, after_expression, token_before_semicolon = nil)
  range = source_range(processed_source.buffer, line, column)

  add_offense(range) do |corrector|
    if after_expression
      corrector.replace(range, "\n")
    else
      # Prevents becoming one range instance with subsequent line when endless range
      # without parentheses.
      # See: https://github.com/rubocop/rubocop/issues/10791
      if token_before_semicolon&.regexp_dots?
        range_node = find_range_node(token_before_semicolon)
        corrector.wrap(range_node, '(', ')') if range_node
      end

      corrector.remove(range)
    end
  end
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 84

def semicolon_position(tokens)
  if tokens.last.semicolon?
    -1
  elsif tokens.first.semicolon?
    0
  elsif exist_semicolon_before_right_curly_brace?(tokens)
    -3
  elsif exist_semicolon_after_left_curly_brace?(tokens) ||
        exist_semicolon_after_left_string_interpolation_brace?(tokens)
    2
  elsif exist_semicolon_after_left_lambda_curly_brace?(tokens)
    3
  elsif exist_semicolon_before_right_string_interpolation_brace?(tokens)
    -4
  end
end

# File 'lib/rubocop/cop/style/semicolon.rb', line 79

def tokens_for_lines
  processed_source.tokens.group_by(&:line)
end