From 8b0cd60019b33488f8819da5c7bdd28d1d3fc737 Mon Sep 17 00:00:00 2001 From: Mark Allen Date: Tue, 12 Nov 2013 23:31:45 -0600 Subject: [PATCH] Pass terminal setting to display utils --- commands.go | 2 +- utils/utils.go | 27 +++++++++++++++++++-------- 2 files changed, 20 insertions(+), 9 deletions(-) diff --git a/commands.go b/commands.go index c514aa28a0..766350409d 100644 --- a/commands.go +++ b/commands.go @@ -2328,7 +2328,7 @@ func (cli *DockerCli) stream(method, path string, in io.Reader, out io.Writer, h } if matchesContentType(resp.Header.Get("Content-Type"), "application/json") { - return utils.DisplayJSONMessagesStream(resp.Body, out) + return utils.DisplayJSONMessagesStream(resp.Body, out, cli.isTerminal) } if _, err := io.Copy(out, resp.Body); err != nil { return err diff --git a/utils/utils.go b/utils/utils.go index 5864add8e1..d1cdff31fd 100644 --- a/utils/utils.go +++ b/utils/utils.go @@ -779,14 +779,19 @@ func NewHTTPRequestError(msg string, res *http.Response) error { } } -func (jm *JSONMessage) Display(out io.Writer) error { +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 } - fmt.Fprintf(out, "%c[2K\r", 27) + endl := "" + if isTerminal { + // [2K = erase entire current line + fmt.Fprintf(out, "%c[2K\r", 27) + endl = "\r" + } if jm.Time != 0 { fmt.Fprintf(out, "[%s] ", time.Unix(jm.Time, 0)) } @@ -797,14 +802,14 @@ func (jm *JSONMessage) Display(out io.Writer) error { fmt.Fprintf(out, "(from %s) ", jm.From) } if jm.Progress != "" { - fmt.Fprintf(out, "%s %s\r", jm.Status, jm.Progress) + fmt.Fprintf(out, "%s %s%s", jm.Status, jm.Progress, endl) } else { - fmt.Fprintf(out, "%s\r\n", jm.Status) + fmt.Fprintf(out, "%s%s\n", jm.Status, endl) } return nil } -func DisplayJSONMessagesStream(in io.Reader, out io.Writer) error { +func DisplayJSONMessagesStream(in io.Reader, out io.Writer, isTerminal bool) error { dec := json.NewDecoder(in) ids := make(map[string]int) diff := 0 @@ -825,11 +830,17 @@ func DisplayJSONMessagesStream(in io.Reader, out io.Writer) error { } else { diff = len(ids) - line } - fmt.Fprintf(out, "%c[%dA", 27, diff) + if isTerminal { + // [{diff}A = move cursor up diff rows + fmt.Fprintf(out, "%c[%dA", 27, diff) + } } - err := jm.Display(out) + err := jm.Display(out, isTerminal) if jm.ID != "" { - fmt.Fprintf(out, "%c[%dB", 27, diff) + if isTerminal { + // [{diff}B = move cursor down diff rows + fmt.Fprintf(out, "%c[%dB", 27, diff) + } } if err != nil { return err