Make tree for folders

2 files changed, 9 insertions, 29 deletions

diff --git a/examples/merkletree_blake3.rs b/examples/merkletree_blake3.rs
index 7f30a76..77ff2a0 100644
--- a/examples/merkletree_blake3.rs
+++ b/examples/merkletree_blake3.rs
@@ -10,19 +10,9 @@ fn main() {
     }
 
     // Read file contents into a vector of bytes
-    let mut file_contents = Vec::new();
-    for filename in &filenames {
-        match std::fs::read(filename) {
-            Ok(contents) => file_contents.push(contents),
-            Err(e) => {
-                eprintln!("Failed to read file '{}': {}", filename, e);
-                std::process::exit(1);
-            }
-        }
-    }
-
     let hasher = Blake3Hasher::new();
-    let tree = MerkleTree::new(hasher.clone(), file_contents.clone());
+
+    let tree = MerkleTree::from_paths(hasher, filenames);
 
     println!("{}", tree.root().hash());
 }
diff --git a/examples/proofer_blake3.rs b/examples/proofer_blake3.rs
index ca38d6f..8db3999 100644
--- a/examples/proofer_blake3.rs
+++ b/examples/proofer_blake3.rs
@@ -1,5 +1,6 @@
 use mt_rs::{
-    hasher::{Blake3Hasher, Hasher},
+    fs,
+    hasher::Blake3Hasher,
     node::Node,
     proof::{DefaultProofer, Proofer},
 };
@@ -21,27 +22,16 @@ fn main() {
         std::process::exit(1);
     }
 
-    let mut nodes: Vec<Node> = Vec::new();
-    for filename in &filenames {
-        match std::fs::read(filename) {
-            Ok(contents) => nodes.push(Node::new_leaf(Blake3Hasher::new().hash(&contents))),
-            Err(e) => {
-                eprintln!("Failed to read file '{}': {}", filename, e);
-                std::process::exit(1);
-            }
-        }
-    }
-
     let hasher = Blake3Hasher::new();
+
+    let nodes: Vec<Node> = fs::hash_dir(hasher.clone(), filenames.clone());
+    let first_node = nodes[0].clone();
+
     let proofer = DefaultProofer::new(hasher, nodes);
     let proof = proofer.generate(0).expect("Couldn't generate proof");
 
     println!(
         "{}",
-        proofer.verify(
-            &proof,
-            std::fs::read(&filenames[0]).unwrap(),
-            &root_hash[..]
-        )
+        proofer.verify_hash(&proof, first_node.hash().to_string(), &root_hash[..])
     );
 }