Refs.tsx

class Refs extends React.PureComponent<IProps> {
  private references: React.ReactInstance[] = []

  componentDidMount () {
    console.log('references', this.references)
  }

  setRef = (ref: React.ReactInstance) => {
    this.references.push(ref)
  }

  renderChildren = (children: React.ReactNode): React.ReactNode => {
    if (Array.isArray(children)) {
      return React.Children.map(children, this.renderChildren)
    }

    if (React.isValidElement(children) && typeof children.type === 'symbol') {
      return React.Children.map((children.props as { children: React.ReactNode }).children, this.renderChildren)
    }

    if (React.isValidElement<{ ref: (ref: React.ReactInstance) => void}>(children)) {
      return React.cloneElement(children, { ref: this.setRef })
    }

    return children
  }

  render () {
    return this.renderChildren(this.props.children) || null
  }
}