feat(prints): Add a WithProgressBar method.

Signed-off-by: Flc゛ <[email protected]>
go-fries · Feb 7, 2024 · 3ac3f26 · 3ac3f26
1 parent 7b869cf
commit 3ac3f26
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/prints/progressbar.go b/prints/progressbar.go
@@ -41,3 +41,9 @@ func NewProgressBar(total int, opts ...ProgressBarOption) *ProgressBar {
 
 	return p
 }
+
+func WithProgressBar(total int, fc func(bar *ProgressBar), opts ...ProgressBarOption) {
+	bar := NewProgressBar(total, opts...)
+	fc(bar)
+	bar.Finish()
+}
diff --git a/prints/progressbar_test.go b/prints/progressbar_test.go
@@ -24,3 +24,11 @@ func TestNewProgressBar(t *testing.T) {
 	}
 	p2.Finish()
 }
+
+func TestWithProgressBar(t *testing.T) {
+	WithProgressBar(100, func(p *ProgressBar) {
+		for i := 1; i <= 100; i++ {
+			p.Increment()
+		}
+	})
+}