set gamerule

friendlyhj · Jan 16, 2025 · 5e87800 · 5e87800
1 parent f0b687f
commit 5e87800
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/src/main/java/youyihj/zenutils/api/world/GameRuleHelper.java b/src/main/java/youyihj/zenutils/api/world/GameRuleHelper.java
@@ -1,6 +1,8 @@
 package youyihj.zenutils.api.world;
 
 import crafttweaker.annotations.ZenRegister;
+import crafttweaker.mc1120.CraftTweaker;
+import net.minecraft.command.CommandGameRule;
 import net.minecraft.world.GameRules;
 import stanhebben.zenscript.annotations.ZenClass;
 import stanhebben.zenscript.annotations.ZenMethod;
@@ -41,6 +43,36 @@ public boolean hasRule(String name) {
         return rules.hasRule(name);
     }
 
+    @ZenMethod
+    public void setInt(String name, int value) {
+        if (rules.areSameType(name, GameRules.ValueType.NUMERICAL_VALUE)) {
+            rules.setOrCreateGameRule(name, Integer.toString(value));
+            CommandGameRule.notifyGameRuleChange(rules, name, CraftTweaker.server);
+        } else {
+            throw new IllegalArgumentException("GameRule " + name + " is not numerical value.");
+        }
+    }
+
+    @ZenMethod
+    public void setBoolean(String name, boolean value) {
+        if (rules.areSameType(name, GameRules.ValueType.BOOLEAN_VALUE)) {
+            rules.setOrCreateGameRule(name, Boolean.toString(value));
+            CommandGameRule.notifyGameRuleChange(rules, name, CraftTweaker.server);
+        } else {
+            throw new IllegalArgumentException("GameRule " + name + " is not boolean value.");
+        }
+    }
+
+    @ZenMethod
+    public void setString(String name, String value) {
+        if (rules.areSameType(name, GameRules.ValueType.ANY_VALUE)) {
+            rules.setOrCreateGameRule(name, value);
+            CommandGameRule.notifyGameRuleChange(rules, name, CraftTweaker.server);
+        } else {
+            throw new IllegalArgumentException("GameRule " + name + " is not string value.");
+        }
+    }
+
     @ZenMethod
     public void addGameRule(String key, String value, String type) {
         if(hasRule(key)) return;