Class: RSpec::Support::Source::Node

Inherits:
Object
  • Object
show all
Includes:
Enumerable
Defined in:
lib/rspec/support/source/node.rb

Overview

A wrapper for Ripper AST node which is generated with ‘Ripper.sexp`.

Direct Known Subclasses

ExpressionSequenceNode

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(ripper_sexp, parent = nil) ⇒ Node

Returns a new instance of Node.



19
20
21
22
 
# File 'lib/rspec/support/source/node.rb', line 19

def initialize(ripper_sexp, parent=nil)
  @sexp = ripper_sexp.freeze
  @parent = parent
end

Instance Attribute Details

#parent ⇒ Object (readonly)

Returns the value of attribute parent.



13
14
15
 
# File 'lib/rspec/support/source/node.rb', line 13

def parent
  @parent
end

#sexp ⇒ Object (readonly)

Returns the value of attribute sexp.



13
14
15
 
# File 'lib/rspec/support/source/node.rb', line 13

def sexp
  @sexp
end

Class Method Details

.sexp?(array) ⇒ Boolean

Returns:

  • (Boolean) 


15
16
17
 
# File 'lib/rspec/support/source/node.rb', line 15

def self.sexp?(array)
  array.is_a?(Array) && array.first.is_a?(Symbol)
end

Instance Method Details

#args ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
 
# 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

#children ⇒ Object 



42
43
44
 
# File 'lib/rspec/support/source/node.rb', line 42

def children
  @children ||= args.select { |arg| arg.is_a?(Node) }.freeze
end

#each ⇒ Object

We use a loop here (instead of recursion) to prevent SystemStackError



51
52
53
54
55
56
57
58
59
60
61
 
# 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

#each_ancestor ⇒ Object 



63
64
65
66
67
68
69
70
71
 
# 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

#inspect ⇒ Object 



73
74
75
 
# File 'lib/rspec/support/source/node.rb', line 73

def inspect
  "#<#{self.class} #{type}>"
end

#location ⇒ Object 



46
47
48
 
# File 'lib/rspec/support/source/node.rb', line 46

def location
  @location ||= args.find { |arg| arg.is_a?(Location) }
end

#type ⇒ Object 



24
25
26
 
# File 'lib/rspec/support/source/node.rb', line 24

def type
  sexp[0]
end