feat(starknet_batcher): add communication for get_height (#2581)

crates/starknet_batcher/src/batcher.rs

@@ -13,6 +13,7 @@ use starknet_api::state::ThinStateDiff;
 use starknet_batcher_types::batcher_types::{
     BatcherResult,
     DecisionReachedInput,
+    GetHeightResponse,
     GetProposalContent,
     GetProposalContentInput,
     GetProposalContentResponse,
@@ -301,6 +302,12 @@ impl Batcher {
         Ok(())
     }
 
+    // TODO(Arni): Implement this function.
+    #[instrument(skip(self), err)]
+    pub async fn get_height(&mut self) -> BatcherResult<GetHeightResponse> {
+        todo!("Implement this function");
+    }
+
     #[instrument(skip(self), err)]
     pub async fn get_proposal_content(
         &mut self,

crates/starknet_batcher/src/communication.rs

@@ -15,6 +15,9 @@ impl ComponentRequestHandler<BatcherRequest, BatcherResponse> for Batcher {
             BatcherRequest::ProposeBlock(input) => {
                 BatcherResponse::ProposeBlock(self.propose_block(input).await)
             }
+            BatcherRequest::GetCurrentHeight => {
+                BatcherResponse::GetCurrentHeight(self.get_height().await)
+            }
             BatcherRequest::GetProposalContent(input) => {
                 BatcherResponse::GetProposalContent(self.get_proposal_content(input).await)
             }

crates/starknet_batcher_types/src/batcher_types.rs

@@ -44,6 +44,11 @@ pub struct GetProposalContentInput {
     pub proposal_id: ProposalId,
 }
 
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
+pub struct GetHeightResponse {
+    pub height: BlockNumber,
+}
+
 #[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
 pub struct GetProposalContentResponse {
     pub content: GetProposalContent,

crates/starknet_batcher_types/src/communication.rs

@@ -20,6 +20,7 @@ use thiserror::Error;
 use crate::batcher_types::{
     BatcherResult,
     DecisionReachedInput,
+    GetHeightResponse,
     GetProposalContentInput,
     GetProposalContentResponse,
     ProposeBlockInput,
@@ -44,6 +45,8 @@ pub type SharedBatcherClient = Arc<dyn BatcherClient>;
 pub trait BatcherClient: Send + Sync {
     /// Starts the process of building a proposal.
     async fn propose_block(&self, input: ProposeBlockInput) -> BatcherClientResult<()>;
+    /// Gets the first height that is not written in the storage yet.
+    async fn get_height(&self) -> BatcherClientResult<GetHeightResponse>;
     /// Gets the next available content from the proposal stream (only relevant when building a
     /// proposal).
     async fn get_proposal_content(
@@ -80,13 +83,15 @@ pub enum BatcherRequest {
     ValidateBlock(ValidateBlockInput),
     SendProposalContent(SendProposalContentInput),
     StartHeight(StartHeightInput),
+    GetCurrentHeight,
     DecisionReached(DecisionReachedInput),
     AddSyncBlock(SyncBlock),
 }
 
 #[derive(Debug, Serialize, Deserialize, Clone)]
 pub enum BatcherResponse {
     ProposeBlock(BatcherResult<()>),
+    GetCurrentHeight(BatcherResult<GetHeightResponse>),
     GetProposalContent(BatcherResult<GetProposalContentResponse>),
     ValidateBlock(BatcherResult<()>),
     SendProposalContent(BatcherResult<SendProposalContentResponse>),
@@ -154,6 +159,17 @@ where
         handle_response_variants!(BatcherResponse, StartHeight, BatcherClientError, BatcherError)
     }
 
+    async fn get_height(&self) -> BatcherClientResult<GetHeightResponse> {
+        let request = BatcherRequest::GetCurrentHeight;
+        let response = self.send(request).await;
+        handle_response_variants!(
+            BatcherResponse,
+            GetCurrentHeight,
+            BatcherClientError,
+            BatcherError
+        )
+    }
+
     async fn decision_reached(&self, input: DecisionReachedInput) -> BatcherClientResult<()> {
         let request = BatcherRequest::DecisionReached(input);
         let response = self.send(request).await;