wehub-resource-sync
82 lines
1.4 KiB
Go
82 lines
1.4 KiB
Go
package term
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/briandowns/spinner"
|
|
)
|
|
|
|
const withMessageMinDuration = 700 * time.Millisecond
|
|
const withoutMessageMinDuration = 350 * time.Millisecond
|
|
|
|
var s = spinner.New(spinner.CharSets[33], 100*time.Millisecond)
|
|
var startedAt time.Time
|
|
|
|
var lastMessage string
|
|
var active bool
|
|
var currentWarningLoop int32
|
|
|
|
func StartSpinner(msg string) {
|
|
if active {
|
|
if msg == lastMessage {
|
|
return
|
|
}
|
|
|
|
s.Stop()
|
|
}
|
|
|
|
startedAt = time.Now()
|
|
s.Prefix = msg + " "
|
|
lastMessage = msg
|
|
s.Start()
|
|
active = true
|
|
}
|
|
|
|
func StopSpinner() {
|
|
elapsed := time.Since(startedAt)
|
|
|
|
if lastMessage != "" && elapsed < withMessageMinDuration {
|
|
time.Sleep(withMessageMinDuration - elapsed)
|
|
} else if elapsed < withoutMessageMinDuration {
|
|
time.Sleep(withoutMessageMinDuration - elapsed)
|
|
}
|
|
|
|
s.Stop()
|
|
ClearCurrentLine()
|
|
|
|
active = false
|
|
}
|
|
|
|
func ResumeSpinner() {
|
|
if !active {
|
|
StartSpinner(lastMessage)
|
|
}
|
|
}
|
|
|
|
func LongSpinnerWithWarning(msg, warning string) {
|
|
atomic.AddInt32(¤tWarningLoop, 1)
|
|
currentLoop := currentWarningLoop
|
|
|
|
StartSpinner(msg)
|
|
|
|
var flashWarning func()
|
|
flashWarning = func() {
|
|
go func() {
|
|
time.Sleep(3 * time.Second)
|
|
if !active || atomic.LoadInt32(¤tWarningLoop) != currentLoop {
|
|
return
|
|
}
|
|
StartSpinner(warning)
|
|
|
|
time.Sleep(2 * time.Second)
|
|
if !active || atomic.LoadInt32(¤tWarningLoop) != currentLoop {
|
|
return
|
|
}
|
|
StartSpinner(msg)
|
|
flashWarning()
|
|
}()
|
|
}
|
|
flashWarning()
|
|
}
|