Class: RuboCop::Cop::Style::AccessModifierDeclarations

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 214

def inline_style?
  style == :inline
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 222

def access_modifier_is_not_inlined?(node)
  !access_modifier_is_inlined?(node)
end

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

def_node_matcher :access_modifier_with_attr?, <<~PATTERN
  (send nil? {:private :protected :public :module_function}
    (send nil? {:attr :attr_reader :attr_writer :attr_accessor} _+))
PATTERN

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 136

def_node_matcher :access_modifier_with_symbol?, <<~PATTERN
  (send nil? {:private :protected :public :module_function}
    {(sym _)+ (splat {#percent_symbol_array? const send})}
  )
PATTERN

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 193

def allow_modifiers_on_attrs?(node)
  cop_config['AllowModifiersOnAttrs'] && access_modifier_with_attr?(node)
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 189

def allow_modifiers_on_symbols?(node)
  cop_config['AllowModifiersOnSymbols'] && access_modifier_with_symbol?(node)
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 163

def allowed?(node)
  !node.access_modifier? ||
    ALLOWED_NODE_TYPES.include?(node.parent&.type) ||
    allow_modifiers_on_symbols?(node) ||
    allow_modifiers_on_attrs?(node)
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 170

def autocorrect(corrector, node)
  case style
  when :group
    def_nodes = find_corresponding_def_nodes(node)
    return unless def_nodes.any?

    replace_defs(corrector, node, def_nodes)
  when :inline
    remove_nodes(corrector, node)
    select_grouped_def_nodes(node).each do |grouped_def_node|
      insert_inline_modifier(corrector, grouped_def_node, node.method_name)
    end
  end
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 203

def correctable_group_offense?(node)
  return false unless group_style?
  return false if allowed?(node)

  access_modifier_is_inlined?(node) && find_corresponding_def_nodes(node).any?
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 305

def def_source(node, def_nodes)
  [
    *processed_source.ast_with_comments[node].map(&:text),
    *def_nodes.map(&:source)
  ].join("\n")
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 265

def find_argument_less_modifier_node(node)
  return unless (parent = node.parent)

  parent.each_child_node(:send).find do |child|
    child.method?(node.method_name) && child.arguments.empty?
  end
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 246

def find_corresponding_def_nodes(node)
  if access_modifier_with_symbol?(node)
    method_names = node.arguments.filter_map do |argument|
      next unless argument.sym_type?

      argument.respond_to?(:value) && argument.value
    end

    def_nodes = node.parent.each_child_node(:def).select do |child|
      method_names.include?(child.method_name)
    end

    # If there isn't a `def` node for each symbol, we will skip autocorrection.
    def_nodes.size == method_names.size ? def_nodes : []
  else
    [node.first_argument]
  end
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 295

def insert_inline_modifier(corrector, node, modifier_name)
  corrector.insert_before(node, "#{modifier_name} ")
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 236

def message(range)
  access_modifier = range.source

  if group_style?
    format(GROUP_STYLE_MESSAGE, access_modifier: access_modifier)
  elsif inline_style?
    format(INLINE_STYLE_MESSAGE, access_modifier: access_modifier)
  end
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 197

def offense?(node)
  (group_style? && access_modifier_is_inlined?(node) &&
    !node.parent&.if_type? && !right_siblings_same_inline_method?(node)) ||
    (inline_style? && access_modifier_is_not_inlined?(node))
end

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

def on_send(node)
  return if allowed?(node)

  if offense?(node)
    add_offense(node.loc.selector) do |corrector|
      autocorrect(corrector, node)
    end
    opposite_style_detected
  else
    correct_style_detected
  end
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 185

def percent_symbol_array?(node)
  node.array_type? && node.percent_literal?(:symbol)
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 299

def remove_nodes(corrector, *nodes)
  nodes.each do |node|
    corrector.remove(range_with_comments_and_lines(node))
  end
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 279

def replace_defs(corrector, node, def_nodes)
  source = def_source(node, def_nodes)
  argument_less_modifier_node = find_argument_less_modifier_node(node)
  if argument_less_modifier_node
    corrector.insert_after(argument_less_modifier_node, "\n\n#{source}")
  elsif (ancestor = node.each_ancestor(:block, :class, :module).first)

    corrector.insert_before(ancestor.loc.end, "#{node.method_name}\n\n#{source}\n")
  else
    corrector.replace(node, "#{node.method_name}\n\n#{source}")
    return
  end

  remove_nodes(corrector, *def_nodes, node)
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 226

def right_siblings_same_inline_method?(node)
  node.right_siblings.any? do |sibling|
    sibling.send_type? &&
      correctable_group_offense?(sibling) &&
      sibling.method?(node.method_name) &&
      !sibling.arguments.empty? &&
      find_corresponding_def_nodes(sibling).any?
  end
end

# File 'lib/rubocop/cop/style/access_modifier_declarations.rb', line 273

def select_grouped_def_nodes(node)
  node.right_siblings.take_while do |sibling|
    !(sibling.send_type? && sibling.bare_access_modifier_declaration?)
  end.select(&:def_type?)
end