Implement context arguments

Alemiz112 · Mar 25, 2023 · c86958f · c86958f
1 parent ec6fd7c
commit c86958f
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/common/src/main/java/eu/mizerak/alemiz/translationlib/common/string/LocalStringBase.java b/common/src/main/java/eu/mizerak/alemiz/translationlib/common/string/LocalStringBase.java
@@ -80,7 +80,7 @@ public String getText(T object) {
                 formatted = formatted.replaceAll("\\{" + entry.getKey() + "\\}", entry.getValue().apply(ctx));
             }
         }
-        return formatted;
+        return ctx.handlePostFormat(formatted);
     }
 
     @Override

diff --git a/common/src/main/java/eu/mizerak/alemiz/translationlib/common/string/TranslationContext.java b/common/src/main/java/eu/mizerak/alemiz/translationlib/common/string/TranslationContext.java
@@ -1,9 +1,13 @@
 package eu.mizerak.alemiz.translationlib.common.string;
 
+import java.util.HashMap;
 import java.util.Locale;
+import java.util.Map;
+import java.util.function.Function;
 
 public abstract class TranslationContext<T> {
     private final T object;
+    private Map<String, Function<T, String>> localArguments;
 
     public TranslationContext(T object) {
         this.object = object;
@@ -15,6 +19,27 @@ public T getObject() {
         return this.object;
     }
 
+    public TranslationContext<T> argument(String argument, Object value) {
+        return this.argument(argument, t -> String.valueOf(value));
+    }
+
+    public TranslationContext<T> argument(String argument, Function<T, String> function) {
+        if (this.localArguments == null) {
+            this.localArguments = new HashMap<>();
+        }
+        this.localArguments.put(argument, function);
+        return this;
+    }
+
+    public String handlePostFormat(String string) {
+        if (this.localArguments != null) {
+            for (Map.Entry<String, Function<T, String>> entry : this.localArguments.entrySet()) {
+                string = string.replaceAll("\\{" + entry.getKey() + "\\}", entry.getValue().apply(this.object));
+            }
+        }
+        return string;
+    }
+
     public interface Factory<T> {
         TranslationContext<T> create(T object);
     }