go/ast: add Inspect function for easy AST inspection w/o a visitor

R=rsc
CC=golang-dev
https://golang.org/cl/2770044

diff --git a/src/pkg/go/ast/walk.go b/src/pkg/go/ast/walk.go
index 6c9837a01d546e243b65a117d7a3a4530db6819e..296da5652de700eb8726853de37fc3535188adfb 100644 (file)
--- a/src/pkg/go/ast/walk.go
+++ b/src/pkg/go/ast/walk.go
@@ -321,3 +321,22 @@ func Walk(v Visitor, node interface{}) {
 
        v.Visit(nil)
 }
+
+
+type inspector func(node interface{}) bool
+
+func (f inspector) Visit(node interface{}) Visitor {
+       if node != nil && f(node) {
+               return f
+       }
+       return nil
+}
+
+
+// Inspect traverses an AST in depth-first order: If node != nil, it
+// invokes f(node). If f returns true, inspect invokes f for all the
+// non-nil children of node, recursively.
+//
+func Inspect(ast interface{}, f func(node interface{}) bool) {
+       Walk(inspector(f), ast)
+}