snowballtools-base/packages/frontend/src/components/projects/create/CheckBalanceIframe.tsx

import { useEffect } from 'react';

import { Modal } from '@mui/material';

import { VITE_WALLET_IFRAME_URL } from 'utils/constants';
import useCheckBalance from '../../../hooks/useCheckBalance';
import { useAddNetwork } from '../../../hooks/useAddNetwork';

const CHECK_BALANCE_INTERVAL = 5000;
const IFRAME_ID = 'checkBalanceIframe';

const CheckBalanceIframe = ({
  onBalanceChange,
  isPollingEnabled,
  amount,
}: {
  onBalanceChange: (value: boolean | undefined) => void;
  isPollingEnabled: boolean;
  amount: string;
}) => {
  const { isBalanceSufficient, checkBalance } = useCheckBalance(
    amount,
    IFRAME_ID,
  );

  const { isNetworkAvailable, setIframe } = useAddNetwork();
  
  useEffect(() => {
    if (!isNetworkAvailable || isBalanceSufficient) {
      return;
    }

    checkBalance();

    if (!isPollingEnabled) {
      return;
    }
    
    const interval = setInterval(() => {
      checkBalance();
    }, CHECK_BALANCE_INTERVAL);

    return () => {
      clearInterval(interval);
    };
  }, [isBalanceSufficient, isPollingEnabled, checkBalance, isNetworkAvailable]);

  useEffect(() => {
    onBalanceChange(isBalanceSufficient);
  }, [isBalanceSufficient]);

  return (
    <Modal open={false} disableEscapeKeyDown keepMounted>
      <iframe
        onLoad={(event) => setIframe(event.target as HTMLIFrameElement)}
        id={IFRAME_ID}
        src={VITE_WALLET_IFRAME_URL}
        width="100%"
        height="100%"
        sandbox="allow-scripts allow-same-origin"
        className="border rounded-md shadow-sm"
      />
    </Modal>
  );
};

export default CheckBalanceIframe;