arno/clair
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

contrib: Add colors / Modify spacing in the analyze-local-images's output

Browse Source
This commit is contained in:
Quentin Machu 2016-03-17 13:07:28 -04:00
parent 93ffc5a1e5
commit e34171025d
1 changed files with 36 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
contrib/analyze-local-images/main.go
View File

@ -32,6 +32,7 @@ import (
"github.com/coreos/clair/api/v1"
"github.com/coreos/clair/utils/types"
"github.com/fatih/color"
)
const (
@ -40,17 +41,19 @@ const (
httpPort = 9279
)
var (
endpoint = flag.String("endpoint", "http://127.0.0.1:6060", "Address to Clair API")
myAddress = flag.String("my-address", "127.0.0.1", "Address from the point of view of Clair")
minimumSeverity = flag.String("minimum-severity", "Negligible", "Minimum severity of vulnerabilities to show (Unknown, Negligible, Low, Medium, High, Critical, Defcon1)")
flagNoColor = flag.Bool("no-color", false, "Disable color output")
)
func main() {
// Parse command-line arguments.
endpoint := flag.String("endpoint", "http://127.0.0.1:6060", "Address to Clair API")
myAddress := flag.String("my-address", "127.0.0.1", "Address from the point of view of Clair")
minimumSeverity := flag.String("minimum-severity", "Negligible", "Minimum severity of vulnerabilities to show (Unknown, Negligible, Low, Medium, High, Critical, Defcon1)")
flag.Usage = func() {
fmt.Fprintf(os.Stderr, "Usage: %s [options] image-id\n\nOptions:\n", os.Args[0])
flag.PrintDefaults()
}
flag.Parse()
if len(flag.Args()) != 1 {
@ -65,6 +68,10 @@ func main() {
os.Exit(1)
}
if *flagNoColor {
color.NoColor = true
}
// Save image.
fmt.Printf("Saving %s\n", imageName)
path, err := save(imageName)
@ -139,7 +146,9 @@ func main() {
isFirstVulnerability := true
for _, vulnerability := range feature.Vulnerabilities {
if minSeverity.Compare(types.Priority(vulnerability.Severity)) > 0 {
severity := types.Priority(vulnerability.Severity)
if minSeverity.Compare(severity) > 0 {
continue
}
@ -151,7 +160,7 @@ func main() {
fmt.Printf("- Added by layer: %s\n", feature.AddedBy)
}
fmt.Printf("### (%s) %s\n", vulnerability.Severity, vulnerability.Name)
fmt.Printf("### (%s) %s\n", coloredSeverity(severity), vulnerability.Name)
if vulnerability.Description != "" {
fmt.Printf("- Link: %s\n", vulnerability.Link)
@ -349,3 +358,18 @@ func getLayer(endpoint, layerID string) (v1.Layer, error) {
return *apiResponse.Layer, nil
}
func coloredSeverity(severity types.Priority) string {
red := color.New(color.FgRed).SprintFunc()
yellow := color.New(color.FgYellow).SprintFunc()
white := color.New(color.FgWhite).SprintFunc()
switch severity {
case types.High, types.Critical:
return red(severity)
case types.Medium:
return yellow(severity)
default:
return white(severity)
}
}