fix: Defer fake player creation to avoid world load freeze

Author: Rover656

enderio-machines/src/main/java/com/enderio/machines/common/blocks/obelisks/attractor/AttractorObeliskBlockEntity.java

@@ -14,8 +14,7 @@
 import com.enderio.machines.common.init.MachineBlockEntities;
 import com.enderio.machines.common.obelisk.ObeliskAreaManager;
 import com.mojang.authlib.GameProfile;
-import java.util.List;
-import java.util.UUID;
+import me.liliandev.ensure.ensures.EnsureSide;
 import net.minecraft.core.BlockPos;
 import net.minecraft.server.level.ServerLevel;
 import net.minecraft.world.entity.LivingEntity;
@@ -39,6 +38,8 @@
 import net.neoforged.neoforge.common.util.FakePlayer;
 import org.jetbrains.annotations.Nullable;
 
+import java.util.List;
+
 public class AttractorObeliskBlockEntity extends ObeliskBlockEntity<AttractorObeliskBlockEntity> {
 
     private static final QuadraticScalable ENERGY_CAPACITY = new QuadraticScalable(CapacitorModifier.ENERGY_CAPACITY,
@@ -51,6 +52,7 @@ public class AttractorObeliskBlockEntity extends ObeliskBlockEntity<AttractorObe
     private final Vec3 targetPos;
 
     @UseOnly(LogicalSide.SERVER)
+    @Nullable
     private FakePlayer fakePlayer;
 
     public AttractorObeliskBlockEntity(BlockPos worldPosition, BlockState blockState) {
@@ -59,6 +61,20 @@ public AttractorObeliskBlockEntity(BlockPos worldPosition, BlockState blockState
         targetPos = new Vec3(worldPosition.getX() + 0.5, worldPosition.getY() + 0.5, worldPosition.getZ() + 0.5);
     }
 
+    @EnsureSide(EnsureSide.Side.SERVER)
+    private FakePlayer getFakePlayer() {
+        if (!(level instanceof ServerLevel serverLevel)) {
+            throw new IllegalStateException("Level is null");
+        }
+
+        if (fakePlayer == null) {
+            fakePlayer = new FakePlayer(serverLevel, new GameProfile(getMachineOwnerOrRandom(), "enderio:attractor:" + worldPosition));
+            fakePlayer.setPos(targetPos.x, targetPos.y, targetPos.z);
+        }
+
+        return fakePlayer;
+    }
+
     @Override
     protected @Nullable ObeliskAreaManager<AttractorObeliskBlockEntity> getAreaManager(ServerLevel level) {
         return null;
@@ -92,10 +108,6 @@ public String getColor() {
     @Override
     public void setLevel(Level level) {
         super.setLevel(level);
-        if (level instanceof ServerLevel sl) {
-            fakePlayer = new FakePlayer(sl, new GameProfile(getMachineOwnerOrRandom(), "enderio:attractor:" + worldPosition));
-            fakePlayer.setPos(targetPos.x, targetPos.y, targetPos.z);
-        }
     }
 
     @Override
@@ -144,7 +156,7 @@ private boolean useTarget(Mob mob) {
 
     private void setTarget(Mob mob) {
         assert level != null;
-        mob.setTarget(fakePlayer);
+        mob.setTarget(getFakePlayer());
     }
 
     private void attractMob(Mob mob) {
@@ -168,8 +180,12 @@ private void directPull(LivingEntity mob, float speed) {
     @Override
     public void setRemoved() {
         super.setRemoved();
-        if (!level.isClientSide) {
-            fakePlayer.discard();
+
+        if (level != null && !level.isClientSide) {
+            if (fakePlayer != null) {
+                fakePlayer.discard();
+                fakePlayer = null;
+            }
         }
     }
 }