feat: unlock, lock and withdraw funds in sdk

Author: MarcosNicolau

crates/sdk/src/eth/batcher_payment_service.rs

@@ -1,6 +1,6 @@
 use std::sync::Arc;
 
-use ethers::prelude::*;
+use ethers::{core::k256::ecdsa::SigningKey, prelude::*};
 
 use crate::common::errors::VerificationError;
 
@@ -11,6 +11,9 @@ abigen!(
 
 pub type BatcherPaymentService = BatcherPaymentServiceContract<Provider<Http>>;
 
+pub type SignerMiddlewareT = SignerMiddleware<Provider<Http>, Wallet<SigningKey>>;
+pub type BatcherPaymentServiceWithSigner = BatcherPaymentServiceContract<SignerMiddlewareT>;
+
 pub async fn batcher_payment_service(
     provider: Provider<Http>,
     contract_address: H160,

crates/sdk/src/verification_layer/mod.rs

@@ -5,7 +5,7 @@ use crate::{
             DEFAULT_MAX_FEE_BATCH_SIZE, GAS_PRICE_PERCENTAGE_MULTIPLIER,
             INSTANT_MAX_FEE_BATCH_SIZE, PERCENTAGE_DIVIDER,
         },
-        errors::{self, GetNonceError},
+        errors::{self, GetNonceError, PaymentError},
         types::{
             AlignedVerificationData, ClientMessage, FeeEstimationType, GetNonceResponseMessage,
             Network, ProvingSystemId, VerificationData,
@@ -19,7 +19,7 @@ use crate::{
     },
     eth::{
         aligned_service_manager::aligned_service_manager,
-        batcher_payment_service::batcher_payment_service,
+        batcher_payment_service::{batcher_payment_service, BatcherPaymentServiceWithSigner},
     },
 };
 
@@ -750,6 +750,73 @@ pub async fn get_balance_in_aligned(
     }
 }
 
+pub async fn unlock_balance_in_aligned(
+    signer: &SignerMiddleware<Provider<Http>, LocalWallet>,
+    network: Network,
+) -> Result<ethers::types::TransactionReceipt, errors::PaymentError> {
+    let payment_service_address = network.get_batcher_payment_service_address();
+    let payment_service =
+        BatcherPaymentServiceWithSigner::new(payment_service_address, signer.clone().into());
+
+    let receipt = payment_service
+        .unlock()
+        .send()
+        .await
+        .map_err(|e| PaymentError::SendError(e.to_string()))?
+        .await
+        .map_err(|e| PaymentError::SubmitError(e.to_string()))?;
+
+    match receipt {
+        Some(hash) => Ok(hash),
+        None => Err(PaymentError::SubmitError("".into())),
+    }
+}
+
+pub async fn lock_balance_in_aligned(
+    signer: &SignerMiddleware<Provider<Http>, LocalWallet>,
+    network: Network,
+) -> Result<ethers::types::TransactionReceipt, errors::PaymentError> {
+    let payment_service_address = network.get_batcher_payment_service_address();
+    let payment_service =
+        BatcherPaymentServiceWithSigner::new(payment_service_address, signer.clone().into());
+
+    let receipt = payment_service
+        .lock()
+        .send()
+        .await
+        .map_err(|e| PaymentError::SendError(e.to_string()))?
+        .await
+        .map_err(|e| PaymentError::SubmitError(e.to_string()))?;
+
+    match receipt {
+        Some(hash) => Ok(hash),
+        None => Err(PaymentError::SubmitError("".into())),
+    }
+}
+
+pub async fn withdraw_balance_from_aligned(
+    signer: &SignerMiddleware<Provider<Http>, LocalWallet>,
+    network: Network,
+    amount: U256,
+) -> Result<ethers::types::TransactionReceipt, errors::PaymentError> {
+    let payment_service_address = network.get_batcher_payment_service_address();
+    let payment_service =
+        BatcherPaymentServiceWithSigner::new(payment_service_address, signer.clone().into());
+
+    let receipt = payment_service
+        .withdraw(amount)
+        .send()
+        .await
+        .map_err(|e| PaymentError::SendError(e.to_string()))?
+        .await
+        .map_err(|e| PaymentError::SubmitError(e.to_string()))?;
+
+    match receipt {
+        Some(hash) => Ok(hash),
+        None => Err(PaymentError::SubmitError("".into())),
+    }
+}
+
 /// Saves AlignedVerificationData in a file.
 ///
 /// # Arguments