2023-07-13 17:11:48 +02:00
|
|
|
//go:build windows
|
2022-09-21 10:08:33 +02:00
|
|
|
// +build windows
|
2022-05-02 09:06:34 +02:00
|
|
|
|
|
|
|
package pb
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"log"
|
2023-07-17 11:51:56 +02:00
|
|
|
"strings"
|
2022-05-02 09:06:34 +02:00
|
|
|
|
2022-09-21 10:08:33 +02:00
|
|
|
"github.com/cheggaaa/pb/v3/termutil"
|
2022-05-02 09:06:34 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
func (p *Pool) print(first bool) bool {
|
|
|
|
p.m.Lock()
|
|
|
|
defer p.m.Unlock()
|
|
|
|
var out string
|
|
|
|
if !first {
|
|
|
|
coords, err := termutil.GetCursorPos()
|
|
|
|
if err != nil {
|
|
|
|
log.Panic(err)
|
|
|
|
}
|
|
|
|
coords.Y -= int16(p.lastBarsCount)
|
|
|
|
if coords.Y < 0 {
|
|
|
|
coords.Y = 0
|
|
|
|
}
|
|
|
|
coords.X = 0
|
|
|
|
|
2023-07-13 17:11:48 +02:00
|
|
|
err = termutil.SetCursorPos(coords)
|
2022-05-02 09:06:34 +02:00
|
|
|
if err != nil {
|
|
|
|
log.Panic(err)
|
|
|
|
}
|
|
|
|
}
|
2023-07-17 11:51:56 +02:00
|
|
|
cols, err := termutil.TerminalWidth()
|
|
|
|
if err != nil {
|
|
|
|
cols = defaultBarWidth
|
|
|
|
}
|
2022-05-02 09:06:34 +02:00
|
|
|
isFinished := true
|
|
|
|
for _, bar := range p.bars {
|
|
|
|
if !bar.IsFinished() {
|
|
|
|
isFinished = false
|
|
|
|
}
|
2023-07-13 17:11:48 +02:00
|
|
|
result := bar.String()
|
|
|
|
if r := cols - CellCount(result); r > 0 {
|
|
|
|
result += strings.Repeat(" ", r)
|
|
|
|
}
|
|
|
|
out += fmt.Sprintf("\r%s\n", result)
|
2022-05-02 09:06:34 +02:00
|
|
|
}
|
|
|
|
if p.Output != nil {
|
|
|
|
fmt.Fprint(p.Output, out)
|
|
|
|
} else {
|
|
|
|
fmt.Print(out)
|
|
|
|
}
|
|
|
|
p.lastBarsCount = len(p.bars)
|
|
|
|
return isFinished
|
|
|
|
}
|