while node

node/stmt/while.go

@@ -0,0 +1,39 @@
+package stmt
+
+import (
+	"fmt"
+	"io"
+
+	"github.com/z7zmey/php-parser/node"
+	"github.com/z7zmey/php-parser/token"
+)
+
+type While struct {
+	node.SimpleNode
+	token token.Token
+	cond  node.Node
+	stmt  node.Node
+}
+
+func NewWhile(token token.Token, cond node.Node, stmt node.Node) node.Node {
+	return While{
+		node.SimpleNode{Name: "While", Attributes: make(map[string]string)},
+		token,
+		cond,
+		stmt,
+	}
+}
+
+func (n While) Print(out io.Writer, indent string) {
+	fmt.Fprintf(out, "\n%v%v [%d %d] %q", indent, n.Name, n.token.StartLine, n.token.EndLine, n.token.Value)
+
+	if n.cond != nil {
+		fmt.Fprintf(out, "\n%vcond:", indent+"  ")
+		n.cond.Print(out, indent+"    ")
+	}
+
+	if n.stmt != nil {
+		fmt.Fprintf(out, "\n%vstmt:", indent+"  ")
+		n.stmt.Print(out, indent+"    ")
+	}
+}

parser/parser.y

@@ -371,11 +371,7 @@ statement:
     |   if_stmt                                         { $$ = $1; }
     |   alt_if_stmt                                     { $$ = $1; }
     |   T_WHILE '(' expr ')' while_statement
-        {
-            $$ = node.NewSimpleNode("While").
-                Append(node.NewSimpleNode("expr").Append($3)).
-                Append(node.NewSimpleNode("stmt").Append($5));
-        }
+                                                        { $$ = stmt.NewWhile($1, $3, $5) }
     |   T_DO statement T_WHILE '(' expr ')' ';'         { $$ = stmt.NewDo($1, $2, $5) }
     |   T_FOR '(' for_exprs ';' for_exprs ';' for_exprs ')' for_statement
                                                         { $$ = stmt.NewFor($1, $3, $5, $7, $9) }