Introduce TemplateInstance.setLocale

Co-authored-by: Martin Kouba <mkouba@redhat.com>
Dairdevil · Apr 15, 2024 · 9f66268 · 9f66268
1 parent 5871b82
commit 9f66268
Show file tree

Hide file tree

Showing 4 changed files with 44 additions and 7 deletions.
diff --git a/docs/src/main/asciidoc/qute-reference.adoc b/docs/src/main/asciidoc/qute-reference.adoc
@@ -2829,7 +2829,7 @@ public class MyBean {
     Template hello;
 
     String render() {
-       return hello.instance().setAttribute("locale", Locale.forLanguageTag("cs")).render(); <1>
+       return hello.instance().setLocale("cs").render(); <1>
     }
 }
 ----

diff --git a/extensions/qute/runtime/src/main/java/io/quarkus/qute/i18n/MessageBundles.java b/extensions/qute/runtime/src/main/java/io/quarkus/qute/i18n/MessageBundles.java
@@ -28,7 +28,7 @@
 
 public final class MessageBundles {
 
-    public static final String ATTRIBUTE_LOCALE = "locale";
+    public static final String ATTRIBUTE_LOCALE = TemplateInstance.LOCALE;
     public static final String DEFAULT_LOCALE = "<<default>>";
 
     private static final Logger LOGGER = Logger.getLogger(MessageBundles.class);

diff --git a/independent-projects/qute/core/src/main/java/io/quarkus/qute/TemplateInstance.java b/independent-projects/qute/core/src/main/java/io/quarkus/qute/TemplateInstance.java
@@ -1,5 +1,6 @@
 package io.quarkus.qute;
 
+import java.util.Locale;
 import java.util.concurrent.CompletionStage;
 import java.util.function.Consumer;
 import java.util.function.Function;
@@ -31,6 +32,11 @@ public interface TemplateInstance {
      */
     String SELECTED_VARIANT = "selectedVariant";
 
+    /**
+     * Attribute key - locale.
+     */
+    String LOCALE = "locale";
+
     /**
      * Set the the root data object. Invocation of this method removes any data set previously by
      * {@link #data(String, Object)} and {@link #computedData(String, Function)}.
@@ -70,7 +76,6 @@ default TemplateInstance computedData(String key, Function<String, Object> funct
     }
 
     /**
-     *
      * @param key
      * @param value
      * @return self
@@ -80,7 +85,6 @@ default TemplateInstance setAttribute(String key, Object value) {
     }
 
     /**
-     *
      * @param key
      * @return the attribute or null
      */
@@ -142,7 +146,6 @@ default CompletionStage<Void> consume(Consumer<String> consumer) {
     }
 
     /**
-     *
      * @return the timeout
      * @see TemplateInstance#TIMEOUT
      */
@@ -151,15 +154,13 @@ default long getTimeout() {
     }
 
     /**
-     *
      * @return the original template
      */
     default Template getTemplate() {
         throw new UnsupportedOperationException();
     }
 
     /**
-     *
      * @param id
      * @return the fragment or {@code null}
      * @see Template#getFragment(String)
@@ -178,6 +179,28 @@ default TemplateInstance onRendered(Runnable action) {
         throw new UnsupportedOperationException();
     }
 
+    /**
+     * Sets the {@code locale} attribute that can be used to localize parts of the template, i.e. to specify the locale for all
+     * message bundle expressions in the template.
+     *
+     * @param locale a language tag
+     * @return self
+     */
+    default TemplateInstance setLocale(String locale) {
+        return setAttribute(LOCALE, Locale.forLanguageTag(locale));
+    }
+
+    /**
+     * Sets the {@code locale} attribute that can be used to localize parts of the template, i.e. to specify the locale for all
+     * message bundle expressions in the template.
+     *
+     * @param locale a {@link Locale} instance
+     * @return self
+     */
+    default TemplateInstance setLocale(Locale locale) {
+        return setAttribute(LOCALE, locale);
+    }
+
     /**
      * This component can be used to initialize a template instance, i.e. the data and attributes.
      *

diff --git a/independent-projects/qute/core/src/test/java/io/quarkus/qute/TemplateInstanceTest.java b/independent-projects/qute/core/src/test/java/io/quarkus/qute/TemplateInstanceTest.java
@@ -4,6 +4,8 @@
 import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
+import java.util.List;
+import java.util.Locale;
 import java.util.concurrent.atomic.AtomicBoolean;
 
 import org.junit.jupiter.api.Test;
@@ -64,4 +66,16 @@ public void testComputeData() {
         assertTrue(fooUsed.get());
         assertFalse(barUsed.get());
     }
+
+    @Test
+    public void testLocale() throws Exception {
+        Engine engine = Engine.builder().addDefaults()
+                .addValueResolver(ValueResolver.builder()
+                        .applyToName("locale")
+                        .resolveSync(ctx -> ctx.getAttribute(TemplateInstance.LOCALE))
+                        .build())
+                .build();
+        Template hello = engine.parse("Hello {locale}!");
+        assertEquals("Hello fr!", hello.instance().setLocale(Locale.FRENCH).render());
+    }
 }