package config import ( "bytes" "errors" "fmt" "io" "strings" "github.com/gdamore/tcell" ) type KeyStroke struct { Key tcell.Key Rune rune } type Binding struct { Output []KeyStroke Input []KeyStroke } type KeyBindings struct { bindings []*Binding // If false, disable global keybindings in this context Globals bool // Which key opens the ex line (default is :) ExKey KeyStroke } const ( BINDING_FOUND = iota BINDING_INCOMPLETE BINDING_NOT_FOUND ) type BindingSearchResult int func NewKeyBindings() *KeyBindings { return &KeyBindings{ ExKey: KeyStroke{tcell.KeyRune, ':'}, Globals: true, } } func (bindings *KeyBindings) Add(binding *Binding) { // TODO: Search for conflicts? bindings.bindings = append(bindings.bindings, binding) } func (bindings *KeyBindings) GetBinding( input []KeyStroke) (BindingSearchResult, []KeyStroke) { incomplete := false // TODO: This could probably be a sorted list to speed things up // TODO: Deal with bindings that share a prefix for _, binding := range bindings.bindings { if len(binding.Input) < len(input) { continue } for i, stroke := range input { if stroke.Key != binding.Input[i].Key { goto next } if stroke.Key == tcell.KeyRune && stroke.Rune != binding.Input[i].Rune { goto next } } if len(binding.Input) != len(input) { incomplete = true } else { return BINDING_FOUND, binding.Output } next: } if incomplete { return BINDING_INCOMPLETE, nil } return BINDING_NOT_FOUND, nil } var ( keyNames map[string]KeyStroke ) func ParseKeyStrokes(keystrokes string) ([]KeyStroke, error) { var strokes []KeyStroke buf := bytes.NewBufferString(keystrokes) for { tok, _, err := buf.ReadRune() if err == io.EOF { break } else if err != nil { return nil, err } // TODO: make it possible to bind to < or > themselves (and default to // switching accounts) switch tok { case '<': name, err := buf.ReadString(byte('>')) if err == io.EOF { return nil, errors.New("Expecting '>'") } else if err != nil { return nil, err } else if name == ">" { return nil, errors.New("Expected a key name") } name = name[:len(name)-1] if key, ok := keyNames[strings.ToLower(name)]; ok { strokes = append(strokes, key) } else { return nil, errors.New(fmt.Sprintf("Unknown key '%s'", name)) } case '>': return nil, errors.New("Found '>' without '<'") default: strokes = append(strokes, KeyStroke{ Key: tcell.KeyRune, Rune: tok, }) } } return strokes, nil } func ParseBinding(input, output string) (*Binding, error) { in, err := ParseKeyStrokes(input) if err != nil { return nil, err } out, err := ParseKeyStrokes(output) if err != nil { return nil, err } return &Binding{ Input: in, Output: out, }, nil } func init() { keyNames = make(map[string]KeyStroke) keyNames["space"] = KeyStroke{tcell.KeyRune, ' '} keyNames["enter"] = KeyStroke{tcell.KeyEnter, 0} keyNames["up"] = KeyStroke{tcell.KeyUp, 0} keyNames["down"] = KeyStroke{tcell.KeyDown, 0} keyNames["right"] = KeyStroke{tcell.KeyRight, 0} keyNames["left"] = KeyStroke{tcell.KeyLeft, 0} keyNames["upleft"] = KeyStroke{tcell.KeyUpLeft, 0} keyNames["upright"] = KeyStroke{tcell.KeyUpRight, 0} keyNames["downleft"] = KeyStroke{tcell.KeyDownLeft, 0} keyNames["downright"] = KeyStroke{tcell.KeyDownRight, 0} keyNames["center"] = KeyStroke{tcell.KeyCenter, 0} keyNames["pgup"] = KeyStroke{tcell.KeyPgUp, 0} keyNames["pgdn"] = KeyStroke{tcell.KeyPgDn, 0} keyNames["home"] = KeyStroke{tcell.KeyHome, 0} keyNames["end"] = KeyStroke{tcell.KeyEnd, 0} keyNames["insert"] = KeyStroke{tcell.KeyInsert, 0} keyNames["delete"] = KeyStroke{tcell.KeyDelete, 0} keyNames["help"] = KeyStroke{tcell.KeyHelp, 0} keyNames["exit"] = KeyStroke{tcell.KeyExit, 0} keyNames["clear"] = KeyStroke{tcell.KeyClear, 0} keyNames["cancel"] = KeyStroke{tcell.KeyCancel, 0} keyNames["print"] = KeyStroke{tcell.KeyPrint, 0} keyNames["pause"] = KeyStroke{tcell.KeyPause, 0} keyNames["backtab"] = KeyStroke{tcell.KeyBacktab, 0} keyNames["f1"] = KeyStroke{tcell.KeyF1, 0} keyNames["f2"] = KeyStroke{tcell.KeyF2, 0} keyNames["f3"] = KeyStroke{tcell.KeyF3, 0} keyNames["f4"] = KeyStroke{tcell.KeyF4, 0} keyNames["f5"] = KeyStroke{tcell.KeyF5, 0} keyNames["f6"] = KeyStroke{tcell.KeyF6, 0} keyNames["f7"] = KeyStroke{tcell.KeyF7, 0} keyNames["f8"] = KeyStroke{tcell.KeyF8, 0} keyNames["f9"] = KeyStroke{tcell.KeyF9, 0} keyNames["f10"] = KeyStroke{tcell.KeyF10, 0} keyNames["f11"] = KeyStroke{tcell.KeyF11, 0} keyNames["f12"] = KeyStroke{tcell.KeyF12, 0} keyNames["f13"] = KeyStroke{tcell.KeyF13, 0} keyNames["f14"] = KeyStroke{tcell.KeyF14, 0} keyNames["f15"] = KeyStroke{tcell.KeyF15, 0} keyNames["f16"] = KeyStroke{tcell.KeyF16, 0} keyNames["f17"] = KeyStroke{tcell.KeyF17, 0} keyNames["f18"] = KeyStroke{tcell.KeyF18, 0} keyNames["f19"] = KeyStroke{tcell.KeyF19, 0} keyNames["f20"] = KeyStroke{tcell.KeyF20, 0} keyNames["f21"] = KeyStroke{tcell.KeyF21, 0} keyNames["f22"] = KeyStroke{tcell.KeyF22, 0} keyNames["f23"] = KeyStroke{tcell.KeyF23, 0} keyNames["f24"] = KeyStroke{tcell.KeyF24, 0} keyNames["f25"] = KeyStroke{tcell.KeyF25, 0} keyNames["f26"] = KeyStroke{tcell.KeyF26, 0} keyNames["f27"] = KeyStroke{tcell.KeyF27, 0} keyNames["f28"] = KeyStroke{tcell.KeyF28, 0} keyNames["f29"] = KeyStroke{tcell.KeyF29, 0} keyNames["f30"] = KeyStroke{tcell.KeyF30, 0} keyNames["f31"] = KeyStroke{tcell.KeyF31, 0} keyNames["f32"] = KeyStroke{tcell.KeyF32, 0} keyNames["f33"] = KeyStroke{tcell.KeyF33, 0} keyNames["f34"] = KeyStroke{tcell.KeyF34, 0} keyNames["f35"] = KeyStroke{tcell.KeyF35, 0} keyNames["f36"] = KeyStroke{tcell.KeyF36, 0} keyNames["f37"] = KeyStroke{tcell.KeyF37, 0} keyNames["f38"] = KeyStroke{tcell.KeyF38, 0} keyNames["f39"] = KeyStroke{tcell.KeyF39, 0} keyNames["f40"] = KeyStroke{tcell.KeyF40, 0} keyNames["f41"] = KeyStroke{tcell.KeyF41, 0} keyNames["f42"] = KeyStroke{tcell.KeyF42, 0} keyNames["f43"] = KeyStroke{tcell.KeyF43, 0} keyNames["f44"] = KeyStroke{tcell.KeyF44, 0} keyNames["f45"] = KeyStroke{tcell.KeyF45, 0} keyNames["f46"] = KeyStroke{tcell.KeyF46, 0} keyNames["f47"] = KeyStroke{tcell.KeyF47, 0} keyNames["f48"] = KeyStroke{tcell.KeyF48, 0} keyNames["f49"] = KeyStroke{tcell.KeyF49, 0} keyNames["f50"] = KeyStroke{tcell.KeyF50, 0} keyNames["f51"] = KeyStroke{tcell.KeyF51, 0} keyNames["f52"] = KeyStroke{tcell.KeyF52, 0} keyNames["f53"] = KeyStroke{tcell.KeyF53, 0} keyNames["f54"] = KeyStroke{tcell.KeyF54, 0} keyNames["f55"] = KeyStroke{tcell.KeyF55, 0} keyNames["f56"] = KeyStroke{tcell.KeyF56, 0} keyNames["f57"] = KeyStroke{tcell.KeyF57, 0} keyNames["f58"] = KeyStroke{tcell.KeyF58, 0} keyNames["f59"] = KeyStroke{tcell.KeyF59, 0} keyNames["f60"] = KeyStroke{tcell.KeyF60, 0} keyNames["f61"] = KeyStroke{tcell.KeyF61, 0} keyNames["f62"] = KeyStroke{tcell.KeyF62, 0} keyNames["f63"] = KeyStroke{tcell.KeyF63, 0} keyNames["f64"] = KeyStroke{tcell.KeyF64, 0} keyNames["c-space"] = KeyStroke{tcell.KeyCtrlSpace, 0} keyNames["c-a"] = KeyStroke{tcell.KeyCtrlA, 0} keyNames["c-b"] = KeyStroke{tcell.KeyCtrlB, 0} keyNames["c-c"] = KeyStroke{tcell.KeyCtrlC, 0} keyNames["c-d"] = KeyStroke{tcell.KeyCtrlD, 0} keyNames["c-e"] = KeyStroke{tcell.KeyCtrlE, 0} keyNames["c-f"] = KeyStroke{tcell.KeyCtrlF, 0} keyNames["c-g"] = KeyStroke{tcell.KeyCtrlG, 0} keyNames["c-h"] = KeyStroke{tcell.KeyCtrlH, 0} keyNames["c-i"] = KeyStroke{tcell.KeyCtrlI, 0} keyNames["c-j"] = KeyStroke{tcell.KeyCtrlJ, 0} keyNames["c-k"] = KeyStroke{tcell.KeyCtrlK, 0} keyNames["c-l"] = KeyStroke{tcell.KeyCtrlL, 0} keyNames["c-m"] = KeyStroke{tcell.KeyCtrlM, 0} keyNames["c-n"] = KeyStroke{tcell.KeyCtrlN, 0} keyNames["c-o"] = KeyStroke{tcell.KeyCtrlO, 0} keyNames["c-p"] = KeyStroke{tcell.KeyCtrlP, 0} keyNames["c-q"] = KeyStroke{tcell.KeyCtrlQ, 0} keyNames["c-r"] = KeyStroke{tcell.KeyCtrlR, 0} keyNames["c-s"] = KeyStroke{tcell.KeyCtrlS, 0} keyNames["c-t"] = KeyStroke{tcell.KeyCtrlT, 0} keyNames["c-u"] = KeyStroke{tcell.KeyCtrlU, 0} keyNames["c-v"] = KeyStroke{tcell.KeyCtrlV, 0} keyNames["c-w"] = KeyStroke{tcell.KeyCtrlW, 0} keyNames["c-x"] = KeyStroke{tcell.KeyCtrlX, 0} keyNames["c-y"] = KeyStroke{tcell.KeyCtrlY, 0} keyNames["c-z"] = KeyStroke{tcell.KeyCtrlZ, 0} keyNames["c-]"] = KeyStroke{tcell.KeyCtrlLeftSq, 0} keyNames["c-\\"] = KeyStroke{tcell.KeyCtrlBackslash, 0} keyNames["c-["] = KeyStroke{tcell.KeyCtrlRightSq, 0} keyNames["c-^"] = KeyStroke{tcell.KeyCtrlCarat, 0} keyNames["c-_"] = KeyStroke{tcell.KeyCtrlUnderscore, 0} keyNames["NUL"] = KeyStroke{tcell.KeyNUL, 0} keyNames["SOH"] = KeyStroke{tcell.KeySOH, 0} keyNames["STX"] = KeyStroke{tcell.KeySTX, 0} keyNames["ETX"] = KeyStroke{tcell.KeyETX, 0} keyNames["EOT"] = KeyStroke{tcell.KeyEOT, 0} keyNames["ENQ"] = KeyStroke{tcell.KeyENQ, 0} keyNames["ACK"] = KeyStroke{tcell.KeyACK, 0} keyNames["BEL"] = KeyStroke{tcell.KeyBEL, 0} keyNames["BS"] = KeyStroke{tcell.KeyBS, 0} keyNames["TAB"] = KeyStroke{tcell.KeyTAB, 0} keyNames["LF"] = KeyStroke{tcell.KeyLF, 0} keyNames["VT"] = KeyStroke{tcell.KeyVT, 0} keyNames["FF"] = KeyStroke{tcell.KeyFF, 0} keyNames["CR"] = KeyStroke{tcell.KeyCR, 0} keyNames["SO"] = KeyStroke{tcell.KeySO, 0} keyNames["SI"] = KeyStroke{tcell.KeySI, 0} keyNames["DLE"] = KeyStroke{tcell.KeyDLE, 0} keyNames["DC1"] = KeyStroke{tcell.KeyDC1, 0} keyNames["DC2"] = KeyStroke{tcell.KeyDC2, 0} keyNames["DC3"] = KeyStroke{tcell.KeyDC3, 0} keyNames["DC4"] = KeyStroke{tcell.KeyDC4, 0} keyNames["NAK"] = KeyStroke{tcell.KeyNAK, 0} keyNames["SYN"] = KeyStroke{tcell.KeySYN, 0} keyNames["ETB"] = KeyStroke{tcell.KeyETB, 0} keyNames["CAN"] = KeyStroke{tcell.KeyCAN, 0} keyNames["EM"] = KeyStroke{tcell.KeyEM, 0} keyNames["SUB"] = KeyStroke{tcell.KeySUB, 0} keyNames["ESC"] = KeyStroke{tcell.KeyESC, 0} keyNames["FS"] = KeyStroke{tcell.KeyFS, 0} keyNames["GS"] = KeyStroke{tcell.KeyGS, 0} keyNames["RS"] = KeyStroke{tcell.KeyRS, 0} keyNames["US"] = KeyStroke{tcell.KeyUS, 0} keyNames["DEL"] = KeyStroke{tcell.KeyDEL, 0} }