AccessibleHeader.js

import React from "react"
import { withLastLocation } from "react-router-last-location"
import PropTypes from "prop-types"

class AccessibleHeader extends React.Component {
  constructor(props) {
    super(props)
    
    this.headerRef = React.createRef()
    this.state = { tabIndex: null }
  }

  static propTypes = {
    children: PropTypes.any.isRequired,
    lastLocation: PropTypes.object.isRequired,
  }

  componentDidMount() {
    // Don't auto-focus header if there's no lastLocation prop!
    if (!this.props.lastLocation) return
    
    this.setState({ tabIndex: "-1" }, () => {
      this.headerRef.current.focus()
    })
  }

  handleHeaderBlur = () => {
    this.setState({ tabIndex: null })
  }
  
  render() {
    return (
      <h1 tabIndex={this.state.tabIndex} onBlur={this.handleHeaderBlur} ref={this.headerRef}>
        {this.props.children}
      </h1>
    )
  }
}
export default withLastLocation(AccessibleHeader)