mirror of
https://github.com/sstent/aicyclingcoach-go.git
synced 2025-12-06 08:01:38 +00:00
66 lines
1.2 KiB
Go
66 lines
1.2 KiB
Go
package components
|
|
|
|
import (
|
|
"github.com/charmbracelet/bubbles/spinner"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
)
|
|
|
|
var (
|
|
spinnerStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("69"))
|
|
helpStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("241")).MarginTop(1)
|
|
)
|
|
|
|
type Spinner struct {
|
|
spinner spinner.Model
|
|
message string
|
|
quitting bool
|
|
}
|
|
|
|
func NewSpinner(message string) Spinner {
|
|
s := spinner.New()
|
|
s.Spinner = spinner.Dot
|
|
s.Style = spinnerStyle
|
|
return Spinner{
|
|
spinner: s,
|
|
message: message,
|
|
}
|
|
}
|
|
|
|
func (s Spinner) Init() tea.Cmd {
|
|
return s.spinner.Tick
|
|
}
|
|
|
|
func (s Spinner) Update(msg tea.Msg) (Spinner, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case tea.KeyMsg:
|
|
switch msg.String() {
|
|
case "q", "esc", "ctrl+c":
|
|
s.quitting = true
|
|
return s, tea.Quit
|
|
default:
|
|
return s, nil
|
|
}
|
|
default:
|
|
var cmd tea.Cmd
|
|
s.spinner, cmd = s.spinner.Update(msg)
|
|
return s, cmd
|
|
}
|
|
}
|
|
|
|
func (s Spinner) View() string {
|
|
str := lipgloss.JoinHorizontal(lipgloss.Center,
|
|
s.spinner.View(),
|
|
" "+s.message,
|
|
)
|
|
if s.quitting {
|
|
return str + "\n"
|
|
}
|
|
return str
|
|
}
|
|
|
|
func (s Spinner) SetMessage(msg string) Spinner {
|
|
s.message = msg
|
|
return s
|
|
}
|