wip: issue/88

Author: morganthomas

crates/lean_compiler/src/b_compile_intermediate.rs

@@ -212,7 +212,6 @@ fn compile_lines(
             }
 
             SimpleLine::Match { value, arms } => {
-                let saved_stack_pos = compiler.stack_pos;
                 compiler.stack_frame_layout.scopes.push(ScopeLayout::default());
 
                 let match_index = compiler.match_blocks.len();
@@ -221,6 +220,8 @@ fn compile_lines(
                 let value_simplified = IntermediateValue::from_simple_expr(value, compiler);
 
                 let mut compiled_arms = vec![];
+                let saved_stack_pos = compiler.stack_pos;
+                let mut new_stack_pos = saved_stack_pos;
                 for (i, arm) in arms.iter().enumerate() {
                     let mut arm_declared_vars = declared_vars.clone();
                     compiler.stack_pos = saved_stack_pos;
@@ -237,7 +238,9 @@ fn compile_lines(
                     } else {
                         declared_vars.intersection(&arm_declared_vars).cloned().collect()
                     };
+                    new_stack_pos = new_stack_pos.max(compiler.stack_pos);
                 }
+                compiler.stack_pos = new_stack_pos;
                 compiler.match_blocks.push(MatchBlock {
                     function_name: function_name.clone(),
                     match_cases: compiled_arms,