@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
JavaScript
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 };