bubbles/progress/progress.go

82 lines
1.9 KiB
Go
Raw Normal View History

2020-11-04 16:42:44 +03:00
package progress
import (
2020-11-18 22:27:41 +03:00
"fmt"
2020-11-04 16:42:44 +03:00
"strings"
"github.com/lucasb-eyer/go-colorful"
2020-11-18 22:27:41 +03:00
"github.com/muesli/reflow/ansi"
2020-11-04 16:42:44 +03:00
"github.com/muesli/termenv"
)
type Model struct {
// Left side color of progress bar. By default, it's #00dbde
StartColor colorful.Color
// Left side color of progress bar. By default, it's #fc00ff
EndColor colorful.Color
2020-11-05 14:41:53 +03:00
// Width of progress bar in symbols.
Width int
2020-11-04 16:42:44 +03:00
// filament rune of done part of progress bar. it's █ by default
2020-11-18 22:27:41 +03:00
Full rune
2020-11-04 16:42:44 +03:00
// empty rune of pending part of progress bar. it's ░ by default
2020-11-18 22:27:41 +03:00
Empty rune
2020-11-04 16:42:44 +03:00
// if true, gradient will be setted from start to end of filled part. Instead, it'll work
// on all proggress bar length
FullGradientMode bool
2020-11-18 22:27:41 +03:00
ShowPercent bool
PercentFormat string
2020-11-04 16:42:44 +03:00
}
// NewModel returns a model with default values.
func NewModel() Model {
2020-11-04 16:42:44 +03:00
startColor, _ := colorful.Hex("#00dbde")
endColor, _ := colorful.Hex("#fc00ff")
return Model{
2020-11-18 22:27:41 +03:00
StartColor: startColor,
EndColor: endColor,
Width: 40,
Full: '█',
Empty: '░',
ShowPercent: true,
PercentFormat: " %3.0f%%",
2020-11-04 16:42:44 +03:00
}
}
2020-11-18 22:27:41 +03:00
func (m Model) View(percent float64) string {
if m.ShowPercent {
s := fmt.Sprintf(m.PercentFormat, percent*100)
w := ansi.PrintableRuneWidth(s)
return m.bar(percent, w) + s
}
return m.bar(percent, 0)
}
func (m Model) bar(percent float64, textWidth int) string {
w := m.Width - textWidth
ramp := make([]string, int(float64(w)*percent))
2020-11-04 16:42:44 +03:00
for i := 0; i < len(ramp); i++ {
2020-11-18 22:27:41 +03:00
gradientPart := float64(w)
2020-11-04 16:42:44 +03:00
if m.FullGradientMode {
gradientPart = float64(len(ramp))
}
percent := float64(i) / gradientPart
c := m.StartColor.BlendLuv(m.EndColor, percent)
ramp[i] = c.Hex()
}
var fullCells string
for i := 0; i < len(ramp); i++ {
2020-11-18 22:27:41 +03:00
fullCells += termenv.String(string(m.Full)).Foreground(termenv.ColorProfile().Color(ramp[i])).String()
2020-11-04 16:42:44 +03:00
}
2020-11-18 22:27:41 +03:00
fullCells += strings.Repeat(string(m.Empty), w-len(ramp))
2020-11-04 16:42:44 +03:00
return fullCells
}