feat(batcher): get first nonce in queue message

Author: MarcosNicolau

batcher/aligned-batcher/src/lib.rs

@@ -31,9 +31,10 @@ use aligned_sdk::core::constants::{
     RESPOND_TO_TASK_FEE_LIMIT_PERCENTAGE_MULTIPLIER,
 };
 use aligned_sdk::core::types::{
-    ClientMessage, GetNonceResponseMessage, NoncedVerificationData, ProofInvalidReason,
-    ProvingSystemId, SubmitProofMessage, SubmitProofResponseMessage, VerificationCommitmentBatch,
-    VerificationData, VerificationDataCommitment,
+    ClientMessage, GetFirstNonceInQueueResponseMessage, GetNonceResponseMessage,
+    NoncedVerificationData, ProofInvalidReason, ProvingSystemId, SubmitProofMessage,
+    SubmitProofResponseMessage, VerificationCommitmentBatch, VerificationData,
+    VerificationDataCommitment,
 };
 
 use aws_sdk_s3::client::Client as S3Client;
@@ -473,6 +474,11 @@ impl Batcher {
                     .handle_submit_proof_msg(msg, ws_conn_sink)
                     .await
             }
+            ClientMessage::GetFirstNonceInQueueForAddress(address) => {
+                self.clone()
+                    .handle_get_first_nonce_in_queue_for_address_msg(address, ws_conn_sink)
+                    .await
+            }
         }
     }
 
@@ -533,6 +539,65 @@ impl Batcher {
         Ok(())
     }
 
+    async fn handle_get_first_nonce_in_queue_for_address_msg(
+        self: Arc<Self>,
+        mut address: Address,
+        ws_conn_sink: WsMessageSink,
+    ) -> Result<(), Error> {
+        if self.is_nonpaying(&address) {
+            info!("Handling nonpaying message");
+            let Some(non_paying_config) = self.non_paying_config.as_ref() else {
+                warn!(
+                    "There isn't a non-paying configuration loaded. This message will be ignored"
+                );
+                send_message(
+                    ws_conn_sink.clone(),
+                    GetFirstNonceInQueueResponseMessage::InvalidRequest(
+                        "There isn't a non-paying configuration loaded.".to_string(),
+                    ),
+                )
+                .await;
+                return Ok(());
+            };
+            let replacement_addr = non_paying_config.replacement.address();
+            address = replacement_addr;
+        }
+
+        let (cached_user_nonce, proof_count) = {
+            let batch_state_lock = self.batch_state.lock().await;
+            (
+                batch_state_lock.get_user_nonce(&address).await,
+                batch_state_lock.get_user_proof_count(&address).await,
+            )
+        };
+
+        let Some(user_nonce) = cached_user_nonce else {
+            send_message(
+                ws_conn_sink.clone(),
+                GetFirstNonceInQueueResponseMessage::NoProofsInQueue,
+            )
+            .await;
+            return Ok(());
+        };
+
+        let Some(proof_count) = proof_count else {
+            send_message(
+                ws_conn_sink.clone(),
+                GetFirstNonceInQueueResponseMessage::NoProofsInQueue,
+            )
+            .await;
+            return Ok(());
+        };
+
+        send_message(
+            ws_conn_sink.clone(),
+            GetFirstNonceInQueueResponseMessage::Nonce(user_nonce - proof_count),
+        )
+        .await;
+
+        Ok(())
+    }
+
     async fn handle_submit_proof_msg(
         self: Arc<Self>,
         client_msg: Box<SubmitProofMessage>,

batcher/aligned-sdk/src/core/types.rs

@@ -231,6 +231,7 @@ pub enum ClientMessage {
     // Needs to be wrapped in box as the message is 3x bigger than the others
     // see https://rust-lang.github.io/rust-clippy/master/index.html#large_enum_variant
     SubmitProof(Box<SubmitProofMessage>),
+    GetFirstNonceInQueueForAddress(Address),
 }
 
 impl Display for ClientMessage {
@@ -239,6 +240,9 @@ impl Display for ClientMessage {
         match self {
             ClientMessage::GetNonceForAddress(_) => write!(f, "GetNonceForAddress"),
             ClientMessage::SubmitProof(_) => write!(f, "SubmitProof"),
+            ClientMessage::GetFirstNonceInQueueForAddress(_) => {
+                write!(f, "GetFirstNonceInQueueForAddress")
+            }
         }
     }
 }
@@ -396,6 +400,13 @@ pub enum GetNonceResponseMessage {
     InvalidRequest(String),
 }
 
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub enum GetFirstNonceInQueueResponseMessage {
+    Nonce(U256),
+    InvalidRequest(String),
+    NoProofsInQueue,
+}
+
 #[derive(Debug, Clone, Copy)]
 pub enum Network {
     Devnet,