104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package ast_test
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/z7zmey/php-parser/pkg/ast"
|
|
"github.com/z7zmey/php-parser/pkg/ast/traverser"
|
|
"github.com/z7zmey/php-parser/pkg/ast/visitor"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
func ExampleStxTree() {
|
|
stxTree := &ast.Root{
|
|
Stmts: []ast.Vertex{
|
|
&ast.Nullable{
|
|
Expr: &ast.Parameter{
|
|
Type: nil,
|
|
Var: nil,
|
|
DefaultValue: nil,
|
|
},
|
|
},
|
|
&ast.Identifier{},
|
|
&ast.ArgumentList{
|
|
Arguments: []ast.Vertex{
|
|
&ast.Argument{},
|
|
&ast.Argument{
|
|
Expr: &ast.ScalarDnumber{},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
traverser.NewDFS(&testVisitor{}).Traverse(stxTree)
|
|
|
|
//output:
|
|
//=> *ast.Root
|
|
//=> Stmts:
|
|
//=> *ast.Nullable
|
|
//=> Expr:
|
|
//=> *ast.Parameter
|
|
//=> *ast.Identifier
|
|
//=> *ast.ArgumentList
|
|
//=> Arguments:
|
|
//=> *ast.Argument
|
|
//=> *ast.Argument
|
|
//=> Expr:
|
|
//=> *ast.ScalarDnumber
|
|
}
|
|
|
|
type testVisitor struct {
|
|
visitor.Null
|
|
depth int
|
|
}
|
|
|
|
|
|
func (v *testVisitor) Enter(key string, _ bool) {
|
|
v.depth++
|
|
fmt.Fprint(os.Stdout, "=>", strings.Repeat(" ", v.depth), key, ":\n")
|
|
}
|
|
|
|
func (v *testVisitor) Leave(key string, _ bool) {
|
|
v.depth--
|
|
}
|
|
|
|
func (v *testVisitor) EnterNode(n ast.Vertex) bool {
|
|
v.depth++
|
|
n.Accept(v)
|
|
|
|
return true
|
|
}
|
|
|
|
func (v *testVisitor) LeaveNode(_ ast.Vertex) {
|
|
v.depth--
|
|
}
|
|
|
|
func (v *testVisitor) Root(_ *ast.Root) {
|
|
fmt.Fprintln(os.Stdout, "=>", strings.Repeat(" ", v.depth-1), "*ast.Root")
|
|
}
|
|
|
|
func (v *testVisitor) Nullable(_ *ast.Nullable) {
|
|
fmt.Fprintln(os.Stdout, "=>", strings.Repeat(" ", v.depth-1), "*ast.Nullable")
|
|
}
|
|
|
|
func (v *testVisitor) Parameter(_ *ast.Parameter) {
|
|
fmt.Fprintln(os.Stdout, "=>", strings.Repeat(" ", v.depth-1), "*ast.Parameter")
|
|
}
|
|
|
|
func (v *testVisitor) Identifier(_ *ast.Identifier) {
|
|
fmt.Fprintln(os.Stdout, "=>", strings.Repeat(" ", v.depth-1), "*ast.Identifier")
|
|
}
|
|
|
|
func (v *testVisitor) ArgumentList(_ *ast.ArgumentList) {
|
|
fmt.Fprintln(os.Stdout, "=>", strings.Repeat(" ", v.depth-1), "*ast.ArgumentList")
|
|
}
|
|
|
|
func (v *testVisitor) Argument(_ *ast.Argument) {
|
|
fmt.Fprintln(os.Stdout, "=>", strings.Repeat(" ", v.depth-1), "*ast.Argument")
|
|
}
|
|
|
|
func (v *testVisitor) ScalarDnumber(_ *ast.ScalarDnumber) {
|
|
fmt.Fprintln(os.Stdout, "=>", strings.Repeat(" ", v.depth-1), "*ast.ScalarDnumber")
|
|
}
|