actually cast fluid capability

DonovanDMC · Jun 19, 2022 · ad9dc84 · ad9dc84
1 parent 53713dc
commit ad9dc84
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/main/java/cool/furry/mc/forge/projectexpansion/block/entity/BlockEntityEMCLink.java b/src/main/java/cool/furry/mc/forge/projectexpansion/block/entity/BlockEntityEMCLink.java
@@ -30,6 +30,7 @@
 import net.minecraftforge.common.capabilities.Capability;
 import net.minecraftforge.common.util.LazyOptional;
 import net.minecraftforge.fluids.FluidStack;
+import net.minecraftforge.fluids.capability.CapabilityFluidHandler;
 import net.minecraftforge.fluids.capability.IFluidHandler;
 import net.minecraftforge.items.CapabilityItemHandler;
 import net.minecraftforge.items.IItemHandler;
@@ -410,14 +411,16 @@ public InteractionResult handleActivation(Player player, InteractionHand hand) {
     @Override
     public <T> LazyOptional<T> getCapability(@Nonnull Capability<T> cap, @Nullable Direction side) {
         return
-                (cap == PECapabilities.EMC_STORAGE_CAPABILITY) ? emcStorageCapability.cast() :
-                        (cap == CapabilityItemHandler.ITEM_HANDLER_CAPABILITY) ? itemHandlerCapability.cast() :
-                                super.getCapability(cap, side);
+            (cap == PECapabilities.EMC_STORAGE_CAPABILITY) ? emcStorageCapability.cast() :
+                (cap == CapabilityItemHandler.ITEM_HANDLER_CAPABILITY) ? itemHandlerCapability.cast() :
+                    (cap == CapabilityFluidHandler.FLUID_HANDLER_CAPABILITY) ? fluidHandlerCapability.cast() :
+                        super.getCapability(cap, side);
     }
 
     @Override
     public void invalidateCaps() {
         emcStorageCapability.invalidate();
         itemHandlerCapability.invalidate();
+        fluidHandlerCapability.invalidate();
     }
 }