From a33e675a01187db6be5f234510fe58e22bbffaca Mon Sep 17 00:00:00 2001
From: Alexandre Terrasa <alexandre.terrasa@shopify.com>
Date: Fri, 1 Apr 2022 11:34:33 -0400
Subject: [PATCH] Introduce `visit_all` to visit multiple nodes

Signed-off-by: Alexandre Terrasa <alexandre.terrasa@shopify.com>
---
 lib/syntax_tree/visitor.rb | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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.