Implemented progressbar for export/import

cmd/progress.go

@@ -0,0 +1,84 @@
+/*
+Copyright © 2023 Philipp Wolfer <phw@uploadedlobster.com>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+
+package cmd
+
+import (
+	"sync"
+	"time"
+
+	"github.com/fatih/color"
+	"github.com/vbauerster/mpb/v8"
+	"github.com/vbauerster/mpb/v8/decor"
+	"go.uploadedlobster.com/scotty/models"
+)
+
+func progressBar(wg *sync.WaitGroup, exportProgress chan models.Progress, importProgress chan models.Progress) *mpb.Progress {
+	p := mpb.New(
+		mpb.WithWaitGroup(wg),
+		mpb.WithOutput(color.Output),
+		// mpb.WithWidth(64),
+		mpb.WithAutoRefresh(),
+	)
+
+	exportBar := setupProgressBar(p, "exporting")
+	importBar := setupProgressBar(p, "importing")
+	go updateProgressBar(exportBar, wg, exportProgress)
+	go updateProgressBar(importBar, wg, importProgress)
+
+	return p
+}
+
+func setupProgressBar(p *mpb.Progress, name string) *mpb.Bar {
+	green := color.New(color.FgGreen).SprintFunc()
+	return p.New(0,
+		mpb.BarStyle(),
+		mpb.PrependDecorators(
+			decor.Name(" "),
+			decor.OnComplete(
+				decor.Spinner(nil, decor.WC{W: 2, C: decor.DidentRight}),
+				green("✓ "),
+			),
+			decor.Name(name, decor.WCSyncWidthR),
+		),
+		mpb.AppendDecorators(
+			decor.OnComplete(
+				decor.EwmaETA(decor.ET_STYLE_GO, 0, decor.WC{C: decor.DSyncWidth}),
+				"done",
+			),
+			// decor.OnComplete(decor.Percentage(decor.WC{W: 5, C: decor.DSyncWidthR}), "done"),
+			decor.Name(" "),
+		),
+	)
+}
+
+func updateProgressBar(bar *mpb.Bar, wg *sync.WaitGroup, progressChan chan models.Progress) {
+	wg.Add(1)
+	defer wg.Done()
+	lastIterTime := time.Now()
+	for progress := range progressChan {
+		oldIterTime := lastIterTime
+		lastIterTime = time.Now()
+		bar.EwmaSetCurrent(progress.Elapsed, lastIterTime.Sub(oldIterTime))
+		bar.SetTotal(progress.Total, progress.Completed)
+	}
+}