From edee4b0b6b37ebbcab7256a187ad7e35a9538b45 Mon Sep 17 00:00:00 2001
From: dianacarvalho1 <diana@propellerheads.xyz>
Date: Wed, 14 May 2025 10:18:07 +0100
Subject: [PATCH] feat: Add get_block_storage_changes (#203)

* feat: add DCI message definitions

#time 4m

#time 2m

* fix: Make entrypoints into a HashSet instead of Vec

#time 0m

* fix: Rollback on message index change in ContractSlot

#time 5m

* feat: Add get_block_storage_changes

#time 34m


#time 0m

* Update substreams/crates/tycho-substreams/src/block_storage.rs

Co-authored-by: Louise Poole <louisecarmenpoole@gmail.com>

* Update substreams/crates/tycho-substreams/src/block_storage.rs

Co-authored-by: Louise Poole <louisecarmenpoole@gmail.com>

* fix: Filter out calls that are reverted and sort change by ordinal

#time 0m

* fix: Improve performance of get_block_storage_changes

We expect this util function to collect and handle a lot of information repeatedly (on every single block). So top performance optimisation (both memory and computation) is vital to minimise substream lags.

#time 6m

* fix: Use filter and flat_map instead of if and for loops

#time 1m

---------

Co-authored-by: Louise Poole <louise@datarevenue.com>
Co-authored-by: Louise Poole <louisecarmenpoole@gmail.com>
---
 .../tycho-substreams/src/block_storage.rs     | 74 +++++++++++++++++++
 substreams/crates/tycho-substreams/src/lib.rs |  1 +
 2 files changed, 75 insertions(+)
 create mode 100644 substreams/crates/tycho-substreams/src/block_storage.rs

diff --git a/substreams/crates/tycho-substreams/src/block_storage.rs b/substreams/crates/tycho-substreams/src/block_storage.rs
new file mode 100644
index 0000000..17e7bcb
--- /dev/null
+++ b/substreams/crates/tycho-substreams/src/block_storage.rs
@@ -0,0 +1,74 @@
+use std::collections::HashMap;
+
+use substreams_ethereum::pb::{
+    eth,
+    eth::v2::{block::DetailLevel, StorageChange},
+};
+
+use crate::{
+    models::{ContractSlot, StorageChanges, Transaction},
+    pb::tycho::evm::v1::TransactionStorageChanges,
+};
+
+#[allow(dead_code)]
+/// Helper function to extract all storage changes on a block.
+/// The raw block information collected is intended to be used by the DCI (Dynamic Contract Indexer)
+/// to extract and index relevant changes. This is specifically for dynamically identified contracts
+/// that the DCI has chosen to index. Note that core protocol data should still be properly
+/// integrated and indexed by the substreams package as per usual.
+///
+/// ## Panics
+/// Panics if the provided block is not an extended block model, as indicated by its detail level.
+///
+/// ## Warning
+/// ⚠️ This function *only* works if the **extended block model** is available,
+/// more [here](https://streamingfastio.medium.com/new-block-model-to-accelerate-chain-integration-9f65126e5425)
+fn get_block_storage_changes(block: &eth::v2::Block) -> Vec<TransactionStorageChanges> {
+    if block.detail_level != Into::<i32>::into(DetailLevel::DetaillevelExtended) {
+        panic!("Only extended blocks are supported");
+    }
+    let mut block_storage_changes = Vec::with_capacity(block.transaction_traces.len());
+
+    for block_tx in block.transactions() {
+        let transaction: Transaction = block_tx.into();
+
+        let mut changes_by_address: HashMap<Vec<u8>, Vec<StorageChange>> = HashMap::new();
+        for storage_change in block_tx
+            .calls
+            .iter()
+            .filter(|call| !call.state_reverted)
+            .flat_map(|call| call.storage_changes.iter())
+        {
+            changes_by_address
+                .entry(storage_change.address.clone())
+                .or_default()
+                .push(storage_change.clone());
+        }
+
+        // For each address, sort by ordinal and collect latest changes per slot
+        let tx_storage_changes: Vec<StorageChanges> = changes_by_address
+            .into_iter()
+            .map(|(address, mut changes)| {
+                changes.sort_unstable_by_key(|change| change.ordinal);
+
+                // Collect latest change per slot
+                let mut latest_changes: HashMap<Vec<u8>, ContractSlot> = HashMap::new();
+                for change in changes {
+                    latest_changes.insert(
+                        change.key.clone(),
+                        ContractSlot { slot: change.key, value: change.new_value },
+                    );
+                }
+
+                StorageChanges { address, slots: latest_changes.into_values().collect() }
+            })
+            .collect();
+
+        block_storage_changes.push(TransactionStorageChanges {
+            tx: Some(transaction),
+            storage_changes: tx_storage_changes,
+        });
+    }
+
+    block_storage_changes
+}
diff --git a/substreams/crates/tycho-substreams/src/lib.rs b/substreams/crates/tycho-substreams/src/lib.rs
index 8420c7d..ba19414 100644
--- a/substreams/crates/tycho-substreams/src/lib.rs
+++ b/substreams/crates/tycho-substreams/src/lib.rs
@@ -1,6 +1,7 @@
 pub mod abi;
 pub mod attributes;
 pub mod balances;
+pub mod block_storage;
 pub mod contract;
 mod mock_store;
 pub mod models;