FastGPT/src/hooks/useLoading.tsx

import { useState } from 'react';
import { Spinner, Flex } from '@chakra-ui/react';

export const useLoading = (props?: { defaultLoading: boolean }) => {
  const [isLoading, setIsLoading] = useState(props?.defaultLoading || false);

  const Loading = ({
    loading,
    fixed = true
  }: {
    loading?: boolean;
    fixed?: boolean;
  }): JSX.Element | null => {
    return isLoading || loading ? (
      <Flex
        position={fixed ? 'fixed' : 'absolute'}
        zIndex={100}
        backgroundColor={'rgba(255,255,255,0.5)'}
        top={0}
        left={0}
        right={0}
        bottom={0}
        alignItems={'center'}
        justifyContent={'center'}
      >
        <Spinner thickness="4px" speed="0.65s" emptyColor="gray.200" color="blue.500" size="xl" />
      </Flex>
    ) : null;
  };

  return {
    isLoading,
    setIsLoading,
    Loading
  };
};