import {useEffect, useState} from 'react';
import {BigNumber} from 'ethers';
import useUnicornFinance from './useUnicornFinance';

const useTreasuryAmount = () => {
  const [amount, setAmount] = useState(BigNumber.from(0));
  const unicornFinance = useUnicornFinance();

  useEffect(() => {
    if (unicornFinance) {
      const {Treasury} = unicornFinance.contracts;
      unicornFinance.UNICORN.balanceOf(Treasury.address).then(setAmount);
    }
  }, [unicornFinance]);
  return amount;
};

export default useTreasuryAmount;