mirror of
https://github.com/kubescape/kubescape.git
synced 2026-04-15 06:58:11 +00:00
57 lines
1.6 KiB
Go
57 lines
1.6 KiB
Go
package printer
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
logger "github.com/kubescape/go-logger"
|
|
"github.com/kubescape/go-logger/helpers"
|
|
"github.com/kubescape/k8s-interface/workloadinterface"
|
|
"github.com/kubescape/kubescape/v2/core/cautils"
|
|
"github.com/kubescape/kubescape/v2/core/pkg/resultshandling/printer"
|
|
"github.com/kubescape/opa-utils/reporthandling/results/v1/reportsummary"
|
|
"github.com/kubescape/opa-utils/reporthandling/results/v1/resourcesresults"
|
|
)
|
|
|
|
type PrometheusPrinter struct {
|
|
writer *os.File
|
|
verboseMode bool
|
|
}
|
|
|
|
func NewPrometheusPrinter(verboseMode bool) *PrometheusPrinter {
|
|
return &PrometheusPrinter{
|
|
verboseMode: verboseMode,
|
|
}
|
|
}
|
|
|
|
func (pp *PrometheusPrinter) SetWriter(outputFile string) {
|
|
pp.writer = printer.GetWriter(outputFile)
|
|
}
|
|
|
|
func (pp *PrometheusPrinter) Score(score float32) {
|
|
fmt.Printf("\n# Overall risk-score (0- Excellent, 100- 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.setRiskScores(summaryDetails)
|
|
// m.setResourcesCounters(resources, results)
|
|
|
|
return m
|
|
}
|
|
|
|
func (pp *PrometheusPrinter) ActionPrint(opaSessionObj *cautils.OPASessionObj) {
|
|
|
|
metrics := pp.generatePrometheusFormat(opaSessionObj.AllResources, opaSessionObj.ResourcesResult, &opaSessionObj.Report.SummaryDetails)
|
|
|
|
if _, err := pp.writer.Write([]byte(metrics.String())); err != nil {
|
|
logger.L().Error("failed to write results", helpers.Error(err))
|
|
} else {
|
|
printer.LogOutputFile(pp.writer.Name())
|
|
}
|
|
}
|