fix(Overlay): check if instance of React Component

alibaba-fusion · Jun 21, 2019 · c9e6802 · c9e6802
1 parent cff7566
commit c9e6802
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/overlay/overlay.jsx b/src/overlay/overlay.jsx
@@ -697,7 +697,11 @@ export default class Overlay extends Component {
             stateVisible || (cache && this._isMounted) ? propChildren : null;
         if (children) {
             let child = Children.only(children);
-            if (typeof child.type === 'function') {
+            // if chlild is a functional component wrap in a component to allow a ref to be set
+            if (
+                typeof child.type === 'function' &&
+                !(child.type.prototype instanceof Component)
+            ) {
                 child = <div role="none">{child}</div>;
             }
             const childClazz = classnames({