// Copyright 2015 Hajime Hoshi
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// +build ignore

// Note:
//   * Respect GLFW key names
//   * https://developer.mozilla.org/en-US/docs/Web/API/KeyboardEvent.keyCode
//   * It is best to replace keyCode with code, but many browsers don't implement it.

package main

import (
	"log"
	"os"
	"path/filepath"
	"sort"
	"strconv"
	"strings"
	"text/template"

	"github.com/go-gl/glfw/v3.2/glfw"
)

var (
	nameToGLFWKeys    map[string]glfw.Key
	nameToJSKeyCode   map[string]string
	keyCodeToNameEdge map[int]string
)

func init() {
	nameToGLFWKeys = map[string]glfw.Key{
		"Unknown":      glfw.KeyUnknown,
		"Space":        glfw.KeySpace,
		"Apostrophe":   glfw.KeyApostrophe,
		"Comma":        glfw.KeyComma,
		"Minus":        glfw.KeyMinus,
		"Period":       glfw.KeyPeriod,
		"Slash":        glfw.KeySlash,
		"Semicolon":    glfw.KeySemicolon,
		"Equal":        glfw.KeyEqual,
		"LeftBracket":  glfw.KeyLeftBracket,
		"Backslash":    glfw.KeyBackslash,
		"RightBracket": glfw.KeyRightBracket,
		"GraveAccent":  glfw.KeyGraveAccent,
		"World1":       glfw.KeyWorld1,
		"World2":       glfw.KeyWorld2,
		"Escape":       glfw.KeyEscape,
		"Enter":        glfw.KeyEnter,
		"Tab":          glfw.KeyTab,
		"Backspace":    glfw.KeyBackspace,
		"Insert":       glfw.KeyInsert,
		"Delete":       glfw.KeyDelete,
		"Right":        glfw.KeyRight,
		"Left":         glfw.KeyLeft,
		"Down":         glfw.KeyDown,
		"Up":           glfw.KeyUp,
		"PageUp":       glfw.KeyPageUp,
		"PageDown":     glfw.KeyPageDown,
		"Home":         glfw.KeyHome,
		"End":          glfw.KeyEnd,
		"CapsLock":     glfw.KeyCapsLock,
		"ScrollLock":   glfw.KeyScrollLock,
		"NumLock":      glfw.KeyNumLock,
		"PrintScreen":  glfw.KeyPrintScreen,
		"Pause":        glfw.KeyPause,
		"LeftShift":    glfw.KeyLeftShift,
		"LeftControl":  glfw.KeyLeftControl,
		"LeftAlt":      glfw.KeyLeftAlt,
		"LeftSuper":    glfw.KeyLeftSuper,
		"RightShift":   glfw.KeyRightShift,
		"RightControl": glfw.KeyRightControl,
		"RightAlt":     glfw.KeyRightAlt,
		"RightSuper":   glfw.KeyRightSuper,
		"Menu":         glfw.KeyMenu,
		"Last":         glfw.KeyLast,
	}
	nameToJSKeyCode = map[string]string{
		"Comma":        "Comma",
		"Period":       "Period",
		"LeftAlt":      "AltLeft",
		"RightAlt":     "AltRight",
		"CapsLock":     "CapsLock",
		"LeftControl":  "ControlLeft",
		"RightControl": "ControlRight",
		"LeftShift":    "ShiftLeft",
		"RightShift":   "ShiftRight",
		"Enter":        "Enter",
		"Space":        "Space",
		"Tab":          "Tab",
		"Delete":       "Delete",
		"End":          "End",
		"Home":         "Home",
		"Insert":       "Insert",
		"PageDown":     "PageDown",
		"PageUp":       "PageUp",
		"Down":         "ArrowDown",
		"Left":         "ArrowLeft",
		"Right":        "ArrowRight",
		"Up":           "ArrowUp",
		"Escape":       "Escape",
		"Backspace":    "Backspace",
		"Apostrophe":   "Quote",
		"Minus":        "Minus",
		"Slash":        "Slash",
		"Semicolon":    "Semicolon",
		"Equal":        "Equal",
		"LeftBracket":  "BracketLeft",
		"Backslash":    "Backslash",
		"RightBracket": "BracketRight",
		"GraveAccent":  "Backquote",
		"NumLock":      "NumLock",
		"Pause":        "Pause",
		"PrintScreen":  "PrintScreen",
		"ScrollLock":   "ScrollLock",
		"Menu":         "ContextMenu",
	}
	// ASCII: 0 - 9
	for c := '0'; c <= '9'; c++ {
		nameToGLFWKeys[string(c)] = glfw.Key0 + glfw.Key(c) - '0'
		nameToJSKeyCode[string(c)] = "Digit" + string(c)
	}
	// ASCII: A - Z
	for c := 'A'; c <= 'Z'; c++ {
		nameToGLFWKeys[string(c)] = glfw.KeyA + glfw.Key(c) - 'A'
		nameToJSKeyCode[string(c)] = "Key" + string(c)
	}
	// Function keys
	for i := 1; i <= 12; i++ {
		name := "F" + strconv.Itoa(i)
		nameToGLFWKeys[name] = glfw.KeyF1 + glfw.Key(i) - 1
		nameToJSKeyCode[name] = name
	}
	// Numpad
	// https://www.w3.org/TR/uievents-code/#key-numpad-section
	for c := '0'; c <= '9'; c++ {
		name := "KP" + string(c)
		nameToGLFWKeys[name] = glfw.KeyKP0 + glfw.Key(c) - '0'
		nameToJSKeyCode[name] = "Numpad" + string(c)
	}

	nameToGLFWKeys["KPDecimal"] = glfw.KeyKPDecimal
	nameToGLFWKeys["KPDivide"] = glfw.KeyKPDivide
	nameToGLFWKeys["KPMultiply"] = glfw.KeyKPMultiply
	nameToGLFWKeys["KPSubtract"] = glfw.KeyKPSubtract
	nameToGLFWKeys["KPAdd"] = glfw.KeyKPAdd
	nameToGLFWKeys["KPEnter"] = glfw.KeyKPEnter
	nameToGLFWKeys["KPEqual"] = glfw.KeyKPEqual

	nameToJSKeyCode["KPDecimal"] = "NumpadDecimal"
	nameToJSKeyCode["KPDivide"] = "NumpadDivide"
	nameToJSKeyCode["KPMultiply"] = "NumpadMultiply"
	nameToJSKeyCode["KPSubtract"] = "NumpadSubtract"
	nameToJSKeyCode["KPAdd"] = "NumpadAdd"
	nameToJSKeyCode["KPEnter"] = "NumpadEnter"
	nameToJSKeyCode["KPEqual"] = "NumpadEqual"
}

