feat: change string to Check

aquasecurity · Jul 22, 2024 · e65c3df · e65c3df
1 parent 13956c6
commit e65c3df
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/pkg/iac/rules/providers.go b/pkg/iac/rules/providers.go
@@ -21,11 +21,11 @@ type Check struct {
 	Severity    string `json:"severity"`
 }
 
-func GetProvidersHierarchy() (providers map[string]map[string][]string) {
+func GetProvidersHierarchy() (providers map[string]map[string][]Check) {
 
 	registeredRules := GetRegistered()
 
-	provs := make(map[string]map[string][]string)
+	provs := make(map[string]map[string][]Check)
 
 	for _, rule := range registeredRules {
 
@@ -35,16 +35,20 @@ func GetProvidersHierarchy() (providers map[string]map[string][]string) {
 		cName := rule.GetRule().AVDID
 
 		if _, ok := provs[pName]; !ok {
-			provs[pName] = make(map[string][]string)
+			provs[pName] = make(map[string][]Check)
 		}
 
 		if _, ok := provs[pName][sName]; !ok {
-			provs[pName][sName] = make([]string, 0)
+			provs[pName][sName] = make([]Check, 0)
 		}
 
 		if _, ok := cNames[cName]; !ok {
 			cNames[cName] = true
-			provs[pName][sName] = append(provs[pName][sName], cName)
+			provs[pName][sName] = append(provs[pName][sName], Check{
+				Name:        cName,
+				Description: rule.GetRule().Summary,
+				Severity:    string(rule.GetRule().Severity),
+			})
 		}
 	}