import { useVegaWallet } from '@vegaprotocol/wallet';
import { useFundsAvailableQuery } from './__generated__/FundsAvailable';
import compact from 'lodash/compact';
import BigNumber from 'bignumber.js';

/**
 * 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
    ? BigNumber(data.networkParameter?.value || '0')
    : undefined;

  const sumOfFunds =
    fundsAvailable
      ?.filter((fa) => fa.balance)
      .reduce((sum, fa) => sum.plus(BigNumber(fa.balance)), BigNumber(0)) ||
    BigNumber(0);

  if (requiredFunds && sumOfFunds.isGreaterThanOrEqualTo(requiredFunds)) {
    stopPolling();
  }

  return {
    fundsAvailable,
    requiredFunds,
    isEligible:
      fundsAvailable != null &&
      requiredFunds != null &&
      sumOfFunds.isGreaterThanOrEqualTo(requiredFunds),
  };
};