From b10dbcb4dda0c3e9337ac78758d7b5b7a8f6e225 Mon Sep 17 00:00:00 2001
From: Christian Rocha <christian@rocha.is>
Date: Wed, 18 Nov 2020 14:27:41 -0500
Subject: [PATCH] Add option to show percent

---
 progress/progress.go | 42 ++++++++++++++++++++++++++++++------------
 1 file changed, 30 insertions(+), 12 deletions(-)

diff --git a/progress/progress.go b/progress/progress.go
index b918136..3f54fb5 100644
--- a/progress/progress.go
+++ b/progress/progress.go
@@ -1,9 +1,11 @@
 package progress
 
 import (
+	"fmt"
 	"strings"
 
 	"github.com/lucasb-eyer/go-colorful"
+	"github.com/muesli/reflow/ansi"
 	"github.com/muesli/termenv"
 )
 
@@ -18,14 +20,17 @@ type Model struct {
 	Width int
 
 	// filament rune of done part of progress bar. it's █ by default
-	FilamentSymbol rune
+	Full rune
 
 	// empty rune of pending part of progress bar. it's ░ by default
-	EmptySymbol rune
+	Empty rune
 
 	// if true, gradient will be setted from start to end of filled part. Instead, it'll work
 	// on all proggress bar length
 	FullGradientMode bool
+
+	ShowPercent   bool
+	PercentFormat string
 }
 
 // NewModel returns a model with default values.
@@ -33,18 +38,31 @@ func NewModel() Model {
 	startColor, _ := colorful.Hex("#00dbde")
 	endColor, _ := colorful.Hex("#fc00ff")
 	return Model{
-		StartColor:     startColor,
-		EndColor:       endColor,
-		Width:          40,
-		FilamentSymbol: '█',
-		EmptySymbol:    '░',
+		StartColor:    startColor,
+		EndColor:      endColor,
+		Width:         40,
+		Full:          '█',
+		Empty:         '░',
+		ShowPercent:   true,
+		PercentFormat: " %3.0f%%",
 	}
 }
 
-func (m *Model) View(percent float64) string {
-	ramp := make([]string, int(float64(m.Width)*percent))
+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))
 	for i := 0; i < len(ramp); i++ {
-		gradientPart := float64(m.Width)
+		gradientPart := float64(w)
 		if m.FullGradientMode {
 			gradientPart = float64(len(ramp))
 		}
@@ -55,9 +73,9 @@ func (m *Model) View(percent float64) string {
 
 	var fullCells string
 	for i := 0; i < len(ramp); i++ {
-		fullCells += termenv.String(string(m.FilamentSymbol)).Foreground(termenv.ColorProfile().Color(ramp[i])).String()
+		fullCells += termenv.String(string(m.Full)).Foreground(termenv.ColorProfile().Color(ramp[i])).String()
 	}
 
-	fullCells += strings.Repeat(string(m.EmptySymbol), m.Width-len(ramp))
+	fullCells += strings.Repeat(string(m.Empty), w-len(ramp))
 	return fullCells
 }