findallInAST bug

emad-elsaid · Jan 14, 2023 · bb1668e · bb1668e
1 parent db4d1a2
commit bb1668e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/renderer.go b/renderer.go
@@ -56,6 +56,10 @@ func FindInAST[t ast.Node](n ast.Node) (found t, ok bool) {
 // Extract all nodes of a specific type from the AST
 func FindAllInAST[t ast.Node](n ast.Node) (a []t) {
 	ast.Walk(n, func(n ast.Node, entering bool) (ast.WalkStatus, error) {
+		if !entering {
+			return ast.WalkContinue, nil
+		}
+
 		if casted, ok := n.(t); ok {
 			a = append(a, casted)
 		}