added register_traits method

Author: majabbour

program/rust/build.rs

@@ -14,7 +14,6 @@ fn main() {
     parser.register_traits("pc_price_info", borsh_derives.to_vec());
     parser.register_traits("cmd_upd_price", borsh_derives.to_vec());
 
-
     //generate and write bindings
     let bindings = Builder::default()
         .header("./src/bindings.h")

program/rust/build_utils.rs

@@ -24,6 +24,17 @@ impl<'a> DeriveAdderParserCallback<'a> {
     }
 }
 
+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<'_> {}