nswrap/ast/if_stmt.go

50 lines
1.2 KiB
Go
Raw Normal View History

2019-04-09 11:52:21 -04:00
package ast
// IfStmt is node represent 'if'
type IfStmt struct {
Addr Address
Pos Position
2021-06-11 11:24:25 -04:00
HasElse bool
2019-04-09 11:52:21 -04:00
ChildNodes []Node
}
func parseIfStmt(line string) Node {
2019-04-09 11:52:21 -04:00
groups := groupsFromRegex(
2021-06-11 11:24:25 -04:00
"<(?P<position>.*)>(?P<has_else> has_else)?",
2019-04-09 11:52:21 -04:00
line,
)
if groups == nil {
return &Unknown{}
}
2019-04-09 11:52:21 -04:00
return &IfStmt{
Addr: ParseAddress(groups["address"]),
Pos: NewPositionFromString(groups["position"]),
2021-06-11 11:24:25 -04:00
HasElse: len(groups["has_else"]) > 0,
2019-04-09 11:52:21 -04:00
ChildNodes: []Node{},
}
}
// AddChild adds a new child node. Child nodes can then be accessed with the
// Children attribute.
func (n *IfStmt) AddChild(node Node) {
n.ChildNodes = append(n.ChildNodes, node)
}
// Address returns the numeric address of the node. See the documentation for
// the Address type for more information.
func (n *IfStmt) Address() Address {
return n.Addr
}
// Children returns the child nodes. If this node does not have any children or
// this node does not support children it will always return an empty slice.
func (n *IfStmt) Children() []Node {
return n.ChildNodes
}
// Position returns the position in the original source code.
func (n *IfStmt) Position() Position {
return n.Pos
}