mirror of
https://github.com/kubescape/kubescape.git
synced 2026-04-15 06:58:11 +00:00
75 lines
2.0 KiB
Go
75 lines
2.0 KiB
Go
package printer
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/kubescape/go-logger"
|
|
"github.com/kubescape/go-logger/helpers"
|
|
"github.com/kubescape/k8s-interface/workloadinterface"
|
|
"github.com/kubescape/kubescape/v3/core/cautils"
|
|
"github.com/kubescape/kubescape/v3/core/pkg/resultshandling/printer"
|
|
"github.com/kubescape/opa-utils/reporthandling/results/v1/reportsummary"
|
|
"github.com/kubescape/opa-utils/reporthandling/results/v1/resourcesresults"
|
|
)
|
|
|
|
var _ printer.IPrinter = &PrometheusPrinter{}
|
|
|
|
type PrometheusPrinter struct {
|
|
writer *os.File
|
|
verboseMode bool
|
|
}
|
|
|
|
func NewPrometheusPrinter(verboseMode bool) *PrometheusPrinter {
|
|
return &PrometheusPrinter{
|
|
verboseMode: verboseMode,
|
|
}
|
|
}
|
|
|
|
func (pp *PrometheusPrinter) PrintNextSteps() {
|
|
|
|
}
|
|
|
|
func (pp *PrometheusPrinter) SetWriter(ctx context.Context, outputFile string) {
|
|
pp.writer = printer.GetWriter(ctx, outputFile)
|
|
}
|
|
|
|
func (pp *PrometheusPrinter) Score(score float32) {
|
|
// Handle invalid scores
|
|
if score > 100 {
|
|
score = 100
|
|
} else if score < 0 {
|
|
score = 0
|
|
}
|
|
|
|
fmt.Printf("\n# Overall compliance-score (100- Excellent, 0- All failed)\nkubescape_score %d\n", cautils.Float32ToInt(score))
|
|
}
|
|
|
|
func (pp *PrometheusPrinter) generatePrometheusFormat(
|
|
resources map[string]workloadinterface.IMetadata,
|
|
results map[string]resourcesresults.Result,
|
|
summaryDetails *reportsummary.SummaryDetails) *Metrics {
|
|
|
|
m := &Metrics{}
|
|
m.setComplianceScores(summaryDetails)
|
|
// m.setResourcesCounters(resources, results)
|
|
|
|
return m
|
|
}
|
|
|
|
func (pp *PrometheusPrinter) ActionPrint(ctx context.Context, opaSessionObj *cautils.OPASessionObj, imageScanData []cautils.ImageScanData) {
|
|
if opaSessionObj == nil {
|
|
logger.L().Ctx(ctx).Error("failed to print results, missing data")
|
|
return
|
|
}
|
|
|
|
metrics := pp.generatePrometheusFormat(opaSessionObj.AllResources, opaSessionObj.ResourcesResult, &opaSessionObj.Report.SummaryDetails)
|
|
|
|
if _, err := pp.writer.Write([]byte(metrics.String())); err != nil {
|
|
logger.L().Ctx(ctx).Error("failed to write results", helpers.Error(err))
|
|
return
|
|
}
|
|
printer.LogOutputFile(pp.writer.Name())
|
|
}
|