mirror of
https://gitea.com/Lydanne/buildx.git
synced 2025-11-04 01:53:42 +08:00
38 lines
580 B
Go
38 lines
580 B
Go
package progress
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/moby/buildkit/client"
|
|
)
|
|
|
|
func WithPrefix(w Writer, pfx string, force bool) Writer {
|
|
return &prefixed{
|
|
main: w,
|
|
pfx: pfx,
|
|
force: force,
|
|
}
|
|
}
|
|
|
|
type prefixed struct {
|
|
main Writer
|
|
pfx string
|
|
force bool
|
|
}
|
|
|
|
func (p *prefixed) Write(v *client.SolveStatus) {
|
|
if p.force {
|
|
for _, v := range v.Vertexes {
|
|
v.Name = addPrefix(p.pfx, v.Name)
|
|
}
|
|
}
|
|
p.main.Write(v)
|
|
}
|
|
|
|
func addPrefix(pfx, name string) string {
|
|
if strings.HasPrefix(name, "[") {
|
|
return "[" + pfx + " " + name[1:]
|
|
}
|
|
return "[" + pfx + "] " + name
|
|
}
|