department-of-veterans-affairs/vets-website

import React from 'react';
import { Route } from 'react-router-dom';
import { useSelector } from 'react-redux';
import PropTypes from 'prop-types';
import App from '../../containers/App';

const FeatureFlagRoute = ({ featureFlag, children, ...rest }) => {
  const displayRoute = useSelector(state => state.featureToggles[featureFlag]);
  return displayRoute ? (
    <Route {...rest}>
      <App>{children}</App>
    </Route>
  ) : null;
};

export default FeatureFlagRoute;

FeatureFlagRoute.propTypes = {
  children: PropTypes.object,
  featureFlag: PropTypes.string,
};