1
0
Fork 0
mirror of https://github.com/moby/moby.git synced 2022-11-09 12:21:53 -05:00
moby--moby/utils/jsonmessage.go

162 lines
4 KiB
Go
Raw Normal View History

2013-11-28 15:16:57 -05:00
package utils
import (
"encoding/json"
"fmt"
"io"
2013-11-28 15:37:07 -05:00
"strings"
2013-11-28 15:16:57 -05:00
"time"
"github.com/dotcloud/docker/pkg/term"
"github.com/dotcloud/docker/pkg/units"
2013-11-28 15:16:57 -05:00
)
type JSONError struct {
Code int `json:"code,omitempty"`
Message string `json:"message,omitempty"`
}
func (e *JSONError) Error() string {
return e.Message
}
type JSONProgress struct {
2013-12-02 14:53:04 -05:00
terminalFd uintptr
Current int `json:"current,omitempty"`
Total int `json:"total,omitempty"`
Start int64 `json:"start,omitempty"`
2013-11-28 15:16:57 -05:00
}
func (p *JSONProgress) String() string {
2013-12-02 14:53:04 -05:00
var (
width = 200
pbBox string
numbersBox string
timeLeftBox string
)
ws, err := term.GetWinsize(p.terminalFd)
if err == nil {
width = int(ws.Width)
}
2013-12-04 14:57:18 -05:00
if p.Current <= 0 && p.Total <= 0 {
2013-11-28 15:16:57 -05:00
return ""
}
current := units.HumanSize(int64(p.Current))
2013-12-04 14:57:18 -05:00
if p.Total <= 0 {
2013-12-02 14:53:04 -05:00
return fmt.Sprintf("%8v", current)
2013-11-28 15:16:57 -05:00
}
total := units.HumanSize(int64(p.Total))
2013-11-28 15:37:07 -05:00
percentage := int(float64(p.Current)/float64(p.Total)*100) / 2
2013-12-02 14:53:04 -05:00
if width > 110 {
pbBox = fmt.Sprintf("[%s>%s] ", strings.Repeat("=", percentage), strings.Repeat(" ", 50-percentage))
}
numbersBox = fmt.Sprintf("%8v/%v", current, total)
2013-11-28 15:37:07 -05:00
if p.Current > 0 && p.Start > 0 && percentage < 50 {
2013-12-02 14:53:04 -05:00
fromStart := time.Now().UTC().Sub(time.Unix(int64(p.Start), 0))
perEntry := fromStart / time.Duration(p.Current)
left := time.Duration(p.Total-p.Current) * perEntry
left = (left / time.Second) * time.Second
if width > 50 {
timeLeftBox = " " + left.String()
}
}
return pbBox + numbersBox + timeLeftBox
2013-11-28 15:16:57 -05:00
}
type JSONMessage struct {
2013-12-06 14:55:56 -05:00
Stream string `json:"stream,omitempty"`
2013-11-28 15:16:57 -05:00
Status string `json:"status,omitempty"`
Progress *JSONProgress `json:"progressDetail,omitempty"`
ProgressMessage string `json:"progress,omitempty"` //deprecated
ID string `json:"id,omitempty"`
From string `json:"from,omitempty"`
Time int64 `json:"time,omitempty"`
Error *JSONError `json:"errorDetail,omitempty"`
ErrorMessage string `json:"error,omitempty"` //deprecated
}
func (jm *JSONMessage) Display(out io.Writer, isTerminal bool) error {
if jm.Error != nil {
if jm.Error.Code == 401 {
return fmt.Errorf("Authentication is required.")
}
return jm.Error
}
2013-12-02 14:49:11 -05:00
var endl string
if isTerminal && jm.Stream == "" && jm.Progress != nil {
2013-11-28 15:16:57 -05:00
// <ESC>[2K = erase entire current line
fmt.Fprintf(out, "%c[2K\r", 27)
2013-12-06 14:55:56 -05:00
endl = "\r"
2013-12-06 16:56:09 -05:00
} else if jm.Progress != nil { //disable progressbar in non-terminal
return nil
2013-11-28 15:16:57 -05:00
}
if jm.Time != 0 {
fmt.Fprintf(out, "%s ", time.Unix(jm.Time, 0).Format(time.RFC3339Nano))
2013-11-28 15:16:57 -05:00
}
if jm.ID != "" {
fmt.Fprintf(out, "%s: ", jm.ID)
}
if jm.From != "" {
fmt.Fprintf(out, "(from %s) ", jm.From)
}
if jm.Progress != nil {
fmt.Fprintf(out, "%s %s%s", jm.Status, jm.Progress.String(), endl)
} else if jm.ProgressMessage != "" { //deprecated
fmt.Fprintf(out, "%s %s%s", jm.Status, jm.ProgressMessage, endl)
2013-12-06 14:55:56 -05:00
} else if jm.Stream != "" {
fmt.Fprintf(out, "%s%s", jm.Stream, endl)
2013-11-28 15:16:57 -05:00
} else {
2013-12-06 14:55:56 -05:00
fmt.Fprintf(out, "%s%s\n", jm.Status, endl)
2013-11-28 15:16:57 -05:00
}
return nil
}
2013-12-02 14:53:04 -05:00
func DisplayJSONMessagesStream(in io.Reader, out io.Writer, terminalFd uintptr, isTerminal bool) error {
2013-12-02 14:49:11 -05:00
var (
dec = json.NewDecoder(in)
ids = make(map[string]int)
diff = 0
)
2013-11-28 15:16:57 -05:00
for {
2013-12-02 14:49:11 -05:00
var jm JSONMessage
if err := dec.Decode(&jm); err != nil {
if err == io.EOF {
break
}
2013-11-28 15:16:57 -05:00
return err
}
2013-12-02 14:53:04 -05:00
if jm.Progress != nil {
jm.Progress.terminalFd = terminalFd
}
if jm.ID != "" && (jm.Progress != nil || jm.ProgressMessage != "") {
2013-11-28 15:16:57 -05:00
line, ok := ids[jm.ID]
if !ok {
line = len(ids)
ids[jm.ID] = line
fmt.Fprintf(out, "\n")
diff = 0
} else {
diff = len(ids) - line
}
if jm.ID != "" && isTerminal {
2013-11-28 15:16:57 -05:00
// <ESC>[{diff}A = move cursor up diff rows
fmt.Fprintf(out, "%c[%dA", 27, diff)
}
}
err := jm.Display(out, isTerminal)
if jm.ID != "" && isTerminal {
// <ESC>[{diff}B = move cursor down diff rows
fmt.Fprintf(out, "%c[%dB", 27, diff)
2013-11-28 15:16:57 -05:00
}
if err != nil {
return err
}
}
return nil
}