Fix potential leak, compile error

Author: Nightenom

src/main/java/com/ldtteam/blockui/mod/ClientEventSubscriber.java

@@ -10,13 +10,15 @@
 import com.ldtteam.blockui.mod.container.ContainerHook;
 import com.ldtteam.blockui.util.resloc.OutOfJarResourceLocation;
 import com.ldtteam.blockui.views.BOWindow;
+import com.ldtteam.common.util.BlockToItemHelper;
 import com.mojang.blaze3d.platform.InputConstants;
 import net.minecraft.client.Minecraft;
 import net.minecraft.client.gui.screens.Screen;
 import net.minecraft.network.chat.Component;
 import net.minecraft.resources.ResourceLocation;
 import net.neoforged.bus.api.EventPriority;
 import net.neoforged.bus.api.SubscribeEvent;
+import net.neoforged.neoforge.client.event.ClientPlayerNetworkEvent;
 import net.neoforged.neoforge.client.event.InputEvent.MouseScrollingEvent;
 import net.neoforged.neoforge.client.event.RenderGuiOverlayEvent;
 import net.neoforged.neoforge.client.gui.overlay.VanillaGuiOverlay;
@@ -143,4 +145,19 @@ public static void renderOverlay(final RenderGuiOverlayEvent.Pre event)
             event.setCanceled(true);
         }
     }
+
+    @SubscribeEvent
+    public static void onClientPlayerLoggingOut(final ClientPlayerNetworkEvent.LoggingOut event)
+    {
+        BlockToItemHelper.releaseFakeLevelInstance();
+    }
+
+    @SubscribeEvent
+    public static void onClientPlayerDimChange(final ClientPlayerNetworkEvent.Clone event)
+    {
+        if (event.getNewPlayer().level() != event.getOldPlayer().level())
+        {
+            BlockToItemHelper.releaseFakeLevelInstance();
+        }
+    }
 }

src/main/java/com/ldtteam/common/fakelevel/FakeLevel.java

@@ -145,7 +145,7 @@ public void setRealLevel(final Level realLevel)
         }
 
         this.realLevel = realLevel;
-        ((FakeLevelData) this.getLevelData()).vanillaLevelData = realLevel.getLevelData();
+        ((FakeLevelData) this.getLevelData()).vanillaLevelData = realLevel == null ? null : realLevel.getLevelData();
     }
 
     public Level realLevel()
@@ -760,15 +760,15 @@ public void updateSkyBrightness()
 
     @Override
     @Nullable
-    public <T, C> @Nullable T getCapability(BlockCapability<T, C> cap, BlockPos pos, C context)
+    public <T, C> T getCapability(BlockCapability<T, C> cap, BlockPos pos, C context)
     {
         // Noop
         return null;
     }
 
     @Override
     @Nullable
-    public <T, C> @Nullable T getCapability(BlockCapability<T, C> cap,
+    public <T, C> T getCapability(BlockCapability<T, C> cap,
         BlockPos pos,
         @Nullable BlockState state,
         @Nullable BlockEntity blockEntity,

src/main/java/com/ldtteam/common/util/BlockToItemHelper.java

@@ -33,6 +33,15 @@ public class BlockToItemHelper
     private static final HitResult ZERO_POS_HIT_RESULT = new BlockHitResult(Vec3.atCenterOf(BlockPos.ZERO), Direction.NORTH, BlockPos.ZERO, true);
     private static FakeLevel<SingleBlockFakeLevelGetter> fakeLevel;
 
+    public static void releaseFakeLevelInstance()
+    {
+        if (fakeLevel != null)
+        {
+            fakeLevel.setRealLevel(null);
+            fakeLevel = null;
+        }
+    }
+
     /**
      * Mostly for use in UI where you dont have level instance (eg. player selects block, from xml, but not when displaying real world
      * info - see {@link BlockStateRenderingData#of(Level, BlockPos, Player)}). NOT thread safe!