/**
 * Mobile Menu 0.0.5
 * Mobile navigation menu module
 * @author Kyle Foster (@hkfoster)
 * @license MIT (https://www.opensource.org/licenses/mit-license.php/)
 **/

// Public API function
const mobileMenu = (() => {
  
  // Scoped variables
  let focusable
  let firstFocusable
  let lastFocusable

  function keyHandler(event) {
    event = event || window.event
    if (document.documentElement.dataset.menuState !== 'on') return false
    
    const tabKey = 9
    const escKey = 27

    // Key code conditionals
    switch (event.keyCode) {

      // Tab
      case tabKey:
        if (focusable.length === 1) {
          event.preventDefault()
          break
        }

        // Go back if shift is fired, tab backward
        if (event.shiftKey) {
          if (document.activeElement === firstFocusable) {
            event.preventDefault()
            lastFocusable.focus()
          }

        // Otherwise tab forward
        } else {
          if (document.activeElement === lastFocusable) {
            event.preventDefault()
            firstFocusable.focus()
          }
        }
        break

      // Esc
      case escKey:
        document.documentElement.dataset.menuState = 'off'
        allowFocus('[data-menu]', false)
        break

      // Default
      default:
        break
    }
  }

  // Click handler function
  function clickHandler(event) {

    // Only run on menu button
    if (event.target.closest('[data-menu-toggle]')) {

      // Toggle body state attribute
      document.documentElement.dataset.menuState = document.documentElement.dataset.menuState === 'on' ? 'off' : 'on'

      // Set up modal focus trap
      if (document.documentElement.dataset.menuState === 'on') {
        sessionStorage.setItem('scroll-position', document.documentElement.scrollTop)
        document.documentElement.dataset.scrollDisabled = 'on'
        allowFocus('[data-menu]', true)
        trapFocus()

      } else {
        document.documentElement.dataset.scrollDisabled = 'off'
        document.documentElement.scrollTop = sessionStorage.getItem('scroll-position')
        allowFocus('[data-menu]', false)
      }
    }
  }

  function allowFocus(selector, state) {
    const container = document.querySelector(selector)
    const focusable = container.querySelectorAll('button, [href], input, select, textarea')
    focusable.forEach(el => el.setAttribute('tabindex', state ? '' : '-1'))
    container.classList.toggle('hidden')
  }

  // Trap focus to currently open modal
  function trapFocus() {
    focusable = document.querySelector('[data-menu-container]').querySelectorAll('a[href]:not([tabindex="-1"]), area[href]:not([tabindex="-1"]), input:not([disabled]):not([tabindex="-1"]), select:not([disabled]):not([tabindex="-1"]), textarea:not([disabled]):not([tabindex="-1"]), button:not([disabled]):not([tabindex="-1"]), iframe:not([tabindex="-1"]), [tabindex]:not([tabindex="-1"]), [contentEditable=true]:not([tabindex="-1"])')
    firstFocusable = focusable[0]
    lastFocusable = focusable[focusable.length - 1]
  }

  // Attach listeners
  document.addEventListener('click', clickHandler, false)
  document.addEventListener('keydown', keyHandler, false)

})()