feat: Implement closures in the comptime interpreter

compiler/noirc_frontend/src/hir/comptime/interpreter.rs

@@ -219,15 +219,14 @@
             }
         } else {
             let name = self.elaborator.interner.function_name(&function);
             unreachable!("Non-builtin, lowlevel or oracle builtin fn '{name}'")
         }
     }
 
     fn call_closure(
         &mut self,
         closure: HirLambda,
-        // TODO: How to define environment here?
-        _environment: Vec<Value>,
+        environment: Vec<Value>,
         arguments: Vec<(Value, Location)>,
         call_location: Location,
     ) -> IResult<Value> {
@@ -246,6 +245,10 @@
             self.define_pattern(parameter, typ, argument, arg_location)?;
         }
 
+        for (param, arg) in closure.captures.into_iter().zip(environment) {
+            self.define(param.ident.id, arg);
+        }
+
         let result = self.evaluate(closure.body)?;
 
         self.exit_function(previous_state);

test_programs/compile_success_empty/comptime_closures/Nargo.toml

@@ -0,0 +1,7 @@
+[package]
+name = "comptime_closures"
+type = "bin"
+authors = [""]
+compiler_version = ">=0.32.0"
+
+[dependencies]

test_programs/compile_success_empty/comptime_closures/src/main.nr

@@ -0,0 +1,39 @@
+fn main() {
+    comptime
+    {
+        closure_test(0);
+    }
+}
+
+fn closure_test(mut x: Field) {
+    let one = 1;
+    let add1 = |z| {
+        (|| {
+            *z += one;
+        })()
+    };
+
+    let two = 2;
+    let add2 = |z| {
+        *z = *z + two;
+    };
+
+    add1(&mut x);
+    assert(x == 1);
+
+    add2(&mut x);
+    assert(x == 3);
+
+    issue_2120();
+}
+
+fn issue_2120() {
+    let x1 = &mut 42;
+    let set_x1 = |y| { *x1 = y; };
+
+    assert(*x1 == 42);
+    set_x1(44);
+    assert(*x1 == 44);
+    set_x1(*x1);
+    assert(*x1 == 44);
+}