astra
/
doggo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: table formatting

pull/2/head
Karan Sharma 2020-12-13 14:12:24 +05:30
parent 90924eb27a
commit 6bb9a78492
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
cmd/output.go
View File

@ -66,18 +66,28 @@ func (hub *Hub) outputJSON(out []Output, msgs []resolvers.Response) {
func (hub *Hub) outputTerminal(out []Output) { func (hub *Hub) outputTerminal(out []Output) {
green := color.New(color.FgGreen).SprintFunc() green := color.New(color.FgGreen).SprintFunc()
blue := color.New(color.FgBlue).SprintFunc() blue := color.New(color.FgBlue).SprintFunc()
magenta := color.New(color.FgMagenta).SprintFunc()
table := tablewriter.NewWriter(os.Stdout) table := tablewriter.NewWriter(os.Stdout)
header := []string{"Name", "Type", "Class", "TTL", "Address"} header := []string{"Name", "Type", "Class", "TTL", "Address", "Nameserver"}
if hub.QueryFlags.DisplayTimeTaken { if hub.QueryFlags.DisplayTimeTaken {
header = append(header, "Time Taken") header = append(header, "Time Taken")
} }
table.SetHeader(header)
table.SetAutoWrapText(false) table.SetAutoWrapText(false)
table.SetAutoFormatHeaders(true) table.SetAutoFormatHeaders(true)
table.SetHeader(header) table.SetHeaderAlignment(tablewriter.ALIGN_LEFT)
table.SetAlignment(tablewriter.ALIGN_LEFT)
table.SetCenterSeparator("")
table.SetColumnSeparator("")
table.SetRowSeparator("")
table.SetHeaderLine(false)
table.SetBorder(false)
table.SetTablePadding("\t") // pad with tabs
table.SetNoWhiteSpace(true)
for _, o := range out { for _, o := range out {
output := []string{green(o.Name), blue(o.Type), o.Class, o.TTL, o.Address} output := []string{green(o.Name), blue(o.Type), o.Class, o.TTL, magenta(o.Address), o.Nameserver}
// Print how long it took // Print how long it took
if hub.QueryFlags.DisplayTimeTaken { if hub.QueryFlags.DisplayTimeTaken {
output = append(output, o.TimeTaken) output = append(output, o.TimeTaken)