feat: add DecodedProofRetainer

src/proof/decoded_retainer.rs

@@ -0,0 +1,48 @@
+use crate::{proof::DecodedProofNodes, Nibbles};
+use alloy_primitives::Bytes;
+
+use alloc::vec::Vec;
+
+/// Proof retainer is used to store proofs during merkle trie construction.
+/// It is intended to be used within the [`HashBuilder`](crate::HashBuilder).
+#[derive(Default, Clone, Debug)]
+pub struct DecodedProofRetainer {
+    /// The nibbles of the target trie keys to retain proofs for.
+    targets: Vec<Nibbles>,
+    /// The map retained trie node keys to RLP serialized trie nodes.
+    proof_nodes: DecodedProofNodes,
+}
+
+impl FromIterator<Nibbles> for DecodedProofRetainer {
+    fn from_iter<T: IntoIterator<Item = Nibbles>>(iter: T) -> Self {
+        Self::new(FromIterator::from_iter(iter))
+    }
+}
+
+impl DecodedProofRetainer {
+    /// Create new retainer with target nibbles.
+    pub fn new(targets: Vec<Nibbles>) -> Self {
+        Self { targets, proof_nodes: Default::default() }
+    }
+
+    /// Returns `true` if the given prefix matches the retainer target.
+    pub fn matches(&self, prefix: &Nibbles) -> bool {
+        self.targets.iter().any(|target| target.starts_with(prefix))
+    }
+
+    /// Returns all collected proofs.
+    pub fn into_proof_nodes(self) -> DecodedProofNodes {
+        self.proof_nodes
+    }
+
+    /// Retain the proof if the key matches any of the targets.
+    ///
+    /// Returns an error if the proof could not be decoded from the given proof bytes.
+    pub fn retain(&mut self, prefix: &Nibbles, proof: &[u8]) -> Result<(), alloy_rlp::Error> {
+        if prefix.is_empty() || self.matches(prefix) {
+            self.proof_nodes.insert_encoded(prefix.clone(), Bytes::from(proof.to_vec()))?;
+        }
+
+        Ok(())
+    }
+}

src/proof/mod.rs

@@ -12,6 +12,9 @@ pub use error::ProofVerificationError;
 mod decoded_proof_nodes;
 pub use decoded_proof_nodes::DecodedProofNodes;
 
+mod decoded_retainer;
+pub use decoded_retainer::DecodedProofRetainer;
+
 mod proof_nodes;
 pub use proof_nodes::ProofNodes;
 

src/proof/retainer.rs

@@ -1,7 +1,6 @@
 use crate::{proof::ProofNodes, Nibbles};
 use alloy_primitives::Bytes;
 
-#[allow(unused_imports)]
 use alloc::vec::Vec;
 
 /// Proof retainer is used to store proofs during merkle trie construction.