php-parser/node/stmt/if.go

75 lines
1.0 KiB
Go
Raw Normal View History

2017-12-06 22:53:03 +00:00
package stmt
import (
"github.com/z7zmey/php-parser/node"
"github.com/z7zmey/php-parser/token"
)
2017-12-27 23:23:32 +00:00
func (n If) Name() string {
2017-12-27 17:55:58 +00:00
return "If"
}
2017-12-06 22:53:03 +00:00
type If struct {
2017-12-27 17:55:58 +00:00
name string
2017-12-06 22:53:03 +00:00
token token.Token
cond node.Node
2017-12-07 09:35:45 +00:00
stmt node.Node
2017-12-06 22:53:03 +00:00
elseIf []node.Node
_else node.Node
}
2017-12-07 09:35:45 +00:00
func NewIf(token token.Token, cond node.Node, stmt node.Node) node.Node {
2017-12-06 22:53:03 +00:00
return If{
2017-12-27 17:55:58 +00:00
"If",
2017-12-06 22:53:03 +00:00
token,
cond,
2017-12-07 09:35:45 +00:00
stmt,
2017-12-06 22:53:03 +00:00
nil,
nil,
}
}
func (n If) AddElseIf(elseIf node.Node) node.Node {
2017-12-07 09:35:45 +00:00
if n.elseIf == nil {
2017-12-06 22:53:03 +00:00
n.elseIf = make([]node.Node, 0)
}
n.elseIf = append(n.elseIf, elseIf)
return n
}
func (n If) SetElse(_else node.Node) node.Node {
n._else = _else
return n
}
2017-12-27 23:23:32 +00:00
func (n If) Walk(v node.Visitor) {
if v.Visit(n) == false {
return
}
2017-12-06 22:53:03 +00:00
if n.cond != nil {
2017-12-27 23:23:32 +00:00
vv := v.Children("cond")
n.cond.Walk(vv)
2017-12-06 22:53:03 +00:00
}
2017-12-07 09:35:45 +00:00
if n.stmt != nil {
2017-12-27 23:23:32 +00:00
vv := v.Children("stmt")
n.stmt.Walk(vv)
2017-12-06 22:53:03 +00:00
}
2017-12-07 09:35:45 +00:00
2017-12-06 22:53:03 +00:00
if n.elseIf != nil {
2017-12-27 23:23:32 +00:00
vv := v.Children("elseIf")
2017-12-06 22:53:03 +00:00
for _, nn := range n.elseIf {
2017-12-27 23:23:32 +00:00
nn.Walk(vv)
2017-12-06 22:53:03 +00:00
}
}
2017-12-27 23:23:32 +00:00
2017-12-06 22:53:03 +00:00
if n._else != nil {
2017-12-27 23:23:32 +00:00
vv := v.Children("else")
n._else.Walk(vv)
2017-12-06 22:53:03 +00:00
}
}