RestoreScroll.js

RestoreScroll.js

class RestoreScroll extends React.Component {
  rememberScroll () {
    window.requestAnimationFrame(() => {
      console.log('scrollY', window.scrollY)
      window.sessionStorage.setItem(SCROLL_KEY, window.scrollY)
    })
  }

  restoreScroll () {
    const scrollPosition = window.sessionStorage.getItem(SCROLL_KEY)
    console.log('scrollPosition', scrollPosition)

    if (scrollPosition) {
      setTimeout(() => {
        window.scroll(0, scrollPosition)
      }, 100)
    }
  }

  componentDidMount () {
    this.restoreScroll()
    window.addEventListener('scroll', this.rememberScroll)
  }

  componentWillUnmount () {
    window.removeEventListener('scroll', this.rememberScroll)
  }

  render () {
    return this.props.children
  }
}