@lifi/widget/hooks/useGetTokenBalancesWithRetry.js

LI.FI Widget for cross-chain bridging and swapping. It will drive your multi-chain strategy and attract new users from everywhere.

import { isAddress } from '@ethersproject/address';
import { useCallback } from 'react';
import { useLiFi } from '../providers';
export const useGetTokenBalancesWithRetry = (provider) => {
    const lifi = useLiFi();
    const getTokenBalancesWithRetry = useCallback(async (accountAddress, tokens, depth = 0) => {
        try {
            const walletAddress = isAddress(accountAddress)
                ? accountAddress
                : await provider?.resolveName(accountAddress);
            const tokenBalances = await lifi.getTokenBalances(walletAddress, tokens);
            if (!tokenBalances.every((token) => token.blockNumber)) {
                if (depth > 10) {
                    console.warn('Token balance backoff depth exceeded.');
                    return undefined;
                }
                await new Promise((resolve) => {
                    setTimeout(resolve, 1.5 ** depth * 100);
                });
                return getTokenBalancesWithRetry(accountAddress, tokens, depth + 1);
            }
            return tokenBalances;
        }
        catch (error) {
            //
        }
    }, [lifi, provider]);
    return getTokenBalancesWithRetry;
};