supabase/apps/studio/hooks/misc/useSelectedProject.ts

import { useMemo } from 'react'

import { useIsLoggedIn, useParams } from 'common'
import { useProjectDetailQuery } from 'data/projects/project-detail-query'
import { ProjectInfo, useProjectsQuery } from 'data/projects/projects-query'
import { PROVIDERS } from 'lib/constants'

export function useSelectedProject({ enabled = true } = {}) {
  const { ref } = useParams()
  const { data } = useProjectDetailQuery({ ref }, { enabled })

  return useMemo(
    () => data && { ...data, parentRef: data?.parent_project_ref ?? data?.ref },
    [data]
  )
}

export function useProjectByRef(
  ref?: string
): Omit<ProjectInfo, 'organization_slug' | 'preview_branch_refs'> | undefined {
  const isLoggedIn = useIsLoggedIn()

  const { data: project } = useProjectDetailQuery({ ref }, { enabled: isLoggedIn })

  // [Alaister]: This is here for the purpose of improving performance.
  // Chances are, the user will already have the list of projects in the cache.
  // We can't exclusively rely on this method, as useProjectsQuery does not return branch projects.
  const { data: projects } = useProjectsQuery({ enabled: isLoggedIn })

  return useMemo(() => {
    if (!ref) return undefined
    if (project) return project
    return projects?.find((project) => project.ref === ref)
  }, [project, projects, ref])
}

export const useIsAwsCloudProvider = () => {
  const project = useSelectedProject()
  const isAws = project?.cloud_provider === PROVIDERS.AWS.id

  return isAws
}

export const useIsAwsK8sCloudProvider = () => {
  const project = useSelectedProject()
  const isAwsK8s = project?.cloud_provider === PROVIDERS.AWS_K8S.id

  return isAwsK8s
}

export const useIsOrioleDb = () => {
  const project = useSelectedProject()
  const isOrioleDb = project?.dbVersion?.endsWith('orioledb')
  return isOrioleDb
}

export const useIsOrioleDbInAws = () => {
  const project = useSelectedProject()
  const isOrioleDbInAws =
    project?.dbVersion?.endsWith('orioledb') && project?.cloud_provider === PROVIDERS.AWS.id
  return isOrioleDbInAws
}