Merge pull request mui#4033 from antoinerousseau/dialog-title-style

[Dialog] Merge title style when title it a node
echenley · Apr 21, 2016 · 9baaf7e · 9baaf7e
2 parents be3615f + 86b5894
commit 9baaf7e
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)}>