Allow snackbar children to of type function (#114)

iamhosseindhv · Apr 27, 2019 · fa231cc · fa231cc
1 parent 21e5854
commit fa231cc
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/SnackbarItem/SnackbarItem.js b/src/SnackbarItem/SnackbarItem.js
@@ -83,6 +83,11 @@ class SnackbarItem extends Component {
             finalAction = contentProps.action(key);
         }
 
+        let snackChildren = snack.children;
+        if (snackChildren && typeof snackChildren === 'function') {
+            snackChildren = snackChildren(key);
+        }
+
         return (
             <RootRef rootRef={this.ref}>
                 <Snackbar
@@ -103,7 +108,7 @@ class SnackbarItem extends Component {
                     onClose={this.handleClose(key)}
                     onExited={this.handleExited(key)}
                 >
-                    {snack.children || (
+                    {snackChildren || (
                         <SnackbarContent
                             className={classNames(
                                 classes.base,