mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-18 21:52:28 +03:00
137 lines
3.3 KiB
Go
137 lines
3.3 KiB
Go
package progress
|
|
|
|
import (
|
|
"fmt"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/fatih/color"
|
|
mpb "github.com/vbauerster/mpb/v8"
|
|
"github.com/vbauerster/mpb/v8/decor"
|
|
)
|
|
|
|
const (
|
|
msgLength = 36
|
|
barWidth = 28
|
|
boxWidth = 64
|
|
refreshRate = 150 * time.Millisecond
|
|
)
|
|
|
|
// DefaultColors returns the default colors used for the progress bars.
|
|
func DefaultColors() *Colors {
|
|
return &Colors{
|
|
Error: color.New(color.FgRed, color.Bold),
|
|
Filler: color.New(color.FgGreen, color.Bold, color.Faint),
|
|
MemUsage: color.New(color.FgGreen, color.Faint),
|
|
Message: color.New(color.Faint),
|
|
Percent: color.New(color.FgCyan, color.Faint),
|
|
Size: color.New(color.Faint),
|
|
Waiting: color.New(color.FgYellow, color.Faint),
|
|
Warning: color.New(color.FgYellow),
|
|
}
|
|
}
|
|
|
|
// Colors is the set of colors used for the progress bars.
|
|
type Colors struct {
|
|
Error *color.Color
|
|
Filler *color.Color
|
|
MemUsage *color.Color
|
|
Message *color.Color
|
|
Percent *color.Color
|
|
Size *color.Color
|
|
Waiting *color.Color
|
|
Warning *color.Color
|
|
}
|
|
|
|
// EnableColor enables or disables color for the progress bars.
|
|
func (c *Colors) EnableColor(enable bool) {
|
|
if c == nil {
|
|
return
|
|
}
|
|
|
|
if enable {
|
|
c.Error.EnableColor()
|
|
c.Filler.EnableColor()
|
|
c.MemUsage.EnableColor()
|
|
c.Message.EnableColor()
|
|
c.Percent.EnableColor()
|
|
c.Size.EnableColor()
|
|
c.Waiting.EnableColor()
|
|
c.Warning.EnableColor()
|
|
return
|
|
}
|
|
|
|
c.Error.DisableColor()
|
|
c.Filler.DisableColor()
|
|
c.MemUsage.DisableColor()
|
|
c.Message.DisableColor()
|
|
c.Percent.DisableColor()
|
|
c.Size.DisableColor()
|
|
c.Waiting.DisableColor()
|
|
c.Warning.DisableColor()
|
|
}
|
|
|
|
func colorize(decorator decor.Decorator, c *color.Color) decor.Decorator {
|
|
return decor.Meta(decorator, func(s string) string {
|
|
return c.Sprint(s)
|
|
})
|
|
}
|
|
|
|
func spinnerStyle(c *color.Color) mpb.SpinnerStyleComposer {
|
|
// REVISIT: maybe use ascii chars only, in case it's a weird terminal?
|
|
frames := []string{"∙∙∙", "●∙∙", "●∙∙", "∙●∙", "∙●∙", "∙∙●", "∙∙●", "∙∙∙"}
|
|
style := mpb.SpinnerStyle(frames...)
|
|
if c != nil {
|
|
style = style.Meta(func(s string) string {
|
|
return c.Sprint(s)
|
|
})
|
|
}
|
|
return style
|
|
}
|
|
|
|
func barStyle(c *color.Color) mpb.BarStyleComposer {
|
|
clr := func(s string) string {
|
|
return c.Sprint(s)
|
|
}
|
|
|
|
frames := []string{"∙", "●", "●", "●", "∙"}
|
|
return mpb.BarStyle().
|
|
Lbound(" ").Rbound(" ").
|
|
Filler("∙").FillerMeta(clr).
|
|
Padding(" ").
|
|
Tip(frames...).TipMeta(clr)
|
|
}
|
|
|
|
func newElapsedSeconds(c *color.Color, startTime time.Time, wcc ...decor.WC) decor.Decorator {
|
|
var msg string
|
|
producer := func(d time.Duration) string {
|
|
return " " + d.Round(time.Second).String()
|
|
}
|
|
fn := func(s decor.Statistics) string {
|
|
if !s.Completed {
|
|
msg = producer(time.Since(startTime))
|
|
msg = c.Sprint(msg)
|
|
}
|
|
return msg
|
|
}
|
|
return decor.Any(fn, wcc...)
|
|
}
|
|
|
|
// OptMemUsage is an Opt that causes the bar to display program
|
|
// memory usage.
|
|
var OptMemUsage = optMemUsage{}
|
|
|
|
var _ Opt = optMemUsage{}
|
|
|
|
type optMemUsage struct{}
|
|
|
|
func (optMemUsage) apply(p *Progress, cfg *barConfig) {
|
|
fn := func(s decor.Statistics) string {
|
|
stats := &runtime.MemStats{}
|
|
runtime.ReadMemStats(stats)
|
|
msg := fmt.Sprintf(" (% .1f)", decor.SizeB1024(stats.Sys))
|
|
return p.colors.MemUsage.Sprint(msg)
|
|
}
|
|
cfg.decorators = append(cfg.decorators, decor.Any(fn, decor.WCSyncSpace))
|
|
}
|