@openocean.finance/widget-sdk/src/_esm/core/checkBalance.js

OpenOcean Any-to-Any Cross-Chain-Swap SDK

import { formatUnits } from 'viem';
import { BalanceError } from '../errors/errors.js';
import { getTokenBalance } from '../services/balance.js';
import { sleep } from '../utils/sleep.js';
export const checkBalance = async (walletAddress, step, depth = 0) => {
    const token = await getTokenBalance(walletAddress, step.action.fromToken);
    if (token) {
        const currentBalance = token.amount ?? 0n;
        const neededBalance = BigInt(step.action.fromAmount);
        if (currentBalance < neededBalance) {
            if (depth <= 3) {
                await sleep(200);
                await checkBalance(walletAddress, step, depth + 1);
            }
            else if ((neededBalance *
                BigInt((1 - (step.action.slippage ?? 0)) * 1_000_000_000)) /
                1000000000n <=
                currentBalance) {
                // adjust amount in slippage limits
                step.action.fromAmount = currentBalance.toString();
            }
            else {
                const neeeded = formatUnits(neededBalance, token.decimals);
                const current = formatUnits(currentBalance, token.decimals);
                let errorMessage = `Your ${token.symbol} balance is too low, you try to transfer ${neeeded} ${token.symbol}, but your wallet only holds ${current} ${token.symbol}. No funds have been sent.`;
                if (currentBalance !== 0n) {
                    errorMessage += `If the problem consists, please delete this transfer and start a new one with a maximum of ${current} ${token.symbol}.`;
                }
                throw new BalanceError('The balance is too low.', new Error(errorMessage));
            }
        }
    }
};
//# sourceMappingURL=checkBalance.js.map