OpenMetadata/openmetadata-ui/src/main/resources/ui/src/hooks/useAirflowStatus.ts

/*
 *  Copyright 2023 Collate.
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *  http://www.apache.org/licenses/LICENSE-2.0
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

import { AxiosError } from 'axios';
import {
  AirflowResponse,
  AirflowStatus,
} from 'interface/AirflowStatus.interface';
import { useEffect, useState } from 'react';
import { getAirflowStatus } from 'rest/ingestionPipelineAPI';

interface UseAirflowStatusProps {
  isFetchingStatus: boolean;
  isAirflowAvailable: boolean;
  error: AxiosError | undefined;
  reason: AirflowResponse['reason'];
  fetchAirflowStatus: () => Promise<void>;
}

export const useAirflowStatus = (): UseAirflowStatusProps => {
  const [isLoading, setIsLoading] = useState<boolean>(false);
  const [isAirflowAvailable, setIsAirflowAvailable] = useState<boolean>(false);
  const [error, setError] = useState<AxiosError>();
  const [reason, setReason] = useState<AirflowResponse['reason']>();

  const fetchAirflowStatus = async () => {
    setIsLoading(true);
    try {
      const response = await getAirflowStatus();
      setIsAirflowAvailable(response.status === AirflowStatus.HEALTHY);
      setReason(response.reason);
    } catch (error) {
      setError(error as AxiosError);
      setIsAirflowAvailable(false);
    } finally {
      setIsLoading(false);
    }
  };

  useEffect(() => {
    fetchAirflowStatus();
  }, []);

  return {
    isFetchingStatus: isLoading,
    isAirflowAvailable,
    error,
    fetchAirflowStatus,
    reason,
  };
};