mirror of https://github.com/docker/cli.git
95 lines
1.7 KiB
Go
95 lines
1.7 KiB
Go
package progresswriter
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
|
|
"github.com/containerd/console"
|
|
"github.com/moby/buildkit/client"
|
|
"github.com/moby/buildkit/util/progress/progressui"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type printer struct {
|
|
status chan *client.SolveStatus
|
|
done <-chan struct{}
|
|
err error
|
|
}
|
|
|
|
func (p *printer) Done() <-chan struct{} {
|
|
return p.done
|
|
}
|
|
|
|
func (p *printer) Err() error {
|
|
return p.err
|
|
}
|
|
|
|
func (p *printer) Status() chan *client.SolveStatus {
|
|
if p == nil {
|
|
return nil
|
|
}
|
|
return p.status
|
|
}
|
|
|
|
type tee struct {
|
|
Writer
|
|
status chan *client.SolveStatus
|
|
}
|
|
|
|
func (t *tee) Status() chan *client.SolveStatus {
|
|
return t.status
|
|
}
|
|
|
|
func Tee(w Writer, ch chan *client.SolveStatus) Writer {
|
|
st := make(chan *client.SolveStatus)
|
|
t := &tee{
|
|
status: st,
|
|
Writer: w,
|
|
}
|
|
go func() {
|
|
for v := range st {
|
|
w.Status() <- v
|
|
ch <- v
|
|
}
|
|
close(w.Status())
|
|
close(ch)
|
|
}()
|
|
return t
|
|
}
|
|
|
|
func NewPrinter(ctx context.Context, out console.File, mode string) (Writer, error) {
|
|
statusCh := make(chan *client.SolveStatus)
|
|
doneCh := make(chan struct{})
|
|
|
|
pw := &printer{
|
|
status: statusCh,
|
|
done: doneCh,
|
|
}
|
|
|
|
if v := os.Getenv("BUILDKIT_PROGRESS"); v != "" && mode == "auto" {
|
|
mode = v
|
|
}
|
|
|
|
var c console.Console
|
|
switch mode {
|
|
case "auto", "tty", "":
|
|
if cons, err := console.ConsoleFromFile(out); err == nil {
|
|
c = cons
|
|
} else {
|
|
if mode == "tty" {
|
|
return nil, errors.Wrap(err, "failed to get console")
|
|
}
|
|
}
|
|
case "plain":
|
|
default:
|
|
return nil, errors.Errorf("invalid progress mode %s", mode)
|
|
}
|
|
|
|
go func() {
|
|
// not using shared context to not disrupt display but let is finish reporting errors
|
|
pw.err = progressui.DisplaySolveStatus(ctx, "", c, out, statusCh)
|
|
close(doneCh)
|
|
}()
|
|
return pw, nil
|
|
}
|