mirror of https://github.com/docker/compose.git
133 lines
2.5 KiB
Go
133 lines
2.5 KiB
Go
package console
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type resource struct {
|
|
name string
|
|
status string
|
|
details string
|
|
}
|
|
|
|
type progress struct {
|
|
console console
|
|
resources []*resource
|
|
}
|
|
|
|
type ProgressWriter interface {
|
|
ResourceEvent(name string, status string, details string)
|
|
}
|
|
|
|
func NewProgressWriter() ProgressWriter {
|
|
return &progress{
|
|
console: ansiConsole{os.Stdout},
|
|
}
|
|
}
|
|
|
|
const (
|
|
blue = "36;2"
|
|
red = "31;1"
|
|
green = "32;1"
|
|
)
|
|
|
|
func (p *progress) ResourceEvent(name string, status string, details string) {
|
|
if logrus.IsLevelEnabled(logrus.DebugLevel) {
|
|
logrus.Debugf("> %s : %s %s\n", name, status, details)
|
|
return
|
|
}
|
|
p.console.MoveUp(len(p.resources))
|
|
|
|
newResource := true
|
|
for _, r := range p.resources {
|
|
if r.name == name {
|
|
newResource = false
|
|
r.status = status
|
|
r.details = details
|
|
break
|
|
}
|
|
}
|
|
if newResource {
|
|
p.resources = append(p.resources, &resource{name, status, details})
|
|
}
|
|
|
|
var width int
|
|
for _, r := range p.resources {
|
|
l := len(r.name)
|
|
if width < l {
|
|
width = l
|
|
}
|
|
}
|
|
|
|
for _, r := range p.resources {
|
|
s := r.status
|
|
if strings.HasSuffix(s, "_IN_PROGRESS") {
|
|
s = p.console.WiP(s)
|
|
} else if strings.HasSuffix(s, "_COMPLETE") {
|
|
s = p.console.OK(s)
|
|
} else if strings.HasSuffix(s, "_FAILED") {
|
|
s = p.console.KO(s)
|
|
}
|
|
p.console.ClearLine()
|
|
p.console.Printf("%-"+strconv.Itoa(width)+"s ... %s %s", r.name, s, r.details) // nolint:errcheck
|
|
p.console.MoveDown(1)
|
|
}
|
|
}
|
|
|
|
type console interface {
|
|
Printf(format string, a ...interface{})
|
|
MoveUp(int)
|
|
MoveDown(int)
|
|
ClearLine()
|
|
OK(string) string
|
|
KO(string) string
|
|
WiP(string) string
|
|
}
|
|
|
|
type ansiConsole struct {
|
|
out io.Writer
|
|
}
|
|
|
|
func (c ansiConsole) Printf(format string, a ...interface{}) {
|
|
fmt.Fprintf(c.out, format, a...) // nolint:errcheck
|
|
fmt.Fprintf(c.out, "\r")
|
|
}
|
|
|
|
func (c ansiConsole) MoveUp(i int) {
|
|
fmt.Fprintf(c.out, "\033[%dA", i) // nolint:errcheck
|
|
}
|
|
|
|
func (c ansiConsole) MoveDown(i int) {
|
|
fmt.Fprintf(c.out, "\033[%dB", i) // nolint:errcheck
|
|
}
|
|
|
|
func (c ansiConsole) ClearLine() {
|
|
fmt.Fprint(c.out, "\033[2K\r") // nolint:errcheck
|
|
}
|
|
|
|
func (c ansiConsole) OK(s string) string {
|
|
return ansiColor(green, s)
|
|
}
|
|
|
|
func (c ansiConsole) KO(s string) string {
|
|
return ansiColor(red, s)
|
|
}
|
|
|
|
func (c ansiConsole) WiP(s string) string {
|
|
return ansiColor(blue, s)
|
|
}
|
|
|
|
func ansiColor(code, s string) string {
|
|
return fmt.Sprintf("%s%s%s", ansi(code), s, ansi("0"))
|
|
}
|
|
|
|
func ansi(code string) string {
|
|
return fmt.Sprintf("\033[%sm", code)
|
|
}
|