custom property for a Context

ZZZank · Jun 17, 2024 · 461f26d · 461f26d
1 parent 8949119
commit 461f26d
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/common/src/main/java/dev/latvian/mods/rhino/Context.java b/common/src/main/java/dev/latvian/mods/rhino/Context.java
@@ -17,6 +17,7 @@
 import dev.latvian.mods.rhino.util.remapper.Remapper;
 import dev.latvian.mods.rhino.util.wrap.TypeWrappers;
 import lombok.Setter;
+import org.jetbrains.annotations.Nullable;
 
 import java.beans.PropertyChangeEvent;
 import java.beans.PropertyChangeListener;
@@ -276,6 +277,7 @@ public class Context {
     private ClassLoader applicationClassLoader;
     @Setter
     private Remapper remapper;
+    private final Map<String, Object> customProperties;
 
     /**
      * Creates a new context. Provided as a preferred super constructor for
@@ -295,6 +297,7 @@ protected Context(ContextFactory factory) {
         maximumInterpreterStackDepth = Integer.MAX_VALUE;
         optimizationLevel = RhinoProperties.INSTANCE.optimizationLevel;
         remapper = null;
+        customProperties = new HashMap<>();
     }
 
     /**
@@ -790,6 +793,18 @@ public Remapper getRemapper() {
         return remapper == null ? factory.remapper : remapper;
     }
 
+    public Object getCustomProperty(String name) {
+        return customProperties.get(name);
+    }
+
+    public Object getCustomProperty(String name, Object orDefault) {
+        return customProperties.getOrDefault(name, orDefault);
+    }
+
+    public void setCustomProperty(String name, @Nullable Object value) {
+        customProperties.put(name, value);
+    }
+
     /**
      * Return {@link ContextFactory} instance used to create this Context.
      */