fix: check balance of source account

matter-labs · Aug 1, 2023 · ccb8e72 · ccb8e72
1 parent 7ec6b51
commit ccb8e72
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/src/deposit.ts b/src/deposit.ts
@@ -102,11 +102,10 @@ export default async function (
       ? (L1Provider = new ethers.providers.JsonRpcProvider(ethProviderUrl))
       : (L1Provider = ethers.getDefaultProvider(ethProviderUrl));
 
-    await checkBalance(results.to, results.amount, L1Provider);
-
     const zkSyncProvider = new Provider(zksyncProviderUrl);
     // Initialize the wallet.
     const wallet = new Wallet(results.key, zkSyncProvider, L1Provider);
+    await checkBalance(wallet.address, results.amount, L1Provider);
 
     // Deposit funds to L2
     const depositHandle: PriorityOpResponse = await wallet.deposit({

diff --git a/src/utils.ts b/src/utils.ts
@@ -1,10 +1,11 @@
 import { ethers, utils } from "ethers";
+import { Provider } from "zksync-web3";
 import { track } from "./analytics";
 
 export const checkBalance = async function (
   address: string,
   amount: string,
-  provider: ethers.providers.BaseProvider
+  provider: Provider | ethers.providers.BaseProvider
 ) {
   const balance = await provider.getBalance(address);
   if (utils.parseEther(amount).gte(balance)) {

diff --git a/src/withdraw.ts b/src/withdraw.ts
@@ -97,11 +97,11 @@ export default async function (
 
     const zkSyncProvider = new Provider(zksyncProviderUrl);
 
-    await checkBalance(results.to, results.amount, zkSyncProvider);
-
     // Initialize the wallet.
     const wallet = new Wallet(results.key, zkSyncProvider, L1Provider);
 
+    await checkBalance(wallet.address, results.amount, zkSyncProvider);
+
     // Withdraw funds to L1
     const withdrawHandle = await wallet.withdraw({
       to: results.to,