chore(aztec_noir): imply the open keyword (#2508)

noir-lang · Aug 31, 2023 · c344d86 · c344d86
1 parent 825600a
commit c344d86
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/crates/noirc_frontend/src/hir/def_map/aztec_library.rs b/crates/noirc_frontend/src/hir/def_map/aztec_library.rs
@@ -212,8 +212,11 @@ fn transform_function(ty: &str, func: &mut NoirFunction) {
     func.def.return_visibility = Visibility::Public;
 
     // Distinct return types are only required for private functions
-    if ty == "Private" {
-        func.def.return_distinctness = Distinctness::Distinct;
+    // Public functions should have open auto-inferred
+    match ty {
+        "Private" => func.def.return_distinctness = Distinctness::Distinct,
+        "Public" => func.def.is_open = true,
+        _ => (),
     }
 }
 

diff --git a/crates/noirc_frontend/src/hir/def_map/mod.rs b/crates/noirc_frontend/src/hir/def_map/mod.rs
@@ -89,7 +89,7 @@ impl CrateDefMap {
         let ast = parse_file(&mut context.file_manager, root_file_id, errors);
 
         #[cfg(feature = "aztec")]
-        let ast = aztec_library::transform(ast);
+        let ast = aztec_library::transform(ast, &crate_id, context, errors);
 
         // Allocate a default Module for the root, giving it a ModuleId
         let mut modules: Arena<ModuleData> = Arena::default();