Module: RLSL::Prism::IR::Traversal

Defined in:

lib/rlsl/prism/ir/traversal.rb

Class Method Summary

• .child_nodes(node) ⇒ Object
• .each(node) {|node| ... } ⇒ Object

Class Method Details

.child_nodes(node) ⇒ Object

# File 'lib/rlsl/prism/ir/traversal.rb', line 17

def child_nodes(node)
  case node
  when Block
    node.statements
  when VarDecl
    [node.initializer]
  when BinaryOp
    [node.left, node.right]
  when UnaryOp
    [node.operand]
  when FuncCall
    [node.receiver, *node.args]
  when FieldAccess, Swizzle
    [node.receiver]
  when IfStatement
    [node.condition, node.then_branch, node.else_branch]
  when Ternary
    [node.condition, node.then_expr, node.else_expr]
  when Return
    [node.expression]
  when Assignment
    [node.target, node.value]
  when ForLoop
    [node.range_start, node.range_end, node.body]
  when WhileLoop
    [node.condition, node.body]
  when Parenthesized
    [node.expression]
  when ArrayLiteral
    node.elements
  when ArrayIndex
    [node.array, node.index]
  when GlobalDecl
    [node.initializer]
  when FunctionDefinition
    [node.body]
  when MultipleAssignment
    [*node.targets, node.value]
  else
    []
  end.compact
end

.each(node) {|node| ... } ⇒ Object

Yields:

• (node)

# File 'lib/rlsl/prism/ir/traversal.rb', line 9

def each(node, &block)
  return enum_for(:each, node) unless block_given?
  return if node.nil?

  yield node
  child_nodes(node).each { |child| each(child, &block) }
end