aerc/widgets/spinner.go
y0ast 03650474e2 update tcell to v2 and enable TrueColor support
Also update to the tcell v2 PaletteColor api, which should keep the chosen
theme of the user intact.

Note, that if $TRUECOLOR is defined and a truecolor given, aerc will now stop
clipping the value to one of the theme colors.
Generally this is desired behaviour though.
2020-12-18 07:23:22 +01:00

83 lines
1.4 KiB
Go

package widgets
import (
"strings"
"sync/atomic"
"time"
"github.com/gdamore/tcell/v2"
"git.sr.ht/~sircmpwn/aerc/config"
"git.sr.ht/~sircmpwn/aerc/lib/ui"
)
type Spinner struct {
ui.Invalidatable
frame int64 // access via atomic
frames []string
stop chan struct{}
style tcell.Style
}
func NewSpinner(uiConf *config.UIConfig) *Spinner {
spinner := Spinner{
stop: make(chan struct{}),
frame: -1,
frames: strings.Split(uiConf.Spinner, uiConf.SpinnerDelimiter),
style: uiConf.GetStyle(config.STYLE_SPINNER),
}
return &spinner
}
func (s *Spinner) Start() {
if s.IsRunning() {
return
}
atomic.StoreInt64(&s.frame, 0)
go func() {
for {
select {
case <-s.stop:
atomic.StoreInt64(&s.frame, -1)
s.stop <- struct{}{}
return
case <-time.After(200 * time.Millisecond):
atomic.AddInt64(&s.frame, 1)
s.Invalidate()
}
}
}()
}
func (s *Spinner) Stop() {
if !s.IsRunning() {
return
}
s.stop <- struct{}{}
<-s.stop
s.Invalidate()
}
func (s *Spinner) IsRunning() bool {
return atomic.LoadInt64(&s.frame) != -1
}
func (s *Spinner) Draw(ctx *ui.Context) {
if !s.IsRunning() {
s.Start()
}
cur := int(atomic.LoadInt64(&s.frame) % int64(len(s.frames)))
ctx.Fill(0, 0, ctx.Width(), ctx.Height(), ' ', s.style)
col := ctx.Width()/2 - len(s.frames[0])/2 + 1
ctx.Printf(col, 0, s.style, "%s", s.frames[cur])
}
func (s *Spinner) Invalidate() {
s.DoInvalidate(s)
}