added register_traits method

Author: majabbour

program/rust/build.rs

@@ -1,6 +1,5 @@
 mod build_utils;
 use bindgen::Builder;
-use std::collections::HashMap;
 use std::vec::Vec;
 
 
@@ -9,11 +8,9 @@ fn main() {
 
     let borsh_derives: Vec<String> = vec!["BorshSerialize".to_string(), "BorshDeserialize".to_string()];
 
-    let parser = build_utils::DeriveAdderParserCallback{types_to_traits: HashMap::from([
-        ("cmd_hdr", borsh_derives)
-        //map more struct names to traits here, be sure that the definitions of your traites
-        //are included in c_oracle_header.rs
-    ])};
+    //make a parser and to it type, traits pairs
+    let mut parser = build_utils::DeriveAdderParserCallback::new();
+    parser.register_traits("cmd_hdr", borsh_derives);
 
     //generate and write bindings
     let bindings = Builder::default().header("./src/bindings.h").parse_callbacks(Box::new(parser)).rustfmt_bindings(true).generate().expect("Unable to generate bindings");

program/rust/build_utils.rs

@@ -6,11 +6,23 @@ use std::collections::HashMap;
 ///that the traits of each struct are added to its
 ///definition when an instance of this struct
 ///is provided as a ParseCallback for bindgen
-#[derive(Debug)]
+#[derive(Debug, Default)]
 pub struct DeriveAdderParserCallback<'a>{
     pub types_to_traits: HashMap<&'a str, Vec<String>> ,
 }
 
+impl <'a> DeriveAdderParserCallback<'a>{
+    ///create a parser that does not add any traits
+    pub fn new() -> Self {
+        Default::default()
+    }
+    //add pairs of types and their desired traits
+    pub fn register_traits(&mut self, type_name: &'a str ,traits: Vec<String>){
+        self.types_to_traits.insert(&type_name, traits);
+    }
+    
+}
+
 //this is required to implement the callback trait
 impl UnwindSafe for DeriveAdderParserCallback<'_> {}