fix(onLoad): handle pass in new image source (#4)

* fix(onLoad): handle pass in new image source

* update

Author: evenchange4

README.md

@@ -41,6 +41,7 @@ import ImageOnLoad from 'react-image-onload';
 
 ```js
 type Props = {
+  src: string,
   onLoad: (image: HTMLImageElement) => Promise<void> | void,
 };
 ```

src/ImageOnLoad.js

@@ -5,11 +5,13 @@ import omit from 'ramda/src/omit';
 import PropTypes from 'prop-types';
 
 export type Props = {
+  src: string,
   onLoad: (image: HTMLImageElement) => Promise<void> | void,
 };
 
 class InputOnLoad extends React.Component<Props> {
   static propTypes = {
+    src: PropTypes.string,
     onLoad: PropTypes.func.isRequired, // (image: HTMLImageElement) => Promise<void> | void,
   };
   componentDidMount() {
@@ -21,6 +23,11 @@ class InputOnLoad extends React.Component<Props> {
       onLoad(image);
     }
   }
+  componentWillReceiveProps(nextProps: Props) {
+    if (nextProps.src !== this.props.src) {
+      this.isLoaded = false;
+    }
+  }
   componentWillUnmount() {
     this.isLoaded = false;
   }
@@ -37,7 +44,7 @@ class InputOnLoad extends React.Component<Props> {
   props: Props;
   render() {
     const { onRef, onLoad } = this;
-    const otherProps = omit(['onLoad'])(this.props);
+    const otherProps: Object = omit(['onLoad'])(this.props);
 
     return <img ref={onRef} onLoad={onLoad} {...otherProps} />;
   }