supabase/apps/studio/data/config/project-creation-postgres-versions-query.ts

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

import { configKeys } from './keys'
import { handleError, post } from '@/data/fetchers'
import type { ResponseError, UseCustomQueryOptions } from '@/types'

export type ProjectCreationPostgresVersionsVariables = {
  cloudProvider: CloudProvider
  dbRegion: string
  organizationSlug: string | undefined
  highAvailability?: boolean
}

export async function getPostgresCreationVersions(
  {
    cloudProvider,
    dbRegion,
    organizationSlug,
    highAvailability,
  }: ProjectCreationPostgresVersionsVariables,
  signal?: AbortSignal
) {
  if (!organizationSlug) throw new Error('organizationSlug is required')

  const { data, error } = await post('/platform/organizations/{slug}/available-versions', {
    params: { path: { slug: organizationSlug } },
    body: { provider: cloudProvider, region: dbRegion, high_availability: highAvailability },
    signal,
  })

  if (error) handleError(error)
  return data
}

export type ProjectCreationPostgresVersionData = Awaited<
  ReturnType<typeof getPostgresCreationVersions>
>
export type ProjectCreationPostgresVersionError = ResponseError

export const useProjectCreationPostgresVersionsQuery = <TData = ProjectCreationPostgresVersionData>(
  {
    cloudProvider,
    dbRegion,
    organizationSlug,
    highAvailability,
  }: ProjectCreationPostgresVersionsVariables,
  {
    enabled = true,
    ...options
  }: UseCustomQueryOptions<
    ProjectCreationPostgresVersionData,
    ProjectCreationPostgresVersionError,
    TData
  > = {}
) => {
  return useQuery<ProjectCreationPostgresVersionData, ProjectCreationPostgresVersionError, TData>({
    queryKey: configKeys.projectCreationPostgresVersions(
      organizationSlug,
      cloudProvider,
      dbRegion,
      highAvailability
    ),
    queryFn: ({ signal }) =>
      getPostgresCreationVersions(
        { organizationSlug, cloudProvider, dbRegion, highAvailability },
        signal
      ),
    enabled:
      enabled && typeof organizationSlug !== 'undefined' && organizationSlug !== '_' && !!dbRegion,
    ...options,
  })
}

export const useAvailableOrioleImageVersion = (
  {
    cloudProvider,
    dbRegion,
    organizationSlug,
    highAvailability,
  }: ProjectCreationPostgresVersionsVariables,
  { enabled = true }: { enabled?: boolean } = {}
) => {
  const { data } = useProjectCreationPostgresVersionsQuery(
    {
      cloudProvider,
      dbRegion,
      organizationSlug,
      highAvailability,
    },
    {
      enabled,
      select(data) {
        return (data?.available_versions ?? []).find((x) => x.postgres_engine === '17-oriole')
      },
    }
  )

  return data
}