From bd7c62906a4a242fdc7ce0624ce8b4c5b6110760 Mon Sep 17 00:00:00 2001
From: Charpa <102919164+jbcaron@users.noreply.github.com>
Date: Tue, 16 Jul 2024 06:33:20 +0200
Subject: [PATCH] feat(sequencer): add get_state_update_with_block (#597)

---
 starknet-providers/src/sequencer/mod.rs        | 18 ++++++++++++++++++
 starknet-providers/src/sequencer/models/mod.rs |  2 +-
 .../src/sequencer/models/state_update.rs       | 10 ++++++++++
 3 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/starknet-providers/src/sequencer/mod.rs b/starknet-providers/src/sequencer/mod.rs
index 74d05225..4453a3ed 100644
--- a/starknet-providers/src/sequencer/mod.rs
+++ b/starknet-providers/src/sequencer/mod.rs
@@ -313,6 +313,24 @@ impl SequencerGatewayProvider {
             .into()
     }
 
+    #[deprecated(
+        note = "Sequencer-specific functions are deprecated. Use it via the Provider trait instead."
+    )]
+    pub async fn get_state_update_with_block(
+        &self,
+        block_identifier: BlockId,
+    ) -> Result<StateUpdateWithBlock, ProviderError> {
+        let mut request_url = self.extend_feeder_gateway_url("get_state_update");
+        append_block_id(&mut request_url, block_identifier);
+        request_url
+            .query_pairs_mut()
+            .append_pair("includeBlock", "true");
+
+        self.send_get_request::<GatewayResponse<_>>(request_url)
+            .await?
+            .into()
+    }
+
     #[deprecated(
         note = "Sequencer-specific functions are deprecated. Use it via the Provider trait instead."
     )]
diff --git a/starknet-providers/src/sequencer/models/mod.rs b/starknet-providers/src/sequencer/models/mod.rs
index 3e269958..c1c0922e 100644
--- a/starknet-providers/src/sequencer/models/mod.rs
+++ b/starknet-providers/src/sequencer/models/mod.rs
@@ -46,7 +46,7 @@ mod contract;
 pub use contract::{CompressedLegacyContractClass, DeployedClass};
 
 pub mod state_update;
-pub use state_update::StateUpdate;
+pub use state_update::{StateUpdate, StateUpdateWithBlock};
 
 pub mod trace;
 pub use trace::{BlockTraces, TransactionTrace};
diff --git a/starknet-providers/src/sequencer/models/state_update.rs b/starknet-providers/src/sequencer/models/state_update.rs
index 0447b96c..c0e56316 100644
--- a/starknet-providers/src/sequencer/models/state_update.rs
+++ b/starknet-providers/src/sequencer/models/state_update.rs
@@ -3,6 +3,16 @@ use serde_with::serde_as;
 use starknet_core::{serde::unsigned_field_element::UfeHex, types::Felt};
 use std::collections::HashMap;
 
+use super::Block;
+
+#[serde_as]
+#[derive(Debug, Deserialize)]
+#[cfg_attr(feature = "no_unknown_fields", serde(deny_unknown_fields))]
+pub struct StateUpdateWithBlock {
+    pub state_update: StateUpdate,
+    pub block: Block,
+}
+
 #[serde_as]
 #[derive(Debug, Deserialize)]
 #[cfg_attr(feature = "no_unknown_fields", serde(deny_unknown_fields))]