|
|
@@ -0,0 +1,57 @@
|
|
|
+package picker
|
|
|
+
|
|
|
+import (
|
|
|
+ "github.com/rivo/tview"
|
|
|
+ "github.com/sahilm/fuzzy"
|
|
|
+)
|
|
|
+
|
|
|
+type Picker struct {
|
|
|
+ *tview.Flex
|
|
|
+ Input *tview.InputField
|
|
|
+ List *tview.List
|
|
|
+
|
|
|
+ app *tview.Application
|
|
|
+ items Items
|
|
|
+}
|
|
|
+
|
|
|
+func New(app *tview.Application, items Items) *Picker {
|
|
|
+ p := &Picker{
|
|
|
+ Flex: tview.NewFlex(),
|
|
|
+ Input: tview.NewInputField(),
|
|
|
+ List: tview.NewList(),
|
|
|
+
|
|
|
+ app: app,
|
|
|
+ items: items,
|
|
|
+ }
|
|
|
+
|
|
|
+ // Render all of the items initially.
|
|
|
+ p.changed("")
|
|
|
+
|
|
|
+ p.Input.SetChangedFunc(p.changed)
|
|
|
+ p.List.ShowSecondaryText(false).SetFocusFunc(func() {
|
|
|
+ app.SetFocus(p.Input)
|
|
|
+ })
|
|
|
+ p.Flex.
|
|
|
+ SetDirection(tview.FlexRow).
|
|
|
+ AddItem(p.Input, 3, 1, true).
|
|
|
+ AddItem(p.List, 0, 1, false)
|
|
|
+ return p
|
|
|
+}
|
|
|
+
|
|
|
+func (p *Picker) changed(text string) {
|
|
|
+ var fuzzied Items
|
|
|
+ if text == "" {
|
|
|
+ fuzzied = append(fuzzied, p.items...)
|
|
|
+ } else {
|
|
|
+ matches := fuzzy.FindFrom(text, p.items)
|
|
|
+ for _, match := range matches {
|
|
|
+ fuzzied = append(fuzzied, p.items[match.Index])
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ p.List.Clear()
|
|
|
+
|
|
|
+ for _, item := range fuzzied {
|
|
|
+ p.List.AddItem(item.text, "", 0, item.selected)
|
|
|
+ }
|
|
|
+}
|