Skip to content

Commit

Permalink
Add JSON stream progress writer
Browse files Browse the repository at this point in the history
Signed-off-by: Felix Fontein <felix@fontein.de>
  • Loading branch information
felixfontein committed Feb 8, 2024
1 parent aaa7ef6 commit 73e9f16
Show file tree
Hide file tree
Showing 3 changed files with 100 additions and 0 deletions.
3 changes: 3 additions & 0 deletions cmd/compose/compose.go
Expand Up @@ -376,6 +376,8 @@ func RootCommand(dockerCli command.Cli, backend api.Service) *cobra.Command { //
ui.Mode = ui.ModePlain
case ui.ModeQuiet, "none":
ui.Mode = ui.ModeQuiet
case ui.ModeJSON:
ui.Mode = ui.ModeJSON
default:
return fmt.Errorf("unsupported --progress value %q", opts.Progress)
}
Expand Down Expand Up @@ -525,5 +527,6 @@ var printerModes = []string{
ui.ModeAuto,
ui.ModeTTY,
ui.ModePlain,
ui.ModeJSON,
ui.ModeQuiet,
}
88 changes: 88 additions & 0 deletions pkg/progress/json.go
@@ -0,0 +1,88 @@
/*
Copyright 2024 Docker Compose CLI authors
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package progress

import (
"context"
"encoding/json"
"fmt"
"io"
)

type jsonWriter struct {
out io.Writer
done chan bool
dryRun bool
}

type jsonMessage struct {
dryRun bool `json:"dry-run,omitempty"`
tail bool `json:"tail,omitempty"`
id string `json:"id,omitempty"`
text string `json:"text,omitempty"`
status string `json:"status,omitempty"`
}

func (p *jsonWriter) Start(ctx context.Context) error {
select {
case <-ctx.Done():
return ctx.Err()
case <-p.done:
return nil
}
}

func (p *jsonWriter) Event(e Event) {
var message = &jsonMessage{
dryRun: p.dryRun,
tail: false,
id: e.ID,
text: e.Text,
status: e.StatusText,
}
marshal, err := json.Marshal(message)
if err == nil {
fmt.Fprintln(p.out, "%s", string(marshal))
}
}

func (p *jsonWriter) Events(events []Event) {
for _, e := range events {
p.Event(e)
}
}

func (p *jsonWriter) TailMsgf(msg string, args ...interface{}) {
var message = &jsonMessage{
dryRun: p.dryRun,
tail: true,
id: "",
text: fmt.Sprintf(msg, args...),
status: "",
}
marshal, err := json.Marshal(message)
if err == nil {
fmt.Fprintln(p.out, "%s", string(marshal))
}
}

func (p *jsonWriter) Stop() {
p.done <- true
}

func (p *jsonWriter) HasMore(bool) {
}
9 changes: 9 additions & 0 deletions pkg/progress/writer.go
Expand Up @@ -110,6 +110,8 @@ const (
ModePlain = "plain"
// ModeQuiet don't display events
ModeQuiet = "quiet"
// ModeJSON outputs a machine-readable JSON stream
ModeJSON = "json"
)

// Mode define how progress should be rendered, either as ModePlain or ModeTTY
Expand All @@ -136,6 +138,13 @@ func NewWriter(ctx context.Context, out io.Writer, progressTitle string) (Writer
return newTTYWriter(f, dryRun, progressTitle)
}
}
if Mode == ModeJSON {
return &jsonWriter{
out: out,
done: make(chan bool),
dryRun: dryRun,
}, nil
}
return &plainWriter{
out: out,
done: make(chan bool),
Expand Down

0 comments on commit 73e9f16

Please sign in to comment.