diff --git a/core/cautils/datastructures.go b/core/cautils/datastructures.go index 555e2f87..74920988 100644 --- a/core/cautils/datastructures.go +++ b/core/cautils/datastructures.go @@ -62,6 +62,11 @@ func NewOPASessionObjMock() *OPASessionObj { ReportID: "", JobID: "", }, + Metadata: &reporthandlingv2.Metadata{ + ScanMetadata: reporthandlingv2.ScanMetadata{ + ScanningTarget: 0, + }, + }, } } diff --git a/core/cautils/scaninfo.go b/core/cautils/scaninfo.go index 4bd61a39..0b761b4a 100644 --- a/core/cautils/scaninfo.go +++ b/core/cautils/scaninfo.go @@ -266,10 +266,23 @@ func scanInfoToScanMetadata(scanInfo *ScanInfo) *reporthandlingv2.Metadata { if len(scanInfo.InputPatterns) > 0 { inputFiles = scanInfo.InputPatterns[0] } - - metadata.ScanMetadata.ScanningTarget = reporthandlingv2.Cluster - if GetScanningContext(inputFiles) != ContextCluster { + switch GetScanningContext(inputFiles) { + case ContextCluster: + // cluster + metadata.ScanMetadata.ScanningTarget = reporthandlingv2.Cluster + case ContextFile: + // local file metadata.ScanMetadata.ScanningTarget = reporthandlingv2.File + case ContextGitURL: + // url + metadata.ScanMetadata.ScanningTarget = reporthandlingv2.Repo + case ContextGitLocal: + // local-git + metadata.ScanMetadata.ScanningTarget = reporthandlingv2.GitLocal + case ContextDir: + // directory + metadata.ScanMetadata.ScanningTarget = reporthandlingv2.Directory + } setContextMetadata(&metadata.ContextMetadata, inputFiles) diff --git a/core/pkg/opaprocessor/processorhandler.go b/core/pkg/opaprocessor/processorhandler.go index ad14287a..ae8d0836 100644 --- a/core/pkg/opaprocessor/processorhandler.go +++ b/core/pkg/opaprocessor/processorhandler.go @@ -18,6 +18,7 @@ import ( "github.com/armosec/k8s-interface/workloadinterface" + reporthandlingv2 "github.com/armosec/opa-utils/reporthandling/v2" "github.com/armosec/opa-utils/resources" "github.com/open-policy-agent/opa/ast" "github.com/open-policy-agent/opa/rego" @@ -62,7 +63,13 @@ func (opap *OPAProcessor) ProcessRulesListenner() error { } func (opap *OPAProcessor) Process(policies *cautils.Policies) error { - logger.L().Info("Scanning", helpers.String("cluster", cautils.ClusterName)) + targetScan := opap.OPASessionObj.Metadata.ScanMetadata.ScanningTarget.String() + clusterScan := reporthandlingv2.Cluster + if clusterScan.String() == targetScan { + logger.L().Info("Scanning", helpers.String(targetScan, cautils.ClusterName)) + } else { + logger.L().Info("Scanning " + targetScan) + } cautils.StartSpinner() @@ -89,7 +96,12 @@ func (opap *OPAProcessor) Process(policies *cautils.Policies) error { opap.Report.ReportGenerationTime = time.Now().UTC() cautils.StopSpinner() - logger.L().Success("Done scanning", helpers.String("cluster", cautils.ClusterName)) + + if clusterScan.String() == targetScan { + logger.L().Success("Done scanning", helpers.String(targetScan, cautils.ClusterName)) + } else { + logger.L().Info("Done scanning " + targetScan) + } return errs } diff --git a/go.mod b/go.mod index 2bbc51a5..10673db7 100644 --- a/go.mod +++ b/go.mod @@ -6,7 +6,7 @@ require ( github.com/armosec/armoapi-go v0.0.97 github.com/armosec/go-git-url v0.0.13 github.com/armosec/k8s-interface v0.0.78 - github.com/armosec/opa-utils v0.0.159 + github.com/armosec/opa-utils v0.0.160 github.com/armosec/rbac-utils v0.0.14 github.com/armosec/utils-go v0.0.7 github.com/armosec/utils-k8s-go v0.0.7 diff --git a/go.sum b/go.sum index ede08f18..f78bd181 100644 --- a/go.sum +++ b/go.sum @@ -184,8 +184,8 @@ github.com/armosec/go-git-url v0.0.13 h1:kwzHBL7oFqf2UsR0h4Sx+HMYqO9xHHuBFXGIyuM github.com/armosec/go-git-url v0.0.13/go.mod h1:GzfssG3IW9KiURSpK7c/bySBRTlghpObQ7NQ1O4hcMI= github.com/armosec/k8s-interface v0.0.78 h1:zqIzbFQqSxVMjcDsB4+lgGNgkHVqDZ369/WTGGdLKCE= github.com/armosec/k8s-interface v0.0.78/go.mod h1:8NX4xWXh8mwW7QyZdZea1czNdM2azCK9BbUNmiZYXW0= -github.com/armosec/opa-utils v0.0.159 h1:GREyXsM8v1tdIY6FDYjM7roXozt191JTjwpfrjkDJIU= -github.com/armosec/opa-utils v0.0.159/go.mod h1:ce7GrjUmp4A2bkw6ItJxnE1GeDeVQxwb3HiwKBbEZTA= +github.com/armosec/opa-utils v0.0.160 h1:KgYN57Fh4t0+EELE/napAB+ysX1X3kDg28Sh34hxt1Y= +github.com/armosec/opa-utils v0.0.160/go.mod h1:ce7GrjUmp4A2bkw6ItJxnE1GeDeVQxwb3HiwKBbEZTA= github.com/armosec/rbac-utils v0.0.14 h1:CKYKcgqJEXWF2Hen/B1pVGtS3nDAG1wp9dDv6oNtq90= github.com/armosec/rbac-utils v0.0.14/go.mod h1:Ex/IdGWhGv9HZq6Hs8N/ApzCKSIvpNe/ETqDfnuyah0= github.com/armosec/utils-go v0.0.7 h1:YGyIzfo7JKocx0SVrKEpQLSKLNLgm+JOEeH0XkO23yQ=