use ferveo data structures

nucypher · Jul 24, 2024 · da3af4f · da3af4f
1 parent 884a78b
commit da3af4f
Show file tree

Hide file tree

Showing 3 changed files with 86 additions and 18 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/ferveo/src/debug.rs b/ferveo/src/debug.rs
@@ -1,16 +1,20 @@
+use core::str::FromStr;
 use std::{
     fs::File,
     io::{self, BufRead},
     path::PathBuf,
 };
 
 use hex::FromHex;
+use crate::api::{PublicKey, Transcript};
+use ferveo_common::FromBytes;
+use crate::EthereumAddress;
 
 #[derive(Debug)]
-struct ValidatorTranscript {
-    validator_address: String,
-    validator_pk: Vec<u8>,
-    transcript: Vec<u8>,
+struct ValidatorTranscript{
+    validator_address: EthereumAddress,
+    validator_pk: PublicKey,
+    transcript: Transcript
 }
 
 fn parse_file(file_path: &PathBuf) -> io::Result<Vec<ValidatorTranscript>> {
@@ -61,6 +65,10 @@ fn parse_file(file_path: &PathBuf) -> io::Result<Vec<ValidatorTranscript>> {
             ));
         }
 
+        let validator_address = EthereumAddress::from_str(&validator_address).unwrap();
+        let validator_pk = PublicKey::from_bytes(&validator_pk).unwrap();
+        let transcript = Transcript::from_bytes(&transcript).unwrap();
+
         records.push(ValidatorTranscript {
             validator_address,
             validator_pk,

diff --git a/ferveo/src/lib.rs b/ferveo/src/lib.rs
@@ -2,6 +2,7 @@
 
 #[cfg(feature = "bindings-wasm")]
 extern crate alloc;
+extern crate core;
 
 #[cfg(feature = "bindings-python")]
 pub mod bindings_python;