import {useEffect, useState} from 'react';
import useUnicornFinance from './useUnicornFinance';
import {Bank, WalletStats} from '../unicorn-finance/types';
import useRefresh from './useRefresh';

const useWalletStats = (banks: Bank[]) => {
  const [stat, setStat] = useState<WalletStats>();
  const {fastRefresh} = useRefresh();
  const unicornFinance = useUnicornFinance();

  useEffect(() => {
    async function fetchWalletStats() {
      try {
        setStat(await unicornFinance.getWalletStats(banks));
      } catch (err) {
        console.error(err);
      }
    }
    fetchWalletStats();
  }, [setStat, unicornFinance, fastRefresh]);

  return stat;
};

export default useWalletStats;