maride/php-parser
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pretty printer: now stmt node printers doesn`t print new line

Browse Source
This commit is contained in:
z7zmey 2018-04-02 22:50:36 +03:00
parent 46b7396c02
commit c3664b9fbf
2 changed files with 272 additions and 156 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

168
printer/printer.go
View File

@ -30,9 +30,12 @@ func joinPrint(glue string, o io.Writer, nn []node.Node) {
} }
func printNodes(o io.Writer, nn []node.Node) { func printNodes(o io.Writer, nn []node.Node) {
// TODO: handle indentations l := len(nn) - 1
for _, n := range nn { for k, n := range nn {
Print(o, n) Print(o, n)
if k < l {
io.WriteString(o, "\n")
}
} }
} }
@ -1268,19 +1271,23 @@ func printStmtAltElseIf(o io.Writer, n node.Node) {
io.WriteString(o, "elseif (") io.WriteString(o, "elseif (")
Print(o, nn.Cond) Print(o, nn.Cond)
io.WriteString(o, ") :\n") io.WriteString(o, ") :")
s := nn.Stmt.(*stmt.StmtList) if s := nn.Stmt.(*stmt.StmtList).Stmts; len(s) > 0 {
printNodes(o, s.Stmts) io.WriteString(o, "\n")
printNodes(o, s)
}
} }
func printStmtAltElse(o io.Writer, n node.Node) { func printStmtAltElse(o io.Writer, n node.Node) {
nn := n.(*stmt.AltElse) nn := n.(*stmt.AltElse)
io.WriteString(o, "else :\n") io.WriteString(o, "else :")
s := nn.Stmt.(*stmt.StmtList) if s := nn.Stmt.(*stmt.StmtList).Stmts; len(s) > 0 {
printNodes(o, s.Stmts) io.WriteString(o, "\n")
printNodes(o, s)
}
} }
func printStmtAltFor(o io.Writer, n node.Node) { func printStmtAltFor(o io.Writer, n node.Node) {
@ -1296,8 +1303,9 @@ func printStmtAltFor(o io.Writer, n node.Node) {
s := nn.Stmt.(*stmt.StmtList) s := nn.Stmt.(*stmt.StmtList)
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "\n")
io.WriteString(o, "endfor;\n") io.WriteString(o, "endfor;")
} }
func printStmtAltForeach(o io.Writer, n node.Node) { func printStmtAltForeach(o io.Writer, n node.Node) {
@ -1322,8 +1330,9 @@ func printStmtAltForeach(o io.Writer, n node.Node) {
s := nn.Stmt.(*stmt.StmtList) s := nn.Stmt.(*stmt.StmtList)
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "\n")
io.WriteString(o, "endforeach;\n") io.WriteString(o, "endforeach;")
} }
func printStmtAltIf(o io.Writer, n node.Node) { func printStmtAltIf(o io.Writer, n node.Node) {
@ -1337,14 +1346,17 @@ func printStmtAltIf(o io.Writer, n node.Node) {
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
for _, elseif := range nn.ElseIf { for _, elseif := range nn.ElseIf {
io.WriteString(o, "\n")
Print(o, elseif) Print(o, elseif)
} }
if nn.Else != nil { if nn.Else != nil {
io.WriteString(o, "\n")
Print(o, nn.Else) Print(o, nn.Else)
} }
io.WriteString(o, "endif;\n") io.WriteString(o, "\n")
io.WriteString(o, "endif;")
} }
func printStmtAltSwitch(o io.Writer, n node.Node) { func printStmtAltSwitch(o io.Writer, n node.Node) {
@ -1356,8 +1368,9 @@ func printStmtAltSwitch(o io.Writer, n node.Node) {
s := nn.Cases s := nn.Cases
printNodes(o, s) printNodes(o, s)
io.WriteString(o, "\n")
io.WriteString(o, "endswitch;\n") io.WriteString(o, "endswitch;")
} }
func printStmtAltWhile(o io.Writer, n node.Node) { func printStmtAltWhile(o io.Writer, n node.Node) {
@ -1369,8 +1382,9 @@ func printStmtAltWhile(o io.Writer, n node.Node) {
s := nn.Stmt.(*stmt.StmtList) s := nn.Stmt.(*stmt.StmtList)
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "\n")
io.WriteString(o, "endwhile;\n") io.WriteString(o, "endwhile;")
} }
func printStmtBreak(o io.Writer, n node.Node) { func printStmtBreak(o io.Writer, n node.Node) {
@ -1382,7 +1396,7 @@ func printStmtBreak(o io.Writer, n node.Node) {
Print(o, nn.Expr) Print(o, nn.Expr)
} }
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtCase(o io.Writer, n node.Node) { func printStmtCase(o io.Writer, n node.Node) {
@ -1390,8 +1404,12 @@ func printStmtCase(o io.Writer, n node.Node) {
io.WriteString(o, "case ") io.WriteString(o, "case ")
Print(o, nn.Cond) Print(o, nn.Cond)
io.WriteString(o, ":\n") io.WriteString(o, ":")
printNodes(o, nn.Stmts)
if len(nn.Stmts) > 0 {
io.WriteString(o, "\n")
printNodes(o, nn.Stmts)
}
} }
func printStmtCatch(o io.Writer, n node.Node) { func printStmtCatch(o io.Writer, n node.Node) {
@ -1403,7 +1421,7 @@ func printStmtCatch(o io.Writer, n node.Node) {
Print(o, nn.Variable) Print(o, nn.Variable)
io.WriteString(o, ") {\n") io.WriteString(o, ") {\n")
printNodes(o, nn.Stmts) printNodes(o, nn.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
} }
func printStmtClassMethod(o io.Writer, n node.Node) { func printStmtClassMethod(o io.Writer, n node.Node) {
@ -1429,9 +1447,9 @@ func printStmtClassMethod(o io.Writer, n node.Node) {
Print(o, nn.ReturnType) Print(o, nn.ReturnType)
} }
io.WriteString(o, "\n{\n") // TODO: handle indentation io.WriteString(o, "\n{\n")
printNodes(o, nn.Stmts) printNodes(o, nn.Stmts)
io.WriteString(o, "}\n") // TODO: handle indentation io.WriteString(o, "\n}")
} }
func printStmtClass(o io.Writer, n node.Node) { func printStmtClass(o io.Writer, n node.Node) {
@ -1464,9 +1482,9 @@ func printStmtClass(o io.Writer, n node.Node) {
joinPrint(", ", o, nn.Implements) joinPrint(", ", o, nn.Implements)
} }
io.WriteString(o, "\n{\n") // TODO: handle indentation io.WriteString(o, "\n{\n")
printNodes(o, nn.Stmts) printNodes(o, nn.Stmts)
io.WriteString(o, "}\n") // TODO: handle indentation io.WriteString(o, "\n}")
} }
func printStmtClassConstList(o io.Writer, n node.Node) { func printStmtClassConstList(o io.Writer, n node.Node) {
@ -1480,7 +1498,7 @@ func printStmtClassConstList(o io.Writer, n node.Node) {
joinPrint(", ", o, nn.Consts) joinPrint(", ", o, nn.Consts)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtConstant(o io.Writer, n node.Node) { func printStmtConstant(o io.Writer, n node.Node) {
@ -1500,7 +1518,7 @@ func printStmtContinue(o io.Writer, n node.Node) {
Print(o, nn.Expr) Print(o, nn.Expr)
} }
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtDeclare(o io.Writer, n node.Node) { func printStmtDeclare(o io.Writer, n node.Node) {
@ -1517,7 +1535,7 @@ func printStmtDeclare(o io.Writer, n node.Node) {
case *stmt.StmtList: case *stmt.StmtList:
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
default: default:
io.WriteString(o, "\n") io.WriteString(o, "\n")
Print(o, s) Print(o, s)
@ -1526,8 +1544,12 @@ func printStmtDeclare(o io.Writer, n node.Node) {
func printStmtDefault(o io.Writer, n node.Node) { func printStmtDefault(o io.Writer, n node.Node) {
nn := n.(*stmt.Default) nn := n.(*stmt.Default)
io.WriteString(o, "default:\n") io.WriteString(o, "default:")
printNodes(o, nn.Stmts)
if len(nn.Stmts) > 0 {
io.WriteString(o, "\n")
printNodes(o, nn.Stmts)
}
} }
func printStmtDo(o io.Writer, n node.Node) { func printStmtDo(o io.Writer, n node.Node) {
@ -1538,22 +1560,23 @@ func printStmtDo(o io.Writer, n node.Node) {
case *stmt.StmtList: case *stmt.StmtList:
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "} ") io.WriteString(o, "\n} ")
default: default:
io.WriteString(o, "\n") io.WriteString(o, "\n")
Print(o, s) Print(o, s)
io.WriteString(o, "\n")
} }
io.WriteString(o, "while (") io.WriteString(o, "while (")
Print(o, nn.Cond) Print(o, nn.Cond)
io.WriteString(o, ");\n") io.WriteString(o, ");")
} }
func printStmtEcho(o io.Writer, n node.Node) { func printStmtEcho(o io.Writer, n node.Node) {
nn := n.(*stmt.Echo) nn := n.(*stmt.Echo)
io.WriteString(o, "echo ") io.WriteString(o, "echo ")
joinPrint(", ", o, nn.Exprs) joinPrint(", ", o, nn.Exprs)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtElseif(o io.Writer, n node.Node) { func printStmtElseif(o io.Writer, n node.Node) {
@ -1570,7 +1593,7 @@ func printStmtElseif(o io.Writer, n node.Node) {
case *stmt.StmtList: case *stmt.StmtList:
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
default: default:
io.WriteString(o, "\n") io.WriteString(o, "\n")
Print(o, s) Print(o, s)
@ -1589,7 +1612,7 @@ func printStmtElse(o io.Writer, n node.Node) {
case *stmt.StmtList: case *stmt.StmtList:
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
default: default:
io.WriteString(o, "\n") io.WriteString(o, "\n")
Print(o, s) Print(o, s)
@ -1601,7 +1624,7 @@ func printStmtExpression(o io.Writer, n node.Node) {
Print(o, nn.Expr) Print(o, nn.Expr)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtFinally(o io.Writer, n node.Node) { func printStmtFinally(o io.Writer, n node.Node) {
@ -1609,7 +1632,7 @@ func printStmtFinally(o io.Writer, n node.Node) {
io.WriteString(o, "finally {\n") io.WriteString(o, "finally {\n")
printNodes(o, nn.Stmts) printNodes(o, nn.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
} }
func printStmtFor(o io.Writer, n node.Node) { func printStmtFor(o io.Writer, n node.Node) {
@ -1630,7 +1653,7 @@ func printStmtFor(o io.Writer, n node.Node) {
case *stmt.StmtList: case *stmt.StmtList:
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
default: default:
io.WriteString(o, "\n") io.WriteString(o, "\n")
Print(o, s) Print(o, s)
@ -1662,7 +1685,7 @@ func printStmtForeach(o io.Writer, n node.Node) {
case *stmt.StmtList: case *stmt.StmtList:
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
default: default:
io.WriteString(o, "\n") io.WriteString(o, "\n")
Print(o, s) Print(o, s)
@ -1691,7 +1714,7 @@ func printStmtFunction(o io.Writer, n node.Node) {
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, nn.Stmts) printNodes(o, nn.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
} }
func printStmtGlobal(o io.Writer, n node.Node) { func printStmtGlobal(o io.Writer, n node.Node) {
@ -1699,7 +1722,7 @@ func printStmtGlobal(o io.Writer, n node.Node) {
io.WriteString(o, "global ") io.WriteString(o, "global ")
joinPrint(", ", o, nn.Vars) joinPrint(", ", o, nn.Vars)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtGoto(o io.Writer, n node.Node) { func printStmtGoto(o io.Writer, n node.Node) {
@ -1707,7 +1730,7 @@ func printStmtGoto(o io.Writer, n node.Node) {
io.WriteString(o, "goto ") io.WriteString(o, "goto ")
Print(o, nn.Label) Print(o, nn.Label)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtGroupUse(o io.Writer, n node.Node) { func printStmtGroupUse(o io.Writer, n node.Node) {
@ -1723,11 +1746,11 @@ func printStmtGroupUse(o io.Writer, n node.Node) {
Print(o, nn.Prefix) Print(o, nn.Prefix)
io.WriteString(o, "\\{") io.WriteString(o, "\\{")
joinPrint(", ", o, nn.UseList) joinPrint(", ", o, nn.UseList)
io.WriteString(o, "};\n") io.WriteString(o, "};")
} }
func printStmtHaltCompiler(o io.Writer, n node.Node) { func printStmtHaltCompiler(o io.Writer, n node.Node) {
io.WriteString(o, "__halt_compiler();\n") io.WriteString(o, "__halt_compiler();")
} }
func printStmtIf(o io.Writer, n node.Node) { func printStmtIf(o io.Writer, n node.Node) {
@ -1744,17 +1767,19 @@ func printStmtIf(o io.Writer, n node.Node) {
case *stmt.StmtList: case *stmt.StmtList:
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
default: default:
io.WriteString(o, "\n") io.WriteString(o, "\n")
Print(o, s) Print(o, s)
} }
if nn.ElseIf != nil { if nn.ElseIf != nil {
io.WriteString(o, "\n")
printNodes(o, nn.ElseIf) printNodes(o, nn.ElseIf)
} }
if nn.Else != nil { if nn.Else != nil {
io.WriteString(o, "\n")
Print(o, nn.Else) Print(o, nn.Else)
} }
} }
@ -1764,7 +1789,7 @@ func printStmtInlineHTML(o io.Writer, n node.Node) {
io.WriteString(o, "?>") io.WriteString(o, "?>")
io.WriteString(o, nn.Value) io.WriteString(o, nn.Value)
io.WriteString(o, "<?php\n") io.WriteString(o, "<?php")
} }
func printStmtInterface(o io.Writer, n node.Node) { func printStmtInterface(o io.Writer, n node.Node) {
@ -1782,35 +1807,39 @@ func printStmtInterface(o io.Writer, n node.Node) {
joinPrint(", ", o, nn.Extends) joinPrint(", ", o, nn.Extends)
} }
io.WriteString(o, "\n{\n") // TODO: handle indentation io.WriteString(o, "\n{\n")
printNodes(o, nn.Stmts) printNodes(o, nn.Stmts)
io.WriteString(o, "}\n") // TODO: handle indentation io.WriteString(o, "\n}")
} }
func printStmtLabel(o io.Writer, n node.Node) { func printStmtLabel(o io.Writer, n node.Node) {
nn := n.(*stmt.Label) nn := n.(*stmt.Label)
Print(o, nn.LabelName) Print(o, nn.LabelName)
io.WriteString(o, ":\n") io.WriteString(o, ":")
} }
func printStmtNamespace(o io.Writer, n node.Node) { func printStmtNamespace(o io.Writer, n node.Node) {
nn := n.(*stmt.Namespace) nn := n.(*stmt.Namespace)
io.WriteString(o, "namespace ") io.WriteString(o, "namespace")
Print(o, nn.NamespaceName)
if nn.NamespaceName != nil {
io.WriteString(o, " ")
Print(o, nn.NamespaceName)
}
if nn.Stmts != nil { if nn.Stmts != nil {
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, nn.Stmts) printNodes(o, nn.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
} else { } else {
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
} }
func printStmtNop(o io.Writer, n node.Node) { func printStmtNop(o io.Writer, n node.Node) {
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtPropertyList(o io.Writer, n node.Node) { func printStmtPropertyList(o io.Writer, n node.Node) {
@ -1819,7 +1848,7 @@ func printStmtPropertyList(o io.Writer, n node.Node) {
joinPrint(" ", o, nn.Modifiers) joinPrint(" ", o, nn.Modifiers)
io.WriteString(o, " ") io.WriteString(o, " ")
joinPrint(", ", o, nn.Properties) joinPrint(", ", o, nn.Properties)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtProperty(o io.Writer, n node.Node) { func printStmtProperty(o io.Writer, n node.Node) {
@ -1838,7 +1867,7 @@ func printStmtReturn(o io.Writer, n node.Node) {
io.WriteString(o, "return ") io.WriteString(o, "return ")
Print(o, nn.Expr) Print(o, nn.Expr)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtStaticVar(o io.Writer, n node.Node) { func printStmtStaticVar(o io.Writer, n node.Node) {
@ -1856,7 +1885,7 @@ func printStmtStatic(o io.Writer, n node.Node) {
io.WriteString(o, "static ") io.WriteString(o, "static ")
joinPrint(", ", o, nn.Vars) joinPrint(", ", o, nn.Vars)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtStmtList(o io.Writer, n node.Node) { func printStmtStmtList(o io.Writer, n node.Node) {
@ -1874,7 +1903,7 @@ func printStmtSwitch(o io.Writer, n node.Node) {
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, nn.Cases) printNodes(o, nn.Cases)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
} }
func printStmtThrow(o io.Writer, n node.Node) { func printStmtThrow(o io.Writer, n node.Node) {
@ -1882,7 +1911,7 @@ func printStmtThrow(o io.Writer, n node.Node) {
io.WriteString(o, "throw ") io.WriteString(o, "throw ")
Print(o, nn.Expr) Print(o, nn.Expr)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtTraitMethodRef(o io.Writer, n node.Node) { func printStmtTraitMethodRef(o io.Writer, n node.Node) {
@ -1909,7 +1938,7 @@ func printStmtTraitUseAlias(o io.Writer, n node.Node) {
Print(o, nn.Alias) Print(o, nn.Alias)
} }
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtTraitUsePrecedence(o io.Writer, n node.Node) { func printStmtTraitUsePrecedence(o io.Writer, n node.Node) {
@ -1919,7 +1948,7 @@ func printStmtTraitUsePrecedence(o io.Writer, n node.Node) {
io.WriteString(o, " insteadof ") io.WriteString(o, " insteadof ")
joinPrint(", ", o, nn.Insteadof) joinPrint(", ", o, nn.Insteadof)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtTraitUse(o io.Writer, n node.Node) { func printStmtTraitUse(o io.Writer, n node.Node) {
@ -1931,9 +1960,9 @@ func printStmtTraitUse(o io.Writer, n node.Node) {
if nn.Adaptations != nil { if nn.Adaptations != nil {
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, nn.Adaptations) printNodes(o, nn.Adaptations)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
} else { } else {
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
} }
@ -1945,7 +1974,7 @@ func printStmtTrait(o io.Writer, n node.Node) {
io.WriteString(o, "\n{\n") io.WriteString(o, "\n{\n")
printNodes(o, nn.Stmts) printNodes(o, nn.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
} }
func printStmtTry(o io.Writer, n node.Node) { func printStmtTry(o io.Writer, n node.Node) {
@ -1953,10 +1982,17 @@ func printStmtTry(o io.Writer, n node.Node) {
io.WriteString(o, "try {\n") io.WriteString(o, "try {\n")
printNodes(o, nn.Stmts) printNodes(o, nn.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
printNodes(o, nn.Catches) if nn.Catches != nil {
Print(o, nn.Finally) io.WriteString(o, "\n")
printNodes(o, nn.Catches)
}
if nn.Finally != nil {
io.WriteString(o, "\n")
Print(o, nn.Finally)
}
} }
func printStmtUnset(o io.Writer, n node.Node) { func printStmtUnset(o io.Writer, n node.Node) {
@ -1964,7 +2000,7 @@ func printStmtUnset(o io.Writer, n node.Node) {
io.WriteString(o, "unset(") io.WriteString(o, "unset(")
joinPrint(", ", o, nn.Vars) joinPrint(", ", o, nn.Vars)
io.WriteString(o, ");\n") io.WriteString(o, ");")
} }
func printStmtUseList(o io.Writer, n node.Node) { func printStmtUseList(o io.Writer, n node.Node) {
@ -1978,7 +2014,7 @@ func printStmtUseList(o io.Writer, n node.Node) {
} }
joinPrint(", ", o, nn.Uses) joinPrint(", ", o, nn.Uses)
io.WriteString(o, ";\n") io.WriteString(o, ";")
} }
func printStmtUse(o io.Writer, n node.Node) { func printStmtUse(o io.Writer, n node.Node) {
@ -2011,7 +2047,7 @@ func printStmtWhile(o io.Writer, n node.Node) {
case *stmt.StmtList: case *stmt.StmtList:
io.WriteString(o, " {\n") io.WriteString(o, " {\n")
printNodes(o, s.Stmts) printNodes(o, s.Stmts)
io.WriteString(o, "}\n") io.WriteString(o, "\n}")
default: default:
io.WriteString(o, "\n") io.WriteString(o, "\n")
Print(o, s) Print(o, s)

260
printer/printer_test.go
View File

@ -1847,7 +1847,24 @@ func TestPrintAltElseIf(t *testing.T) {
}, },
}) })
expected := "elseif ($a) :\n$b;\n" expected := `elseif ($a) :
$b;`
actual := o.String()
if expected != actual {
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
}
}
func TestPrintAltElseIfEmpty(t *testing.T) {
o := bytes.NewBufferString("")
printer.Print(o, &stmt.AltElseIf{
Cond: &expr.Variable{VarName: &node.Identifier{Value: "a"}},
Stmt: &stmt.StmtList{},
})
expected := `elseif ($a) :`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -1866,7 +1883,23 @@ func TestPrintAltElse(t *testing.T) {
}, },
}) })
expected := "else :\n$b;\n" expected := `else :
$b;`
actual := o.String()
if expected != actual {
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
}
}
func TestPrintAltElseEmpty(t *testing.T) {
o := bytes.NewBufferString("")
printer.Print(o, &stmt.AltElse{
Stmt: &stmt.StmtList{},
})
expected := `else :`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -1894,7 +1927,9 @@ func TestPrintAltFor(t *testing.T) {
}, },
}) })
expected := "for ($a; $b; $c) :\n$d;\nendfor;\n" expected := `for ($a; $b; $c) :
$d;
endfor;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -1917,7 +1952,9 @@ func TestPrintAltForeach(t *testing.T) {
}, },
}) })
expected := "foreach ($var as $key => &$val) :\n$d;\nendforeach;\n" expected := `foreach ($var as $key => &$val) :
$d;
endforeach;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -1965,8 +2002,7 @@ $b;
elseif ($c) : elseif ($c) :
else : else :
$b; $b;
endif; endif;`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2000,8 +2036,7 @@ case 'a':
$a; $a;
case 'b': case 'b':
$b; $b;
endswitch; endswitch;`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2021,7 +2056,9 @@ func TestPrintAltWhile(t *testing.T) {
}, },
}) })
expected := "while ($a) :\n$b;\nendwhile;\n" expected := `while ($a) :
$b;
endwhile;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2036,7 +2073,7 @@ func TestPrintStmtBreak(t *testing.T) {
Expr: &scalar.Lnumber{Value: "1"}, Expr: &scalar.Lnumber{Value: "1"},
}) })
expected := "break 1;\n" expected := "break 1;"
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2054,7 +2091,24 @@ func TestPrintStmtCase(t *testing.T) {
}, },
}) })
expected := "case $a:\n$a;\n" expected := `case $a:
$a;`
actual := o.String()
if expected != actual {
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
}
}
func TestPrintStmtCaseEmpty(t *testing.T) {
o := bytes.NewBufferString("")
printer.Print(o, &stmt.Case{
Cond: &expr.Variable{VarName: &node.Identifier{Value: "a"}},
Stmts: []node.Node{},
})
expected := "case $a:"
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2078,8 +2132,7 @@ func TestPrintStmtCatch(t *testing.T) {
expected := `catch (Exception | \RuntimeException $e) { expected := `catch (Exception | \RuntimeException $e) {
$a; $a;
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2115,8 +2168,7 @@ func TestPrintStmtClassMethod(t *testing.T) {
expected := `public function &foo(?int &$a = null, ...$b): void expected := `public function &foo(?int &$a = null, ...$b): void
{ {
$a; $a;
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2151,8 +2203,7 @@ func TestPrintStmtClass(t *testing.T) {
expected := `abstract class Foo extends Bar implements Baz, Quuz expected := `abstract class Foo extends Bar implements Baz, Quuz
{ {
public const FOO = 'bar'; public const FOO = 'bar';
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2194,8 +2245,7 @@ func TestPrintStmtAnonymousClass(t *testing.T) {
expected := `abstract class($a, $b) extends Bar implements Baz, Quuz expected := `abstract class($a, $b) extends Bar implements Baz, Quuz
{ {
public const FOO = 'bar'; public const FOO = 'bar';
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2220,7 +2270,7 @@ func TestPrintStmtClassConstList(t *testing.T) {
}, },
}) })
expected := "public const FOO = 'a', BAR = 'b';\n" expected := `public const FOO = 'a', BAR = 'b';`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2251,7 +2301,7 @@ func TestPrintStmtContinue(t *testing.T) {
Expr: &scalar.Lnumber{Value: "1"}, Expr: &scalar.Lnumber{Value: "1"},
}) })
expected := "continue 1;\n" expected := `continue 1;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2276,7 +2326,9 @@ func TestPrintStmtDeclareStmts(t *testing.T) {
}, },
}) })
expected := "declare(FOO = 'bar') {\n;\n}\n" expected := `declare(FOO = 'bar') {
;
}`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2297,7 +2349,8 @@ func TestPrintStmtDeclareExpr(t *testing.T) {
Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}}, Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}},
}) })
expected := "declare(FOO = 'bar')\n'bar';\n" expected := `declare(FOO = 'bar')
'bar';`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2318,7 +2371,7 @@ func TestPrintStmtDeclareNop(t *testing.T) {
Stmt: &stmt.Nop{}, Stmt: &stmt.Nop{},
}) })
expected := "declare(FOO = 'bar');\n" expected := `declare(FOO = 'bar');`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2335,7 +2388,23 @@ func TestPrintStmtDefalut(t *testing.T) {
}, },
}) })
expected := "default:\n$a;\n" expected := `default:
$a;`
actual := o.String()
if expected != actual {
t.Errorf("\nexpected: %s\ngot: %s\n", expected, actual)
}
}
func TestPrintStmtDefalutEmpty(t *testing.T) {
o := bytes.NewBufferString("")
printer.Print(o, &stmt.Default{
Stmts: []node.Node{},
})
expected := `default:`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2353,7 +2422,9 @@ func TestPrintStmtDo_Expression(t *testing.T) {
}, },
}) })
expected := "do\n$a;\nwhile (1);\n" expected := `do
$a;
while (1);`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2373,7 +2444,9 @@ func TestPrintStmtDo_StmtList(t *testing.T) {
}, },
}) })
expected := "do {\n$a;\n} while (1);\n" expected := `do {
$a;
} while (1);`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2391,7 +2464,7 @@ func TestPrintStmtEcho(t *testing.T) {
}, },
}) })
expected := "echo $a, $b;\n" expected := `echo $a, $b;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2411,7 +2484,9 @@ func TestPrintStmtElseIfStmts(t *testing.T) {
}, },
}) })
expected := "elseif ($a) {\n;\n}\n" expected := `elseif ($a) {
;
}`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2427,7 +2502,8 @@ func TestPrintStmtElseIfExpr(t *testing.T) {
Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}}, Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}},
}) })
expected := "elseif ($a)\n'bar';\n" expected := `elseif ($a)
'bar';`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2443,7 +2519,7 @@ func TestPrintStmtElseIfNop(t *testing.T) {
Stmt: &stmt.Nop{}, Stmt: &stmt.Nop{},
}) })
expected := "elseif ($a);\n" expected := `elseif ($a);`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2462,7 +2538,9 @@ func TestPrintStmtElseStmts(t *testing.T) {
}, },
}) })
expected := "else {\n;\n}\n" expected := `else {
;
}`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2477,7 +2555,8 @@ func TestPrintStmtElseExpr(t *testing.T) {
Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}}, Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}},
}) })
expected := "else\n'bar';\n" expected := `else
'bar';`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2492,7 +2571,7 @@ func TestPrintStmtElseNop(t *testing.T) {
Stmt: &stmt.Nop{}, Stmt: &stmt.Nop{},
}) })
expected := "else;\n" expected := `else;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2505,7 +2584,7 @@ func TestPrintExpression(t *testing.T) {
printer.Print(o, &stmt.Expression{Expr: &expr.Variable{VarName: &node.Identifier{Value: "a"}}}) printer.Print(o, &stmt.Expression{Expr: &expr.Variable{VarName: &node.Identifier{Value: "a"}}})
expected := "$a;\n" expected := `$a;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2522,7 +2601,9 @@ func TestPrintStmtFinally(t *testing.T) {
}, },
}) })
expected := "finally {\n;\n}\n" expected := `finally {
;
}`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2553,7 +2634,9 @@ func TestPrintStmtForStmts(t *testing.T) {
}, },
}) })
expected := "for ($a, $b; $c, $d; $e, $f) {\n;\n}\n" expected := `for ($a, $b; $c, $d; $e, $f) {
;
}`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2577,7 +2660,8 @@ func TestPrintStmtForExpr(t *testing.T) {
Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}}, Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}},
}) })
expected := "for ($a; $b; $c)\n'bar';\n" expected := `for ($a; $b; $c)
'bar';`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2601,7 +2685,7 @@ func TestPrintStmtForNop(t *testing.T) {
Stmt: &stmt.Nop{}, Stmt: &stmt.Nop{},
}) })
expected := "for ($a; $b; $c);\n" expected := `for ($a; $b; $c);`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2622,7 +2706,9 @@ func TestPrintStmtForeachStmts(t *testing.T) {
}, },
}) })
expected := "foreach ($a as $b) {\n;\n}\n" expected := `foreach ($a as $b) {
;
}`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2640,7 +2726,8 @@ func TestPrintStmtForeachExpr(t *testing.T) {
Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}}, Stmt: &stmt.Expression{Expr: &scalar.String{Value: "bar"}},
}) })
expected := "foreach ($a as $k => $v)\n'bar';\n" expected := `foreach ($a as $k => $v)
'bar';`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2659,7 +2746,7 @@ func TestPrintStmtForeachNop(t *testing.T) {
Stmt: &stmt.Nop{}, Stmt: &stmt.Nop{},
}) })
expected := "foreach ($a as $k => &$v);\n" expected := `foreach ($a as $k => &$v);`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2686,7 +2773,9 @@ func TestPrintStmtFunction(t *testing.T) {
}, },
}) })
expected := "function &foo(&$var): \\Foo {\n;\n}\n" expected := `function &foo(&$var): \Foo {
;
}`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2704,7 +2793,7 @@ func TestPrintStmtGlobal(t *testing.T) {
}, },
}) })
expected := "global $a, $b;\n" expected := `global $a, $b;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2719,7 +2808,7 @@ func TestPrintStmtGoto(t *testing.T) {
Label: &node.Identifier{Value: "FOO"}, Label: &node.Identifier{Value: "FOO"},
}) })
expected := "goto FOO;\n" expected := `goto FOO;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2744,7 +2833,7 @@ func TestPrintStmtGroupUse(t *testing.T) {
}, },
}) })
expected := "use function Foo\\{Bar as Baz, Quuz};\n" expected := `use function Foo\{Bar as Baz, Quuz};`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2757,7 +2846,7 @@ func TestPrintHaltCompiler(t *testing.T) {
printer.Print(o, &stmt.HaltCompiler{}) printer.Print(o, &stmt.HaltCompiler{})
expected := "__halt_compiler();\n" expected := `__halt_compiler();`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2803,8 +2892,7 @@ $d;
} }
elseif ($e); elseif ($e);
else else
$f; $f;`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2828,8 +2916,7 @@ func TestPrintIfStmtList(t *testing.T) {
expected := `if ($a) { expected := `if ($a) {
$b; $b;
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2845,8 +2932,7 @@ func TestPrintIfNop(t *testing.T) {
Stmt: &stmt.Nop{}, Stmt: &stmt.Nop{},
}) })
expected := `if ($a); expected := `if ($a);`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2861,8 +2947,7 @@ func TestPrintInlineHtml(t *testing.T) {
Value: "test", Value: "test",
}) })
expected := `?>test<?php expected := `?>test<?php`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2897,8 +2982,7 @@ public function foo()
{ {
$a; $a;
} }
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2913,8 +2997,7 @@ func TestPrintLabel(t *testing.T) {
LabelName: &node.Identifier{Value: "FOO"}, LabelName: &node.Identifier{Value: "FOO"},
}) })
expected := `FOO: expected := `FOO:`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2929,8 +3012,7 @@ func TestPrintNamespace(t *testing.T) {
NamespaceName: &name.Name{Parts: []node.Node{&name.NamePart{Value: "Foo"}}}, NamespaceName: &name.Name{Parts: []node.Node{&name.NamePart{Value: "Foo"}}},
}) })
expected := `namespace Foo; expected := `namespace Foo;`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2950,8 +3032,7 @@ func TestPrintNamespaceWithStmts(t *testing.T) {
expected := `namespace Foo { expected := `namespace Foo {
$a; $a;
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2964,7 +3045,7 @@ func TestPrintNop(t *testing.T) {
printer.Print(o, &stmt.Nop{}) printer.Print(o, &stmt.Nop{})
expected := ";\n" expected := `;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -2990,7 +3071,7 @@ func TestPrintPropertyList(t *testing.T) {
}, },
}) })
expected := "public static $a, $b;\n" expected := `public static $a, $b;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3006,7 +3087,7 @@ func TestPrintProperty(t *testing.T) {
Expr: &scalar.Lnumber{Value: "1"}, Expr: &scalar.Lnumber{Value: "1"},
}) })
expected := "$a = 1" expected := `$a = 1`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3021,7 +3102,7 @@ func TestPrintReturn(t *testing.T) {
Expr: &scalar.Lnumber{Value: "1"}, Expr: &scalar.Lnumber{Value: "1"},
}) })
expected := "return 1;\n" expected := `return 1;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3037,7 +3118,7 @@ func TestPrintStaticVar(t *testing.T) {
Expr: &scalar.Lnumber{Value: "1"}, Expr: &scalar.Lnumber{Value: "1"},
}) })
expected := "$a = 1" expected := `$a = 1`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3059,7 +3140,7 @@ func TestPrintStatic(t *testing.T) {
}, },
}) })
expected := "static $a, $b;\n" expected := `static $a, $b;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3077,7 +3158,8 @@ func TestPrintStmtList(t *testing.T) {
}, },
}) })
expected := "$a;\n$b;\n" expected := `$a;
$b;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3111,8 +3193,7 @@ case 'a':
$a; $a;
case 'b': case 'b':
$b; $b;
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3127,7 +3208,7 @@ func TestPrintStmtThrow(t *testing.T) {
Expr: &expr.Variable{VarName: &node.Identifier{Value: "var"}}, Expr: &expr.Variable{VarName: &node.Identifier{Value: "var"}},
}) })
expected := "throw $var;\n" expected := `throw $var;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3143,7 +3224,7 @@ func TestPrintStmtTraitMethodRef(t *testing.T) {
Method: &node.Identifier{Value: "a"}, Method: &node.Identifier{Value: "a"},
}) })
expected := "Foo::a" expected := `Foo::a`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3163,7 +3244,7 @@ func TestPrintStmtTraitUseAlias(t *testing.T) {
Alias: &node.Identifier{Value: "b"}, Alias: &node.Identifier{Value: "b"},
}) })
expected := "Foo::a as public b;\n" expected := `Foo::a as public b;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3185,7 +3266,7 @@ func TestPrintStmtTraitUsePrecedence(t *testing.T) {
}, },
}) })
expected := "Foo::a insteadof Bar, Baz;\n" expected := `Foo::a insteadof Bar, Baz;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3203,7 +3284,7 @@ func TestPrintStmtTraitUse(t *testing.T) {
}, },
}) })
expected := "use Foo, Bar;\n" expected := `use Foo, Bar;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3232,8 +3313,7 @@ func TestPrintStmtTraitAdaptations(t *testing.T) {
expected := `use Foo, Bar { expected := `use Foo, Bar {
Foo::a as b; Foo::a as b;
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3264,8 +3344,7 @@ public function foo()
{ {
$a; $a;
} }
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3307,8 +3386,7 @@ $b;
} }
finally { finally {
; ;
} }`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3326,8 +3404,7 @@ func TestPrintStmtUset(t *testing.T) {
}, },
}) })
expected := `unset($a, $b); expected := `unset($a, $b);`
`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3351,7 +3428,7 @@ func TestPrintStmtUseList(t *testing.T) {
}, },
}) })
expected := "use function Foo as Bar, Baz;\n" expected := `use function Foo as Bar, Baz;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3368,7 +3445,7 @@ func TestPrintUse(t *testing.T) {
Alias: &node.Identifier{Value: "Bar"}, Alias: &node.Identifier{Value: "Bar"},
}) })
expected := "function Foo as Bar" expected := `function Foo as Bar`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3388,7 +3465,9 @@ func TestPrintWhileStmtList(t *testing.T) {
}, },
}) })
expected := "while ($a) {\n$a;\n}\n" expected := `while ($a) {
$a;
}`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3404,7 +3483,8 @@ func TestPrintWhileExpression(t *testing.T) {
Stmt: &stmt.Expression{Expr: &expr.Variable{VarName: &node.Identifier{Value: "a"}}}, Stmt: &stmt.Expression{Expr: &expr.Variable{VarName: &node.Identifier{Value: "a"}}},
}) })
expected := "while ($a)\n$a;\n" expected := `while ($a)
$a;`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {
@ -3420,7 +3500,7 @@ func TestPrintWhileNop(t *testing.T) {
Stmt: &stmt.Nop{}, Stmt: &stmt.Nop{},
}) })
expected := "while ($a);\n" expected := `while ($a);`
actual := o.String() actual := o.String()
if expected != actual { if expected != actual {