feat(helmExecute): Allow custom delimiter (SAP#4312)

Co-authored-by: Ralf Pannemans <[email protected]> Co-authored-by: Johannes Dillmann <[email protected]> Co-authored-by: Jan von Loewenstein <[email protected]>
maxatsap · Jul 23, 2024 · d4bb209 · d4bb209
1 parent 3d6fae2
commit d4bb209
Show file tree

Hide file tree

Showing 5 changed files with 74 additions and 2 deletions.
diff --git a/cmd/helmExecute.go b/cmd/helmExecute.go
@@ -175,7 +175,7 @@ func parseAndRenderCPETemplate(config helmExecuteOptions, rootPath string, utils
 		if err != nil {
 			return fmt.Errorf("failed to read file: %v", err)
 		}
-		generated, err := cpe.ParseTemplate(string(cpeTemplate))
+		generated, err := cpe.ParseTemplateWithDelimiter(string(cpeTemplate), config.TemplateStartDelimiter, config.TemplateEndDelimiter)
 		if err != nil {
 			return fmt.Errorf("failed to parse template: %v", err)
 		}

diff --git a/cmd/helmExecute_generated.go b/cmd/helmExecute_generated.go
diff --git a/pkg/piperenv/templating.go b/pkg/piperenv/templating.go
@@ -7,9 +7,16 @@ import (
 	"text/template"
 )
 
+const DEFAULT_START_DELIMITER = "{{"
+const DEFAULT_END_DELIMITER = "}}"
+
 // ParseTemplate allows to parse a template which contains references to the CPE
 // Utility functions make it simple to access specific parts of the CPE
 func (c *CPEMap) ParseTemplate(cpeTemplate string) (*bytes.Buffer, error) {
+	return c.ParseTemplateWithDelimiter(cpeTemplate, DEFAULT_START_DELIMITER, DEFAULT_END_DELIMITER)
+}
+
+func (c *CPEMap) ParseTemplateWithDelimiter(cpeTemplate string, startDelimiter string, endDelimiter string) (*bytes.Buffer, error) {
 	funcMap := template.FuncMap{
 		"cpe":         c.cpe,
 		"cpecustom":   c.custom,
@@ -21,7 +28,7 @@ func (c *CPEMap) ParseTemplate(cpeTemplate string) (*bytes.Buffer, error) {
 		// This requires alignment on artifact handling before, though
 	}
 
-	tmpl, err := template.New("cpetemplate").Funcs(funcMap).Parse(cpeTemplate)
+	tmpl, err := template.New("cpetemplate").Delims(startDelimiter, endDelimiter).Funcs(funcMap).Parse(cpeTemplate)
 	if err != nil {
 		return nil, fmt.Errorf("failed to parse cpe template '%v': %w", cpeTemplate, err)
 	}

diff --git a/pkg/piperenv/templating_test.go b/pkg/piperenv/templating_test.go
@@ -40,6 +40,35 @@ func TestParseTemplate(t *testing.T) {
 	}
 }
 
+func TestParseTemplateWithDelimiter(t *testing.T) {
+	tt := []struct {
+		template      string
+		cpe           CPEMap
+		expected      string
+		expectedError error
+	}{
+		{template: `version: [[index .CPE "artifactVersion"]], sha: [[git "commitId"]]`, expected: "version: 1.2.3, sha: thisIsMyTestSha"},
+		{template: "version: [[", expectedError: fmt.Errorf("failed to parse cpe template 'version: [['")},
+		{template: `version: [[index .CPE "artifactVersion"]], release: {{ .RELEASE }}`, expected: "version: 1.2.3, release: {{ .RELEASE }}"},
+	}
+
+	cpe := CPEMap{
+		"artifactVersion": "1.2.3",
+		"git/commitId":    "thisIsMyTestSha",
+	}
+
+	for _, test := range tt {
+		res, err := cpe.ParseTemplateWithDelimiter(test.template, "[[", "]]")
+		if test.expectedError != nil {
+			assert.Contains(t, fmt.Sprint(err), fmt.Sprint(test.expectedError))
+		} else {
+			assert.NoError(t, err)
+			assert.Equal(t, test.expected, (*res).String())
+		}
+
+	}
+}
+
 func TestTemplateFunctionCpe(t *testing.T) {
 	t.Run("CPE from object", func(t *testing.T) {
 		tt := []struct {

diff --git a/resources/metadata/helmExecute.yaml b/resources/metadata/helmExecute.yaml
@@ -345,6 +345,20 @@ spec:
           - PARAMETERS
           - STAGES
           - STEPS
+      - name: templateStartDelimiter
+        type: string
+        description: When templating value files, use this start delimiter.
+        default: "{{"
+        scope:
+          - STEPS
+          - PARAMETERS
+      - name: templateEndDelimiter
+        type: string
+        description: When templating value files, use this end delimiter.
+        default: "}}"
+        scope:
+          - STEPS
+          - PARAMETERS
   containers:
     - image: dtzar/helm-kubectl:3
       workingDir: /config