init

internal/ui/search.go

@@ -0,0 +1,66 @@
+package app
+
+import (
+	"strings"
+
+	"charm.land/bubbles/v2/key"
+	tea "charm.land/bubbletea/v2"
+
+	"github.com/anotherhadi/proton-vpn-tui/internal/backend"
+	"github.com/anotherhadi/proton-vpn-tui/internal/config"
+)
+
+var searchEnterKey = key.NewBinding(key.WithKeys("enter"))
+
+func (m Model) updateSearch(msg tea.KeyPressMsg) (Model, tea.Cmd) {
+	gk := config.Global.Keybindings.Global
+
+	switch {
+	case key.Matches(msg, gk.Escape):
+		return m.exitSearch(true)
+	case key.Matches(msg, searchEnterKey):
+		return m.submitSearch()
+	}
+
+	prev := m.searchInput.Value()
+	var cmd tea.Cmd
+	m.searchInput, cmd = m.searchInput.Update(msg)
+	if m.searchInput.Value() != prev {
+		m.expandedCountries = map[string]bool{}
+		m.expandedCities = map[string]bool{}
+	}
+	return m, tea.Batch(cmd, m.refreshList())
+}
+
+func (m Model) submitSearch() (Model, tea.Cmd) {
+	q := strings.TrimSpace(m.searchInput.Value())
+	if q == "" {
+		return m.exitSearch(true)
+	}
+
+	m.searching = false
+	m.searchInput.Blur()
+	if id := strings.ToUpper(q); backend.IsServerID(id) {
+		m.selectServerByID(id)
+	}
+	return m, nil
+}
+
+func (m *Model) selectServerByID(id string) {
+	for i, it := range m.list.Items() {
+		if r, ok := it.(row); ok && r.kind == rowServer && strings.EqualFold(r.server.Name, id) {
+			m.list.Select(i)
+			return
+		}
+	}
+}
+
+func (m Model) exitSearch(clear bool) (Model, tea.Cmd) {
+	m.searching = false
+	m.searchInput.Blur()
+	if clear {
+		m.searchInput.Reset()
+	}
+	cmd := m.refreshList()
+	return m, cmd
+}