supabase/apps/studio/data/usage/org-usage-query.ts

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

import { usageKeys } from './keys'
import type { components } from '@/data/api'
import { get, handleError } from '@/data/fetchers'
import { IS_PLATFORM } from '@/lib/constants'
import type { ResponseError, UseCustomQueryOptions } from '@/types'

export type OrgUsageVariables = {
  orgSlug?: string
  projectRef?: string | null
  start?: Date
  end?: Date
}

export type OrgUsageResponse = components['schemas']['OrgUsageResponse']
export type OrgMetricsUsage = components['schemas']['OrgUsageResponse']['usages'][0]

export async function getOrgUsage(
  { orgSlug, projectRef, start, end }: OrgUsageVariables,
  signal?: AbortSignal
): Promise<OrgUsageResponse> {
  if (!orgSlug) throw new Error('orgSlug is required')
  const { data, error } = await get(`/platform/organizations/{slug}/usage`, {
    params: {
      path: { slug: orgSlug },
      query: {
        project_ref: projectRef ?? undefined,
        start: start?.toISOString(),
        end: end?.toISOString(),
      },
    },
    signal,
  })
  if (error) handleError(error)
  return data
}

export type OrgUsageData = Awaited<ReturnType<typeof getOrgUsage>>
export type OrgUsageError = ResponseError

export const useOrgUsageQuery = <TData = OrgUsageData>(
  { orgSlug, projectRef, start, end }: OrgUsageVariables,
  { enabled = true, ...options }: UseCustomQueryOptions<OrgUsageData, OrgUsageError, TData> = {}
) =>
  useQuery<OrgUsageData, OrgUsageError, TData>({
    queryKey: usageKeys.orgUsage(
      orgSlug,
      projectRef ?? undefined,
      start?.toISOString(),
      end?.toISOString()
    ),
    queryFn: ({ signal }) => getOrgUsage({ orgSlug, projectRef, start, end }, signal),
    enabled: enabled && IS_PLATFORM && typeof orgSlug !== 'undefined',
    staleTime: 1000 * 60 * 60,
    ...options,
  })