Fix crash when copy_tank_data loot function is applied to wrong block

Closes #1056
CyclopsMC · Sep 5, 2024 · 983fe9b · 983fe9b
1 parent a2e00ca
commit 983fe9b
Showing 1 changed file with 12 additions and 11 deletions.
diff --git a/src/main/java/org/cyclops/evilcraft/loot/functions/LootFunctionCopyTankData.java b/src/main/java/org/cyclops/evilcraft/loot/functions/LootFunctionCopyTankData.java
@@ -31,17 +31,18 @@ protected LootFunctionCopyTankData(LootItemCondition[] conditionsIn) {
 
  @Override
  public ItemStack run(ItemStack itemStack, LootContext lootContext) {
- BlockEntityTankInventory tile = (BlockEntityTankInventory) lootContext.getParamOrNull(LootContextParams.BLOCK_ENTITY);
- SingleUseTank fluidHandlerTile = tile.getTank();
- itemStack.getCapability(CapabilityFluidHandler.FLUID_HANDLER_ITEM_CAPABILITY)
- .ifPresent(fluidHandlerItem -> {
- if (fluidHandlerItem instanceof IFluidHandlerMutable) {
- ((IFluidHandlerMutable) fluidHandlerItem).setFluidInTank(0, fluidHandlerTile.getFluidInTank(0));
- }
- if (fluidHandlerItem instanceof IFluidHandlerItemCapacity) {
- ((IFluidHandlerItemCapacity) fluidHandlerItem).setCapacity(fluidHandlerTile.getTankCapacity(0));
- }
- });
+ if (lootContext.getParamOrNull(LootContextParams.BLOCK_ENTITY) instanceof BlockEntityTankInventory tile) {
+ SingleUseTank fluidHandlerTile = tile.getTank();
+ itemStack.getCapability(CapabilityFluidHandler.FLUID_HANDLER_ITEM_CAPABILITY)
+ .ifPresent(fluidHandlerItem -> {
+ if (fluidHandlerItem instanceof IFluidHandlerMutable) {
+ ((IFluidHandlerMutable) fluidHandlerItem).setFluidInTank(0, fluidHandlerTile.getFluidInTank(0));
+ }
+ if (fluidHandlerItem instanceof IFluidHandlerItemCapacity) {
+ ((IFluidHandlerItemCapacity) fluidHandlerItem).setCapacity(fluidHandlerTile.getTankCapacity(0));
+ }
+ });
+ }
  return itemStack;
  }