jump.js

export function jumpKeyUX() {
  return window => {
    let jumps = []

    function focus(next) {
      let current = window.document.activeElement
      if (current && current !== window.document.body) {
        jumps.push(new WeakRef(current))
      }
      next.focus({ focusVisible: true })
    }

    function back() {
      let ref = jumps.pop()
      if (!ref) {
        window.document.activeElement.blur()
        return
      }
      let el = ref.deref()
      if (el && el.isConnected) {
        el.focus()
      } else {
        back()
      }
    }

    let tries = 0
    let finding

    function jump(from) {
      clearInterval(finding)
      let ariaControls = from.getAttribute('aria-controls')
      finding = setInterval(() => {
        if (tries++ > 50) {
          clearInterval(finding)
          return
        }
        let area = window.document.getElementById(ariaControls)
        if (area) {
          let next = area.querySelector(
            'a, button, select, textarea, ' +
              'input:not([type=radio]), [type=radio]:checked, ' +
              '[tabindex]:not([tabindex="-1"])'
          )
          if (next) {
            clearInterval(finding)
            area.dispatchEvent(
              new window.CustomEvent('keyuxJump', { bubbles: true })
            )
            focus(next)
          }
        }
      }, 50)
    }

    function keyDown(event) {
      if (event.target.getAttribute('aria-controls')) {
        if (event.key === 'Enter') {
          jump(event.target)
        }
      }
      if (event.key === 'Escape') {
        back()
      }
    }

    window.addEventListener('keydown', keyDown)
    return () => {
      window.removeEventListener('keydown', keyDown)
    }
  }
}