From 4936823662d1239cfe62c7e9f984101f75e828be Mon Sep 17 00:00:00 2001
From: Jean Mertz <jean@mertz.fm>
Date: Fri, 23 Feb 2018 16:22:30 +0100
Subject: [PATCH] Allow printing output in JSON formated style

---
 Gopkg.lock        |  2 +-
 README.md         |  4 ++++
 config/cli.go     |  3 +++
 config/options.go |  2 ++
 main.go           | 28 ++++++++++++++++++++++++++++
 5 files changed, 38 insertions(+), 1 deletion(-)

diff --git a/Gopkg.lock b/Gopkg.lock
index d13250f..2a601b2 100644
--- a/Gopkg.lock
+++ b/Gopkg.lock
@@ -150,6 +150,6 @@
 [solve-meta]
   analyzer-name = "dep"
   analyzer-version = 1
-  inputs-digest = "ff73565cd1dfcba890d7c70059b84ababc57754a4c63d7ecd3b8d85ac313b798"
+  inputs-digest = "f304bb7f4f11d1f4c7f19999b95148d555f75ba09e2b996398ccc72f23758a48"
   solver-name = "gps-cdcl"
   solver-version = 1
diff --git a/README.md b/README.md
index 05da972..9e00777 100644
--- a/README.md
+++ b/README.md
@@ -67,6 +67,10 @@ Options:
   -r NAME=URL, --repo=NAME=URL,... List of NAME=URL pairs of repositories to add
                                    to the index before compiling charts config
   -p DIR, --partials-dir=DIR       Path from which to load partial templates
+                                   [default: config/deploy/partials]
+  -j, --json                       Print resources formatted as JSON instead of
+                                   YAML. Each resource is printed on a single
+                                   line.
   --example-config                 Print an example charts.yaml, including
                                    extended documentation on the tunables
 ```
diff --git a/config/cli.go b/config/cli.go
index d0b1e77..85c8fb4 100644
--- a/config/cli.go
+++ b/config/cli.go
@@ -44,6 +44,9 @@ Options:
                                    to the index before compiling charts config
   -p DIR, --partials-dir=DIR       Path from which to load partial templates
                                    [default: config/deploy/partials]
+  -j, --json                       Print resources formatted as JSON instead of
+                                   YAML. Each resource is printed on a single
+                                   line.
   --example-config                 Print an example charts.yaml, including
                                    extended documentation on the tunables
 `
diff --git a/config/options.go b/config/options.go
index e0ac7d7..e7b2a47 100644
--- a/config/options.go
+++ b/config/options.go
@@ -10,6 +10,7 @@ type CLIOptions struct {
 	ChartsConfigurationPath    string
 	PartialTemplatesPath       string
 	ChartsConfigurationOptions *ChartsConfigurationOptions
+	OutputJSON                 bool
 }
 
 // ChartsConfigurationOptions contains the CLI options relevant for the charts
@@ -30,6 +31,7 @@ func NewCLIOptions(cli map[string]interface{}) (*CLIOptions, error) {
 	namespace, _ := cli["--namespace"].(string)
 
 	c := &CLIOptions{
+		OutputJSON:              cli["--json"].(bool),
 		ChartsConfigurationPath: path,
 		ChartsConfigurationOptions: &ChartsConfigurationOptions{
 			Name:      name,
diff --git a/main.go b/main.go
index c508888..b33a2ad 100644
--- a/main.go
+++ b/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"bytes"
 	"fmt"
 	"io/ioutil"
 	"os"
@@ -9,6 +10,7 @@ import (
 	"github.com/blendle/kubecrt/chartsconfig"
 	"github.com/blendle/kubecrt/config"
 	"github.com/blendle/kubecrt/helm"
+	"github.com/ghodss/yaml"
 )
 
 func main() {
@@ -70,6 +72,14 @@ func main() {
 		os.Exit(1)
 	}
 
+	if opts.OutputJSON {
+		out, err = toJSON(out)
+		if err != nil {
+			fmt.Printf("error converting chart to JSON format: %s\n", err)
+			os.Exit(1)
+		}
+	}
+
 	if cli["--output"] == nil {
 		fmt.Print(string(out))
 		return
@@ -88,3 +98,21 @@ func readInput(input string) ([]byte, error) {
 
 	return ioutil.ReadFile(input)
 }
+
+func toJSON(input []byte) ([]byte, error) {
+	var err error
+
+	bs := bytes.Split(input, []byte("---"))
+	for i := range bs {
+		if len(bs[i]) == 0 {
+			continue
+		}
+
+		bs[i], err = yaml.YAMLToJSON(bs[i])
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	return bytes.Join(bs, []byte("\n")), nil
+}