Set function internal flag based on attribute during macro run

AztecProtocol · Mar 4, 2024 · e3130fd · e3130fd
1 parent bd4b955
commit e3130fd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/noir/noir-repo/aztec_macros/src/lib.rs b/noir/noir-repo/aztec_macros/src/lib.rs
@@ -684,10 +684,11 @@ fn transform_function(
     let inputs_name = format!("{}ContextInputs", ty);
     let return_type_name = format!("{}CircuitPublicInputs", ty);
 
-    // Add check that msg sender equals this address
+    // Add check that msg sender equals this address and flag function as internal
     if is_internal {
         let is_internal_check = create_internal_check(func.name());
         func.def.body.0.insert(0, is_internal_check);
+        func.def.is_internal = true;
     }
 
     // Add initialization check