From 20261a80daf02ad40caaa889708d7342125375ae Mon Sep 17 00:00:00 2001
From: mbrookes <github@nospam.33m.co>
Date: Thu, 24 Mar 2016 01:51:15 +0000
Subject: [PATCH] [GridList] Replace `rootClass` with `containerElement`

---
 .../components/GridList/ExampleComplex.js     |  2 +-
 .../components/GridList/ExampleSimple.js      |  3 +-
 src/GridList/GridTile.js                      | 44 +++++++++----------
 3 files changed, 23 insertions(+), 26 deletions(-)

diff --git a/docs/src/app/components/pages/components/GridList/ExampleComplex.js b/docs/src/app/components/pages/components/GridList/ExampleComplex.js
index ae36986901e830..858b9f4e68efaf 100644
--- a/docs/src/app/components/pages/components/GridList/ExampleComplex.js
+++ b/docs/src/app/components/pages/components/GridList/ExampleComplex.js
@@ -11,7 +11,7 @@ const styles = {
   },
   gridList: {
     width: 500,
-    height: 400,
+    height: 450,
     overflowY: 'auto',
     marginBottom: 24,
   },
diff --git a/docs/src/app/components/pages/components/GridList/ExampleSimple.js b/docs/src/app/components/pages/components/GridList/ExampleSimple.js
index 2c5a9f0f3d96c9..67573ef89756ce 100644
--- a/docs/src/app/components/pages/components/GridList/ExampleSimple.js
+++ b/docs/src/app/components/pages/components/GridList/ExampleSimple.js
@@ -12,7 +12,7 @@ const styles = {
   },
   gridList: {
     width: 500,
-    height: 400,
+    height: 500,
     overflowY: 'auto',
     marginBottom: 24,
   },
@@ -61,7 +61,6 @@ const tilesData = [
   },
 ];
 
-
 const GridListExampleSimple = () => (
   <div style={styles.root}>
     <GridList
diff --git a/src/GridList/GridTile.js b/src/GridList/GridTile.js
index b72f8c9a6e15df..17b2faaa4219b4 100644
--- a/src/GridList/GridTile.js
+++ b/src/GridList/GridTile.js
@@ -85,15 +85,15 @@ const GridTile = React.createClass({
     cols: React.PropTypes.number,
 
     /**
-     * Either a string used as tag name for the tile root element, or a ReactComponent.
+     * Either a string used as tag name for the tile root element, or a ReactElement.
      * This is useful when you have, for example, a custom implementation of
-     * a navigation link (that knowsabout your routes) and you want to use it as primary tile action.
-     * In case you pass a ReactComponent, please make sure that it passes all props,
+     * a navigation link (that knows about your routes) and you want to use it as the primary tile action.
+     * In case you pass a ReactElement, please ensure that it passes all props,
      * accepts styles overrides and render it's children.
      */
-    rootClass: React.PropTypes.oneOfType([
+    containerElement: React.PropTypes.oneOfType([
       React.PropTypes.string,
-      React.PropTypes.object,
+      React.PropTypes.element,
     ]),
 
     /**
@@ -143,7 +143,7 @@ const GridTile = React.createClass({
       actionPosition: 'right',
       cols: 1,
       rows: 1,
-      rootClass: 'div',
+      containerElement: 'div',
     };
   },
 
@@ -207,7 +207,7 @@ const GridTile = React.createClass({
       actionPosition,
       style,
       children,
-      rootClass,
+      containerElement,
       ...other,
     } = this.props;
 
@@ -223,28 +223,25 @@ const GridTile = React.createClass({
 
     if (title) {
       titleBar = (
-        <div style={prepareStyles(styles.titleBar)}>
+        <div key="titlebar" style={prepareStyles(styles.titleBar)}>
           <div style={prepareStyles(styles.titleWrap)}>
             <div style={prepareStyles(styles.title)}>{title}</div>
-            {
-              subtitle ? (<div style={prepareStyles(styles.subtitle)}>{subtitle}</div>) : null
-            }
+            {subtitle ? (<div style={prepareStyles(styles.subtitle)}>{subtitle}</div>) : null}
           </div>
-          {
-            actionIcon ? (<div style={prepareStyles(styles.actionIcon)}>{actionIcon}</div>) : null
-          }
+          {actionIcon ? (<div style={prepareStyles(styles.actionIcon)}>{actionIcon}</div>) : null}
         </div>
       );
     }
 
     let newChildren = children;
 
-    // if there is an image passed as children
-    // clone it an put our styles
+    // if there is a single image passed as children
+    // clone it and add our styles
     if (React.Children.count(children) === 1) {
       newChildren = React.Children.map(children, (child) => {
         if (child.type === 'img') {
           return React.cloneElement(child, {
+            key: 'img',
             ref: 'img',
             style: prepareStyles(Object.assign({}, styles.childImg, child.props.style)),
           });
@@ -254,13 +251,14 @@ const GridTile = React.createClass({
       });
     }
 
-    const RootTag = rootClass;
-    return (
-      <RootTag style={prepareStyles(mergedRootStyles)} {...other}>
-        {newChildren}
-        {titleBar}
-      </RootTag>
-    );
+    const containerProps = {
+      style: prepareStyles(mergedRootStyles),
+      ...other,
+    };
+
+    return React.isValidElement(containerElement) ?
+      React.cloneElement(containerElement, containerProps, [newChildren, titleBar]) :
+      React.createElement(containerElement, containerProps, [newChildren, titleBar]);
   },
 });