Expand proptypes

treyhuffine · Oct 30, 2018 · 1641f0b · 1641f0b
1 parent 2f2062b
commit 1641f0b
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/lightbox-react.js b/src/lightbox-react.js
@@ -1563,21 +1563,28 @@ class ReactImageLightbox extends Component {
   }
 }
 
+const srcTypes = PropTypes.oneOfType([
+  PropTypes.string,
+  PropTypes.func,
+  PropTypes.element,
+  PropTypes.node,
+]);
+
 ReactImageLightbox.propTypes = {
   //-----------------------------
   // Image sources
   //-----------------------------
 
   // Main display image url
-  mainSrc: PropTypes.oneOfType([PropTypes.string, PropTypes.func]).isRequired, // eslint-disable-line react/no-unused-prop-types
+  mainSrc: srcTypes.isRequired, // eslint-disable-line react/no-unused-prop-types
 
   // Previous display image url (displayed to the left)
   // If left undefined, movePrev actions will not be performed, and the button not displayed
-  prevSrc: PropTypes.oneOfType([PropTypes.string, PropTypes.func]),
+  prevSrc: srcTypes,
 
   // Next display image url (displayed to the right)
   // If left undefined, moveNext actions will not be performed, and the button not displayed
-  nextSrc: PropTypes.oneOfType([PropTypes.string, PropTypes.func]),
+  nextSrc: srcTypes,
 
   //-----------------------------
   // Image thumbnail sources