/
printers.go
56 lines (49 loc) · 1.44 KB
/
printers.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
package cmd
import (
"io"
"log"
"github.com/fatih/color"
"github.com/metal-stack/metal-lib/pkg/genericcli/printers"
"github.com/metal-stack/metalctl/cmd/tableprinters"
"github.com/spf13/viper"
)
func newPrinterFromCLI(out io.Writer) printers.Printer {
var printer printers.Printer
switch format := viper.GetString("output-format"); format {
case "yaml":
printer = printers.NewYAMLPrinter().WithOut(out)
case "json":
printer = printers.NewJSONPrinter().WithOut(out)
case "table", "wide", "markdown":
tp := tableprinters.New()
cfg := &printers.TablePrinterConfig{
ToHeaderAndRows: tp.ToHeaderAndRows,
Wide: format == "wide",
Markdown: format == "markdown",
NoHeaders: viper.GetBool("no-headers"),
}
tablePrinter := printers.NewTablePrinter(cfg).WithOut(out)
tp.SetPrinter(tablePrinter)
tp.SetLastEventErrorThreshold(viper.GetDuration("last-event-error-threshold"))
printer = tablePrinter
case "template":
printer = printers.NewTemplatePrinter(viper.GetString("template")).WithOut(out)
default:
log.Fatalf("unknown output format: %q", format)
}
if viper.IsSet("force-color") {
enabled := viper.GetBool("force-color")
if enabled {
color.NoColor = false
} else {
color.NoColor = true
}
}
return printer
}
func defaultToYAMLPrinter(out io.Writer) printers.Printer {
if viper.IsSet("output-format") {
return newPrinterFromCLI(out)
}
return printers.NewYAMLPrinter().WithOut(out)
}