supabase/studio/data/profile/profile-query.ts

import { useQuery, useQueryClient, UseQueryOptions } from '@tanstack/react-query'
import { get } from 'lib/common/fetch'
import { API_URL } from 'lib/constants'
import { useCallback } from 'react'
import { profileKeys } from './keys'

export type Profile = {
  id: number
  auth0_id: string
  primary_email: string
  username: string
  first_name: string
  last_name: string
  mobile: string | null
  is_alpha_user: boolean
  gotrue_id: string
  free_project_limit: number
}

export type ProfileResponse = Profile

export async function getProfile(signal?: AbortSignal) {
  const response = await get(`${API_URL}/profile`, {
    signal,
  })
  if (response.error) {
    throw response.error
  }

  return response as ProfileResponse
}

export type ProfileData = Awaited<ReturnType<typeof getProfile>>
export type ProfileError = unknown

export const useProfileQuery = <TData = ProfileData>({
  enabled = true,
  ...options
}: UseQueryOptions<ProfileData, ProfileError, TData> = {}) =>
  useQuery<ProfileData, ProfileError, TData>(
    profileKeys.profile(),
    ({ signal }) => getProfile(signal),
    {
      staleTime: 1000 * 60 * 30, // default good for 30 mins
      ...options,
    }
  )

export const useProfilePrefetch = () => {
  const client = useQueryClient()

  return useCallback(() => {
    client.prefetchQuery(profileKeys.profile(), ({ signal }) => getProfile(signal))
  }, [])
}