Add stopwatch example

This commit is contained in:
Christian Rocha 2021-12-17 10:10:38 -05:00
parent 03d1aa6c08
commit bdb04767ff

103
examples/stopwatch/main.go Normal file
View File

@ -0,0 +1,103 @@
package main
import (
"fmt"
"os"
"time"
"github.com/charmbracelet/bubbles/help"
"github.com/charmbracelet/bubbles/key"
"github.com/charmbracelet/bubbles/stopwatch"
tea "github.com/charmbracelet/bubbletea"
)
type model struct {
stopwatch stopwatch.Model
keymap keymap
help help.Model
quitting bool
}
type keymap struct {
start key.Binding
stop key.Binding
reset key.Binding
quit key.Binding
}
func (m model) Init() tea.Cmd {
return m.stopwatch.Init()
}
func (m model) View() string {
// Note: you could further customize the time output by getting the
// duration from m.stopwatch.Elapsed(), which returns a time.Duration, and
// skip m.stopwatch.View() altogether.
s := m.stopwatch.View() + "\n"
if !m.quitting {
s = "Elapsed: " + s
s += m.helpView()
}
return s
}
func (m model) helpView() string {
return "\n" + m.help.ShortHelpView([]key.Binding{
m.keymap.start,
m.keymap.stop,
m.keymap.reset,
m.keymap.quit,
})
}
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
switch msg := msg.(type) {
case tea.KeyMsg:
switch {
case key.Matches(msg, m.keymap.quit):
m.quitting = true
return m, tea.Quit
case key.Matches(msg, m.keymap.reset):
return m, m.stopwatch.Reset()
case key.Matches(msg, m.keymap.start, m.keymap.stop):
m.keymap.stop.SetEnabled(!m.stopwatch.Running())
m.keymap.start.SetEnabled(m.stopwatch.Running())
return m, m.stopwatch.Toggle()
}
}
var cmd tea.Cmd
m.stopwatch, cmd = m.stopwatch.Update(msg)
return m, cmd
}
func main() {
m := model{
stopwatch: stopwatch.NewWithInterval(time.Millisecond),
keymap: keymap{
start: key.NewBinding(
key.WithKeys("s"),
key.WithHelp("s", "start"),
),
stop: key.NewBinding(
key.WithKeys("s"),
key.WithHelp("s", "stop"),
),
reset: key.NewBinding(
key.WithKeys("r"),
key.WithHelp("r", "reset"),
),
quit: key.NewBinding(
key.WithKeys("ctrl+c", "q"),
key.WithHelp("q", "quit"),
),
},
help: help.NewModel(),
}
m.keymap.start.SetEnabled(false)
if err := tea.NewProgram(m).Start(); err != nil {
fmt.Println("Oh no, it didn't work:", err)
os.Exit(1)
}
}