bubbletea/examples/spinner/main.go

package main

import (
	"fmt"
	"log"

	"github.com/charmbracelet/boba"
	"github.com/charmbracelet/teaparty/spinner"
	"github.com/muesli/termenv"
)

var (
	color = termenv.ColorProfile()
)

type Model struct {
	spinner spinner.Model
	err     error
}

type errMsg error

func main() {
	p := boba.NewProgram(initialize, update, view, subscriptions)
	if err := p.Start(); err != nil {
		log.Fatal(err)
	}
}

func initialize() (boba.Model, boba.Cmd) {
	m := spinner.NewModel()
	m.Type = spinner.Dot

	return Model{
		spinner: m,
	}, nil
}

func update(msg boba.Msg, model boba.Model) (boba.Model, boba.Cmd) {
	m, ok := model.(Model)
	if !ok {
		return model, nil
	}

	switch msg := msg.(type) {

	case boba.KeyMsg:
		switch msg.String() {
		case "q":
			fallthrough
		case "esc":
			fallthrough
		case "ctrl+c":
			return m, boba.Quit
		default:
			return m, nil
		}

	case errMsg:
		m.err = msg
		return m, nil

	default:
		m.spinner, _ = spinner.Update(msg, m.spinner)
		return m, nil
	}

}

func view(model boba.Model) string {
	m, ok := model.(Model)
	if !ok {
		return "could not perform assertion on model in view"
	}
	if m.err != nil {
		return m.err.Error()
	}
	s := termenv.
		String(spinner.View(m.spinner)).
		Foreground(color.Color("205")).
		String()
	return fmt.Sprintf("\n\n   %s Loading forever...press q to quit\n\n", s)
}

func subscriptions(model boba.Model) boba.Subs {
	m, ok := model.(Model)
	if !ok {
		return nil
	}

	sub, err := spinner.MakeSub(m.spinner)
	if err != nil {
		return nil
	}
	return boba.Subs{
		"tick": sub,
	}
}