loop: fix loop after block no terminator

cell-labs · Aug 6, 2024 · 251819b · 251819b
1 parent 1cbaeb6
commit 251819b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/compiler/compiler/for.go b/compiler/compiler/for.go
@@ -17,6 +17,7 @@ func (c *Compiler) compileForNode(v *parser.ForNode) {
 }
 
 func (c *Compiler) compileForThreeType(v *parser.ForNode) {
+	forParent := c.contextBlock
 	// TODO: create a new context-block for code running inside the for loop
 	c.compile([]parser.Node{
 		v.BeforeLoop,
@@ -56,6 +57,7 @@ func (c *Compiler) compileForThreeType(v *parser.ForNode) {
 
 	// Set context to the new block after the loop
 	c.contextBlock = afterLoopBlock
+	afterLoopBlock.NewBr(forParent)
 
 	// Pop break and continue
 	c.contextLoopBreak = c.contextLoopBreak[0 : len(c.contextLoopBreak)-1]