mirror of https://github.com/portainer/portainer
27 lines
691 B
TypeScript
27 lines
691 B
TypeScript
import { ComponentType } from 'react';
|
|
|
|
import { FeatureFlag, useFeatureFlag } from './useRedirectFeatureFlag';
|
|
|
|
export function withFeatureFlag<T>(
|
|
WrappedComponent: ComponentType<T>,
|
|
flag: FeatureFlag
|
|
): ComponentType<T> {
|
|
// Try to create a nice displayName for React Dev Tools.
|
|
const displayName =
|
|
WrappedComponent.displayName || WrappedComponent.name || 'Component';
|
|
|
|
function WrapperComponent(props: T) {
|
|
const featureFlagQuery = useFeatureFlag(flag);
|
|
|
|
if (!featureFlagQuery.data) {
|
|
return null;
|
|
}
|
|
|
|
return <WrappedComponent {...props} />;
|
|
}
|
|
|
|
WrapperComponent.displayName = `with${flag}FeatureFlag(${displayName})`;
|
|
|
|
return WrapperComponent;
|
|
}
|