fix: preceding simple statement in IfStmt should have its own block

shenyih0ng · Apr 16, 2024 · 70f6a20 · 70f6a20
1 parent 33d16a4
commit 70f6a20
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/go-slang/goroutine.ts b/src/go-slang/goroutine.ts
@@ -184,7 +184,14 @@ const Interpreter: {
 
   IfStatement: ({ stmt, cond, cons, alt }: IfStatement, { C, H }) => {
     const branchOp: BranchOp = { type: CommandType.BranchOp, cons, alt }
-    stmt ? C.pushR(...H.allocM([stmt, cond, branchOp])) : C.pushR(...H.allocM([cond, branchOp]))
+    stmt
+      ? C.push(
+          H.alloc({
+            type: NodeType.Block,
+            statements: [stmt, { type: NodeType.IfStatement, cond, cons, alt }]
+          })
+        )
+      : C.pushR(...H.allocM([cond, branchOp]))
   },
 
   ForStatement: (inst: ForStatement, { C, H }) => {