mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-12-23 22:49:43 -05:00
c82d5ccba9
* Hide cursor and fix dots spinner * Allow restarting of spinner * Don't spin on non TTY output Signed-off-by: Daniel Weiße <dw@edgeless.systems>
114 lines
2.4 KiB
Go
114 lines
2.4 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
tty "github.com/mattn/go-isatty"
|
|
)
|
|
|
|
const (
|
|
// hideCursor and showCursor are ANSI escape sequences to hide and show the cursor.
|
|
hideCursor = "\033[?25l"
|
|
showCursor = "\033[?25h"
|
|
)
|
|
|
|
var (
|
|
spinnerStates = []string{"⣷", "⣯", "⣟", "⡿", "⢿", "⣻", "⣽", "⣾"}
|
|
dotsStates = []string{". ", ".. ", "..."}
|
|
)
|
|
|
|
type spinnerInterf interface {
|
|
Start(text string, showDots bool)
|
|
Stop()
|
|
}
|
|
|
|
type spinner struct {
|
|
out io.Writer
|
|
delay time.Duration
|
|
wg *sync.WaitGroup
|
|
stop int32
|
|
spinFunc func(out io.Writer, wg *sync.WaitGroup, stop *int32, delay time.Duration, text string, showDots bool)
|
|
}
|
|
|
|
func newSpinner(writer io.Writer) *spinner {
|
|
s := &spinner{
|
|
out: writer,
|
|
wg: &sync.WaitGroup{},
|
|
delay: time.Millisecond * 100,
|
|
}
|
|
|
|
s.spinFunc = spinTTY
|
|
//
|
|
if !(writer == os.Stdout && tty.IsTerminal(os.Stdout.Fd())) {
|
|
s.spinFunc = spinNoTTY
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
// Start starts the spinner using the given text.
|
|
func (s *spinner) Start(text string, showDots bool) {
|
|
s.wg.Add(1)
|
|
|
|
go s.spinFunc(s.out, s.wg, &s.stop, s.delay, text, showDots)
|
|
}
|
|
|
|
// Stop stops the spinner.
|
|
func (s *spinner) Stop() {
|
|
atomic.StoreInt32(&s.stop, 1)
|
|
s.wg.Wait()
|
|
}
|
|
|
|
// Write stops the spinner and writes the given bytes to the underlying writer.
|
|
func (s *spinner) Write(p []byte) (n int, err error) {
|
|
s.Stop()
|
|
return s.out.Write(p)
|
|
}
|
|
|
|
func spinTTY(out io.Writer, wg *sync.WaitGroup, stop *int32, delay time.Duration, text string, showDots bool) {
|
|
defer wg.Done()
|
|
|
|
fmt.Fprint(out, hideCursor)
|
|
|
|
for i := 0; ; i = (i + 1) % len(spinnerStates) {
|
|
if atomic.LoadInt32(stop) != 0 {
|
|
break
|
|
}
|
|
dotsState := ""
|
|
if showDots {
|
|
dotsState = dotsStates[i%len(dotsStates)]
|
|
}
|
|
state := fmt.Sprintf("\r%s %s%s", spinnerStates[i], text, dotsState)
|
|
fmt.Fprint(out, state)
|
|
time.Sleep(delay)
|
|
}
|
|
dotsState := ""
|
|
if showDots {
|
|
dotsState = dotsStates[len(dotsStates)-1]
|
|
}
|
|
finalState := fmt.Sprintf("\r%s%s ", text, dotsState)
|
|
fmt.Fprintln(out, finalState)
|
|
fmt.Fprint(out, showCursor)
|
|
}
|
|
|
|
func spinNoTTY(out io.Writer, wg *sync.WaitGroup, _ *int32, _ time.Duration, text string, _ bool) {
|
|
defer wg.Done()
|
|
fmt.Fprintln(out, text+"...")
|
|
}
|
|
|
|
type nopSpinner struct{}
|
|
|
|
func (s nopSpinner) Start(string, bool) {}
|
|
func (s nopSpinner) Stop() {}
|