feat: add support for reading boolean arrays from toml

noir-lang · Feb 23, 2023 · ba5b0f3 · ba5b0f3
1 parent 575436f
commit ba5b0f3
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/crates/nargo/tests/test_data/main_bool_arg/Prover.toml b/crates/nargo/tests/test_data/main_bool_arg/Prover.toml
@@ -1 +1,2 @@
 x = true
+y = [true, false]
diff --git a/crates/nargo/tests/test_data/main_bool_arg/src/main.nr b/crates/nargo/tests/test_data/main_bool_arg/src/main.nr
@@ -1,6 +1,8 @@
-fn main(x : bool) {
+fn main(x : bool, y: [bool;2]) {
     if x {
         constrain 1 != 2;
     }
+
     constrain x;
+    constrain y[0] != y[1];
 }
diff --git a/crates/noirc_abi/src/input_parser/toml.rs b/crates/noirc_abi/src/input_parser/toml.rs
@@ -60,6 +60,8 @@ enum TomlTypes {
     ArrayNum(Vec<u64>),
     // Array of hexadecimal integers
     ArrayString(Vec<String>),
+    // Array of booleans
+    ArrayBool(Vec<bool>),
     // Struct of TomlTypes
     Table(BTreeMap<String, TomlTypes>),
 }
@@ -119,6 +121,18 @@ impl InputValue {
 
                 InputValue::Vec(array_elements)
             }
+            TomlTypes::ArrayBool(arr_bool) => {
+                let array_elements = vecmap(arr_bool, |elem_bool| {
+                    if elem_bool {
+                        FieldElement::one()
+                    } else {
+                        FieldElement::zero()
+                    }
+                });
+
+                InputValue::Vec(array_elements)
+            }
+
             TomlTypes::Table(table) => {
                 let fields = match param_type {
                     AbiType::Struct { fields } => fields,