Add keybindings and help menu

sevenkeys/ui/model.go

@@ -1,10 +1,69 @@
 package ui
 
+import (
+	"github.com/charmbracelet/bubbles/help"
+	"github.com/charmbracelet/bubbles/key"
+)
+
+type KeyMap struct {
+	Update         key.Binding
+	SearchCriteria key.Binding
+	Search         key.Binding
+	StorageOptions key.Binding
+	Find           key.Binding
+	Quit           key.Binding
+}
+
+func (k KeyMap) ShortHelp() []key.Binding {
+	return []key.Binding{k.Update, k.SearchCriteria, k.Search, k.StorageOptions, k.Find, k.Quit}
+}
+
+func (k KeyMap) FullHelp() [][]key.Binding {
+	return [][]key.Binding{
+		{k.Update, k.SearchCriteria, k.Search},
+		{k.StorageOptions, k.Find, k.Quit},
+	}
+}
+
+var DefaultKeyMap = KeyMap{
+	Update: key.NewBinding(
+		key.WithKeys("u"),
+		key.WithHelp("u", "update database"),
+	),
+	SearchCriteria: key.NewBinding(
+		key.WithKeys("c"),
+		key.WithHelp("c", "card printing search criteria"),
+	),
+	Search: key.NewBinding(
+		key.WithKeys("s"),
+		key.WithHelp("s", "card printing search"),
+	),
+	StorageOptions: key.NewBinding(
+		key.WithKeys("o"),
+		key.WithHelp("o", "card storage options"),
+	),
+	Find: key.NewBinding(
+		key.WithKeys("f"),
+		key.WithHelp("f", "find in storage"),
+	),
+	Quit: key.NewBinding(
+		key.WithKeys("ctrl+c", "q"),
+		key.WithHelp("q", "quit program"),
+	),
+}
+
 type Model struct {
 	WindowHeight int
 	WindowWidth  int
+
+	Help help.Model
 }
 
 func NewModel() Model {
-	return Model{}
+	help := help.New()
+	help.ShortSeparator = help.FullSeparator
+
+	return Model{
+		Help: help,
+	}
 }

sevenkeys/ui/update.go

@@ -1,6 +1,9 @@
 package ui
 
-import tea "github.com/charmbracelet/bubbletea"
+import (
+	"github.com/charmbracelet/bubbles/key"
+	tea "github.com/charmbracelet/bubbletea"
+)
 
 func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	var cmd tea.Cmd
@@ -9,10 +12,11 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	case tea.WindowSizeMsg:
 		m.WindowHeight = msg.Height
 		m.WindowWidth = msg.Width
+		m.Help.Width = msg.Width
 		return m, nil
 	case tea.KeyMsg:
-		switch msg.String() {
-		case "ctrl+c":
+		switch {
+		case key.Matches(msg, DefaultKeyMap.Quit):
 			return m, tea.Quit
 		}
 		break

sevenkeys/ui/view.go

@@ -3,6 +3,7 @@ package ui
 import (
 	"os"
 	"path/filepath"
+	"strings"
 
 	"github.com/lukesampson/figlet/figletlib"
 )
@@ -10,16 +11,23 @@ import (
 func (m Model) View() string {
 	var ui string
 
-	if m.WindowWidth > 0 {
+	if m.WindowWidth <= 0 {
+		return ui
+	}
+
+	// Get fonts for figlet
 	cwd, _ := os.Getwd()
 	fontsdir := filepath.Join(cwd, "fonts")
-
 	slantFont, _ := figletlib.GetFontByName(fontsdir, "slant")
 	termFont, _ := figletlib.GetFontByName(fontsdir, "term")
 
+	// Display splash screen
 	ui += figletlib.SprintMsg("SEVENKEYS", slantFont, m.WindowWidth, slantFont.Settings(), "center")
 	ui += figletlib.SprintMsg("the ultimate Magic: the Gathering trading card storage system", termFont, m.WindowWidth, termFont.Settings(), "center")
-	}
+
+	// Display help
+	ui += strings.Repeat("\n", m.WindowHeight-8) // TODO: Avoid hardcoding height somehow
+	ui += m.Help.View(DefaultKeyMap)
 
 	return ui
 }