47 lines
1.3 KiB
TypeScript
47 lines
1.3 KiB
TypeScript
import { useVegaWallet } from '@vegaprotocol/wallet';
|
|
import { useFundsAvailableQuery } from './__generated__/FundsAvailable';
|
|
import compact from 'lodash/compact';
|
|
import sum from 'lodash/sum';
|
|
|
|
/**
|
|
* Gets the funds for given public key and required min for
|
|
* the referral program.
|
|
*
|
|
* (Uses currently connected public key if left empty)
|
|
*/
|
|
export const useFundsAvailable = (pubKey?: string) => {
|
|
const { pubKey: currentPubKey } = useVegaWallet();
|
|
const partyId = pubKey || currentPubKey;
|
|
const { data, stopPolling } = useFundsAvailableQuery({
|
|
variables: { partyId: partyId || '' },
|
|
skip: !partyId,
|
|
fetchPolicy: 'network-only',
|
|
errorPolicy: 'ignore',
|
|
pollInterval: 5000,
|
|
});
|
|
|
|
const fundsAvailable = data
|
|
? compact(data.party?.accountsConnection?.edges?.map((e) => e?.node))
|
|
: undefined;
|
|
const requiredFunds = data
|
|
? BigInt(data.networkParameter?.value || '0')
|
|
: undefined;
|
|
|
|
const sumOfFunds = sum(
|
|
fundsAvailable?.filter((fa) => fa.balance).map((fa) => BigInt(fa.balance))
|
|
);
|
|
|
|
if (requiredFunds && sumOfFunds >= requiredFunds) {
|
|
stopPolling();
|
|
}
|
|
|
|
return {
|
|
fundsAvailable,
|
|
requiredFunds,
|
|
isEligible:
|
|
fundsAvailable != null &&
|
|
requiredFunds != null &&
|
|
sumOfFunds >= requiredFunds,
|
|
};
|
|
};
|