feat: helper functions for control catalogs

Signed-off-by: Eddie Knight <knight@linux.com>

Author: eddie-knight

control_yaml.go control.gocontrol_yaml.go renamed to control.go

@@ -43,3 +43,13 @@ func (c *Control) UnmarshalYAML(data []byte) error {
 
 	return nil
 }
+
+func (c *Control) GetMappingReferences() (refs []string) {
+	for _, ref := range c.Guidelines {
+		refs = append(refs, ref.ReferenceId)
+	}
+	for _, ref := range c.Threats {
+		refs = append(refs, ref.ReferenceId)
+	}
+	return refs
+}

control_catalog_yaml.go control_catalog.gocontrol_catalog_yaml.go renamed to control_catalog.go

@@ -1,6 +1,10 @@
 package gemara
 
-import "github.com/goccy/go-yaml"
+import (
+	"slices"
+
+	"github.com/goccy/go-yaml"
+)
 
 // UnmarshalYAML allows decoding control catalogs from older/alternate YAML schemas.
 // It supports mapping `families` -> `groups`.
@@ -38,3 +42,30 @@ func (c *ControlCatalog) UnmarshalYAML(data []byte) error {
 
 	return nil
 }
+
+func (c *ControlCatalog) GetGroupNames() (groups []string) {
+	for _, group := range c.Groups {
+		groups = append(groups, group.Title)
+	}
+	return groups
+}
+
+func (c *ControlCatalog) GetControlsForGroup(group string) (controls []Control) {
+	for _, control := range c.Controls {
+		if control.Group == group {
+			controls = append(controls, control)
+		}
+	}
+	return controls
+}
+
+func (c *ControlCatalog) GetRequirementForApplicability(applicability string) (reqs []AssessmentRequirement) {
+	for _, control := range c.Controls {
+		for _, assessment := range control.AssessmentRequirements {
+			if slices.Contains(assessment.Applicability, applicability) {
+				reqs = append(reqs, assessment)
+			}
+		}
+	}
+	return reqs
+}