Set display name on HOC

adamgraham · Jan 8, 2021 · 832cee4 · 832cee4
1 parent 9b9fbc9
commit 832cee4
Showing 1 changed file with 24 additions and 18 deletions.
diff --git a/src/utils/hoc.js b/src/utils/hoc.js
@@ -3,21 +3,27 @@ import { useDispatch } from 'react-redux';
 import { useBaseColor, useSecondaryBaseColor } from './hooks';
 import { setBaseColor, setSecondaryBaseColor } from '../actions';
 
-export const withBaseColor = (Component) => (props) => {
-  const dispatch = useDispatch();
-  const baseColor = useBaseColor();
-  const secondaryBaseColor = useSecondaryBaseColor();
-  return (
-    <Component
-      {...props}
-      baseColor={baseColor}
-      secondaryBaseColor={secondaryBaseColor}
-      setBaseColor={(color) => {
-        dispatch(setBaseColor(color));
-      }}
-      setSecondaryBaseColor={(color) => {
-        dispatch(setSecondaryBaseColor(color));
-      }}
-    />
-  );
-};
+export function withBaseColor(WrappedComponent) {
+  const WithBaseColor = (props) => {
+    const dispatch = useDispatch();
+    const baseColor = useBaseColor();
+    const secondaryBaseColor = useSecondaryBaseColor();
+    return (
+      <WrappedComponent
+        {...props}
+        baseColor={baseColor}
+        secondaryBaseColor={secondaryBaseColor}
+        setBaseColor={(color) => {
+          dispatch(setBaseColor(color));
+        }}
+        setSecondaryBaseColor={(color) => {
+          dispatch(setSecondaryBaseColor(color));
+        }}
+      />
+    );
+  };
+  WithBaseColor.displayName = `WithBaseColor(${
+    WrappedComponent.displayName || WrappedComponent.name || 'Component'
+  })`;
+  return WithBaseColor;
+}