php-parser/node/stmt/function.go

76 lines
1.3 KiB
Go
Raw Normal View History

2017-12-07 16:39:34 +00:00
package stmt
import (
"github.com/z7zmey/php-parser/node"
)
type Function struct {
2017-12-29 15:20:24 +00:00
name string
attributes map[string]interface{}
functionName node.Node
params []node.Node
returnType node.Node
stmts []node.Node
2017-12-07 16:39:34 +00:00
}
2017-12-29 15:20:24 +00:00
func NewFunction(functionName node.Node, returnsRef bool, params []node.Node, returnType node.Node, stmts []node.Node) node.Node {
2017-12-07 16:39:34 +00:00
return Function{
2017-12-27 17:55:58 +00:00
"Function",
2017-12-29 15:20:24 +00:00
map[string]interface{}{
"returnsRef": returnsRef,
},
functionName,
2017-12-07 16:39:34 +00:00
params,
returnType,
stmts,
}
}
2017-12-29 15:20:24 +00:00
func (n Function) Name() string {
return "Function"
}
func (n Function) Attributes() map[string]interface{} {
return n.attributes
}
2017-12-31 09:57:55 +00:00
func (n Function) Attribute(key string) interface{} {
return n.attributes[key]
}
func (n Function) SetAttribute(key string, value interface{}) {
n.attributes[key] = value
}
2017-12-27 23:23:32 +00:00
func (n Function) Walk(v node.Visitor) {
2017-12-28 11:36:27 +00:00
if v.EnterNode(n) == false {
2017-12-27 23:23:32 +00:00
return
}
2017-12-07 16:39:34 +00:00
2017-12-29 15:20:24 +00:00
if n.functionName != nil {
vv := v.GetChildrenVisitor("functionName")
n.functionName.Walk(vv)
}
2017-12-07 16:39:34 +00:00
if n.params != nil {
2017-12-28 11:36:27 +00:00
vv := v.GetChildrenVisitor("params")
2017-12-07 16:39:34 +00:00
for _, nn := range n.params {
2017-12-27 23:23:32 +00:00
nn.Walk(vv)
2017-12-07 16:39:34 +00:00
}
}
if n.returnType != nil {
2017-12-28 11:36:27 +00:00
vv := v.GetChildrenVisitor("returnType")
2017-12-27 23:23:32 +00:00
n.returnType.Walk(vv)
2017-12-07 16:39:34 +00:00
}
if n.stmts != nil {
2017-12-28 11:36:27 +00:00
vv := v.GetChildrenVisitor("stmts")
2017-12-07 16:39:34 +00:00
for _, nn := range n.stmts {
2017-12-27 23:23:32 +00:00
nn.Walk(vv)
2017-12-07 16:39:34 +00:00
}
}
2017-12-28 11:36:27 +00:00
v.LeaveNode(n)
2017-12-07 16:39:34 +00:00
}