output.go

package utils

import (
	"bytes"
	"encoding/json"
	"fmt"
	"os"

	"github.com/olekukonko/tablewriter"
	"gopkg.in/yaml.v2"
)

type Output struct {
	Format string
}

func Println(a ...interface{}) (n int, err error) {
	return fmt.Println(a...)
}

func (o *Output) Write(header []string, row Row) {
	if o.Format == "table" {
		writeTable(header, []Row{row})
	} else if o.Format == "yaml" {
		y, _ := yaml.Marshal(row)
		fmt.Println(string(y))
	} else {
		buf := new(bytes.Buffer)
		enc := json.NewEncoder(buf)
		enc.SetEscapeHTML(false)
		enc.SetIndent("", "  ")
		if err := enc.Encode(row); err != nil {
			LogErrorAndExit(err)
		}
		fmt.Println(buf.String())
	}
}

func (o *Output) WriteList(header []string, tableRows []Row) {
	if tableRows == nil {
		tableRows = []Row{}
	}
	if o.Format == "table" {
		writeTable(header, tableRows)
	} else if o.Format == "yaml" {
		y, _ := yaml.Marshal(tableRows)
		fmt.Println(string(y))
	} else {
		j, _ := json.MarshalIndent(tableRows, "", "  ")
		fmt.Println(string(j))
	}
}

type Row interface {
	DataAsStringArray() []string
}

func writeTable(header []string, tableRows []Row) {
	table := tablewriter.NewWriter(os.Stdout)
	table.SetHeader(header)
	table.SetAutoWrapText(false)
	for _, v := range tableRows {
		table.Append(v.DataAsStringArray())
	}
	table.Render()
}