feat: Extend replaceAll to all stringish types

chezmoi · Mar 5, 2024 · 994c83e · 994c83e
1 parent 9965c51
commit 994c83e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/templatefuncs.go b/templatefuncs.go
@@ -43,8 +43,10 @@ func NewFuncMap() template.FuncMap {
 		"prefixLines":      prefixLinesTemplateFunc,
 		"quote":            eachString(strconv.Quote),
 		"regexpReplaceAll": regexpReplaceAllTemplateFunc,
-		"replaceAll": func(old, new, s string) string {
-			return strings.ReplaceAll(s, old, new)
+		"replaceAll": func(old, new_ string, v any) any {
+			return eachString(func(s string) any {
+				return strings.ReplaceAll(s, old, new_)
+			})(v)
 		},
 		"stat":      eachString(statTemplateFunc),
 		"toJSON":    toJSONTemplateFunc,

diff --git a/templatefuncs_test.go b/templatefuncs_test.go
@@ -122,6 +122,10 @@ func TestFuncMap(t *testing.T) {
 			template: `{{ "abcba" | replaceAll "b" "d" }}`,
 			expected: `adcda`,
 		},
+		{
+			template: `{{ list "abc" "cba" | replaceAll "b" "d" }}`,
+			expected: "[adc cda]",
+		},
 		{
 			template: `{{ quote "a" }}`,
 			expected: `"a"`,