Class RefetchConnect

src/components/connect.js:197–430

  return function wrapWithConnect(WrappedComponent) {
    class RefetchConnect extends Component {
      constructor(props) {
        super(props)
        this.version = version

        // To avoid undefined data at mount, pre-populated pending PromiseStates
        // TODO: de-dupe with update code
        const mappings = finalMapPropsToRequestsToProps(omitChildren(props)) || {}
        const initDate = Object.keys(mappings).reduce((data, prop) => {
          const mapping = mappings[prop]
          if (Function.prototype.isPrototypeOf(mapping)) {
            data[prop] = (...args) => {
              this.refetchDataFromMappings(mapping(...args))
            }
          } else {
            data[prop] = PromiseState.create(mapping.meta)
          }
          return data
        }, {})

        this.state = {
          mappings: {},
          startedAts: {},
          data: initDate,
          refreshTimeouts: {}
        }
      }

      componentDidMount() {
        this.refetchDataFromProps()
      }

      componentDidUpdate(prevProps) {
        if (dependsOnProps && !shallowEqual(omitChildren(this.props), omitChildren(prevProps))) {
          this.refetchDataFromProps()
        }
      }

      shouldComponentUpdate(nextProps, nextState) {
        return this.state.data !== nextState.data || !shallowEqual(this.props, nextProps)
      }

      componentWillUnmount() {
        this.clearAllRefreshTimeouts()
        this._unmounted = true
      }

      render() {
        const ref = options.withRef ? 'wrappedInstance' : null
        return (
          <WrappedComponent { ...this.state.data } { ...this.props } ref={ref}/>
        )
      }

      getWrappedInstance() {
        invariant(options.withRef,
          'To access the wrapped instance, you need to specify { withRef: true } in .options().'
        )