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

import dayjs from 'dayjs'
import utc from 'dayjs/plugin/utc'
import { useEffect, useMemo, useRef } from 'react'

import { usePHFlag } from 'hooks/ui/useFlag'
import { IS_PLATFORM } from 'lib/constants'
import { useTrack } from 'lib/telemetry/track'

dayjs.extend(utc)

export type TableCreateGeneratePoliciesVariant = 'control' | 'variation'

const VALID_VARIANTS: TableCreateGeneratePoliciesVariant[] = ['control', 'variation']

export function isValidExperimentVariant(
  value: unknown
): value is TableCreateGeneratePoliciesVariant {
  return (
    typeof value === 'string' &&
    VALID_VARIANTS.includes(value as TableCreateGeneratePoliciesVariant)
  )
}

interface UseTableCreateGeneratePoliciesOptions {
  /**
   * Whether this is a new table being created
   */
  isNewRecord?: boolean
  /**
   * Project creation timestamp
   */
  projectInsertedAt?: string
}

interface UseTableCreateGeneratePoliciesResult {
  /**
   * Whether the generate policies feature is enabled
   */
  enabled: boolean
}

/**
 * Hook to manage the table create generate policies feature flag.
 * Handles feature flag determination and exposure tracking.
 *
 * @param options Configuration for feature targeting
 * @returns Feature state including whether it's enabled
 */
export function useTableCreateGeneratePolicies({
  isNewRecord = false,
  projectInsertedAt,
}: UseTableCreateGeneratePoliciesOptions): UseTableCreateGeneratePoliciesResult {
  const track = useTrack()
  const tableCreateGeneratePoliciesFlag = usePHFlag<string>('tableCreateGeneratePolicies')
  const hasTrackedExposure = useRef(false)

  const enabled = useMemo(() => {
    if (!IS_PLATFORM) return false
    if (tableCreateGeneratePoliciesFlag !== 'variation') return false
    return true
  }, [tableCreateGeneratePoliciesFlag])

  useEffect(() => {
    if (!IS_PLATFORM) return
    if (hasTrackedExposure.current) return
    if (!isNewRecord) return
    if (!isValidExperimentVariant(tableCreateGeneratePoliciesFlag)) return
    if (!projectInsertedAt) return

    try {
      const insertedDate = dayjs.utc(projectInsertedAt)
      if (!insertedDate.isValid()) return

      const daysSinceCreation = dayjs.utc().diff(insertedDate, 'day')
      track('table_create_generate_policies_experiment_exposed', {
        experiment_id: 'tableCreateGeneratePolicies',
        variant: tableCreateGeneratePoliciesFlag,
        days_since_project_creation: daysSinceCreation,
      })
      hasTrackedExposure.current = true
    } catch {
      hasTrackedExposure.current = false
    }
  }, [isNewRecord, tableCreateGeneratePoliciesFlag, projectInsertedAt, track])

  return {
    enabled,
  }
}