class RSpec::Support::Source::Node

# File lib/rspec/support/source/node.rb, line 19
def initialize(ripper_sexp, parent=nil)
  @sexp = ripper_sexp.freeze
  @parent = parent
end

# File lib/rspec/support/source/node.rb, line 15
def self.sexp?(array)
  array.is_a?(Array) && array.first.is_a?(Symbol)
end

# File lib/rspec/support/source/node.rb, line 28
def args
  @args ||= raw_args.map do |raw_arg|
    if Node.sexp?(raw_arg)
      Node.new(raw_arg, self)
    elsif Location.location?(raw_arg)
      Location.new(*raw_arg)
    elsif raw_arg.is_a?(Array)
      ExpressionSequenceNode.new(raw_arg, self)
    else
      raw_arg
    end
  end.freeze
end

# File lib/rspec/support/source/node.rb, line 42
def children
  @children ||= args.select { |arg| arg.is_a?(Node) }.freeze
end

# File lib/rspec/support/source/node.rb, line 51
def each
  return to_enum(__method__) unless block_given?

  node_queue = []
  node_queue << self

  while (current_node = node_queue.shift)
    yield current_node
    node_queue.concat(current_node.children)
  end
end

# File lib/rspec/support/source/node.rb, line 63
def each_ancestor
  return to_enum(__method__) unless block_given?

  current_node = self

  while (current_node = current_node.parent)
    yield current_node
  end
end

# File lib/rspec/support/source/node.rb, line 73
def inspect
  "#<#{self.class} #{type}>"
end

# File lib/rspec/support/source/node.rb, line 46
def location
  @location ||= args.find { |arg| arg.is_a?(Location) }
end

# File lib/rspec/support/source/node.rb, line 24
def type
  sexp[0]
end

# File lib/rspec/support/source/node.rb, line 79
def raw_args
  sexp[1..-1] || []
end