Skip to content

Commit

Permalink
fix jsonmessage in build
Browse files Browse the repository at this point in the history
  • Loading branch information
vieux committed Dec 6, 2013
1 parent 7fd64e0 commit 05f416d
Show file tree
Hide file tree
Showing 4 changed files with 19 additions and 6 deletions.
4 changes: 2 additions & 2 deletions buildfile.go
Original file line number Diff line number Diff line change
Expand Up @@ -375,7 +375,7 @@ type StdoutFormater struct {
}

func (sf *StdoutFormater) Write(buf []byte) (int, error) {
formattedBuf := sf.StreamFormatter.FormatStatus("", "%s", string(buf))
formattedBuf := sf.StreamFormatter.FormatStream(string(buf))
n, err := sf.Writer.Write(formattedBuf)
if n != len(formattedBuf) {
return n, io.ErrShortWrite
Expand All @@ -389,7 +389,7 @@ type StderrFormater struct {
}

func (sf *StderrFormater) Write(buf []byte) (int, error) {
formattedBuf := sf.StreamFormatter.FormatStatus("", "%s", "\033[91m"+string(buf)+"\033[0m")
formattedBuf := sf.StreamFormatter.FormatStream("\033[91m" + string(buf) + "\033[0m")
n, err := sf.Writer.Write(formattedBuf)
if n != len(formattedBuf) {
return n, io.ErrShortWrite
Expand Down
2 changes: 0 additions & 2 deletions commands.go
Original file line number Diff line number Diff line change
Expand Up @@ -229,8 +229,6 @@ func (cli *DockerCli) CmdBuild(args ...string) error {
if context != nil {
headers.Set("Content-Type", "application/tar")
}
// Temporary hack to fix displayJSON behavior
cli.isTerminal = false
err = cli.stream("POST", fmt.Sprintf("/build?%s", v.Encode()), body, cli.out, headers)
if jerr, ok := err.(*utils.JSONError); ok {
return &utils.StatusError{Status: jerr.Message, StatusCode: jerr.Code}
Expand Down
7 changes: 5 additions & 2 deletions utils/jsonmessage.go
Original file line number Diff line number Diff line change
Expand Up @@ -66,6 +66,7 @@ func (p *JSONProgress) String() string {
}

type JSONMessage struct {
Stream string `json:"stream,omitempty"`
Status string `json:"status,omitempty"`
Progress *JSONProgress `json:"progressDetail,omitempty"`
ProgressMessage string `json:"progress,omitempty"` //deprecated
Expand All @@ -87,7 +88,7 @@ func (jm *JSONMessage) Display(out io.Writer, isTerminal bool) error {
if isTerminal {
// <ESC>[2K = erase entire current line
fmt.Fprintf(out, "%c[2K\r", 27)
endl = "\r\n"
endl = "\r"
}
if jm.Time != 0 {
fmt.Fprintf(out, "[%s] ", time.Unix(jm.Time, 0))
Expand All @@ -102,8 +103,10 @@ func (jm *JSONMessage) Display(out io.Writer, isTerminal bool) error {
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)
} else if jm.Stream != "" {
fmt.Fprintf(out, "%s%s", jm.Stream, endl)
} else {
fmt.Fprintf(out, "%s%s", jm.Status, endl)
fmt.Fprintf(out, "%s%s\n", jm.Status, endl)
}
return nil
}
Expand Down
12 changes: 12 additions & 0 deletions utils/streamformatter.go
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,18 @@ func NewStreamFormatter(json bool) *StreamFormatter {
return &StreamFormatter{json, false}
}

func (sf *StreamFormatter) FormatStream(str string) []byte {
sf.used = true
if sf.json {
b, err := json.Marshal(&JSONMessage{Stream: str})
if err != nil {
return sf.FormatError(err)
}
return b
}
return []byte(str + "\r")
}

func (sf *StreamFormatter) FormatStatus(id, format string, a ...interface{}) []byte {
sf.used = true
str := fmt.Sprintf(format, a...)
Expand Down

0 comments on commit 05f416d

Please sign in to comment.