[refactoring] update ast structure of "If", "ElseIf", "Else" nodes

pkg/ast/visitor/dump.go

@@ -252,18 +252,6 @@ func (v *Dump) Argument(n *ast.Argument) {
 	}
 }
 
-func (v *Dump) StmtAltElse(n *ast.StmtAltElse) {
-	v.printIndentIfNotSingle(v.indent - 1)
-	v.print("&ast.StmtAltElse{\n")
-	v.printNode(n.GetNode())
-}
-
-func (v *Dump) StmtAltElseIf(n *ast.StmtAltElseIf) {
-	v.printIndentIfNotSingle(v.indent - 1)
-	v.print("&ast.StmtAltElseIf{\n")
-	v.printNode(n.GetNode())
-}
-
 func (v *Dump) StmtAltFor(n *ast.StmtAltFor) {
 	v.printIndentIfNotSingle(v.indent - 1)
 	v.print("&ast.StmtAltFor{\n")
@@ -276,12 +264,6 @@ func (v *Dump) StmtAltForeach(n *ast.StmtAltForeach) {
 	v.printNode(n.GetNode())
 }
 
-func (v *Dump) StmtAltIf(n *ast.StmtAltIf) {
-	v.printIndentIfNotSingle(v.indent - 1)
-	v.print("&ast.StmtAltIf{\n")
-	v.printNode(n.GetNode())
-}
-
 func (v *Dump) StmtAltSwitch(n *ast.StmtAltSwitch) {
 	v.printIndentIfNotSingle(v.indent - 1)
 	v.print("&ast.StmtAltSwitch{\n")
@@ -404,12 +386,22 @@ func (v *Dump) StmtElse(n *ast.StmtElse) {
 	v.printIndentIfNotSingle(v.indent - 1)
 	v.print("&ast.StmtElse{\n")
 	v.printNode(n.GetNode())
+
+	if n.Alt {
+		v.printIndent(v.indent)
+		v.print("Alt: true,\n")
+	}
 }
 
 func (v *Dump) StmtElseIf(n *ast.StmtElseIf) {
 	v.printIndentIfNotSingle(v.indent - 1)
 	v.print("&ast.StmtElseIf{\n")
 	v.printNode(n.GetNode())
+
+	if n.Alt {
+		v.printIndent(v.indent)
+		v.print("Alt: true,\n")
+	}
 }
 
 func (v *Dump) StmtExpression(n *ast.StmtExpression) {
@@ -469,6 +461,11 @@ func (v *Dump) StmtIf(n *ast.StmtIf) {
 	v.printIndentIfNotSingle(v.indent - 1)
 	v.print("&ast.StmtIf{\n")
 	v.printNode(n.GetNode())
+
+	if n.Alt {
+		v.printIndent(v.indent)
+		v.print("Alt: true,\n")
+	}
 }
 
 func (v *Dump) StmtInlineHtml(n *ast.StmtInlineHtml) {

pkg/ast/visitor/filter_parser_nodes.go

@@ -13,60 +13,6 @@ func (v *FilterParserNodes) EnterNode(n ast.Vertex) bool {
 	return true
 }
 
-func (v *FilterParserNodes) StmtAltIf(n *ast.StmtAltIf) {
-	for {
-		if nn, ok := n.Cond.(*ast.ParserBrackets); ok {
-			n.Cond = nn.Child
-		} else {
-			break
-		}
-	}
-
-	if nn, ok := n.Stmt.(*ast.ParserBrackets); ok {
-		n.Stmt = nn.Child
-	}
-}
-
-func (v *FilterParserNodes) StmtAltElseIf(n *ast.StmtAltElseIf) {
-	for {
-		if nn, ok := n.Cond.(*ast.ParserBrackets); ok {
-			n.Cond = nn.Child
-		} else {
-			break
-		}
-	}
-
-	if nn, ok := n.Stmt.(*ast.ParserBrackets); ok {
-		n.Stmt = nn.Child
-	}
-}
-
-func (v *FilterParserNodes) StmtAltElse(n *ast.StmtAltElse) {
-	if nn, ok := n.Stmt.(*ast.ParserBrackets); ok {
-		n.Stmt = nn.Child
-	}
-}
-
-func (v *FilterParserNodes) StmtIf(n *ast.StmtIf) {
-	for {
-		if nn, ok := n.Cond.(*ast.ParserBrackets); ok {
-			n.Cond = nn.Child
-		} else {
-			break
-		}
-	}
-}
-
-func (v *FilterParserNodes) StmtElseIf(n *ast.StmtElseIf) {
-	for {
-		if nn, ok := n.Cond.(*ast.ParserBrackets); ok {
-			n.Cond = nn.Child
-		} else {
-			break
-		}
-	}
-}
-
 func (v *FilterParserNodes) StmtWhile(n *ast.StmtWhile) {
 	for {
 		if nn, ok := n.Cond.(*ast.ParserBrackets); ok {

pkg/ast/visitor/filter_tokens.go

@@ -87,3 +87,29 @@ func (v *FilterTokens) StmtStmtList(n *ast.StmtStmtList) {
 	n.OpenCurlyBracket = nil
 	n.CloseCurlyBracket = nil
 }
+
+func (v *FilterTokens) StmtIf(n *ast.StmtIf) {
+	n.IfTkn = nil
+	n.OpenParenthesisTkn = nil
+	n.CloseParenthesisTkn = nil
+	n.ColonTkn = nil
+	n.EndIfTkn = nil
+	n.SemiColonTkn = nil
+}
+
+func (v *FilterTokens) StmtElseIf(n *ast.StmtElseIf) {
+	n.ElseIfTkn = nil
+	n.OpenParenthesisTkn = nil
+	n.CloseParenthesisTkn = nil
+	n.ColonTkn = nil
+}
+
+func (v *FilterTokens) StmtElse(n *ast.StmtElse) {
+	n.ElseTkn = nil
+	n.ColonTkn = nil
+}
+
+func (v *FilterTokens) ParserBrackets(n *ast.ParserBrackets) {
+	n.OpenBracketTkn = nil
+	n.CloseBracketTkn = nil
+}

pkg/ast/visitor/null.go

@@ -54,14 +54,6 @@ func (v *Null) Argument(_ *ast.Argument) {
 	// do nothing
 }
 
-func (v *Null) StmtAltElse(_ *ast.StmtAltElse) {
-	// do nothing
-}
-
-func (v *Null) StmtAltElseIf(_ *ast.StmtAltElseIf) {
-	// do nothing
-}
-
 func (v *Null) StmtAltFor(_ *ast.StmtAltFor) {
 	// do nothing
 }
@@ -70,10 +62,6 @@ func (v *Null) StmtAltForeach(_ *ast.StmtAltForeach) {
 	// do nothing
 }
 
-func (v *Null) StmtAltIf(_ *ast.StmtAltIf) {
-	// do nothing
-}
-
 func (v *Null) StmtAltSwitch(_ *ast.StmtAltSwitch) {
 	// do nothing
 }