47 lines
975 B
Go
47 lines
975 B
Go
|
package format
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"github.com/olekukonko/tablewriter"
|
||
|
|
"gopkg.in/yaml.v2"
|
||
|
|
"io"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type consoleWriterFactory func(io.Writer) ConsoleWriter
|
||
|
|
|
||
|
|
var (
|
||
|
|
writers = map[string]consoleWriterFactory{
|
||
|
|
"table": func(writer io.Writer) ConsoleWriter {
|
||
|
|
tw := tablewriter.NewWriter(writer)
|
||
|
|
tw.SetBorders(tablewriter.Border{Left: true, Top: false, Right: true, Bottom: false})
|
||
|
|
tw.SetCenterSeparator("|")
|
||
|
|
|
||
|
|
return &tblWriter{
|
||
|
|
tableWriter: tw,
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"json": func(writer io.Writer) ConsoleWriter {
|
||
|
|
return &jsonWriter{
|
||
|
|
encoder: json.NewEncoder(writer),
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"yaml": func(writer io.Writer) ConsoleWriter {
|
||
|
|
return &yamlWriter{
|
||
|
|
encoder: yaml.NewEncoder(writer),
|
||
|
|
}
|
||
|
|
},
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func Writer(format string, writer io.Writer) ConsoleWriter {
|
||
|
|
if cw, ok := writers[strings.ToLower(format)]; ok {
|
||
|
|
return cw(writer)
|
||
|
|
}
|
||
|
|
return writers["table"](writer)
|
||
|
|
}
|
||
|
|
|
||
|
|
type ConsoleWriter interface {
|
||
|
|
Write(in interface{}) (err error)
|
||
|
|
}
|