mirror of
https://github.com/kubescape/kubescape.git
synced 2026-04-15 06:58:11 +00:00
44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
package score
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/armosec/opa-utils/score"
|
|
|
|
"github.com/armosec/kubescape/v2/core/cautils"
|
|
)
|
|
|
|
/* provides a wrapper for scoreUtils, since there's no common interface between postureReportV1 and PostureReportV2
|
|
and the need of concrete objects
|
|
I've decided to create scoreWrapper that will allow calculating score regardless (as long as opaSessionObj is there)
|
|
*/
|
|
type ScoreWrapper struct {
|
|
scoreUtil *score.ScoreUtil
|
|
opaSessionObj *cautils.OPASessionObj
|
|
}
|
|
|
|
type PostureReportVersion string
|
|
|
|
const (
|
|
EPostureReportV1 PostureReportVersion = "v1"
|
|
EPostureReportV2 PostureReportVersion = "V2"
|
|
)
|
|
|
|
func (su *ScoreWrapper) Calculate(reportVersion PostureReportVersion) error {
|
|
switch reportVersion {
|
|
case EPostureReportV1:
|
|
return su.scoreUtil.Calculate(su.opaSessionObj.PostureReport.FrameworkReports)
|
|
case EPostureReportV2:
|
|
return su.scoreUtil.CalculatePostureReportV2(su.opaSessionObj.Report)
|
|
}
|
|
|
|
return fmt.Errorf("unsupported score calculator")
|
|
}
|
|
|
|
func NewScoreWrapper(opaSessionObj *cautils.OPASessionObj) *ScoreWrapper {
|
|
return &ScoreWrapper{
|
|
scoreUtil: score.NewScore(opaSessionObj.AllResources),
|
|
opaSessionObj: opaSessionObj,
|
|
}
|
|
}
|