adds a new argment in the main struct to pass in command source to Main module

1 files changed, 4 insertions, 2 deletions

diff --git a/src/main/java/los/numeros/HidersAndDiggers/Main.java b/src/main/java/los/numeros/HidersAndDiggers/Main.java
index 7e82dd8..30e40e2 100644
--- a/src/main/java/los/numeros/HidersAndDiggers/Main.java
+++ b/src/main/java/los/numeros/HidersAndDiggers/Main.java
@@ -11,6 +11,7 @@ import los.numeros.GameEngine.Module;
 import net.fabricmc.fabric.api.networking.v1.PlayerLookup;
 import net.minecraft.network.MessageType;
 import net.minecraft.server.MinecraftServer;
+import net.minecraft.server.command.ServerCommandSource;
 import net.minecraft.server.network.ServerPlayerEntity;
 import net.minecraft.text.LiteralText;
 import los.numeros.HidersAndDiggers.modules.*;
@@ -19,8 +20,9 @@ public class Main implements Engine {
   BooleanSupplier running = () -> { return true; };
   Module modules[] = new Module[10];
 
-  public Main(MinecraftServer server, int numDiggers) {
+  public Main(ServerCommandSource source, int numDiggers) {
     int i;
+    MinecraftServer server = source.getMinecraftServer();
     ServerPlayerEntity[] players, diggers, hiders;
     Random r = new Random(new Date().getTime());
 
@@ -59,7 +61,7 @@ public class Main implements Engine {
     }
 
     /* populate the modules */
-    modules[0] = new MainModule(diggers, hiders, players);
+    modules[0] = new MainModule(diggers, hiders, players, source);
 
     /* start the game */
     Engine.StartEngine(running, modules);