react-photo-view/src/Photo.tsx

import React from 'react';
import classNames from 'classnames';
import throttle from './utils/throttle';
import Spinner from './components/Spinner';
import getSuitableImageSize from './utils/getSuitableImageSize';
import './Photo.less';

export interface IPhotoProps extends React.HTMLAttributes<any> {
  src: string;
  className?: string;
  onPhotoResize: () => void;
  loadingElement?: JSX.Element;
  brokenElement?: JSX.Element;
}

type PhotoState = {
  loaded: boolean;
  broken: boolean;
  naturalWidth: number;
  naturalHeight: number;
  width: number;
  height: number;
};

export default class Photo extends React.PureComponent<
  IPhotoProps,
  PhotoState
> {
  static displayName = 'Photo';

  readonly state = {
    loaded: false,
    broken: false,
    naturalWidth: 1,
    naturalHeight: 1,
    width: 1,
    height: 1,
  };

  private isMount = true;

  constructor(props: IPhotoProps) {
    super(props);
    this.handleResize = throttle(this.handleResize, 8);
  }

  componentDidMount() {
    const currPhoto = new Image();
    currPhoto.src = this.props.src;
    currPhoto.onload = this.handleImageLoaded;
    currPhoto.onerror = this.handleImageBroken;

    window.addEventListener('resize', this.handleResize);
  }

  componentWillUnmount() {
    this.isMount = false;
    window.removeEventListener('resize', this.handleResize);
  }

  handleImageLoaded = e => {
    const { naturalWidth, naturalHeight } = e.target;
    if (this.isMount) {
      this.setState({
        loaded: true,
        naturalWidth,
        naturalHeight,
        ...getSuitableImageSize(naturalWidth, naturalHeight),
      });
    }
  };

  handleImageBroken = () => {
    if (this.isMount) {
      this.setState({
        broken: true,
      });
    }
  };

  handleResize = () => {
    const { loaded, naturalWidth, naturalHeight } = this.state;
    if (loaded && this.isMount) {
      this.setState(
        getSuitableImageSize(naturalWidth, naturalHeight),
        this.props.onPhotoResize,
      );
    }
  };

  render() {
    const {
      src,
      className,
      loadingElement,
      brokenElement,
      onPhotoResize,
      ...restProps
    } = this.props;
    const { loaded, broken, width, height } = this.state;

    if (src && !broken) {
      if (loaded) {
        return (
          <img
            className={classNames('PhotoView__Photo', className)}
            src={src}
            width={width}
            height={height}
            alt=""
            {...restProps}
          />
        );
      }
      return loadingElement || <Spinner />;
    }
    return brokenElement || null;
  }
}