import { isFeatureEnabled, type Feature } from 'common'

import { useEnabledFeaturesOverrideQuery } from '@/data/misc/enabled-features-override-query'
import { useProfile } from '@/lib/profile'

function useIsFeatureEnabled<T extends Feature[]>(
  features: T
): ReturnType<typeof isFeatureEnabled<T>>
function useIsFeatureEnabled(features: Feature): ReturnType<typeof isFeatureEnabled>
function useIsFeatureEnabled<T extends Feature | Feature[]>(features: T) {
  const { profile } = useProfile()
  const { data: override } = useEnabledFeaturesOverrideQuery()

  const disabledFeatures = [
    ...(profile?.disabled_features ?? []),
    ...((override?.disabled_features ?? []) as Feature[]),
  ]

  // This code branch is to make the type checker happy, it's intentionally
  // the same as the isFeatureEnabled function call below.
  if (Array.isArray(features)) {
    return isFeatureEnabled(features, disabledFeatures)
  }

  return isFeatureEnabled(features, disabledFeatures)
}

export { useIsFeatureEnabled }