Allow for early termination of the visitor

flowr-analysis · Aug 30, 2024 · 0e46211 · 0e46211
1 parent f836948
commit 0e46211
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/R/visit.R b/R/visit.R
@@ -23,7 +23,11 @@ visit_node <- function(node, callback) {
     return()
   }
 
-  callback(node)
+  res <- callback(node)
+  # Exit early if the callback returns FALSE
+  if (isFALSE(res)) {
+    return()
+  }
 
   # same logic as the builtin visitor (while explicitly specifying if an entry is a single node or a list)
   # https://github.com/Code-Inspect/flowr/blob/main/src/r-bridge/lang-4.x/ast/model/processing/visitor.ts#L22