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

import { projectKeys } from './keys'
import { InfraInstanceSize } from '@/components/interfaces/DiskManagement/DiskManagement.types'
import { get, handleError } from '@/data/fetchers'
import type { ResponseError, UseCustomQueryOptions } from '@/types'

export async function getCloneBackups(projectRef?: string) {
  if (!projectRef) throw new Error('Project ref is required')

  const { data, error } = await get(`/platform/database/{ref}/clone`, {
    params: { path: { ref: projectRef } },
  })

  if (error) handleError(error)
  return data
}

export type CloneBackupsData = Awaited<ReturnType<typeof getCloneBackups>>
export type CloneBackupsError = ResponseError

export const useCloneBackupsQuery = <
  TData = CloneBackupsData & {
    target_volume_size_gb: number
    target_compute_size: InfraInstanceSize
  },
>(
  { projectRef }: { projectRef?: string },
  options: UseCustomQueryOptions<CloneBackupsData, CloneBackupsError, TData> = {}
) => {
  return useQuery<CloneBackupsData, CloneBackupsError, TData>({
    queryKey: projectKeys.listCloneBackups(projectRef),
    queryFn: () => getCloneBackups(projectRef),
    enabled: !!projectRef,
    ...options,
  })
}