mirror of
https://github.com/lima-vm/lima.git
synced 2026-10-28 11:44:35 +08:00
Akihiro Suda
fd1552da70
The boilerplate is added for Go files, shell scripts, and Makefiles. https://github.com/containerd/ltag is used for validating the boilerplates. - hack/ltag: ltag templates - .github/workflows/test.yml: ltag is executed for validation Signed-off-by: Akihiro Suda <akihiro.suda.cz@hco.ntt.co.jp>
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
// SPDX-FileCopyrightText: Copyright The Lima Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package progressbar
|
|
|
|
import (
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/cheggaaa/pb/v3"
|
|
"github.com/mattn/go-isatty"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// ProgressBar adapts pb.ProgressBar to go-qcow2reader.convert.Updater interface.
|
|
type ProgressBar struct {
|
|
*pb.ProgressBar
|
|
}
|
|
|
|
func (b *ProgressBar) Update(n int64) {
|
|
b.Add64(n)
|
|
}
|
|
|
|
func New(size int64) (*ProgressBar, error) {
|
|
bar := &ProgressBar{pb.New64(size)}
|
|
|
|
bar.Set(pb.Bytes, true)
|
|
|
|
if showProgress() {
|
|
bar.SetTemplateString(`{{counters . }} {{bar . | green }} {{percent .}} {{speed . "%s/s"}}`)
|
|
bar.SetRefreshRate(200 * time.Millisecond)
|
|
} else {
|
|
bar.Set(pb.Static, true)
|
|
}
|
|
|
|
bar.SetWidth(80)
|
|
if err := bar.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return bar, nil
|
|
}
|
|
|
|
func showProgress() bool {
|
|
// Progress supports only text format fow now.
|
|
if _, ok := logrus.StandardLogger().Formatter.(*logrus.TextFormatter); !ok {
|
|
return false
|
|
}
|
|
|
|
// Both logrus and pb use stderr by default.
|
|
logFd := os.Stderr.Fd()
|
|
return isatty.IsTerminal(logFd) || isatty.IsCygwinTerminal(logFd)
|
|
}
|