fix(zk_toolbox): Add chain id for local wallet

zk_toolbox/crates/common/src/ethereum.rs

@@ -1,5 +1,6 @@
 use std::{ops::Add, time::Duration};
 
+use ethers::prelude::Signer;
 use ethers::{
     core::k256::ecdsa::SigningKey,
     middleware::MiddlewareBuilder,
@@ -14,8 +15,12 @@ use crate::wallets::Wallet;
 pub fn create_ethers_client(
     private_key: H256,
     l1_rpc: String,
+    chain_id: Option<u32>,
 ) -> anyhow::Result<SignerMiddleware<Provider<Http>, ethers::prelude::Wallet<SigningKey>>> {
-    let wallet = LocalWallet::from_bytes(private_key.as_bytes())?;
+    let mut wallet = LocalWallet::from_bytes(private_key.as_bytes())?;
+    if let Some(chain_id) = chain_id {
+        wallet = wallet.with_chain_id(chain_id);
+    }
     let client = Provider::<Http>::try_from(l1_rpc)?.with_signer(wallet);
     Ok(client)
 }
@@ -27,7 +32,7 @@ pub async fn distribute_eth(
     chain_id: u32,
     amount: u128,
 ) -> anyhow::Result<()> {
-    let client = create_ethers_client(main_wallet.private_key.unwrap(), l1_rpc)?;
+    let client = create_ethers_client(main_wallet.private_key.unwrap(), l1_rpc, Some(chain_id))?;
     let mut pending_txs = vec![];
     let mut nonce = client.get_transaction_count(client.address(), None).await?;
     for address in addresses {

zk_toolbox/crates/common/src/forge.rs

@@ -134,7 +134,7 @@ impl ForgeScript {
         let Some(private_key) = self.private_key() else {
             return Ok(true);
         };
-        let client = create_ethers_client(private_key, rpc_url)?;
+        let client = create_ethers_client(private_key, rpc_url, None)?;
         let balance = client.get_balance(client.address(), None).await?;
         Ok(balance > minimum_value)
     }