diff --git a/cmd/doggo/help.go b/cmd/doggo/help.go index c714b7a..abd3f98 100644 --- a/cmd/doggo/help.go +++ b/cmd/doggo/help.go @@ -100,7 +100,7 @@ func renderCustomHelp() { // should ideally never happen. panic(err) } - err = tmpl.Execute(os.Stdout, helpTmplVars) + err = tmpl.Execute(color.Output, helpTmplVars) if err != nil { // should ideally never happen. panic(err) diff --git a/internal/app/output.go b/internal/app/output.go index 1c3415a..6ea0413 100644 --- a/internal/app/output.go +++ b/internal/app/output.go @@ -3,7 +3,6 @@ package app import ( "encoding/json" "fmt" - "os" "github.com/fatih/color" "github.com/miekg/dns" @@ -45,7 +44,7 @@ func (app *App) outputTerminal(rsp []resolvers.Response) { } // Conditional Time column. - table := tablewriter.NewWriter(os.Stdout) + table := tablewriter.NewWriter(color.Output) header := []string{"Name", "Type", "Class", "TTL", "Address", "Nameserver"} if app.QueryFlags.DisplayTimeTaken { header = append(header, "Time Taken")