supabase/apps/studio/data/config/disk-attributes-query.ts

import { useQuery } from '@tanstack/react-query'
import dayjs from 'dayjs'
import { useMemo } from 'react'

import { COOLDOWN_DURATION } from './disk-attributes-update-mutation'
import { configKeys } from './keys'
import type { components } from '@/data/api'
import { get, handleError } from '@/data/fetchers'
import type { ResponseError, UseCustomQueryOptions } from '@/types'

export type DiskAttributesVariables = {
  projectRef?: string
}

export type DiskAttribute = components['schemas']['DiskResponse_Output']

export async function getDiskAttributes(
  { projectRef }: DiskAttributesVariables,
  signal?: AbortSignal
) {
  if (!projectRef) throw new Error('Project ref is required')

  const { data, error } = await get(`/platform/projects/{ref}/disk`, {
    params: { path: { ref: projectRef } },
    signal,
  })

  if (error) handleError(error)
  return data
}

export type DiskAttributesData = Awaited<ReturnType<typeof getDiskAttributes>>
export type DiskAttributesError = ResponseError

export const useDiskAttributesQuery = <TData = DiskAttributesData>(
  { projectRef }: DiskAttributesVariables,
  {
    enabled = true,
    ...options
  }: UseCustomQueryOptions<DiskAttributesData, DiskAttributesError, TData> = {}
) =>
  useQuery<DiskAttributesData, DiskAttributesError, TData>({
    queryKey: configKeys.diskAttributes(projectRef),
    queryFn: ({ signal }) => getDiskAttributes({ projectRef }, signal),
    enabled: enabled && typeof projectRef !== 'undefined',
    ...options,
  })

export const useRemainingDurationForDiskAttributeUpdate = ({
  projectRef,
  enabled = true,
}: {
  projectRef?: string
  enabled?: boolean
}) => {
  const {
    data,
    isPending: isLoading,
    isError,
    isSuccess,
    error,
  } = useDiskAttributesQuery({ projectRef }, { enabled })

  const lastModifiedAtString = dayjs(data?.last_modified_at ?? '').utc()
  const secondsFromNow = Math.max(
    0,
    dayjs().utc().diff(dayjs(lastModifiedAtString).utc(), 'second')
  )

  const remainingDuration = useMemo(() => {
    return lastModifiedAtString === undefined || COOLDOWN_DURATION - secondsFromNow < 0
      ? 0
      : COOLDOWN_DURATION - secondsFromNow
  }, [lastModifiedAtString, secondsFromNow])

  if (data?.last_modified_at === undefined)
    return {
      remainingDuration: 0,
      isWithinCooldownWindow: false,
      isLoading,
      isError,
      error,
      isSuccess,
    }

  return {
    remainingDuration,
    isWithinCooldownWindow: remainingDuration > 0,
    isLoading,
    isError,
    error,
    isSuccess,
  }
}