supabase/apps/studio/data/organizations/organizations-query.ts

import { QueryClient, useQuery, useQueryClient, UseQueryOptions } from '@tanstack/react-query'
import { get } from 'lib/common/fetch'
import { API_URL } from 'lib/constants'
import { useCallback } from 'react'
import type { Organization, ResponseError } from 'types'
import { organizationKeys } from './keys'

export async function getOrganizations(signal?: AbortSignal): Promise<Organization[]> {
  const data = await get(`${API_URL}/organizations`, { signal })
  if (data.error) throw data.error

  if (!Array.isArray(data)) {
    return []
  }

  const sorted = (data as Organization[]).sort((a, b) => a.name.localeCompare(b.name))
  return sorted
}

export type OrganizationsData = Awaited<ReturnType<typeof getOrganizations>>
export type OrganizationsError = ResponseError

export const useOrganizationsQuery = <TData = OrganizationsData>({
  enabled = true,
  ...options
}: UseQueryOptions<OrganizationsData, OrganizationsError, TData> = {}) =>
  useQuery<OrganizationsData, OrganizationsError, TData>(
    organizationKeys.list(),
    ({ signal }) => getOrganizations(signal),
    { enabled: enabled, ...options, staleTime: 30 * 60 * 1000 }
  )

export function invalidateOrganizationsQuery(client: QueryClient) {
  return client.invalidateQueries(organizationKeys.list())
}