Class SizeAwareComponent

src/with-size.js:157–292

    const SizeMeRenderWrapper = renderWrapper(WrappedComponent)

    class SizeAwareComponent extends React.Component {
      static displayName = `SizeMe(${getDisplayName(WrappedComponent)})`

      domEl = null

      state = {
        width: undefined,
        height: undefined,
      }

      componentDidMount() {
        this.detector = resizeDetector(resizeDetectorStrategy)
        this.determineStrategy(this.props)
        this.handleDOMNode()
      }

      componentDidUpdate() {
        this.determineStrategy(this.props)
        this.handleDOMNode()
      }

      componentWillUnmount() {
        // Change our size checker to a noop just in case we have some
        // late running events.
        this.hasSizeChanged = () => undefined
        this.checkIfSizeChanged = () => undefined
        this.uninstall()
      }

      uninstall = () => {
        if (this.domEl) {
          try {
            this.detector.uninstall(this.domEl)
          } catch (err) {
            // eslint-disable-next-line no-console
            console.warn(errMsg)
          }
          this.domEl = null
        }
      }

      determineStrategy = (props) => {
        if (props.onSize) {
          if (!this.callbackState) {
            this.callbackState = {
              ...this.state,
            }
          }
          this.strategy = 'callback'
        } else {
          this.strategy = 'render'
        }
      }

      strategisedSetState = (state) => {
        if (this.strategy === 'callback') {
          this.callbackState = state
          this.props.onSize(state)