Fixes #84 - Support ref using React.forwardRef

iamhosseindhv · Apr 8, 2019 · 5dff56c · 5dff56c
1 parent e87e83d
commit 5dff56c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/withSnackbar.js b/src/withSnackbar.js
@@ -4,17 +4,18 @@ import getDisplayName from './utils/getDisplayName';
 import SnackbarContext from './SnackbarContext';
 
 const withSnackbar = (Component) => {
-    const WrappedComponent = props => (
+    const WrappedComponent = React.forwardRef((props, ref) => (
         <SnackbarContext.Consumer>
             {context => (
                 <Component
                     {...props}
+                    ref={ref}
                     enqueueSnackbar={context.handleEnqueueSnackbar}
                     closeSnackbar={context.handleCloseSnackbar}
                 />
             )}
         </SnackbarContext.Consumer>
-    );
+    ));
 
     if (process.env.NODE_ENV !== 'production') {
         WrappedComponent.displayName = `WithSnackbar(${getDisplayName(Component)})`;