package ui

import (
	"strconv"

	"github.com/gdamore/tcell/v2"
)

type keyData struct {
	name string
	key  tcell.Key
}

type KeyMaps map[string]keyData

func NewKeyData(chStr string) keyData {
	var (
		name string
		key  tcell.Key
	)

	if len(chStr) > 1 && chStr[0] == 'F' {
		name = chStr
		fNum, err := strconv.Atoi(chStr[1:])
		if err != nil {
			panic(err)
		}
		key = tcell.KeyF1 + tcell.Key(fNum-1)
	} else {
		switch chStr[0] {
		case ' ':
			name = "C-Space"
			key = tcell.KeyCtrlSpace
		case '\\':
			name = "C-\\"
			key = tcell.KeyCtrlBackslash
		case ']':
			name = "C-]"
			key = tcell.KeyCtrlRightSq
		case '^':
			name = "C-^"
			key = tcell.KeyCtrlCarat
		case '_':
			name = "C-_"
			key = tcell.KeyCtrlUnderscore
		default:
			// This should be a to z
			name = "C-" + chStr
			key = tcell.KeyCtrlA + tcell.Key(chStr[0]-'a')
		}
	}

	return keyData{
		name: name,
		key:  key,
	}
}

func (k keyData) GetName() string {
	return k.name
}

func (k keyData) GetKey() tcell.Key {
	return k.key
}