Fixed #2403

- Added the onRemove method to the Wireless Transmitter
Lothrazar · Apr 19, 2024 · a681f80 · a681f80
1 parent ba04e8a
commit a681f80
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/main/java/com/lothrazar/cyclic/block/wireless/redstone/BlockWirelessTransmit.java b/src/main/java/com/lothrazar/cyclic/block/wireless/redstone/BlockWirelessTransmit.java
@@ -5,6 +5,7 @@
 import com.lothrazar.cyclic.registry.TileRegistry;
 import net.minecraft.client.gui.screens.MenuScreens;
 import net.minecraft.core.BlockPos;
+import net.minecraft.world.Containers;
 import net.minecraft.world.level.Level;
 import net.minecraft.world.level.block.Block;
 import net.minecraft.world.level.block.entity.BlockEntity;
@@ -44,4 +45,18 @@ public <T extends BlockEntity> BlockEntityTicker<T> getTicker(Level world, Block
   protected void createBlockStateDefinition(StateDefinition.Builder<Block, BlockState> builder) {
     builder.add(POWERED);
   }
+
+  @Override // was onReplaced
+  public void onRemove(BlockState state, Level worldIn, BlockPos pos, BlockState newState, boolean isMoving) {
+    if (state.getBlock() != newState.getBlock()) {
+      TileWirelessTransmit tileentity = (TileWirelessTransmit) worldIn.getBlockEntity(pos);
+      if (tileentity != null) {
+        for (int i = 0; i < tileentity.inventory.getSlots(); ++i) {
+          // was  InventoryHelper.spawnItemStack
+          Containers.dropItemStack(worldIn, pos.getX(), pos.getY(), pos.getZ(), tileentity.inventory.getStackInSlot(i));
+        }
+      }
+      super.onRemove(state, worldIn, pos, newState, isMoving);
+    }
+  }
 }