"use client";

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

import { getTest } from "@/lib/sitespeed";

import { speedKeys } from "./keys";

export function useTestQuery(id: string) {
  return useQuery({
    queryKey: speedKeys.test(id),
    queryFn: () => getTest(id),
    refetchInterval: (query) => {
      const status = query.state.data?.status;
      if (status === "completed" || status === "failed") return false;
      return 5000;
    },
  });
}