78c6151519
We want all kpod subcommands to use the formats code to output formats like json. Altering kpod diff --json to kpod diff --format json like the kpod images command. Signed-off-by: baude <bbaude@redhat.com>
126 lines
2.4 KiB
Go
126 lines
2.4 KiB
Go
package formats
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/ghodss/yaml"
|
|
"os"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// JSONString const to save on duplicate variable names
|
|
const JSONString string = "json"
|
|
|
|
// Writer interface for outputs
|
|
type Writer interface {
|
|
Out() error
|
|
}
|
|
|
|
// JSONStructArray for JSON output
|
|
type JSONStructArray struct {
|
|
Output []interface{}
|
|
}
|
|
|
|
// StdoutTemplateArray for Go template output
|
|
type StdoutTemplateArray struct {
|
|
Output []interface{}
|
|
Template string
|
|
Fields map[string]string
|
|
}
|
|
|
|
// JSONStruct for JSON output
|
|
type JSONStruct struct {
|
|
Output interface{}
|
|
}
|
|
|
|
// StdoutTemplate for Go template output
|
|
type StdoutTemplate struct {
|
|
Output interface{}
|
|
Template string
|
|
Fields map[string]string
|
|
}
|
|
|
|
// YAMLStruct for YAML output
|
|
type YAMLStruct struct {
|
|
Output interface{}
|
|
}
|
|
|
|
// Out method for JSON Arrays
|
|
func (j JSONStructArray) Out() error {
|
|
data, err := json.MarshalIndent(j.Output, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Printf("%s\n", data)
|
|
return nil
|
|
}
|
|
|
|
// Out method for Go templates
|
|
func (t StdoutTemplateArray) Out() error {
|
|
if strings.HasPrefix(t.Template, "table") {
|
|
t.Template = strings.TrimSpace(t.Template[5:])
|
|
headerTmpl, err := template.New("header").Funcs(headerFunctions).Parse(t.Template)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "Template parsing error")
|
|
}
|
|
err = headerTmpl.Execute(os.Stdout, t.Fields)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Println()
|
|
}
|
|
tmpl, err := template.New("image").Funcs(basicFunctions).Parse(t.Template)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "Template parsing error")
|
|
}
|
|
|
|
for _, img := range t.Output {
|
|
basicTmpl := tmpl.Funcs(basicFunctions)
|
|
err = basicTmpl.Execute(os.Stdout, img)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Println()
|
|
}
|
|
return nil
|
|
|
|
}
|
|
|
|
// Out method for JSON struct
|
|
func (j JSONStruct) Out() error {
|
|
data, err := json.MarshalIndent(j.Output, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Printf("%s\n", data)
|
|
return nil
|
|
}
|
|
|
|
//Out method for Go templates
|
|
func (t StdoutTemplate) Out() error {
|
|
tmpl, err := template.New("image").Parse(t.Template)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "template parsing error")
|
|
}
|
|
err = tmpl.Execute(os.Stdout, t.Output)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Println()
|
|
return nil
|
|
}
|
|
|
|
// Out method for YAML
|
|
func (y YAMLStruct) Out() error {
|
|
var buf []byte
|
|
var err error
|
|
buf, err = yaml.Marshal(y.Output)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Println(string(buf))
|
|
return nil
|
|
}
|