fix(iam): pass all document metadata fields to rego

aquasecurity · Jan 18, 2024 · 8fb98dc · 8fb98dc
1 parent 87a2031
commit 8fb98dc
Showing 1 changed file with 16 additions and 8 deletions.
diff --git a/pkg/providers/aws/iam/iam.go b/pkg/providers/aws/iam/iam.go
@@ -36,15 +36,23 @@ type Document struct {
 func (d Document) ToRego() interface{} {
 	m := d.Metadata
 	doc, _ := d.Parsed.MarshalJSON()
-	return map[string]interface{}{
-		"filepath":  m.Range().GetFilename(),
-		"startline": m.Range().GetStartLine(),
-		"endline":   m.Range().GetEndLine(),
-		"managed":   m.IsManaged(),
-		"explicit":  m.IsExplicit(),
-		"value":     string(doc),
-		"fskey":     defsecTypes.CreateFSKey(m.Range().GetFS()),
+	input := map[string]interface{}{
+		"filepath":     m.Range().GetFilename(),
+		"startline":    m.Range().GetStartLine(),
+		"endline":      m.Range().GetEndLine(),
+		"managed":      m.IsManaged(),
+		"explicit":     m.IsExplicit(),
+		"value":        string(doc),
+		"sourceprefix": m.Range().GetSourcePrefix(),
+		"fskey":        defsecTypes.CreateFSKey(m.Range().GetFS()),
+		"resource":     m.Reference(),
 	}
+
+	if m.Parent() != nil {
+		input["parent"] = m.Parent().ToRego()
+	}
+
+	return input
 }
 
 type Group struct {