diff --git a/lib/syntax_tree/visitor.rb b/lib/syntax_tree/visitor.rb index 87e6ef76..27561cdf 100644 --- a/lib/syntax_tree/visitor.rb +++ b/lib/syntax_tree/visitor.rb @@ -57,8 +57,12 @@ def visit(node) node&.accept(self) end + def visit_all(nodes) + nodes.each { |node| visit(node) } + end + def visit_child_nodes(node) - node.child_nodes.each { |child_node| visit(child_node) } + visit_all(node.child_nodes) end # Visit an ARef node.