fix: nodeRef prop type for cross-realm elements (#732)

shaikdev2 · May 29, 2021 · c4faae4 · c4faae4
1 parent e6c5db3
commit c4faae4
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/Transition.js b/src/Transition.js
@@ -391,9 +391,18 @@ Transition.propTypes = {
    *     [test/CSSTransition-test.js](https://github.com/reactjs/react-transition-group/blob/13435f897b3ab71f6e19d724f145596f5910581c/test/CSSTransition-test.js#L362-L437)).
    */
   nodeRef: PropTypes.shape({
-    current: typeof Element === 'undefined'
-      ? PropTypes.any
-      : PropTypes.instanceOf(Element)
+    current:
+      typeof Element === 'undefined'
+        ? PropTypes.any
+        : (propValue, key, componentName, location, propFullName, secret) => {
+            const value = propValue[key];
+
+            return PropTypes.instanceOf(
+              value && 'ownerDocument' in value
+                ? value.ownerDocument.defaultView.Element
+                : Element
+            )(propValue, key, componentName, location, propFullName, secret);
+          },
   }),
 
   /**