-
Notifications
You must be signed in to change notification settings - Fork 3
/
render.go
105 lines (89 loc) · 2.41 KB
/
render.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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package qbcli
import (
"context"
"errors"
"fmt"
"reflect"
"github.com/QuickBase/quickbase-cli/qbclient"
"github.com/QuickBase/quickbase-cli/qberrors"
"github.com/cpliakas/cliutil"
"github.com/jedib0t/go-pretty/v6/table"
"github.com/spf13/cobra"
)
// Render renders the output in JSON, or writes an error log.
func Render(
ctx context.Context,
logger *cliutil.LeveledLogger,
cmd *cobra.Command,
cfg GlobalConfig,
v interface{},
err error,
) {
// Render the error.
if err != nil {
ctx = cliutil.ContextWithLogTag(ctx, "code", fmt.Sprintf("%v", qberrors.StatusCode(err)))
HandleError(ctx, logger, qberrors.SafeMessage(err), errors.New(qberrors.SafeDetail(err)))
}
// Do not return output.
if cfg.Quiet() {
return
}
// Try to render a table.
if cfg.Format() == "table" || cfg.Format() == "csv" || cfg.Format() == "markdown" {
rerr := renderTable(v, cfg.Format())
HandleError(ctx, logger, "error rendering table", rerr)
return
}
// Default to rendering JSON.
rerr := cliutil.PrintJSONWithFilter(v, cfg.JMESPathFilter())
HandleError(ctx, logger, "JMESPath filter not valid", rerr)
}
func renderTable(a interface{}, format string) error {
tw := table.NewWriter()
// Only pointers!
// This will panic otherwise. This is an internal function, but we whould
// be a little more defensive to prevent that from happening.
rv := reflect.ValueOf(a).Elem()
rt := reflect.TypeOf(a).Elem()
for idx := 0; idx < rt.NumField(); idx++ {
rvf := rv.Field(idx)
if !rvf.CanInterface() {
continue
}
// Look for the embedded Report field.
i := rvf.Interface()
switch i.(type) {
case qbclient.Records:
r := i.(qbclient.Records)
// map of field ids to index position in the table.
fmap := make(map[int]int, len(r.Fields))
// Add the header.
header := make(table.Row, len(r.Fields))
for idx, f := range r.Fields {
header[idx] = f.Label
fmap[f.FieldID] = idx
}
tw.AppendHeader(header)
// Add the table data.
data := make([]table.Row, len(r.Data))
for idx, row := range r.Data {
data[idx] = make(table.Row, len(row))
for fid, record := range row {
data[idx][fmap[fid]] = record.Value.String()
}
}
tw.AppendRows(data)
}
}
switch format {
case "table":
fmt.Println(tw.Render())
case "csv":
fmt.Println(tw.RenderCSV())
case "markdown":
fmt.Println(tw.RenderMarkdown())
default:
return fmt.Errorf("%s: format not valid", format)
}
return nil
}