sync

internal/tui/components/spinner.go

@@ -0,0 +1,65 @@
+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
+}