Class: RuboCop::Cop::Style::SuperArguments

# File 'lib/rubocop/cop/style/super_arguments.rb', line 108

def arguments_identical?(def_node, super_node, def_args, super_args)
  return false if argument_list_size_differs?(def_args, super_args, super_node)

  def_args.zip(super_args).each do |def_arg, super_arg|
    next if positional_arg_same?(def_arg, super_arg)
    next if positional_rest_arg_same(def_arg, super_arg)
    next if keyword_arg_same?(def_arg, super_arg)
    next if keyword_rest_arg_same?(def_arg, super_arg)
    next if block_arg_same?(def_node, super_node, def_arg, super_arg)
    next if forward_arg_same?(def_arg, super_arg)

    return false
  end

  true
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 181

def block_arg_same?(def_node, super_node, def_arg, super_arg)
  return false unless def_arg.blockarg_type?
  return true if block_sends_to_super?(super_node)
  return false unless super_arg.block_pass_type?

  # anonymous forwarding
  return true if (block_pass_child = super_arg.children.first).nil? && def_arg.name.nil?

  block_arg_name = block_pass_child.children.first
  def_arg.name == block_arg_name && !block_reassigned?(def_node, block_arg_name)
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 195

def block_reassigned?(def_node, block_arg_name)
  def_node.each_node(*ASSIGN_TYPES).any? do |assign_node|
    # TODO: Since `Symbol#name` is supported from Ruby 3.0, the inheritance check for
    # `AST::Node` can be removed when requiring Ruby 3.0+.
    lhs = assign_node.node_parts[0]
    next if lhs.is_a?(AST::Node) && !lhs.respond_to?(:name)

    assign_node.name == block_arg_name
  end
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 135

def block_sends_to_super?(super_node, parent_node = super_node.parent)
  # Checks if the send node of a block is the given super node,
  # or a method chain containing it.
  return false unless parent_node
  return false unless parent_node.type?(:block, :numblock)

  parent_node.send_node.each_node(:super).any?(super_node)
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 96

def find_def_node(super_node)
  super_node.ancestors.find do |node|
    # When defining dynamic methods, implicitly calling `super` is not possible.
    # Since there is a possibility of delegation to `define_method`,
    # `super` used within the block is always allowed.
    break if node.block_type? && !block_sends_to_super?(super_node, node)

    break node if DEF_TYPES.include?(node.type)
  end
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 206

def forward_arg_same?(def_arg, super_arg)
  def_arg.forward_arg_type? && super_arg.forwarded_args_type?
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 161

def keyword_arg_same?(def_arg, super_arg)
  return false unless def_arg.kwarg_type? || def_arg.kwoptarg_type?
  return false unless (pair_node = super_arg).pair_type?
  return false unless (sym_node = pair_node.key).sym_type?
  return false unless (lvar_node = pair_node.value).lvar_type?
  return false unless sym_node.source == lvar_node.source

  def_arg.name == sym_node.value
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 171

def keyword_rest_arg_same?(def_arg, super_arg)
  return false unless def_arg.kwrestarg_type?
  # anonymous forwarding
  return true if def_arg.name.nil? && super_arg.forwarded_kwrestarg_type?
  return false unless super_arg.kwsplat_type?
  return false unless (lvar_node = super_arg.children.first).lvar_type?

  def_arg.name == lvar_node.children.first
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 78

def on_super(super_node)
  return unless (def_node = find_def_node(super_node))

  def_node_args = def_node.arguments.argument_list
  super_args = preprocess_super_args(super_node.arguments)

  return unless arguments_identical?(def_node, super_node, def_node_args, super_args)

  # If the number of arguments to the def node and super node are different here,
  # it's because the block argument is not forwarded.
  message = def_node_args.size == super_args.size ? MSG : MSG_INLINE_BLOCK
  add_offense(super_node, message: message) do |corrector|
    corrector.replace(super_node, 'super')
  end
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 144

def positional_arg_same?(def_arg, super_arg)
  return false unless def_arg.arg_type? || def_arg.optarg_type?
  return false unless super_arg.lvar_type?

  def_arg.name == super_arg.children.first
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 151

def positional_rest_arg_same(def_arg, super_arg)
  return false unless def_arg.restarg_type?
  # anonymous forwarding
  return true if def_arg.name.nil? && super_arg.forwarded_restarg_type?
  return false unless super_arg.splat_type?
  return false unless (lvar_node = super_arg.children.first).lvar_type?

  def_arg.name == lvar_node.children.first
end

# File 'lib/rubocop/cop/style/super_arguments.rb', line 210

def preprocess_super_args(super_args)
  super_args.flat_map do |node|
    if node.hash_type? && !node.braces?
      node.children
    else
      node
    end
  end
end