use-codegate-status.ts

import { queryOptions, useQuery } from "@tanstack/react-query";

import { healthCheckHealthGet, v1VersionCheck } from "@/api/generated";
import { HealthStatus, VersionResponse } from "../types";

type HealthResponse = { status: "healthy" | unknown } | null;

const getCodeGateHealth = async (): Promise<HealthStatus | null> => {
  const data = (await healthCheckHealthGet()).data;

  if ((data as HealthResponse)?.status === "healthy")
    return HealthStatus.HEALTHY;
  if ((data as HealthResponse)?.status !== "healthy")
    return HealthStatus.UNHEALTHY;

  return null;
};

const getVersion = async (): Promise<VersionResponse | null> => {
  return ((await v1VersionCheck()).data as VersionResponse) ?? null;
};

export function getQueryOptionsCodeGateStatus() {
  return queryOptions({
    queryFn: async () => {
      const health = await getCodeGateHealth();
      const version = await getVersion();

      return {
        health,
        version: version as VersionResponse | null,
      };
    },
    queryKey: ["useHealthCheck"],
    refetchInterval: 5_000,
    staleTime: Infinity,
    gcTime: Infinity,
    refetchIntervalInBackground: true,
    refetchOnMount: true,
    refetchOnReconnect: true,
    refetchOnWindowFocus: true,
    retry: false,
  });
}

export const useCodeGateStatus = () =>
  useQuery({
    ...getQueryOptionsCodeGateStatus(),
  });