fix: forward ref to defaultStyle wrapped component

signavio · Sep 12, 2022 · 6a3ac3c · 6a3ac3c
1 parent 4fdfdd9
commit 6a3ac3c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/utils/defaultStyle.js b/src/utils/defaultStyle.js
@@ -3,21 +3,21 @@ import useStyles from 'substyle'
 
 function createDefaultStyle(defaultStyle, getModifiers) {
   const enhance = (ComponentToWrap) => {
-    const DefaultStyleEnhancer = ({
+    const DefaultStyleEnhancer = React.forwardRef(({
       style,
       className,
       classNames,
       ...rest
-    }) => {
+    }, forwardedRef) => {
       const modifiers = getModifiers ? getModifiers(rest) : undefined
       const styles = useStyles(
         defaultStyle,
         { style, className, classNames },
         modifiers
       )
 
-      return <ComponentToWrap {...rest} style={styles} />
-    }
+      return <ComponentToWrap {...rest} ref={forwardedRef} style={styles} />
+    })
     const displayName =
       ComponentToWrap.displayName || ComponentToWrap.name || 'Component'
     DefaultStyleEnhancer.displayName = `defaultStyle(${displayName})`