[Dialog] merge title style when it's a node, fixes mui#3373

echenley · Apr 20, 2016 · 86b5894 · 86b5894
1 parent f34ddf0
commit 86b5894
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/src/Dialog/Dialog.js b/src/Dialog/Dialog.js
@@ -295,10 +295,19 @@ class DialogInline extends React.Component {
       </div>
     );
 
-    const titleElement = typeof title === 'string' ?
-      <h3 className={titleClassName} style={prepareStyles(styles.title)}>
-        {title}
-      </h3> : title;
+    let titleElement = title;
+    if (React.isValidElement(title)) {
+      titleElement = React.cloneElement(title, {
+        className: title.props.className || titleClassName,
+        style: prepareStyles(Object.assign(styles.title, title.props.style)),
+      });
+    } else if (typeof title === 'string') {
+      titleElement = (
+        <h3 className={titleClassName} style={prepareStyles(styles.title)}>
+          {title}
+        </h3>
+      );
+    }
 
     return (
       <div className={className} style={prepareStyles(styles.root)}>