add option to skip template execution

nicksnyder · Jan 27, 2024 · 0f449a0 · 0f449a0
1 parent 5257e26
commit 0f449a0
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 2 deletions.
diff --git a/v2/i18n/localizer.go b/v2/i18n/localizer.go
@@ -68,6 +68,10 @@ type LocalizeConfig struct {
 
 	// Funcs is used to extend the Go template engine's built in functions
 	Funcs template.FuncMap
+
+	// NoTemplateExec is used to disable template execution.
+	// The localized string will be returned as is.
+	NoTemplateExec bool
 }
 
 type invalidPluralCountErr struct {
@@ -152,15 +156,15 @@ func (l *Localizer) LocalizeWithTag(lc *LocalizeConfig) (string, language.Tag, e
 	}
 
 	pluralForm := l.pluralForm(tag, operands)
-	msg, err2 := template.Execute(pluralForm, templateData, lc.Funcs)
+	msg, err2 := getOrExecute(template, pluralForm, templateData, lc)
 	if err2 != nil {
 		if err == nil {
 			err = err2
 		}
 
 		// Attempt to fallback to "Other" pluralization in case translations are incomplete.
 		if pluralForm != plural.Other {
-			msg2, err3 := template.Execute(plural.Other, templateData, lc.Funcs)
+			msg2, err3 := getOrExecute(template, plural.Other, templateData, lc)
 			if err3 == nil {
 				msg = msg2
 			}
@@ -169,6 +173,13 @@ func (l *Localizer) LocalizeWithTag(lc *LocalizeConfig) (string, language.Tag, e
 	return msg, tag, err
 }
 
+func getOrExecute(template *MessageTemplate, pluralForm plural.Form, data interface{}, lc *LocalizeConfig) (string, error) {
+	if lc.NoTemplateExec {
+		return template.Get(pluralForm)
+	}
+	return template.Execute(pluralForm, data, lc.Funcs)
+}
+
 func (l *Localizer) getMessageTemplate(id string, defaultMessage *Message) (language.Tag, *MessageTemplate, error) {
 	_, i, _ := l.bundle.matcher.Match(l.tags...)
 	tag := l.bundle.tags[i]

diff --git a/v2/i18n/message_template.go b/v2/i18n/message_template.go
@@ -63,3 +63,15 @@ func (mt *MessageTemplate) Execute(pluralForm plural.Form, data interface{}, fun
 	}
 	return t.Execute(funcs, data)
 }
+
+// Get returns unprocessed template string for the plural form.
+func (mt *MessageTemplate) Get(pluralForm plural.Form) (string, error) {
+	t := mt.PluralTemplates[pluralForm]
+	if t == nil {
+		return "", pluralFormNotFoundError{
+			pluralForm: pluralForm,
+			messageID:  mt.Message.ID,
+		}
+	}
+	return t.Src, nil
+}