func init() {
	// TODO: How should we treat modifier keys? Now 'left' modifier keys are available.
	keyCodeToNameEdge = map[int]string{
		0xbc: "Comma",
		0xbe: "Period",
		0x12: "LeftAlt",
		0x14: "CapsLock",
		0x11: "LeftControl",
		0x10: "LeftShift",
		0x0D: "Enter",
		0x20: "Space",
		0x09: "Tab",
		0x2E: "Delete",
		0x23: "End",
		0x24: "Home",
		0x2D: "Insert",
		0x22: "PageDown",
		0x21: "PageUp",
		0x28: "Down",
		0x25: "Left",
		0x27: "Right",
		0x26: "Up",
		0x1B: "Escape",
		0xde: "Apostrophe",
		0xbd: "Minus",
		0xbf: "Slash",
		0xba: "Semicolon",
		0xbb: "Equal",
		0xdb: "LeftBracket",
		0xdc: "Backslash",
		0xdd: "RightBracket",
		0xc0: "GraveAccent",
		0x08: "Backspace",
		0x90: "NumLock",
		0x6e: "KPDecimal",
		0x6f: "KPDivide",
		0x6a: "KPMultiply",
		0x6d: "KPSubtract",
		0x6b: "KPAdd",
		0x13: "Pause",
		0x91: "ScrollLock",
		0x5d: "Menu",

		// On Edge, this key does not work. PrintScreen works only on keyup event.
		// 0x2C: "PrintScreen",

		// On Edge, it is impossible to tell KPEnter and Enter / KPEqual and Equal.
		// 0x0d: "KPEnter",
		// 0x0c: "KPEqual",
	}
	// ASCII: 0 - 9
	for c := '0'; c <= '9'; c++ {
		keyCodeToNameEdge[int(c)] = string(c)
	}
	// ASCII: A - Z
	for c := 'A'; c <= 'Z'; c++ {
		keyCodeToNameEdge[int(c)] = string(c)
	}
	// Function keys
	for i := 1; i <= 12; i++ {
		keyCodeToNameEdge[0x70+i-1] = "F" + strconv.Itoa(i)
	}
	// Numpad keys
	for c := '0'; c <= '9'; c++ {
		keyCodeToNameEdge[0x60+int(c-'0')] = "KP" + string(c)
	}
}

