v3/packages/design-system/atoms/feature-gate/FeatureGate.tsx

import {
  FeatureFlag,
  FeatureFlagsContext,
} from '@roleypoly/misc-utils/featureFlags/react';
import * as React from 'react';

export type FeatureGateProps = {
  featureFlag: FeatureFlag;
  children: () => React.ReactNode;
};

export const FeatureGate = (props: FeatureGateProps) => {
  const featureContext = React.useContext(FeatureFlagsContext);

  if (featureContext.has(props.featureFlag)) {
    return props.children();
  } else {
    return <></>;
  }
};