List local topics with Json format

xitonix · Jul 1, 2020 · 6f59427 · 6f59427
1 parent c318403
commit 6f59427
Showing 1 changed file with 21 additions and 6 deletions.
diff --git a/commands/list/local_topics.go b/commands/list/local_topics.go
@@ -1,6 +1,7 @@
 package list
 
 import (
+	"encoding/json"
 	"fmt"
 	"regexp"
 	"sort"
@@ -52,17 +53,19 @@ func (l *listLocalTopics) run(_ *kingpin.ParseContext) error {
 	}
 
 	switch l.format {
-	case commands.ListFormat:
-		l.printAsList(localStore, false)
+	case commands.JsonFormat:
+		return l.printAsJson(localStore)
 	case commands.TableFormat:
-		l.printAsTable(localStore)
+		return l.printAsTable(localStore)
+	case commands.ListFormat:
+		return l.printAsList(localStore, false)
 	case commands.PlainTextFormat:
-		l.printAsList(localStore, true)
+		return l.printAsList(localStore, true)
 	}
 	return nil
 }
 
-func (l *listLocalTopics) printAsTable(store map[string][]string) {
+func (l *listLocalTopics) printAsTable(store map[string][]string) error {
 	for env, topics := range store {
 		table := tabular.NewTable(l.globalParams.EnableColor, tabular.C(format.WithCount(env, len(topics))).Align(tabular.AlignLeft).MinWidth(60))
 		sort.Strings(topics)
@@ -73,9 +76,10 @@ func (l *listLocalTopics) printAsTable(store map[string][]string) {
 		table.Render()
 		output.NewLines(1)
 	}
+	return nil
 }
 
-func (l *listLocalTopics) printAsList(store map[string][]string, plain bool) {
+func (l *listLocalTopics) printAsList(store map[string][]string, plain bool) error {
 	b := list.New(plain)
 	b.AsTree()
 	for env, topics := range store {
@@ -88,4 +92,15 @@ func (l *listLocalTopics) printAsList(store map[string][]string, plain bool) {
 		b.UnIntend()
 	}
 	b.Render()
+	return nil
+}
+
+func (l *listLocalTopics) printAsJson(store map[string][]string) error {
+	result, err := json.MarshalIndent(store, "", "  ")
+	if err != nil {
+		return err
+	}
+	h := internal.NewJsonHighlighter(l.style, l.globalParams.EnableColor)
+	fmt.Println(string(h.Highlight(result)))
+	return nil
 }