const ebitenKeysTmpl = `{{.License}}

{{.DoNotEdit}}

package ebiten

import (
	"strings"

	"github.com/hajimehoshi/ebiten/internal/driver"
)

// A Key represents a keyboard key.
// These keys represent pysical keys of US keyboard.
// For example, KeyQ represents Q key on US keyboards and ' (quote) key on Dvorak keyboards.
type Key int

// Keys.
const (
{{range $index, $name := .EbitenKeyNamesWithoutMods}}Key{{$name}} Key = Key(driver.Key{{$name}})
{{end}}	KeyAlt Key = Key(driver.KeyReserved0)
	KeyControl Key = Key(driver.KeyReserved1)
	KeyShift Key = Key(driver.KeyReserved2)
	KeyMax Key = KeyShift
)

func (k Key) isValid() bool {
	switch k {
	{{range $name := .EbitenKeyNames}}case Key{{$name}}:
		return true
	{{end}}
	default:
		return false
	}
}

// String returns a string representing the key.
//
// If k is an undefined key, String returns an empty string.
func (k Key) String() string {
	switch k {
	{{range $name := .EbitenKeyNames}}case Key{{$name}}:
		return {{$name | printf "%q"}}
	{{end}}}
	return ""
}

func keyNameToKey(name string) (Key, bool) {
	switch strings.ToLower(name) {
	{{range $name := .EbitenKeyNames}}case {{$name | printf "%q" | ToLower}}:
		return Key{{$name}}, true
	{{end}}}
	return 0, false
}
`

const driverKeysTmpl = `{{.License}}

{{.DoNotEdit}}

package driver

type Key int

const (
{{range $index, $name := .DriverKeyNames}}Key{{$name}}{{if eq $index 0}} Key = iota{{end}}
{{end}}	KeyReserved0
	KeyReserved1
	KeyReserved2
)
`

const eventKeysTmpl = `{{.License}}

{{.DoNotEdit}}

package event

import (
	"github.com/hajimehoshi/ebiten/internal/driver"
)

type Key int

const (
{{range $index, $name := .DriverKeyNames}}Key{{$name}} Key = Key(driver.Key{{$name}})
{{end}}
)
`

const uidriverGlfwKeysTmpl = `{{.License}}

{{.DoNotEdit}}

{{.BuildTag}}

package glfw

import (
	"github.com/hajimehoshi/ebiten/internal/driver"
	"github.com/hajimehoshi/ebiten/internal/glfw"
)

var glfwKeyCodeToKey = map[glfw.Key]driver.Key{
{{range $index, $name := .DriverKeyNames}}glfw.Key{{$name}}: driver.Key{{$name}},
{{end}}
}
`

const uidriverJsKeysTmpl = `{{.License}}

{{.DoNotEdit}}

{{.BuildTag}}

package js

import (
	"github.com/hajimehoshi/ebiten/internal/driver"
)

var keyToCode = map[driver.Key]string{
{{range $name, $code := .NameToJSKeyCode}}driver.Key{{$name}}: {{$code | printf "%q"}},
{{end}}
}

var keyCodeToKeyEdge = map[int]driver.Key{
{{range $code, $name := .KeyCodeToNameEdge}}{{$code}}: driver.Key{{$name}},
{{end}}
}
`

const glfwKeysTmpl = `{{.License}}

{{.DoNotEdit}}

{{.BuildTag}}

package glfw

const (
{{range $name, $key := .NameToGLFWKeys}}Key{{$name}} = Key({{$key}})
{{end}}
)
`

func digitKey(name string) int {
	if len(name) != 1 {
		return -1
	}
	c := name[0]
	if c < '0' || '9' < c {
		return -1
	}
	return int(c - '0')
}

func alphabetKey(name string) rune {
	if len(name) != 1 {
		return -1
	}
	c := rune(name[0])
	if c < 'A' || 'Z' < c {
		return -1
	}
	return c
}

func functionKey(name string) int {
	if len(name) < 2 {
		return -1
	}
	if name[0] != 'F' {
		return -1
	}
	i, err := strconv.Atoi(name[1:])
	if err != nil {
		return -1
	}
	return i
}

