From 48702bb6ea0378f7b77101a83923a4c604081c2f Mon Sep 17 00:00:00 2001
From: pedro bufulin <pedro@semiotic.ai>
Date: Wed, 6 Nov 2024 11:55:18 -0300
Subject: [PATCH] feat: add proof verification example

---
 Cargo.lock                                    |  1 +
 crates/forrestrie-examples/Cargo.toml         |  1 +
 .../examples/receipts_proof.rs                | 25 +++++++++++++++++--
 3 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/Cargo.lock b/Cargo.lock
index 4d0c01bb..92b12672 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -2375,6 +2375,7 @@ dependencies = [
  "prost-wkt-types",
  "reqwest",
  "reth-primitives",
+ "reth-trie-common",
  "serde",
  "serde_json",
  "ssz_types 0.6.0",
diff --git a/crates/forrestrie-examples/Cargo.toml b/crates/forrestrie-examples/Cargo.toml
index 6cdeec0d..ad8dc2c1 100644
--- a/crates/forrestrie-examples/Cargo.toml
+++ b/crates/forrestrie-examples/Cargo.toml
@@ -18,6 +18,7 @@ prost-wkt.workspace = true
 prost-wkt-types.workspace = true
 reqwest = { workspace = true, features = ["json"] }
 reth-primitives.workspace = true
+reth-trie-common.workspace = true
 serde = { workspace = true, features = ["derive"] }
 serde_json.workspace = true
 ssz_types.workspace = true
diff --git a/crates/forrestrie-examples/examples/receipts_proof.rs b/crates/forrestrie-examples/examples/receipts_proof.rs
index a75d874c..5936bb9b 100644
--- a/crates/forrestrie-examples/examples/receipts_proof.rs
+++ b/crates/forrestrie-examples/examples/receipts_proof.rs
@@ -5,8 +5,9 @@
 //!
 use firehose_client::client::{Chain, FirehoseClient};
 use firehose_protos::ethereum_v2::{self, eth_block::FullReceipt, Block};
-use forrestrie::execution_layer::build_trie_with_proofs;
+use forrestrie::execution_layer::{build_trie_with_proofs, TargetLeaf, TargetLeaves};
 use reth_primitives::ReceiptWithBloom;
+use reth_trie_common::proof::verify_proof;
 
 const EXECUTION_BLOCK_NUMBER: u64 = 20759937;
 
@@ -34,7 +35,10 @@ async fn main() {
         .map(|full_receipt| full_receipt.receipt.clone())
         .collect();
 
-    let mut hb = build_trie_with_proofs(&receipts_with_bloom, &[1, 2, 3]);
+    // These are de indexes of receipts on which proofs have to be generated
+    let target_idxs = &[1, 2, 3];
+    let targets = TargetLeaves::from_indices(target_idxs, &receipts_with_bloom).unwrap();
+    let mut hb = build_trie_with_proofs(&receipts_with_bloom, target_idxs);
 
     // produces the root, which matches the root of the blocks.
     // hb.root() also calculates the proofs and store them in the HashBuilder.
@@ -42,4 +46,21 @@ async fn main() {
 
     let calc_root = eth1_block.calculate_receipt_root();
     println!("roots: {:?},  {:?}", root, calc_root);
+
+    // proofs can be taken and sorted, so each proof matches one of the target.
+    // each proof of a specific target receipt is provided in `take_proof_nodes()`
+    // and can be stored or verified singularly
+    let proof = hb.take_proof_nodes();
+    for target in targets {
+        let _verification = verify_proof(
+            hb.root(),
+            target.nibbles.clone(),
+            Some(target.value.to_vec()),
+            proof
+                .clone()
+                .matching_nodes_sorted(&target.nibbles)
+                .iter()
+                .map(|(_, node)| node),
+        );
+    }
 }