+ onLoad prop for images

ethanselzer · Oct 16, 2017 · 77d0906 · 77d0906
1 parent e07f632
commit 77d0906
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/src/EnlargedImage.js b/src/EnlargedImage.js
@@ -187,7 +187,8 @@ export default class extends React.Component {
                     src: largeImage.src,
                     srcSet: largeImage.srcSet,
                     sizes: largeImage.sizes,
-                    style: objectAssign({}, imageStyle, computedImageStyle)
+                    style: objectAssign({}, imageStyle, computedImageStyle),
+                    onLoad: largeImage.onLoad
                 }}/>
             </div>
         );

diff --git a/src/ImageShape.js b/src/ImageShape.js
@@ -6,5 +6,6 @@ export default PropTypes.shape({
     srcSet: PropTypes.string,
     sizes: PropTypes.string,
     width: PropTypes.number.required,
-    height: PropTypes.number.required
+    height: PropTypes.number.required,
+    onLoad: PropTypes.func,
 });
diff --git a/src/ReactImageMagnify.js b/src/ReactImageMagnify.js
@@ -52,7 +52,8 @@ class ReactImageMagnify extends React.Component {
             srcSet: PropTypes.string,
             sizes: PropTypes.string,
             width: requiredIf(PropTypes.number, props => !props.isFluidWidth),
-            height: requiredIf(PropTypes.number, props => !props.isFluidWidth)
+            height: requiredIf(PropTypes.number, props => !props.isFluidWidth),
+            onLoad: PropTypes.func
         }),
         style: PropTypes.object,
         enlargedImagePosition: PropTypes.oneOf(['beside', 'over'])
@@ -66,10 +67,17 @@ class ReactImageMagnify extends React.Component {
 
     onSmallImageLoad() {
         if (!this.props.smallImage.isFluidWidth) {
+            if (this.props.smallImage.onLoad) {
+                this.props.smallImage.onLoad.apply(null, arguments);
+            }
             return;
         }
 
         this.setSmallImageDimensionState();
+
+        if (this.props.smallImage.onLoad) {
+            this.props.smallImage.onLoad.apply(null, arguments);
+        }
     }
 
     setSmallImageDimensionState() {