chore: add storage_slots helper (#1884)

alloy-rs · Jan 4, 2025 · c927bd8 · c927bd8
1 parent 3539a02
commit c927bd8
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/crates/genesis/src/lib.rs b/crates/genesis/src/lib.rs
@@ -246,6 +246,14 @@ impl GenesisAccount {
         self
     }
 
+    /// Returns an iterator over the storage slots in (`B256`, `U256`) format.
+    pub fn storage_slots(&self) -> impl Iterator<Item = (B256, U256)> + '_ {
+        self.storage.as_ref().into_iter().flat_map(|storage| storage.iter()).map(|(key, value)| {
+            let value = U256::from_be_bytes(value.0);
+            (*key, value)
+        })
+    }
+
     /// Convert the genesis account into the [`TrieAccount`] format.
     pub fn into_trie_account(self) -> TrieAccount {
         self.into()