mirrors/crowdsec
1
0
Fork 0
mirror of https://github.com/crowdsecurity/crowdsec.git synced 2025-05-11 12:25:53 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Alert inspect improvement / Use correct CSV output when listing in raw format (#1127)

Browse source
This commit is contained in:
AlteredCoder 2021-12-29 14:08:47 +01:00 • committed by GitHub
parent 3105897f37
commit 9c8ca5c73a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
10 changed files with 109 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

34
cmd/crowdsec-cli/utils.go
View file

@ -1,6 +1,7 @@
package main
import (
"encoding/csv"
"encoding/json"
"fmt"
"io/ioutil"
@ -101,7 +102,7 @@ func setHubBranch() error {
return nil
}
func ListItem(itemType string, args []string) {
func ListItem(itemType string, args []string, showType bool, showHeader bool) {
var hubStatus []map[string]string
@ -131,13 +132,40 @@ func ListItem(itemType string, args []string) {
}
fmt.Printf("%s", string(x))
} else if csConfig.Cscli.Output == "raw" {
fmt.Printf("name,status,version,description\n")
csvwriter := csv.NewWriter(os.Stdout)
if showHeader {
if showType {
err := csvwriter.Write([]string{"name", "status", "version", "description", "type"})
if err != nil {
log.Fatalf("failed to write header: %s", err)
}
} else {
err := csvwriter.Write([]string{"name", "status", "version", "description"})
if err != nil {
log.Fatalf("failed to write header: %s", err)
}
}
}
for _, v := range hubStatus {
if v["local_version"] == "" {
v["local_version"] = "n/a"
}
fmt.Printf("%s,%s,%s,%s\n", v["name"], v["status"], v["local_version"], v["description"])
row := []string{
v["name"],
v["status"],
v["local_version"],
v["description"],
}
if showType {
row = append(row, itemType)
}
err := csvwriter.Write(row)
if err != nil {
log.Fatalf("failed to write raw output : %s", err)
}
}
csvwriter.Flush()
}
}