feat: add block_id field inside the ContractReader

cartridge-gg · Dec 15, 2023 · 8bba7d3 · 8bba7d3
1 parent 028fc11
commit 8bba7d3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/crates/rs/src/expand/contract.rs b/crates/rs/src/expand/contract.rs
@@ -40,14 +40,15 @@ impl CairoContract {
             pub struct #reader<P: #snrs_providers::Provider + Sync> {
                 pub address: #snrs_types::FieldElement,
                 pub provider: P,
+                pub block_id: Option<#snrs_types::BlockId>,
             }
 
             impl<P: #snrs_providers::Provider + Sync> #reader<P> {
                 pub fn new(
                     address: #snrs_types::FieldElement,
                     provider: P,
                 ) -> Self {
-                    Self { address, provider }
+                    Self { address, provider, block_id: None }
                 }
 
                 pub fn set_contract_address(mut self, address: #snrs_types::FieldElement) {
@@ -57,6 +58,10 @@ impl CairoContract {
                 pub fn provider(&self) -> &P {
                     &self.provider
                 }
+
+                pub fn with_block(self, block_id: #snrs_types::BlockId) -> Self {
+                    Self { block_id: Some(block_id), ..self }
+                }
             }
         };