Add recursion limit

Author: GeorgeRNG

src/main/java/io/github/techstreet/dfscript/commands/misc/ScriptsCommand.java

@@ -1,16 +1,17 @@
 package io.github.techstreet.dfscript.commands.misc;
 
 import com.mojang.brigadier.CommandDispatcher;
+import com.mojang.brigadier.arguments.IntegerArgumentType;
 import com.mojang.brigadier.arguments.StringArgumentType;
 import io.github.techstreet.dfscript.DFScript;
 import io.github.techstreet.dfscript.commands.Command;
 import io.github.techstreet.dfscript.commands.arguments.StringFuncArgumentType;
+import io.github.techstreet.dfscript.event.system.EventManager;
 import io.github.techstreet.dfscript.screen.script.ScriptListScreen;
 import io.github.techstreet.dfscript.script.Script;
 import io.github.techstreet.dfscript.script.ScriptManager;
 import io.github.techstreet.dfscript.script.values.ScriptValue;
 import io.github.techstreet.dfscript.util.chat.ChatUtil;
-import java.util.ArrayList;
 import java.util.List;
 import java.util.Map.Entry;
 import net.fabricmc.fabric.api.client.command.v2.FabricClientCommandSource;
@@ -49,6 +50,14 @@ public void register(CommandDispatcher<FabricClientCommandSource> cd) {
                         )
                     )
                 )
+                .then(literal("recursion")
+                    .then(argument("limit", IntegerArgumentType.integer(-1)).executes(ctx -> {
+                        int limit = IntegerArgumentType.getInteger(ctx,"limit");
+                        EventManager.getInstance().setEventLimit(limit);
+                        ChatUtil.info("Set events per tick limit to " + limit);
+                        return 0;
+                    }))
+                )
         );
     }
 

src/main/java/io/github/techstreet/dfscript/event/system/EventManager.java

@@ -8,6 +8,8 @@
 public class EventManager {
 
     private static EventManager instance;
+    private int events = 0;
+    private int eventLimit = 100;
     private final HashMap<Class<? extends Event>, List<Consumer<Event>>> listeners = new HashMap<>();
 
     public EventManager() {
@@ -31,7 +33,17 @@ public <T extends Event> void register(Class<T> type, Consumer<T> consumer) {
 
     public void dispatch(Event event) {
         for (Consumer<Event> consumer : listeners.getOrDefault(event.getClass(), new ArrayList<>())) {
+            if(events == eventLimit) return;
+            events++;
             consumer.accept(event);
         }
     }
+
+    public void resetEvents() {
+        this.events = 0;
+    }
+
+    public void setEventLimit(int eventLimit) {
+        this.eventLimit = eventLimit;
+    }
 }

src/main/java/io/github/techstreet/dfscript/mixin/player/MLocalPlayer.java

@@ -45,7 +45,9 @@ private void command2(String command, Text preview, CallbackInfo ci) {
 
     @Inject(method = "tick", at = @At("HEAD"))
     private void tick(CallbackInfo ci) {
-        EventManager.getInstance().dispatch(new TickEvent());
+        EventManager em = EventManager.getInstance();
+        em.dispatch(new TickEvent());
+        em.resetEvents();
     }
 
 }