2020-02-10 19:40:52 +03:00
|
|
|
package spinner
|
|
|
|
|
|
|
|
import (
|
|
|
|
"time"
|
|
|
|
|
2020-05-26 02:57:58 +03:00
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
2020-03-27 21:10:09 +03:00
|
|
|
"github.com/muesli/termenv"
|
2020-02-10 19:40:52 +03:00
|
|
|
)
|
|
|
|
|
2020-05-26 02:57:58 +03:00
|
|
|
// Spinner is a set of frames used in animating the spinner.
|
2020-02-10 19:51:08 +03:00
|
|
|
type Spinner = int
|
|
|
|
|
|
|
|
// Available types of spinners
|
|
|
|
const (
|
|
|
|
Line Spinner = iota
|
|
|
|
Dot
|
|
|
|
)
|
|
|
|
|
2020-05-26 02:57:58 +03:00
|
|
|
const (
|
2020-06-22 21:49:21 +03:00
|
|
|
defaultFPS = time.Second / 10
|
2020-05-26 02:57:58 +03:00
|
|
|
)
|
|
|
|
|
2020-02-10 19:40:52 +03:00
|
|
|
var (
|
2020-02-10 19:51:08 +03:00
|
|
|
// Spinner frames
|
|
|
|
spinners = map[Spinner][]string{
|
2020-04-22 21:29:30 +03:00
|
|
|
Line: {"|", "/", "-", "\\"},
|
|
|
|
Dot: {"⣾ ", "⣽ ", "⣻ ", "⢿ ", "⡿ ", "⣟ ", "⣯ ", "⣷ "},
|
2020-02-10 19:51:08 +03:00
|
|
|
}
|
|
|
|
|
2020-03-27 21:10:09 +03:00
|
|
|
color = termenv.ColorProfile().Color
|
2020-02-10 19:40:52 +03:00
|
|
|
)
|
|
|
|
|
|
|
|
// Model contains the state for the spinner. Use NewModel to create new models
|
|
|
|
// rather than using Model as a struct literal.
|
|
|
|
type Model struct {
|
2020-05-26 02:57:58 +03:00
|
|
|
|
|
|
|
// Type is the set of frames to use. See Spinner.
|
|
|
|
Type Spinner
|
|
|
|
|
|
|
|
// FPS is the speed at which the ticker should tick
|
2020-06-22 21:49:21 +03:00
|
|
|
FPS time.Duration
|
2020-05-26 02:57:58 +03:00
|
|
|
|
|
|
|
// ForegroundColor sets the background color of the spinner. It can be a
|
|
|
|
// hex code or one of the 256 ANSI colors. If the terminal emulator can't
|
|
|
|
// doesn't support the color specified it will automatically degrade
|
|
|
|
// (per github.com/muesli/termenv).
|
2020-03-27 21:10:09 +03:00
|
|
|
ForegroundColor string
|
2020-05-26 02:57:58 +03:00
|
|
|
|
|
|
|
// BackgroundColor sets the background color of the spinner. It can be a
|
|
|
|
// hex code or one of the 256 ANSI colors. If the terminal emulator can't
|
|
|
|
// doesn't support the color specified it will automatically degrade
|
|
|
|
// (per github.com/muesli/termenv).
|
2020-03-27 21:10:09 +03:00
|
|
|
BackgroundColor string
|
|
|
|
|
2020-02-10 19:40:52 +03:00
|
|
|
frame int
|
|
|
|
}
|
|
|
|
|
2020-05-26 02:57:58 +03:00
|
|
|
// NewModel returns a model with default values.
|
2020-02-10 19:40:52 +03:00
|
|
|
func NewModel() Model {
|
|
|
|
return Model{
|
2020-05-26 02:57:58 +03:00
|
|
|
Type: Line,
|
|
|
|
FPS: defaultFPS,
|
2020-02-10 19:40:52 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-05-26 02:57:58 +03:00
|
|
|
// TickMsg indicates that the timer has ticked and we should render a frame.
|
|
|
|
type TickMsg struct{}
|
2020-02-10 19:40:52 +03:00
|
|
|
|
2020-05-26 02:57:58 +03:00
|
|
|
// Update is the Tea update function. This will advance the spinner one frame
|
|
|
|
// every time it's called, regardless the message passed, so be sure the logic
|
|
|
|
// is setup so as not to call this Update needlessly.
|
2020-02-10 19:40:52 +03:00
|
|
|
func Update(msg tea.Msg, m Model) (Model, tea.Cmd) {
|
2020-06-22 21:49:21 +03:00
|
|
|
if _, ok := msg.(TickMsg); ok {
|
|
|
|
m.frame++
|
|
|
|
if m.frame >= len(spinners[m.Type]) {
|
|
|
|
m.frame = 0
|
|
|
|
}
|
2020-05-26 02:57:58 +03:00
|
|
|
return m, Tick(m)
|
2020-02-10 19:40:52 +03:00
|
|
|
}
|
2020-06-22 21:49:21 +03:00
|
|
|
return m, nil
|
2020-02-10 19:40:52 +03:00
|
|
|
}
|
|
|
|
|
2020-05-26 02:57:58 +03:00
|
|
|
// View renders the model's view.
|
2020-02-10 19:40:52 +03:00
|
|
|
func View(model Model) string {
|
2020-02-10 19:51:08 +03:00
|
|
|
s := spinners[model.Type]
|
|
|
|
if model.frame >= len(s) {
|
2020-02-10 19:40:52 +03:00
|
|
|
return "[error]"
|
|
|
|
}
|
2020-03-27 21:10:09 +03:00
|
|
|
|
|
|
|
str := s[model.frame]
|
|
|
|
|
|
|
|
if model.ForegroundColor != "" || model.BackgroundColor != "" {
|
|
|
|
return termenv.
|
|
|
|
String(str).
|
|
|
|
Foreground(color(model.ForegroundColor)).
|
|
|
|
Background(color(model.BackgroundColor)).
|
|
|
|
String()
|
|
|
|
}
|
|
|
|
|
|
|
|
return str
|
2020-02-10 19:40:52 +03:00
|
|
|
}
|
|
|
|
|
2020-05-26 02:57:58 +03:00
|
|
|
// Tick is the command used to advance the spinner one frame.
|
2020-06-22 21:49:21 +03:00
|
|
|
func Tick(m Model) tea.Cmd {
|
|
|
|
return tea.Tick(m.FPS, func(time.Time) tea.Msg {
|
2020-05-26 02:57:58 +03:00
|
|
|
return TickMsg{}
|
2020-06-22 21:49:21 +03:00
|
|
|
})
|
2020-02-10 19:40:52 +03:00
|
|
|
}
|