UNPKG

@knine-sdk/react

Version:

React helpers for Knine Finance projects. Part of Knine JS SDK

57 lines (50 loc) 2.65 kB
'use strict'; Object.defineProperty(exports, '__esModule', { value: true }); var invariant = require('tiny-invariant'); var warning = require('tiny-warning'); var react = require('react'); var contracts = require('@knine-sdk/contracts'); var useContractSWR = require('./useContractSWR.js'); var useSDK = require('./useSDK.js'); var useDebounceCallback = require('./useDebounceCallback.js'); function _interopDefaultLegacy (e) { return e && typeof e === 'object' && 'default' in e ? e : { 'default': e }; } var invariant__default = /*#__PURE__*/_interopDefaultLegacy(invariant); var warning__default = /*#__PURE__*/_interopDefaultLegacy(warning); const useTokenBalance = (token, account, config, providerRpc, web3Provider) => { const { providerRpc: sdkProviderRpc, providerWeb3: sdkProviderWeb3, account: sdkAccount, } = useSDK.useSDK(); const mergedAccount = account !== null && account !== void 0 ? account : sdkAccount; const chosenRpcProvider = providerRpc !== null && providerRpc !== void 0 ? providerRpc : sdkProviderRpc; const chosenWeb3Provider = web3Provider !== null && web3Provider !== void 0 ? web3Provider : sdkProviderWeb3; invariant__default["default"](token != null, 'Token is required'); const contractRpc = contracts.getERC20Contract(token, chosenRpcProvider); const contractWeb3 = chosenWeb3Provider ? contracts.getERC20Contract(token, chosenWeb3Provider) : null; const result = useContractSWR.useContractSWR({ shouldFetch: !!mergedAccount, contract: contractRpc, method: 'balanceOf', params: [mergedAccount], config, }); const updateBalanceDebounced = useDebounceCallback.useDebounceCallback(result.update, 1000); react.useEffect(() => { if (!mergedAccount || !chosenWeb3Provider || !contractWeb3) return; try { const fromMe = contractWeb3.filters.Transfer(mergedAccount, null); const toMe = contractWeb3.filters.Transfer(null, mergedAccount); chosenWeb3Provider.on(fromMe, updateBalanceDebounced); chosenWeb3Provider.on(toMe, updateBalanceDebounced); return () => { chosenWeb3Provider.off(fromMe, updateBalanceDebounced); chosenWeb3Provider.off(toMe, updateBalanceDebounced); }; } catch (error) { return warning__default["default"](false, 'Cannot subscribe to events'); } }, [chosenWeb3Provider, contractWeb3, mergedAccount, updateBalanceDebounced]); return result; }; exports.useTokenBalance = useTokenBalance;