57 lines
1.1 KiB
TypeScript
57 lines
1.1 KiB
TypeScript
"use client"
|
|
|
|
import { useRef } from "react"
|
|
|
|
import { trpc } from "@/lib/trpc/client"
|
|
|
|
import type { BookingStatusEnum } from "@/constants/booking"
|
|
|
|
export function useHandleBookingStatus({
|
|
refId,
|
|
expectedStatuses,
|
|
maxRetries,
|
|
retryInterval,
|
|
enabled,
|
|
}: {
|
|
refId: string
|
|
expectedStatuses: BookingStatusEnum[]
|
|
maxRetries: number
|
|
retryInterval: number
|
|
enabled: boolean
|
|
}) {
|
|
const retries = useRef(0)
|
|
|
|
const query = trpc.booking.status.useQuery(
|
|
{ refId },
|
|
{
|
|
enabled,
|
|
refetchInterval: (query) => {
|
|
retries.current = query.state.dataUpdateCount
|
|
|
|
if (query.state.error || query.state.dataUpdateCount >= maxRetries) {
|
|
return false
|
|
}
|
|
|
|
if (
|
|
expectedStatuses.includes(
|
|
query.state.data?.reservationStatus as BookingStatusEnum
|
|
)
|
|
) {
|
|
return false
|
|
}
|
|
|
|
return retryInterval
|
|
},
|
|
refetchIntervalInBackground: true,
|
|
refetchOnWindowFocus: false,
|
|
refetchOnMount: false,
|
|
retry: false,
|
|
}
|
|
)
|
|
|
|
return {
|
|
...query,
|
|
isTimeout: retries.current >= maxRetries,
|
|
}
|
|
}
|