Skip to content

Commit

Permalink
[wip] add cataloger list command
Browse files Browse the repository at this point in the history
Signed-off-by: Alex Goodman <wagoodman@users.noreply.github.com>
  • Loading branch information
wagoodman committed Dec 18, 2023
1 parent 2550e62 commit 4ec140f
Show file tree
Hide file tree
Showing 7 changed files with 265 additions and 229 deletions.
1 change: 1 addition & 0 deletions cmd/syft/cli/cli.go
Original file line number Diff line number Diff line change
Expand Up @@ -85,6 +85,7 @@ func create(id clio.Identification, out io.Writer) (clio.Application, *cobra.Com
// add sub-commands
rootCmd.AddCommand(
packagesCmd,
commands.Cataloger(app),
commands.Attest(app),
commands.Convert(app),
clio.VersionCommand(id),
Expand Down
2 changes: 1 addition & 1 deletion cmd/syft/cli/commands/attest.go
Original file line number Diff line number Diff line change
Expand Up @@ -231,7 +231,7 @@ func attestCommand(sbomFilepath string, opts *attestOptions, userInput string) (
}

func predicateType(outputName string) string {
// Select Cosign predicate type based on defined output type
// select the Cosign predicate type based on defined output type
// As orientation, check: https://github.com/sigstore/cosign/blob/main/pkg/cosign/attestation/attestation.go
switch strings.ToLower(outputName) {
case "cyclonedx-json":
Expand Down
20 changes: 20 additions & 0 deletions cmd/syft/cli/commands/cataloger.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
package commands

import (
"github.com/spf13/cobra"

"github.com/anchore/clio"
)

func Cataloger(app clio.Application) *cobra.Command {
cmd := &cobra.Command{
Use: "cataloger",
Short: "Show available catalogers and configuration",
}

cmd.AddCommand(
CatalogerList(app),
)

return cmd
}
235 changes: 235 additions & 0 deletions cmd/syft/cli/commands/cataloger_list.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,235 @@
package commands

import (
"encoding/json"
"fmt"
"sort"
"strings"

"github.com/charmbracelet/lipgloss"
"github.com/jedib0t/go-pretty/v6/table"
"github.com/scylladb/go-set/strset"
"github.com/spf13/cobra"

"github.com/anchore/clio"
"github.com/anchore/syft/internal/bus"
"github.com/anchore/syft/internal/task"
)

type catalogerListOptions struct {
Output string `yaml:"output" json:"output" mapstructure:"output"`
DefaultCatalogers []string `yaml:"default-catalogers" json:"default-catalogers" mapstructure:"default-catalogers"`
SelectCatalogers []string `yaml:"select-catalogers" json:"select-catalogers" mapstructure:"select-catalogers"`
ShowHidden bool `yaml:"show-hidden" json:"show-hidden" mapstructure:"show-hidden"`
}

func (o *catalogerListOptions) AddFlags(flags clio.FlagSet) {
flags.StringVarP(&o.Output, "output", "o", "format to output the cataloger list (available: table, json)")

flags.StringArrayVarP(&o.DefaultCatalogers, "override-default-catalogers", "", "override the default catalogers with an expression")

flags.StringArrayVarP(&o.SelectCatalogers, "select-catalogers", "", "select catalogers with an expression")

flags.BoolVarP(&o.ShowHidden, "show-hidden", "s", "show catalogers that have been de-selected")
}

func CatalogerList(app clio.Application) *cobra.Command {
opts := &catalogerListOptions{
DefaultCatalogers: []string{"all"},
}

return app.SetupCommand(&cobra.Command{
Use: "list [OPTIONS]",
Short: "List available catalogers",
RunE: func(cmd *cobra.Command, args []string) error {
return runCatalogerList(opts)
},
}, opts)
}

func runCatalogerList(opts *catalogerListOptions) error {
factories := task.DefaultPackageTaskFactories()
allTasks, err := factories.Tasks(task.DefaultCatalogingFactoryConfig())
if err != nil {
return fmt.Errorf("unable to create cataloger tasks: %w", err)
}

selectedTasks, selectionEvidence, err := task.Select(allTasks, opts.DefaultCatalogers, opts.SelectCatalogers)
if err != nil {
return fmt.Errorf("unable to select catalogers: %w", err)
}

var report string

switch opts.Output {
case "json":
report, err = renderCatalogerListJSON(selectedTasks, opts.SelectCatalogers)
case "table", "":
if opts.ShowHidden {
report = renderCatalogerListTable(allTasks, &selectionEvidence, opts.SelectCatalogers)
} else {
report = renderCatalogerListTable(selectedTasks, &selectionEvidence, opts.SelectCatalogers)
}
}

if err != nil {
return fmt.Errorf("unable to render cataloger list: %w", err)
}

bus.Report(report)

return nil
}

func renderCatalogerListJSON(tasks []task.Task, expressions []string) (string, error) {
type node struct {
Name string `json:"name"`
Tags []string `json:"tags"`
}

names, tagsByName := extractTaskInfo(tasks)

nodesByName := make(map[string]node)

for name, tags := range tagsByName {
if tags == nil {
// ensure collections are not null
tags = []string{}
}
nodesByName[name] = node{
Name: name,
Tags: tags,
}
}

type document struct {
Expressions []string `json:"expressions"`
Catalogers []node `json:"catalogers"`
}

if expressions == nil {
// ensure collections are not null
expressions = []string{}
}

doc := document{
Expressions: expressions,
}

for _, name := range names {
doc.Catalogers = append(doc.Catalogers, nodesByName[name])
}

by, err := json.Marshal(doc)

return string(by), err
}

func renderCatalogerListTable(tasks []task.Task, selection *task.Selection, expressions []string) string {
t := table.NewWriter()
t.SetStyle(table.StyleLight)
t.AppendHeader(table.Row{"Cataloger", "Tags"})

names, tagsByName := extractTaskInfo(tasks)

rowsByName := make(map[string]table.Row)

for name, tags := range tagsByName {
rowsByName[name] = formatRow(name, tags, selection)
}

for _, name := range names {
t.AppendRow(rowsByName[name])
}

report := t.Render()

if len(expressions) > 0 {
header := "Selected by expressions:\n"
for _, expr := range expressions {
header += fmt.Sprintf(" - %q\n", expr)
}
report = header + report
}

return report
}

func formatRow(name string, tags []string, selection *task.Selection) table.Row {
if selection == nil {
return table.Row{name, strings.Join(tags, ", ")}
}

isIncluded := selection.Result.Has(name)
var selections *task.TokenSelection
if s, exists := selection.TokensByTask[name]; exists {
selections = &s
}

var formattedTags []string
for _, tag := range tags {
formattedTags = append(formattedTags, formatToken(tag, selections, isIncluded))
}

var tagStr string
if isIncluded {
tagStr = strings.Join(formattedTags, ", ")
} else {
tagStr = strings.Join(formattedTags, grey.Render(", "))
}

// TODO: selection should keep warnings (non-selections) in struct

return table.Row{
formatToken(name, selections, isIncluded),
tagStr,
}
}

var (
green = lipgloss.NewStyle().Foreground(lipgloss.Color("10")) // hi green
grey = lipgloss.NewStyle().Foreground(lipgloss.Color("8")) // dark grey
red = lipgloss.NewStyle().Foreground(lipgloss.Color("9")) // high red
)

func formatToken(token string, selection *task.TokenSelection, included bool) string {
if included && selection != nil {
// format all tokens in selection in green
if selection.SelectedOn.Has(token) {
return green.Render(token)
}

return token
}

// format all tokens in selection in red, all others in grey
if selection != nil && selection.DeselectedOn.Has(token) {
return red.Render(token)
}

return grey.Render(token)
}

func extractTaskInfo(tasks []task.Task) ([]string, map[string][]string) {
tagsByName := make(map[string][]string)
var names []string

for _, tsk := range tasks {
var tags []string
name := tsk.Name()

if s, ok := tsk.(task.Selector); ok {
set := strset.New(s.Selectors()...)
set.Remove(name)
tags = set.List()
sort.Strings(tags)
}

tagsByName[name] = tags
names = append(names, name)
}

sort.Strings(names)

return names, tagsByName
}
Loading

0 comments on commit 4ec140f

Please sign in to comment.