TheGame/packages/web/lib/hooks/useProfileImageOnload.ts

import { useEffect, useMemo, useState } from 'react';

import { Player } from '#graphql/autogen/hasura-sdk';
import { GuildPlayer } from '#graphql/types';
import { getPlayerImage } from '#utils/playerHelpers';

export const useProfileImageOnload = ({
  player,
}: {
  player: Player | GuildPlayer | undefined;
}) => {
  const avatarImg = useMemo(() => getPlayerImage(player), [player]);

  const [imageURL, setImageURL] = useState(avatarImg.profileIcon);
  const profileImage = avatarImg.link;

  useEffect(() => {
    if (!profileImage) return;
    const image = new Image();
    image.src = profileImage;

    image.onload = () => {
      setImageURL(profileImage);
    };

    image.onerror = () => {
      console.error(`Original image, ${profileImage}, failed to load.`);
    };
  }, [profileImage]);

  return imageURL;
};