Merge pull request #6 from ahmetalpbalkan/json

replication-status: add --json flag

Author: ahmetb

main.go

@@ -63,6 +63,9 @@ var (
 		Name:   "region-2",
 		Usage:  "Primary pilot location to roll out the extension (e.g. 'Brazil South')",
 		EnvVar: "REGION2"}
+	flJSON = cli.BoolFlag{
+		Name:  "json",
+		Usage: "Print output as JSON"}
 )
 
 func main() {
@@ -126,7 +129,7 @@ func main() {
 			Action: listVersions},
 		{Name: "replication-status",
 			Usage:  "Retrieves replication status for an uploaded extension package",
-			Flags:  []cli.Flag{flMgtURL, flSubsID, flSubsCert, flNamespace, flName, flVersion},
+			Flags:  []cli.Flag{flMgtURL, flSubsID, flSubsCert, flNamespace, flName, flVersion, flJSON},
 			Action: replicationStatus},
 		{Name: "unpublish-version",
 			Usage:  "Marks the specified version of the extension internal. Does not delete.",

replicationstatus.go

@@ -1,6 +1,8 @@
 package main
 
 import (
+	"encoding/json"
+	"fmt"
 	"os"
 
 	log "github.com/Sirupsen/logrus"
@@ -11,17 +13,41 @@ import (
 func replicationStatus(c *cli.Context) {
 	cl := mkClient(checkFlag(c, flMgtURL.Name), checkFlag(c, flSubsID.Name), checkFlag(c, flSubsCert.Name))
 	ns, name, version := checkFlag(c, flNamespace.Name), checkFlag(c, flName.Name), checkFlag(c, flVersion.Name)
+	json := c.Bool(flJSON.Name)
 	log.Debug("Requesting replication status.")
 	rs, err := cl.GetReplicationStatus(ns, name, version)
 	if err != nil {
 		log.Fatalf("Cannot fetch replication status: %v", err)
 	}
+
+	var f func(_ ReplicationStatusResponse) error
+	if json {
+		f = printAsJson
+	} else {
+		f = printAsTable
+	}
+	if err := f(rs); err != nil {
+		log.Fatal(err)
+	}
+}
+
+func printAsJson(r ReplicationStatusResponse) error {
+	b, err := json.MarshalIndent(r.Statuses, "", "  ")
+	if err != nil {
+		return fmt.Errorf("failed to format as json: %+v", err)
+	}
+	fmt.Fprintf(os.Stdout, "%s", string(b))
+	return nil
+}
+
+func printAsTable(r ReplicationStatusResponse) error {
 	table := tablewriter.NewWriter(os.Stdout)
 	table.SetHeader([]string{"Location", "Status"})
 	data := [][]string{}
-	for _, s := range rs.Statuses {
+	for _, s := range r.Statuses {
 		data = append(data, []string{s.Location, s.Status})
 	}
 	table.AppendBulk(data)
 	table.Render()
+	return nil
 }