Partial fixes for chunk loading, made machines autoload if they have a tunnel. (#580)

* Updates chunk loading upgrade, made machines autoload if they have a tunnel.

Author: RealMangorage

src/main/java/dev/compactmods/machines/machine/CompactMachineBlockEntity.java

@@ -9,6 +9,7 @@
 import dev.compactmods.machines.machine.graph.CompactMachineNode;
 import dev.compactmods.machines.machine.graph.legacy.LegacyMachineConnections;
 import dev.compactmods.machines.room.graph.CompactMachineRoomNode;
+import dev.compactmods.machines.tunnel.TunnelHelper;
 import dev.compactmods.machines.tunnel.TunnelWallEntity;
 import dev.compactmods.machines.tunnel.graph.TunnelConnectionGraph;
 import dev.compactmods.machines.tunnel.graph.TunnelNode;
@@ -237,6 +238,7 @@ public void syncConnectedRoom() {
                     .ifPresent(roomNode -> this.roomNode = new WeakReference<>(roomNode));
 
             this.setChanged();
+            forceLoadIfHasTunnels();
         }
     }
 
@@ -259,6 +261,18 @@ public void disconnect() {
         }
     }
 
+    public void forceLoadIfHasTunnels() {
+        if(level == null || roomChunk == null) return;
+
+        if (level instanceof ServerLevel sl) {
+            final var compactDim = CompactDimension.forServer(sl.getServer());
+            if (compactDim == null) return;
+            final var tunnelGraph = TunnelConnectionGraph.forRoom(compactDim, roomChunk);
+            TunnelHelper.setChunkMode(compactDim, roomChunk, tunnelGraph.tunnels().findAny().isPresent());
+        }
+    }
+
+
     public Stream<BlockPos> getTunnels(Direction dir) {
         if(level == null || roomChunk == null) return Stream.empty();
 

src/main/java/dev/compactmods/machines/tunnel/TunnelHelper.java

@@ -1,14 +1,20 @@
 package dev.compactmods.machines.tunnel;
 
+import dev.compactmods.machines.api.core.Constants;
 import net.minecraft.core.Direction;
+import net.minecraft.server.level.ServerLevel;
+import net.minecraft.server.level.TicketType;
+import net.minecraft.world.level.ChunkPos;
 
 import javax.annotation.Nonnull;
 import java.util.Collection;
+import java.util.Comparator;
 import java.util.Optional;
 import java.util.Set;
 import java.util.stream.Stream;
 
 public class TunnelHelper {
+    private static final TicketType<ChunkPos> CM4_TUUNEL_LOAD_TYPE = TicketType.create(Constants.MOD_ID + ":tunnels", Comparator.comparingLong(ChunkPos::toLong));
 
     @Nonnull
     public static Direction getNextDirection(Direction in) {
@@ -33,4 +39,14 @@ public static Optional<Direction> getNextDirection(Direction current, Set<Direct
 
         return stream.skip(found + 1).filter(dir -> !used.contains(dir)).findFirst();
     }
+
+    public static void setChunkMode(ServerLevel level, ChunkPos room, boolean forceLoad) {
+        final var chunks = level.getChunkSource();
+        level.setChunkForced(room.x, room.z, forceLoad);
+        if (forceLoad)
+            chunks.addRegionTicket(CM4_TUUNEL_LOAD_TYPE, room, 2, room);
+        else
+            chunks.removeRegionTicket(CM4_TUUNEL_LOAD_TYPE, room, 2, room);
+        chunks.save(true);
+    }
 }

src/main/java/dev/compactmods/machines/upgrade/ChunkloadUpgrade.java

@@ -6,6 +6,7 @@
 import dev.compactmods.machines.api.room.upgrade.ILevelLoadedUpgradeListener;
 import dev.compactmods.machines.api.room.upgrade.RoomUpgrade;
 import net.minecraft.resources.ResourceLocation;
+import net.minecraft.server.commands.ForceLoadCommand;
 import net.minecraft.server.level.ServerLevel;
 import net.minecraft.server.level.TicketType;
 import net.minecraft.world.level.ChunkPos;
@@ -50,14 +51,14 @@ private void forceLoad(ServerLevel level, ChunkPos room) {
         final var chunks = level.getChunkSource();
         level.setChunkForced(room.x, room.z, true);
         chunks.addRegionTicket(CM4_LOAD_TYPE, room, 2, room);
-        chunks.save(false);
+        chunks.save(true);
     }
 
     private void normalLoad(ServerLevel level, ChunkPos room) {
         final var chunks = level.getChunkSource();
         level.setChunkForced(room.x, room.z, false);
         chunks.removeRegionTicket(CM4_LOAD_TYPE, room, 2, room);
-        chunks.save(false);
+        chunks.save(true);
     }
 }