timeline-frontend/src/components/Hooks/useFetchImageUrl.ts

import { fetchImage } from '@/pages/story/service';
import { useRequest } from '@umijs/max';
import { useEffect, useState } from 'react';

const useFetchImageUrl = (imageInstanceId: string) => {
  const [imageUrl, setImageUrl] = useState("error");
  const { data: response, run, loading } = useRequest(
    () => {
      return fetchImage(imageInstanceId);
    },
    {
      manual: true,
      onSuccess: (data) => {
        console.log(data);
      },
    },
  );
  useEffect(() => {
    if (response) {
      setImageUrl(URL.createObjectURL(response));
    } else {
      setImageUrl("error");
    }
  }, [response]);
  useEffect(() => {
    if (imageInstanceId) {
      run();
    }
  }, [imageInstanceId]);
  return {imageUrl, loading};
};
export default useFetchImageUrl;