forked from Mirrors/bubbletea
123 lines
2.3 KiB
Go
123 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/charmbracelet/bubbles/help"
|
|
"github.com/charmbracelet/bubbles/key"
|
|
"github.com/charmbracelet/bubbles/timer"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
)
|
|
|
|
const timeout = time.Second * 5
|
|
|
|
type model struct {
|
|
timer timer.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.timer.Init()
|
|
}
|
|
|
|
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case timer.TickMsg:
|
|
var cmd tea.Cmd
|
|
m.timer, cmd = m.timer.Update(msg)
|
|
return m, cmd
|
|
|
|
case timer.StartStopMsg:
|
|
var cmd tea.Cmd
|
|
m.timer, cmd = m.timer.Update(msg)
|
|
m.keymap.stop.SetEnabled(m.timer.Running())
|
|
m.keymap.start.SetEnabled(!m.timer.Running())
|
|
return m, cmd
|
|
|
|
case timer.TimeoutMsg:
|
|
m.quitting = true
|
|
return m, tea.Quit
|
|
|
|
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):
|
|
m.timer.Timeout = timeout
|
|
case key.Matches(msg, m.keymap.start, m.keymap.stop):
|
|
return m, m.timer.Toggle()
|
|
}
|
|
}
|
|
|
|
return m, nil
|
|
}
|
|
|
|
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) View() string {
|
|
// For a more detailed timer view you could read m.timer.Timeout to get
|
|
// the remaining time as a time.Duration and skip calling m.timer.View()
|
|
// entirely.
|
|
s := m.timer.View()
|
|
|
|
if m.timer.Timedout() {
|
|
s = "All done!"
|
|
}
|
|
s += "\n"
|
|
if !m.quitting {
|
|
s = "Exiting in " + s
|
|
s += m.helpView()
|
|
}
|
|
return s
|
|
}
|
|
|
|
func main() {
|
|
m := model{
|
|
timer: timer.NewWithInterval(timeout, 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("q", "ctrl+c"),
|
|
key.WithHelp("q", "quit"),
|
|
),
|
|
},
|
|
help: help.NewModel(),
|
|
}
|
|
m.keymap.start.SetEnabled(false)
|
|
|
|
if _, err := tea.NewProgram(m).Run(); err != nil {
|
|
fmt.Println("Uh oh, we encountered an error:", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|