Merge pull request #91 from x-tyger/do

Implements do-while statement transpilation

ast/do_stmt.go

@@ -3,7 +3,7 @@ package ast
 type DoStmt struct {
 	Address  string
 	Position string
-	Children []interface{}
+	Children []Node
 }
 
 func parseDoStmt(line string) *DoStmt {
@@ -15,7 +15,7 @@ func parseDoStmt(line string) *DoStmt {
 	return &DoStmt{
 		Address:  groups["address"],
 		Position: groups["position"],
-		Children: []interface{}{},
+		Children: []Node{},
 	}
 }
 

ast/do_stmt_test.go

@@ -9,7 +9,7 @@ func TestDoStmt(t *testing.T) {
 		`0x7ff36d0a0938 <line:11:5, line:14:23>`: &DoStmt{
 			Address:  "0x7ff36d0a0938",
 			Position: "line:11:5, line:14:23",
-			Children: []interface{}{},
+			Children: []Node{},
 		},
 	}
 

tests/do.c

@@ -0,0 +1,10 @@
+#include <stdio.h>
+
+int main() {
+	int i = 0;
+
+	do {
+		printf("do loop %d\n", i);
+		i = i + 1;
+	} while( i < 4 );
+}

transpiler/branch.go

@@ -154,3 +154,32 @@ func transpileWhileStmt(n *ast.WhileStmt, p *program.Program) (*goast.ForStmt, e
 		Body: body,
 	}, nil
 }
+
+func transpileDoStmt(n *ast.DoStmt, p *program.Program) (*goast.ForStmt, error) {
+	children := n.Children
+
+	body, err := transpileToBlockStmt(children[0], p)
+	if err != nil {
+		return nil, err
+	}
+
+	condition, conditionType, err := transpileToExpr(children[1], p)
+	if err != nil {
+		return nil, err
+	}
+
+	// Add IfStmt to the end of the loop to check the condition
+	body.List = append(body.List, &goast.IfStmt{
+		Cond: &goast.UnaryExpr{
+			Op: token.NOT,
+			X:  types.CastExpr(p, condition, conditionType, "bool"),
+		},
+		Body: &goast.BlockStmt{
+			List: []goast.Stmt{&goast.BranchStmt{Tok: token.BREAK}},
+		},
+	})
+
+	return &goast.ForStmt{
+		Body: body,
+	}, nil
+}

transpiler/transpiler.go

@@ -137,6 +137,9 @@ func transpileToStmt(node ast.Node, p *program.Program) (goast.Stmt, error) {
 	case *ast.WhileStmt:
 		return transpileWhileStmt(n, p)
 
+	case *ast.DoStmt:
+		return transpileDoStmt(n, p)
+
 	case *ast.IfStmt:
 		return transpileIfStmt(n, p)