supabase/studio/hooks/queries/useProjectPaygStatistics.ts

import dayjs from 'dayjs'
import useSWR from 'swr'
import { get as _get, maxBy } from 'lodash'
import { get } from 'lib/common/fetch'
import { API_URL, DATE_FORMAT, PRICING_TIER_PRODUCT_IDS } from 'lib/constants'
import { chargeableProducts } from 'components/interfaces/Billing'
import { PaygStats } from 'components/interfaces/Billing/PAYGUsage/PAYGUsage.types'

/**
 * Get project payg statistics
 *
 * @param ref project reference
 * @param tierProdId tier product supabase_prod_id
 */
export function useProjectPaygStatistics(ref?: string, tierProdId?: string) {
  const startDate = dayjs().utc().startOf('month').format(DATE_FORMAT)
  const endDate = dayjs().utc().endOf('month').format(DATE_FORMAT)
  const attributes =
    'total_egress,total_db_size_bytes,total_db_egress_bytes,total_storage_size_bytes,total_storage_egress'
  const url = `${API_URL}/projects/${ref}/daily-stats?attribute=${attributes}&startDate=${encodeURIComponent(
    startDate
  )}&endDate=${encodeURIComponent(endDate)}&interval='1d'`
  /**
   * data is a dictionary
   * {
   *   data: array,
   *   totalAverage: number,
   *   total: number,
   *   maximum: number,
   *   format: string,
   * }
   */
  const { data, error } = useSWR<any>(
    ref && tierProdId === PRICING_TIER_PRODUCT_IDS.PAYG ? url : null,
    get
  )
  const anyError = data?.error || error

  return {
    paygStats: anyError ? undefined : getPaygStats(data?.data),
    error: anyError,
    isLoading: !anyError && !data,
    isError: !!anyError,
  }
}

function getPaygStats(data: any): PaygStats {
  const paygStats: any = {}
  if (!data) return paygStats

  let dataModified: any = []
  data.map((x: any) => {
    dataModified.push({
      ...x,
      total_egress_modified:
        (x.total_egress == undefined ? 0 : x.total_egress) -
        (x.total_storage_egress == undefined ? 0 : x.total_storage_egress),
    })
  })

  chargeableProducts.forEach((product) => {
    product.features.forEach((feature) => {
      // calculate max
      const max = _get(maxBy(dataModified, feature.attribute), feature.attribute) ?? 0

      // calculate sum
      let sum = 0
      dataModified.map((x: any) => {
        const featureAttribute = x[feature.attribute] == undefined ? 0 : x[feature.attribute]
        sum = sum + featureAttribute
      })

      // set sum and max under attribute name
      paygStats[feature.attribute] = {
        sum,
        max,
      }
    })
  })

  return paygStats
}