yield yield_from nodes

node/expr/yield.go

@@ -0,0 +1,36 @@
+package expr
+
+import (
+	"fmt"
+	"io"
+
+	"github.com/z7zmey/php-parser/node"
+)
+
+type Yield struct {
+	node.SimpleNode
+	key   node.Node
+	value node.Node
+}
+
+func NewYield(key node.Node, value node.Node) node.Node {
+	return Yield{
+		node.SimpleNode{Name: "Yield", Attributes: make(map[string]string)},
+		key,
+		value,
+	}
+}
+
+func (n Yield) Print(out io.Writer, indent string) {
+	fmt.Fprintf(out, "\n%v%v [- -]", indent, n.Name)
+
+	if n.key != nil {
+		fmt.Fprintf(out, "\n%vkey:", indent+"  ")
+		n.key.Print(out, indent+"    ")
+	}
+
+	if n.value != nil {
+		fmt.Fprintf(out, "\n%vvalue:", indent+"  ")
+		n.value.Print(out, indent+"    ")
+	}
+}

node/expr/yield_from.go

@@ -0,0 +1,29 @@
+package expr
+
+import (
+	"fmt"
+	"io"
+
+	"github.com/z7zmey/php-parser/node"
+)
+
+type YieldFrom struct {
+	node.SimpleNode
+	expr node.Node
+}
+
+func NewYieldFrom(expression node.Node) node.Node {
+	return YieldFrom{
+		node.SimpleNode{Name: "YieldFrom", Attributes: make(map[string]string)},
+		expression,
+	}
+}
+
+func (n YieldFrom) Print(out io.Writer, indent string) {
+	fmt.Fprintf(out, "\n%v%v [- -]", indent, n.Name)
+
+	if n.expr != nil {
+		fmt.Fprintf(out, "\n%vexpr:", indent+"  ")
+		n.expr.Print(out, indent+"    ")
+	}
+}

parser/parser.go

@@ -4221,25 +4221,25 @@ yydefault:
 		yyDollar = yyS[yypt-1 : yypt+1]
 		//line parser/parser.y:894
 		{
-			yyVAL.node = node.NewSimpleNode("Yield")
+			yyVAL.node = expr.NewYield(nil, nil)
 		}
 	case 364:
 		yyDollar = yyS[yypt-2 : yypt+1]
 		//line parser/parser.y:895
 		{
-			yyVAL.node = node.NewSimpleNode("Yield").Append(yyDollar[2].node)
+			yyVAL.node = expr.NewYield(nil, yyDollar[2].node)
 		}
 	case 365:
 		yyDollar = yyS[yypt-4 : yypt+1]
 		//line parser/parser.y:896
 		{
-			yyVAL.node = node.NewSimpleNode("Yield").Append(yyDollar[2].node).Append(yyDollar[4].node)
+			yyVAL.node = expr.NewYield(yyDollar[2].node, yyDollar[4].node)
 		}
 	case 366:
 		yyDollar = yyS[yypt-2 : yypt+1]
 		//line parser/parser.y:897
 		{
-			yyVAL.node = node.NewSimpleNode("YieldFrom").Append(yyDollar[2].node)
+			yyVAL.node = expr.NewYieldFrom(yyDollar[2].node)
 		}
 	case 367:
 		yyDollar = yyS[yypt-10 : yypt+1]

parser/parser.y

@@ -891,10 +891,10 @@ expr_without_variable:
     |   scalar                                          { $$ = $1; }
     |   '`' backticks_expr '`'                          { $$ = expr.NewShellExec($2) }
     |   T_PRINT expr                                    { $$ = expr.NewPrint($2) }
-    |   T_YIELD                                         { $$ = node.NewSimpleNode("Yield"); }
+    |   T_YIELD                                         { $$ = expr.NewYield(nil, nil) }
-    |   T_YIELD expr                                    { $$ = node.NewSimpleNode("Yield").Append($2); }
+    |   T_YIELD expr                                    { $$ = expr.NewYield(nil, $2) }
-    |   T_YIELD expr T_DOUBLE_ARROW expr                { $$ = node.NewSimpleNode("Yield").Append($2).Append($4); }
+    |   T_YIELD expr T_DOUBLE_ARROW expr                { $$ = expr.NewYield($2, $4) }
-    |   T_YIELD_FROM expr                               { $$ = node.NewSimpleNode("YieldFrom").Append($2); }
+    |   T_YIELD_FROM expr                               { $$ = expr.NewYieldFrom($2) }
     |   T_FUNCTION returns_ref '(' parameter_list ')' lexical_vars return_type '{' inner_statement_list '}'
             {
                 $$ = expr.NewClosure($4.(node.SimpleNode).Children, $6, $7, $9.(node.SimpleNode).Children, false, $2 == "true")