Same clean up on feature flag component

influxdata · Feb 16, 2019 · f53f0c9 · f53f0c9
1 parent d471b49
commit f53f0c9
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/ui/src/shared/components/FeatureFlag.tsx b/ui/src/shared/components/FeatureFlag.tsx
@@ -1,16 +1,19 @@
-import {SFC} from 'react'
+import {PureComponent} from 'react'
 
 interface Props {
   name?: string
-  children?: any
 }
 
-const FeatureFlag: SFC<Props> = props => {
-  if (process.env.NODE_ENV === 'development') {
-    return props.children
+export default class extends PureComponent<Props> {
+  public render() {
+    if (this.isHidden) {
+      return null
+    }
+
+    return this.props.children
   }
 
-  return null
+  private get isHidden(): boolean {
+    return process.env.NODE_ENV !== 'development'
+  }
 }
-
-export default FeatureFlag