import type { Directive } from 'vue'

const findFocusable = (element: Element | null | undefined, programmatic = false) => {
    if (!element) {
        return null
    }
    if (programmatic) {
        return element.querySelectorAll('*[tabindex="-1"]')
    }
    return element.querySelectorAll(`a[href]:not([tabindex="-1"]),
                                     area[href],
                                     input:not([disabled]),
                                     select:not([disabled]),
                                     textarea:not([disabled]),
                                     button:not([disabled]),
                                     iframe,
                                     object,
                                     embed,
                                     *[tabindex]:not([tabindex="-1"]),
                                     *[contenteditable]`)
}

let onKeyDown: (event: KeyboardEvent) => void

const beforeMount = (el: HTMLElement, { value = true }) => {
    if (value) {
        let focusable = findFocusable(el)
        let focusableProg = findFocusable(el, true)

        if (focusable && focusable.length > 0) {
            onKeyDown = (event) => {
                // Need to get focusable each time since it can change between key events
                // ex. changing month in a datepicker
                focusable = findFocusable(el)!
                focusableProg = findFocusable(el, true)
                const firstFocusable = focusable[0]
                const lastFocusable = focusable[focusable.length - 1]

                if (event.target === firstFocusable && event.shiftKey && event.key === 'Tab') {
                    event.preventDefault();
                    (lastFocusable! as HTMLElement).focus()
                } else if ((event.target === lastFocusable || Array.from(focusableProg!).indexOf(event.target as Element) >= 0) && !event.shiftKey && event.key === 'Tab') {
                    event.preventDefault();
                    (firstFocusable! as HTMLElement).focus()
                }
            }
            el.addEventListener('keydown', onKeyDown)
        }
    }
}

const unmounted = (el: HTMLElement) => {
    el.removeEventListener('keydown', onKeyDown)
}

const directive: Directive<HTMLElement, boolean | undefined> = {
    beforeMount,
    unmounted
}

export default directive