fix(helm): escape consecutive commas in cleanSetParameters (#19269) (#…

…20113) Signed-off-by: KangManJoo <[email protected]> Signed-off-by: daengdaengLee <[email protected]> Co-authored-by: daengdaengLee <[email protected]>
argoproj · Oct 2, 2024 · 3070297 · 3070297
1 parent 1a84db5
commit 3070297
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 6 deletions.
diff --git a/util/helm/cmd.go b/util/helm/cmd.go
@@ -342,19 +342,36 @@ type TemplateOpts struct {
 	SkipCrds    bool
 }
 
-var (
-	re                 = regexp.MustCompile(`([^\\]),`)
-	apiVersionsRemover = regexp.MustCompile(`(--api-versions [^ ]+ )+`)
-)
-
 func cleanSetParameters(val string) string {
 	// `{}` equal helm list parameters format, so don't escape `,`.
 	if strings.HasPrefix(val, `{`) && strings.HasSuffix(val, `}`) {
 		return val
 	}
-	return re.ReplaceAllString(val, `$1\,`)
+
+	val = replaceAllWithLookbehind(val, ',', `\,`, '\\')
+	return val
 }
 
+func replaceAllWithLookbehind(val string, old rune, new string, lookbehind rune) string {
+	var result strings.Builder
+	var prevR rune
+	for _, r := range val {
+		if r == old {
+			if prevR != lookbehind {
+				result.WriteString(new)
+			} else {
+				result.WriteRune(old)
+			}
+		} else {
+			result.WriteRune(r)
+		}
+		prevR = r
+	}
+	return result.String()
+}
+
+var apiVersionsRemover = regexp.MustCompile(`(--api-versions [^ ]+ )+`)
+
 func (c *Cmd) template(chartPath string, opts *TemplateOpts) (string, string, error) {
 	if callback, err := cleanupChartLockFile(filepath.Clean(path.Join(c.WorkDir, chartPath))); err == nil {
 		defer callback()

diff --git a/util/helm/helm_test.go b/util/helm/helm_test.go
@@ -166,6 +166,8 @@ func TestHelmArgCleaner(t *testing.T) {
 		`not, clean`: `not\, clean`,
 		`a\,b,c`:     `a\,b\,c`,
 		`{a,b,c}`:    `{a,b,c}`,
+		`,,,,,\,`:    `\,\,\,\,\,\,`,
+		`\,,\\,,`:    `\,\,\\,\,`,
 	} {
 		cleaned := cleanSetParameters(input)
 		assert.Equal(t, expected, cleaned)