-
Notifications
You must be signed in to change notification settings - Fork 0
/
mode_autocomplete.go
109 lines (83 loc) · 2.7 KB
/
mode_autocomplete.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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package shell
import (
"strings"
"github.com/DomBlack/bubble-shell/pkg/tui/autocomplete"
"github.com/charmbracelet/bubbles/key"
tea "github.com/charmbracelet/bubbletea"
)
type AutoCompleteMode struct{}
var _ Mode = (*AutoCompleteMode)(nil)
func (a *AutoCompleteMode) Enter(m Model) (Model, tea.Cmd) {
m.input.Blur()
return m, m.autocomplete.AutoComplete(m.input.Value(), m.input.Position())
}
func (a *AutoCompleteMode) Leave(m Model) (Model, tea.Cmd) {
return m, m.autocomplete.Clear()
}
func (a *AutoCompleteMode) Update(m Model, msg tea.Msg) (Model, tea.Cmd) {
switch msg := msg.(type) {
case autocomplete.SingleAutoCompleteOptionMsg:
if m.id.Matches(msg) {
return a.AcceptOption(m)
}
case tea.KeyMsg:
switch {
case key.Matches(msg, m.cfg.KeyMap.AutoComplete):
return m, m.autocomplete.NextResult()
case key.Matches(msg, m.cfg.KeyMap.PreviousAutoComplete):
return m, m.autocomplete.PreviousResult()
case key.Matches(msg, m.cfg.KeyMap.Up):
return m, m.autocomplete.PreviousRow()
case key.Matches(msg, m.cfg.KeyMap.Down):
return m, m.autocomplete.NextRow()
case key.Matches(msg, m.cfg.KeyMap.Left):
return m, m.autocomplete.PreviousColumn()
case key.Matches(msg, m.cfg.KeyMap.Right):
return m, m.autocomplete.NextColumn()
case key.Matches(msg, m.cfg.KeyMap.Cancel):
return m, m.Enter(&CommandEntryMode{KeepInputContent: true})
case key.Matches(msg, m.cfg.KeyMap.ExecuteCommand):
return a.AcceptOption(m)
case msg.Type == tea.KeySpace:
return a.AcceptOption(m)
default:
// default is to exit the mode and re-send the key message
return m, tea.Sequence(
m.Enter(&CommandEntryMode{KeepInputContent: true}),
func() tea.Msg { return msg }, // re-send the key message
)
}
}
return m, nil
}
func (a *AutoCompleteMode) AdditionalView(m Model) string {
return m.autocomplete.View()
}
func (a *AutoCompleteMode) AcceptOption(m Model) (Model, tea.Cmd) {
if suggestion := m.autocomplete.Accept(); suggestion != "" {
input := m.input.Value()
// Cut off the last word so we can replace it with the autocomplete
if idx := strings.LastIndexByte(input, ' '); idx == -1 {
input = ""
} else {
input = input[:idx+1]
}
m.input.SetValue(input + suggestion + " ")
m.input.CursorEnd()
}
return m, m.Enter(&CommandEntryMode{KeepInputContent: true})
}
func (a *AutoCompleteMode) ShortHelp(m Model, keyMap KeyMap) []key.Binding {
execute := keyMap.ExecuteCommand
execute.SetHelp(execute.Help().Key, "Accept")
return []key.Binding{
keyMap.AutoComplete, keyMap.PreviousAutoComplete,
execute,
keyMap.Cancel,
}
}
func (a *AutoCompleteMode) FullHelp(m Model, keyMap KeyMap) [][]key.Binding {
return [][]key.Binding{
a.ShortHelp(m, keyMap),
}
}