import Keyboard from './novnc.js'

export interface NoVncKeyboardInterface {
  // original functions
  onkeyevent: (keysym: number | null, code: string, down: boolean) => boolean
  grab: () => void
  ungrab: () => void

  // custom functions
  onkeydown?: (keysym: number) => boolean
  onkeyup?: (keysym: number) => void
  release: (keysym: number) => void
  listenTo: (element: Element | Document) => void
  removeListener: () => void
}

export default function (element?: Element): NoVncKeyboardInterface {
  // @ts-ignore
  const keyboard = new Keyboard(element)

  // map on key event to onkeydown and onkeyup
  keyboard.onkeyevent = function (keysym: number | null, code: string, down: boolean) {
    if (keysym === null) return false
    if (down) return this.onkeydown(keysym, code)
    this.onkeyup(keysym, code)
  }

  // add release function
  keyboard.release = function (keysym: number) {
    for (const code in this._keyDownList) {
      if (this._keyDownList[code] === keysym) {
        this._sendKeyEvent(keysym, code, false)
        break
      }
    }
  }

  // add listenTo function
  keyboard.listenTo = function (element: Element | Document) {
    if (element) this._target = element
    this.grab()
  }

  // add removeListener function
  keyboard.removeListener = function () {
    this.ungrab()
  }

  return keyboard
}