mirror of
https://git.sr.ht/~phw/scotty
synced 2025-04-10 23:49:28 +02:00
79 lines
2.3 KiB
Go
79 lines
2.3 KiB
Go
/*
|
|
Copyright © 2023 Philipp Wolfer <phw@uploadedlobster.com>
|
|
|
|
This file is part of Scotty.
|
|
|
|
Scotty is free software: you can redistribute it and/or modify it under the
|
|
terms of the GNU General Public License as published by the Free Software
|
|
Foundation, either version 3 of the License, or (at your option) any later version.
|
|
|
|
Scotty is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
|
A PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along with
|
|
Scotty. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
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)
|
|
}
|
|
}
|