Class: RuboCop::Cop::Style::AccessorGrouping

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 143

def separated_style?
  style == :separated
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 76

def check(send_node)
  return if previous_line_comment?(send_node) || !groupable_accessor?(send_node)
  return unless (grouped_style? && groupable_sibling_accessors(send_node).size > 1) ||
                (separated_style? && send_node.arguments.size > 1)

  message = message(send_node)
  add_offense(send_node, message: message) do |corrector|
    autocorrect(corrector, send_node)
  end
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 127

def class_send_elements(class_node)
  class_def = class_node.body

  if !class_def || class_def.def_type?
    []
  elsif class_def.send_type?
    [class_def]
  else
    class_def.each_child_node(:send).to_a
  end
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 184

def group_accessors(node, accessors)
  accessor_names = accessors.flat_map { |accessor| accessor.arguments.map(&:source) }.uniq

  "#{node.method_name} #{accessor_names.join(', ')}"
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 104

def groupable_accessor?(node)
  return true unless (previous_expression = node.left_siblings.last)

  # Accessors with Sorbet `sig { ... }` blocks shouldn't be groupable.
  if previous_expression.block_type?
    previous_expression.child_nodes.each do |child_node|
      break previous_expression = child_node if child_node.send_type?
    end
  end

  return true unless previous_expression.send_type?

  # Accessors with RBS::Inline annotations shouldn't be groupable.
  return false if processed_source.comments.any? do |c|
    same_line?(c, previous_expression) && c.text.start_with?('#:')
  end

  previous_expression.attribute_accessor? ||
    previous_expression.access_modifier? ||
    node.first_line - previous_expression.last_line > 1 # there is a space between nodes
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 147

def groupable_sibling_accessor?(node, sibling)
  sibling.attribute_accessor? &&
    sibling.method?(node.method_name) &&
    node_visibility(sibling) == node_visibility(node) &&
    groupable_accessor?(sibling) && !previous_line_comment?(sibling)
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 154

def groupable_sibling_accessors(send_node)
  send_node.parent.each_child_node(:send).select do |sibling|
    groupable_sibling_accessor?(send_node, sibling)
  end
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 160

def message(send_node)
  msg = grouped_style? ? GROUPED_MSG : SEPARATED_MSG
  format(msg, accessor: send_node.method_name)
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 64

def on_class(node)
  class_send_elements(node).each do |macro|
    next unless macro.attribute_accessor?

    check(macro)
  end
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 72

alias on_module on_class

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 71

alias on_sclass on_class

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 165

def preferred_accessors(node)
  if grouped_style?
    return if skip_for_grouping?(node)

    accessors = groupable_sibling_accessors(node)
    if node.loc == accessors.first.loc || skip_for_grouping?(accessors.first)
      group_accessors(node, accessors)
    end
  else
    separate_accessors(node)
  end
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 99

def previous_line_comment?(node)
  comment_line?(processed_source[node.first_line - 2])
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 205

def range_with_trailing_argument_comment(node)
  comment = processed_source.ast_with_comments[node.last_argument].last
  if comment
    add_range(node.source_range, comment.source_range)
  else
    node
  end
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 190

def separate_accessors(node)
  node.arguments.flat_map do |arg|
    lines = [
      *processed_source.ast_with_comments[arg].map(&:text),
      "#{node.method_name} #{arg.source}"
    ]
    if arg == node.first_argument
      lines
    else
      indent = ' ' * node.loc.column
      lines.map { |line| "#{indent}#{line}" }
    end
  end.join("\n")
end

# File 'lib/rubocop/cop/style/accessor_grouping.rb', line 179

def skip_for_grouping?(node)
  node.right_siblings.any?(&:casgn_type?) &&
    node.right_siblings.any? { |n| n.send_type? && groupable_sibling_accessor?(node, n) }
end