From 51df90ef0473cc82aa91c8e930a42746207a132b Mon Sep 17 00:00:00 2001
From: Kevin Newton <kddnewton@gmail.com>
Date: Tue, 27 Jan 2026 15:08:55 -0500
Subject: [PATCH] Add Prism::Node#find_all

---
 templates/lib/prism/node.rb.erb               | 22 ++++++++++++++++++-
 templates/rbi/prism/node.rbi.erb              |  3 +++
 templates/sig/prism/node.rbs.erb              |  3 +++
 .../prism/result/breadth_first_search_test.rb | 11 ++++++++++
 4 files changed, 38 insertions(+), 1 deletion(-)

diff --git a/templates/lib/prism/node.rb.erb b/templates/lib/prism/node.rb.erb
index 6f8e8b0acc..d14a06961a 100644
--- a/templates/lib/prism/node.rb.erb
+++ b/templates/lib/prism/node.rb.erb
@@ -200,7 +200,7 @@ module Prism
     end
 
     # Returns the first node that matches the given block when visited in a
-    # depth-first search. This is useful for finding a node that matches a
+    # breadth-first search. This is useful for finding a node that matches a
     # particular condition.
     #
     #     node.breadth_first_search { |node| node.node_id == node_id }
@@ -215,6 +215,26 @@ module Prism
 
       nil
     end
+    alias find breadth_first_search
+
+    # Returns all of the nodes that match the given block when visited in a
+    # breadth-first search. This is useful for finding all nodes that match a
+    # particular condition.
+    #
+    #     node.breadth_first_search_all { |node| node.is_a?(Prism::CallNode) }
+    #
+    def breadth_first_search_all(&block)
+      queue = [self] #: Array[Prism::node]
+      results = [] #: Array[Prism::node]
+
+      while (node = queue.shift)
+        results << node if yield node
+        queue.concat(node.compact_child_nodes)
+      end
+
+      results
+    end
+    alias find_all breadth_first_search_all
 
     # Returns a list of the fields that exist for this node class. Fields
     # describe the structure of the node. This kind of reflection is useful for
diff --git a/templates/rbi/prism/node.rbi.erb b/templates/rbi/prism/node.rbi.erb
index c0474bd208..0c89b4524c 100644
--- a/templates/rbi/prism/node.rbi.erb
+++ b/templates/rbi/prism/node.rbi.erb
@@ -49,6 +49,9 @@ class Prism::Node
   sig { params(block: T.proc.params(node: Prism::Node).returns(T::Boolean)).returns(T.nilable(Prism::Node)) }
   def breadth_first_search(&block); end
 
+  sig { params(block: T.proc.params(node: Prism::Node).returns(T::Boolean)).returns(T::Array[Prism::Node]) }
+  def breadth_first_search_all(&block); end
+
   sig { abstract.params(visitor: Prism::Visitor).returns(T.untyped) }
   def accept(visitor); end
 
diff --git a/templates/sig/prism/node.rbs.erb b/templates/sig/prism/node.rbs.erb
index 01914b571c..138edc11f4 100644
--- a/templates/sig/prism/node.rbs.erb
+++ b/templates/sig/prism/node.rbs.erb
@@ -25,6 +25,9 @@ module Prism
     def to_dot: () -> String
     def tunnel: (Integer line, Integer column) -> Array[Prism::node]
     def breadth_first_search: () { (Prism::node) -> bool } -> Prism::node?
+    alias find breadth_first_search
+    def breadth_first_search_all: () { (Prism::node) -> bool } -> Array[Prism::node]
+    alias find_all breadth_first_search_all
     def newline!: (Array[untyped]) -> void
 
     def save: (_Repository repository) -> void
diff --git a/test/prism/result/breadth_first_search_test.rb b/test/prism/result/breadth_first_search_test.rb
index e2e043a902..7e7962f172 100644
--- a/test/prism/result/breadth_first_search_test.rb
+++ b/test/prism/result/breadth_first_search_test.rb
@@ -14,5 +14,16 @@ def test_breadth_first_search
       refute_nil found
       assert_equal 8, found.start_offset
     end
+
+    def test_breadth_first_search_all
+      result = Prism.parse("[1 + 2, 2]")
+      found_nodes =
+        result.value.breadth_first_search_all do |node|
+          node.is_a?(IntegerNode)
+        end
+
+      assert_equal 3, found_nodes.size
+      assert_equal 8, found_nodes[0].start_offset
+    end
   end
 end