mirror of
https://github.com/regclient/regclient.git
synced 2025-04-18 22:44:00 +03:00
95c9152941
- Use "any" instead of an empty interface. - Use range over an integer for for loops. - Remove shadow variables in loops now that Go no longer reuses the variable. - Use "slices.Contains", "slices.Delete", "slices.Equal", "slices.Index", "slices.SortFunc". - Use "cmp.Or", "min", and "max". - Use "fmt.Appendf" instead of "Sprintf" for generating a byte slice. - Use "errors.Join" or "fmt.Errorf" with multiple "%w" for multiple errors. Additionally, use modern regclient features: - Use "ref.SetTag", "ref.SetDigest", and "ref.AddDigest". - Call "regclient.ManifestGet" using "WithManifestDesc" instead of setting the digest on the reference. Signed-off-by: Brandon Mitchell <git@bmitch.net>
60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package ascii
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
type ProgressBar struct {
|
|
Width, Min, Max int
|
|
Start, Done, Active, Pending, End byte
|
|
Out io.Writer
|
|
}
|
|
|
|
func NewProgressBar(w io.Writer) *ProgressBar {
|
|
width := 0
|
|
if wFd, ok := w.(interface{ Fd() uintptr }); ok && term.IsTerminal(int(wFd.Fd())) {
|
|
w, _, err := term.GetSize(int(wFd.Fd()))
|
|
if err == nil {
|
|
width = w
|
|
}
|
|
}
|
|
|
|
return &ProgressBar{
|
|
Width: width,
|
|
Min: 10,
|
|
Max: 40,
|
|
Out: w,
|
|
Start: '[',
|
|
Done: '=',
|
|
Active: '>',
|
|
Pending: ' ',
|
|
End: ']',
|
|
}
|
|
}
|
|
|
|
func (p *ProgressBar) Generate(pct float64, pre, post string) []byte {
|
|
if pct < 0 {
|
|
pct = 0
|
|
} else if pct > 1 {
|
|
pct = 1
|
|
}
|
|
curWidth := p.Width - (len(pre) + len(post) + 2)
|
|
curWidth = min(max(curWidth, p.Min), p.Max)
|
|
buf := make([]byte, curWidth)
|
|
|
|
doneLen := int(float64(curWidth) * pct)
|
|
for i := range doneLen {
|
|
buf[i] = p.Done
|
|
}
|
|
if doneLen < curWidth {
|
|
buf[doneLen] = p.Active
|
|
}
|
|
for i := doneLen + 1; i < curWidth; i++ {
|
|
buf[i] = p.Pending
|
|
}
|
|
return fmt.Appendf(nil, "%s%c%s%c%s\n", pre, p.Start, buf, p.End, post)
|
|
}
|