UNPKG

@lido-sdk/react

Version:

This project is being slowly deprecated and may not receive further updates. Check out [modern Lido SDK](https://github.com/lidofinance/lido-ethereum-sdk/pulls) to access latest functionality. It is actively maintained and is built for interacting with Li

46 lines (43 loc) 1.86 kB
import invariant from 'tiny-invariant'; import warning from 'tiny-warning'; import { useEffect } from 'react'; import { getERC20Contract } from '@lido-sdk/contracts'; import { useContractSWR } from './useContractSWR.js'; import { useSDK } from './useSDK.js'; import { useDebounceCallback } from './useDebounceCallback.js'; const useTokenBalance = (token, account, config) => { const { providerRpc, providerWeb3, account: sdkAccount } = useSDK(); const mergedAccount = account !== null && account !== void 0 ? account : sdkAccount; invariant(token != null, 'Token is required'); const contractRpc = getERC20Contract(token, providerRpc); const contractWeb3 = providerWeb3 ? getERC20Contract(token, providerWeb3) : null; const result = useContractSWR({ shouldFetch: !!mergedAccount, contract: contractRpc, method: 'balanceOf', params: [mergedAccount], config, }); const updateBalanceDebounced = useDebounceCallback(result.update, 1000); useEffect(() => { if (!mergedAccount || !providerWeb3 || !contractWeb3) return; try { const fromMe = contractWeb3.filters.Transfer(mergedAccount, null); const toMe = contractWeb3.filters.Transfer(null, mergedAccount); providerWeb3.on(fromMe, updateBalanceDebounced); providerWeb3.on(toMe, updateBalanceDebounced); return () => { providerWeb3.off(fromMe, updateBalanceDebounced); providerWeb3.off(toMe, updateBalanceDebounced); }; } catch (error) { return warning(false, 'Cannot subscribe to events'); } }, [providerWeb3, contractWeb3, mergedAccount, updateBalanceDebounced]); return result; }; export { useTokenBalance };