abstract find_matching_nodes

Author: flyerhzm

lib/synvert/core/rewriter/scope/within_scope.rb

@@ -22,24 +22,31 @@ def process
       current_node = @instance.current_node
       return unless current_node
 
+      matching_nodes = find_matching_nodes(current_node)
       @instance.process_with_node current_node do
-        matching_nodes = []
-        matching_nodes << current_node if current_node.match? @rules
-        if @options[:recursive]
-          current_node.recursive_children do |child_node|
-            matching_nodes << child_node if child_node.match? @rules
-          end
-        else
-          current_node.children do |child_node|
-            matching_nodes << child_node if child_node.match? @rules
-          end
-        end
         matching_nodes.each do |matching_node|
           @instance.process_with_node matching_node do
             @instance.instance_eval &@block
           end
         end
       end
     end
+
+    private
+
+    def find_matching_nodes(current_node)
+      matching_nodes = []
+      if @options[:recursive]
+        matching_nodes << current_node if current_node.match? @rules
+        current_node.recursive_children do |child_node|
+          matching_nodes << child_node if child_node.match? @rules
+        end
+      else
+        current_node.each do |child_node|
+          matching_nodes << child_node if child_node.match? @rules
+        end
+      end
+      matching_nodes
+    end
   end
 end