pretty printer: print For, Foreach, Function
This commit is contained in:
z7zmey
parent
5cc62dd8ee
commit
483dc5a16e
@ -316,6 +316,12 @@ func getPrintFuncByNode(n node.Node) func(o io.Writer, n node.Node) {
|
||||
return printStmtExpression
|
||||
case *stmt.Finally:
|
||||
return printStmtFinally
|
||||
case *stmt.For:
|
||||
return printStmtFor
|
||||
case *stmt.Foreach:
|
||||
return printStmtForeach
|
||||
case *stmt.Function:
|
||||
return printStmtFunction
|
||||
|
||||
case *stmt.StmtList:
|
||||
return printStmtStmtList
|
||||
@ -1555,6 +1561,88 @@ func printStmtFinally(o io.Writer, n node.Node) {
|
||||
io.WriteString(o, "}\n")
|
||||
}
|
||||
|
||||
func printStmtFor(o io.Writer, n node.Node) {
|
||||
nn := n.(*stmt.For)
|
||||
|
||||
io.WriteString(o, "for (")
|
||||
joinPrint(", ", o, nn.Init)
|
||||
io.WriteString(o, "; ")
|
||||
joinPrint(", ", o, nn.Cond)
|
||||
io.WriteString(o, "; ")
|
||||
joinPrint(", ", o, nn.Loop)
|
||||
io.WriteString(o, ")")
|
||||
|
||||
switch s := nn.Stmt.(type) {
|
||||
case *stmt.Nop:
|
||||
Print(o, s)
|
||||
break
|
||||
case *stmt.StmtList:
|
||||
io.WriteString(o, " {\n")
|
||||
printNodes(o, s.Stmts)
|
||||
io.WriteString(o, "}\n")
|
||||
default:
|
||||
io.WriteString(o, "\n")
|
||||
Print(o, s)
|
||||
}
|
||||
}
|
||||
|
||||
func printStmtForeach(o io.Writer, n node.Node) {
|
||||
nn := n.(*stmt.Foreach)
|
||||
|
||||
io.WriteString(o, "foreach (")
|
||||
Print(o, nn.Expr)
|
||||
io.WriteString(o, " as ")
|
||||
|
||||
if nn.Key != nil {
|
||||
Print(o, nn.Key)
|
||||
io.WriteString(o, " => ")
|
||||
}
|
||||
|
||||
if nn.ByRef {
|
||||
io.WriteString(o, "&")
|
||||
}
|
||||
Print(o, nn.Variable)
|
||||
io.WriteString(o, ")")
|
||||
|
||||
switch s := nn.Stmt.(type) {
|
||||
case *stmt.Nop:
|
||||
Print(o, s)
|
||||
break
|
||||
case *stmt.StmtList:
|
||||
io.WriteString(o, " {\n")
|
||||
printNodes(o, s.Stmts)
|
||||
io.WriteString(o, "}\n")
|
||||
default:
|
||||
io.WriteString(o, "\n")
|
||||
Print(o, s)
|
||||
}
|
||||
}
|
||||
|
||||
func printStmtFunction(o io.Writer, n node.Node) {
|
||||
nn := n.(*stmt.Function)
|
||||
|
||||
io.WriteString(o, "function ")
|
||||
|
||||
if nn.ReturnsRef {
|
||||
io.WriteString(o, "&")
|
||||
}
|
||||
|
||||
Print(o, nn.FunctionName)
|
||||
|
||||
io.WriteString(o, "(")
|
||||
joinPrint(", ", o, nn.Params)
|
||||
io.WriteString(o, ")")
|
||||
|
||||
if nn.ReturnType != nil {
|
||||
io.WriteString(o, ": ")
|
||||
Print(o, nn.ReturnType)
|
||||
}
|
||||
|
||||
io.WriteString(o, " {\n")
|
||||
printNodes(o, nn.Stmts)
|
||||
io.WriteString(o, "}\n")
|
||||
}
|
||||
|
||||
func printStmtStmtList(o io.Writer, n node.Node) {
|
||||
nn := n.(*stmt.StmtList)
|
||||
|
||||
|
||||
@ -2530,6 +2530,169 @@ func TestPrintStmtFinally(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestPrintStmtForStmts(t *testing.T) {
|
||||
o := bytes.NewBufferString("")
|
||||
|
||||
printer.Print(o, &stmt.For{
|
||||
Init: []node.Node{
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "a"}},
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "b"}},
|
||||
},
|
||||
Cond: []node.Node{
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "c"}},
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "d"}},
|
||||
},
|
||||
Loop: []node.Node{
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "e"}},
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "f"}},
|
||||
},
|
||||
Stmt: &stmt.StmtList{
|
||||
Stmts: []node.Node{
|
||||
&stmt.Nop{},
|
||||
},
|
||||
},
|
||||
})
|
||||
|
||||
expected := "for ($a, $b; $c, $d; $e, $f) {\n;\n}\n"
|
||||
actual := o.String()
|
||||
|
||||
if expected != actual {
|
||||
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPrintStmtForExpr(t *testing.T) {
|
||||
o := bytes.NewBufferString("")
|
||||
|
||||
printer.Print(o, &stmt.For{
|
||||
Init: []node.Node{
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "a"}},
|
||||
},
|
||||
Cond: []node.Node{
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "b"}},
|
||||
},
|
||||
Loop: []node.Node{
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "c"}},
|
||||
},
|
||||
Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}},
|
||||
})
|
||||
|
||||
expected := "for ($a; $b; $c)\n'bar';\n"
|
||||
actual := o.String()
|
||||
|
||||
if expected != actual {
|
||||
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPrintStmtForNop(t *testing.T) {
|
||||
o := bytes.NewBufferString("")
|
||||
|
||||
printer.Print(o, &stmt.For{
|
||||
Init: []node.Node{
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "a"}},
|
||||
},
|
||||
Cond: []node.Node{
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "b"}},
|
||||
},
|
||||
Loop: []node.Node{
|
||||
&expr.Variable{VarName: &node.Identifier{Value: "c"}},
|
||||
},
|
||||
Stmt: &stmt.Nop{},
|
||||
})
|
||||
|
||||
expected := "for ($a; $b; $c);\n"
|
||||
actual := o.String()
|
||||
|
||||
if expected != actual {
|
||||
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPrintStmtForeachStmts(t *testing.T) {
|
||||
o := bytes.NewBufferString("")
|
||||
|
||||
printer.Print(o, &stmt.Foreach{
|
||||
Expr: &expr.Variable{VarName: &node.Identifier{Value: "a"}},
|
||||
Variable: &expr.Variable{VarName: &node.Identifier{Value: "b"}},
|
||||
Stmt: &stmt.StmtList{
|
||||
Stmts: []node.Node{
|
||||
&stmt.Nop{},
|
||||
},
|
||||
},
|
||||
})
|
||||
|
||||
expected := "foreach ($a as $b) {\n;\n}\n"
|
||||
actual := o.String()
|
||||
|
||||
if expected != actual {
|
||||
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPrintStmtForeachExpr(t *testing.T) {
|
||||
o := bytes.NewBufferString("")
|
||||
|
||||
printer.Print(o, &stmt.Foreach{
|
||||
Expr: &expr.Variable{VarName: &node.Identifier{Value: "a"}},
|
||||
Key: &expr.Variable{VarName: &node.Identifier{Value: "k"}},
|
||||
Variable: &expr.Variable{VarName: &node.Identifier{Value: "v"}},
|
||||
Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}},
|
||||
})
|
||||
|
||||
expected := "foreach ($a as $k => $v)\n'bar';\n"
|
||||
actual := o.String()
|
||||
|
||||
if expected != actual {
|
||||
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPrintStmtForeachNop(t *testing.T) {
|
||||
o := bytes.NewBufferString("")
|
||||
|
||||
printer.Print(o, &stmt.Foreach{
|
||||
ByRef: true,
|
||||
Expr: &expr.Variable{VarName: &node.Identifier{Value: "a"}},
|
||||
Key: &expr.Variable{VarName: &node.Identifier{Value: "k"}},
|
||||
Variable: &expr.Variable{VarName: &node.Identifier{Value: "v"}},
|
||||
Stmt: &stmt.Nop{},
|
||||
})
|
||||
|
||||
expected := "foreach ($a as $k => &$v);\n"
|
||||
actual := o.String()
|
||||
|
||||
if expected != actual {
|
||||
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
|
||||
}
|
||||
}
|
||||
func TestPrintStmtFunction(t *testing.T) {
|
||||
o := bytes.NewBufferString("")
|
||||
|
||||
printer.Print(o, &stmt.Function{
|
||||
ReturnsRef: true,
|
||||
FunctionName: &node.Identifier{Value: "foo"},
|
||||
Params: []node.Node{
|
||||
&node.Parameter{
|
||||
ByRef: true,
|
||||
Variadic: false,
|
||||
Variable: &expr.Variable{VarName: &node.Identifier{Value: "var"}},
|
||||
},
|
||||
},
|
||||
ReturnType: &name.FullyQualified{Parts: []node.Node{&name.NamePart{Value: "Foo"}}},
|
||||
Stmts: []node.Node{
|
||||
&stmt.Nop{},
|
||||
},
|
||||
})
|
||||
|
||||
expected := "function &foo(&$var): \\Foo {\n;\n}\n"
|
||||
actual := o.String()
|
||||
|
||||
if expected != actual {
|
||||
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
|
||||
}
|
||||
}
|
||||
|
||||
func TestPrintStmtList(t *testing.T) {
|
||||
o := bytes.NewBufferString("")
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user