Add template functions (#110)

lunarway · Nov 24, 2023 · d8ded4b · d8ded4b
1 parent b009ef3
commit d8ded4b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/template.go b/template.go
@@ -20,8 +20,13 @@ func BotMessage(data BotMessageData) (string, error) {
 		return "", errors.New("template is empty")
 	}
 
+	templateFuncs := template.FuncMap{
+		"contains":   strings.Contains,
+		"replaceAll": strings.ReplaceAll,
+	}
+
 	template := template.New("test")
-	template, err := template.Parse(data.Template)
+	template, err := template.Funcs(templateFuncs).Parse(data.Template)
 	if err != nil {
 		return "", errors.Wrapf(err, "parsing template: '%s'", data.Template)
 	}