internal/progress: Add progress meter

1 files changed, 30 insertions, 0 deletions

diff --git a/internal/progress/set.go b/internal/progress/set.go
new file mode 100644
index 00000000..3a580a63
--- /dev/null
+++ b/internal/progress/set.go
@@ -0,0 +1,30 @@
+package progress
+
+import "time"
+
+// Set records current progress and renders when percent changed or the 1s tick
+// elapsed.
+func (meter *Meter) Set(done uint64, bytes uint64) {
+	meter.lastDone = done
+	meter.lastBytes = bytes
+	meter.sawValue = true
+
+	if meter.writer == nil {
+		return
+	}
+
+	now := time.Now()
+	forced := meter.consumeUpdateTick(now)
+
+	percentChanged := false
+	if meter.total > 0 {
+		percent := int(done * 100 / meter.total)
+		percentChanged = percent != meter.lastPercent
+	}
+
+	if !percentChanged && !forced {
+		return
+	}
+
+	meter.render(now, "\r")
+}