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

feat: Add short output flag

Browse source 
Ref https://github.com/mr-karan/doggo/issues/35
This commit is contained in:
Karan Sharma 2022-05-17 10:34:55 +05:30
parent 7619cbdeb0
commit eec8374e6f
6 changed files with 14 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
internal/app/output.go
View file

@ -21,6 +21,14 @@ func (app *App) outputJSON(rsp []resolvers.Response) {
fmt.Printf("%s", res)
}
func (app *App) outputShort(rsp []resolvers.Response) {
for _, r := range rsp {
for _, a := range r.Answers {
fmt.Printf("%s\n", a.Address)
}
}
}
func (app *App) outputTerminal(rsp []resolvers.Response) {
var (
green = color.New(color.FgGreen, color.Bold).SprintFunc()
@ -134,6 +142,8 @@ func (app *App) outputTerminal(rsp []resolvers.Response) {
func (app *App) Output(responses []resolvers.Response) {
if app.QueryFlags.ShowJSON {
app.outputJSON(responses)
} else if app.QueryFlags.ShortOutput {
app.outputShort(responses)
} else {
app.outputTerminal(responses)
}