class SyntaxTree::Search

Provides an interface for searching for a pattern of nodes against a subtree of an AST.

Attributes

Public Class Methods

Source

# File lib/syntax_tree/search.rb, line 9
def initialize(pattern)
  @pattern = pattern
end

Public Instance Methods

scan (root) { |node| ... }

Source

# File lib/syntax_tree/search.rb, line 13
def scan(root)
  return to_enum(__method__, root) unless block_given?
  queue = [root]

  until queue.empty?
    node = queue.shift
    next unless node

    yield node if pattern.call(node)
    queue += node.child_nodes
  end
end