-
-
Notifications
You must be signed in to change notification settings - Fork 1.7k
/
action.go
90 lines (77 loc) · 1.91 KB
/
action.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package ui
import (
"sort"
"github.com/derailed/k9s/internal/model"
"github.com/gdamore/tcell/v2"
"github.com/rs/zerolog/log"
)
type (
// ActionHandler handles a keyboard command.
ActionHandler func(*tcell.EventKey) *tcell.EventKey
// KeyAction represents a keyboard action.
KeyAction struct {
Description string
Action ActionHandler
Visible bool
Shared bool
}
// KeyActions tracks mappings between keystrokes and actions.
KeyActions map[tcell.Key]KeyAction
)
// NewKeyAction returns a new keyboard action.
func NewKeyAction(d string, a ActionHandler, display bool) KeyAction {
return KeyAction{Description: d, Action: a, Visible: display}
}
// NewSharedKeyAction returns a new shared keyboard action.
func NewSharedKeyAction(d string, a ActionHandler, display bool) KeyAction {
return KeyAction{Description: d, Action: a, Visible: display, Shared: true}
}
// Add sets up keyboard action listener.
func (a KeyActions) Add(aa KeyActions) {
for k, v := range aa {
a[k] = v
}
}
// Clear remove all actions.
func (a KeyActions) Clear() {
for k := range a {
delete(a, k)
}
}
// Set replace actions with new ones.
func (a KeyActions) Set(aa KeyActions) {
for k, v := range aa {
a[k] = v
}
}
// Delete deletes actions by the given keys.
func (a KeyActions) Delete(kk ...tcell.Key) {
for _, k := range kk {
delete(a, k)
}
}
// Hints returns a collection of hints.
func (a KeyActions) Hints() model.MenuHints {
kk := make([]int, 0, len(a))
for k := range a {
if !a[k].Shared {
kk = append(kk, int(k))
}
}
sort.Ints(kk)
hh := make(model.MenuHints, 0, len(kk))
for _, k := range kk {
if name, ok := tcell.KeyNames[tcell.Key(k)]; ok {
hh = append(hh,
model.MenuHint{
Mnemonic: name,
Description: a[tcell.Key(k)].Description,
Visible: a[tcell.Key(k)].Visible,
},
)
} else {
log.Error().Msgf("Unable to locate KeyName for %#v", k)
}
}
return hh
}