fix #422, allow consumer templating to also carry information about w…

pkg/templating/template_description.go

@@ -20,13 +20,14 @@ const (
 type (
 	enrichedIssue struct {
 		*v1.EnrichedIssue
-		ToolName       string
-		ScanStartTime  string
-		ScanID         string
-		ConfidenceText string
-		SeverityText   string
-		Count          uint
-		FirstFound     string
+		ToolName            string
+		ScanStartTime       string
+		ScanID              string
+		ConfidenceText      string
+		SeverityText        string
+		Count               uint
+		FirstFound          string
+		FixedCodeAnnotation string
 	}
 
 	enrichedIssueOption func(*enrichedIssue) error
@@ -126,6 +127,14 @@ func EnrichedIssueWithFirstFound(firstFound time.Time) enrichedIssueOption {
 	}
 }
 
+// EnrichedIssueWithFixedCodeAnnotation allows to define which of the annotations contains fixed code
+func EnrichedIssueWithFixedCodeAnnotation(fixedCodeAnnotation string) enrichedIssueOption {
+	return func(ei *enrichedIssue) error {
+		ei.FixedCodeAnnotation = fixedCodeAnnotation
+		return nil
+	}
+}
+
 // TemplateStringEnriched applies the provided go template to the Enriched Issue provided and returns the resulting str
 func TemplateStringEnriched(inputTemplate string, issue *v1.EnrichedIssue, opts ...enrichedIssueOption) (*string, error) {
 	enrichedIssue := &enrichedIssue{

pkg/templating/template_description_test.go

@@ -242,6 +242,14 @@ func TestDescriptionOptions(t *testing.T) {
 				ScanID: "scan-1",
 			},
 		},
+		{
+			name:    "valid code scan annotation returns no err",
+			option:  EnrichedIssueWithFixedCodeAnnotation("fixedCode"),
+			wantErr: false,
+			expectedEnrichedIssue: &enrichedIssue{
+				FixedCodeAnnotation: "fixedCode",
+			},
+		},
 	} {
 		t.Run(tt.name, func(t *testing.T) {
 			var ei enrichedIssue