php-parser/pkg/ast/ast_test.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")
}