feat: add labels flag

Signed-off-by: Charles-Edouard Brétéché <charles.edouard@nirmata.com>

pkg/commands/scan/command.go

@@ -17,5 +17,6 @@ func Command() *cobra.Command {
 	cmd.Flags().StringVar(&command.payload, "payload", "", "Path to payload (json or yaml file)")
 	cmd.Flags().StringSliceVar(&command.preprocessors, "pre-process", nil, "JmesPath expression used to pre process payload")
 	cmd.Flags().StringSliceVar(&command.policies, "policy", nil, "Path to kyverno-json policies")
+	cmd.Flags().StringSliceVar(&command.selectors, "labels", nil, "Labels selectors for policies")
 	return cmd
 }

pkg/commands/scan/options.go

@@ -4,20 +4,24 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"strings"
 
+	"github.com/kyverno/kyverno-json/pkg/apis/v1alpha1"
 	"github.com/kyverno/kyverno-json/pkg/engine/template"
 	jsonengine "github.com/kyverno/kyverno-json/pkg/json-engine"
 	"github.com/kyverno/kyverno-json/pkg/payload"
 	"github.com/kyverno/kyverno-json/pkg/policy"
 	"github.com/kyverno/kyverno/cmd/cli/kubectl-kyverno/output/pluralize"
 	"github.com/spf13/cobra"
 	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
+	"k8s.io/apimachinery/pkg/labels"
 )
 
 type options struct {
 	payload       string
 	preprocessors []string
 	policies      []string
+	selectors     []string
 }
 
 func (c *options) run(cmd *cobra.Command, _ []string) error {
@@ -27,6 +31,23 @@ func (c *options) run(cmd *cobra.Command, _ []string) error {
 	if err != nil {
 		return err
 	}
+	selector := labels.Everything()
+	if len(c.selectors) != 0 {
+		parsed, err := labels.Parse(strings.Join(c.selectors, ","))
+		if err != nil {
+			return err
+		}
+		selector = parsed
+	}
+	{
+		var filteredPolicies []*v1alpha1.Policy
+		for _, policy := range policies {
+			if selector.Matches(labels.Set(policy.Labels)) {
+				filteredPolicies = append(filteredPolicies, policy)
+			}
+		}
+		policies = filteredPolicies
+	}
 	fmt.Fprintln(out, "Loading payload ...")
 	payload, err := payload.Load(c.payload)
 	if err != nil {