func keyNamesLess(k []string) func(i, j int) bool {
	return func(i, j int) bool {
		k0, k1 := k[i], k[j]
		d0, d1 := digitKey(k0), digitKey(k1)
		a0, a1 := alphabetKey(k0), alphabetKey(k1)
		f0, f1 := functionKey(k0), functionKey(k1)
		if d0 != -1 {
			if d1 != -1 {
				return d0 < d1
			}
			return true
		}
		if a0 != -1 {
			if d1 != -1 {
				return false
			}
			if a1 != -1 {
				return a0 < a1
			}
			return true
		}
		if d1 != -1 {
			return false
		}
		if a1 != -1 {
			return false
		}
		if f0 != -1 && f1 != -1 {
			return f0 < f1
		}
		return k0 < k1
	}
}

const license = `// Copyright 2013 The Ebiten Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
`

func main() {
	// Follow the standard comment rule (https://golang.org/s/generatedcode).
	doNotEdit := "// Code generated by genkeys.go using 'go generate'. DO NOT EDIT."

	ebitenKeyNames := []string{}
	ebitenKeyNamesWithoutMods := []string{}
	driverKeyNames := []string{}
	for name := range nameToJSKeyCode {
		driverKeyNames = append(driverKeyNames, name)
		if !strings.HasSuffix(name, "Alt") && !strings.HasSuffix(name, "Control") && !strings.HasSuffix(name, "Shift") {
			ebitenKeyNames = append(ebitenKeyNames, name)
			ebitenKeyNamesWithoutMods = append(ebitenKeyNamesWithoutMods, name)
			continue
		}
		if name == "LeftAlt" {
			ebitenKeyNames = append(ebitenKeyNames, "Alt")
			continue
		}
		if name == "LeftControl" {
			ebitenKeyNames = append(ebitenKeyNames, "Control")
			continue
		}
		if name == "LeftShift" {
			ebitenKeyNames = append(ebitenKeyNames, "Shift")
			continue
		}
	}

	sort.Slice(ebitenKeyNames, keyNamesLess(ebitenKeyNames))
	sort.Slice(ebitenKeyNamesWithoutMods, keyNamesLess(ebitenKeyNamesWithoutMods))
	sort.Slice(driverKeyNames, keyNamesLess(driverKeyNames))

	for path, tmpl := range map[string]string{
		filepath.Join("event", "keys.go"):                        eventKeysTmpl,
		filepath.Join("internal", "driver", "keys.go"):           driverKeysTmpl,
		filepath.Join("internal", "glfw", "keys.go"):             glfwKeysTmpl,
		filepath.Join("internal", "uidriver", "glfw", "keys.go"): uidriverGlfwKeysTmpl,
		filepath.Join("internal", "uidriver", "js", "keys.go"):   uidriverJsKeysTmpl,
		filepath.Join("keys.go"):                                 ebitenKeysTmpl,
	} {
		f, err := os.Create(path)
		if err != nil {
			log.Fatal(err)
		}
		defer f.Close()

		funcs := template.FuncMap{
			"ToLower": strings.ToLower,
		}
		tmpl, err := template.New(path).Funcs(funcs).Parse(tmpl)
		if err != nil {
			log.Fatal(err)
		}

		// The build tag can't be included in the templates because of `go vet`.
		// Pass the build tag and extract this in the template to make `go vet` happy.
		buildTag := ""
		switch path {
		case "internal/uidriver/glfw/keys.go":
			buildTag = "// +build darwin freebsd linux windows" +
				"\n// +build !js" +
				"\n// +build !android" +
				"\n// +build !ios"
		case "internal/uidriver/js/keys.go":
			buildTag = "// +build js"
		}
		// NOTE: According to godoc, maps are automatically sorted by key.
		if err := tmpl.Execute(f, struct {
			License                   string
			DoNotEdit                 string
			BuildTag                  string
			NameToJSKeyCode           map[string]string
			KeyCodeToNameEdge         map[int]string
			EbitenKeyNames            []string
			EbitenKeyNamesWithoutMods []string
			DriverKeyNames            []string
			NameToGLFWKeys            map[string]glfw.Key
		}{
			License:                   license,
			DoNotEdit:                 doNotEdit,
			BuildTag:                  buildTag,
			NameToJSKeyCode:           nameToJSKeyCode,
			KeyCodeToNameEdge:         keyCodeToNameEdge,
			EbitenKeyNames:            ebitenKeyNames,
			EbitenKeyNamesWithoutMods: ebitenKeyNamesWithoutMods,
			DriverKeyNames:            driverKeyNames,
			NameToGLFWKeys:            nameToGLFWKeys,
		}); err != nil {
			log.Fatal(err)
		}
	}
}