Make capability invalidation callbacks less strict

Forge!! *shakes fist*.

src/main/java/dan200/computercraft/shared/Peripherals.java

@@ -38,13 +38,13 @@ public static synchronized void register( @Nonnull IPeripheralProvider provider
     }
 
     @Nullable
-    public static IPeripheral getPeripheral( World world, BlockPos pos, Direction side, NonNullConsumer<LazyOptional<IPeripheral>> invalidate )
+    public static IPeripheral getPeripheral( World world, BlockPos pos, Direction side, NonNullConsumer<Object> invalidate )
     {
         return World.isInWorldBounds( pos ) && !world.isClientSide ? getPeripheralAt( world, pos, side, invalidate ) : null;
     }
 
     @Nullable
-    private static IPeripheral getPeripheralAt( World world, BlockPos pos, Direction side, NonNullConsumer<LazyOptional<IPeripheral>> invalidate )
+    private static IPeripheral getPeripheralAt( World world, BlockPos pos, Direction side, NonNullConsumer<? super Object> invalidate )
     {
         TileEntity block = world.getBlockEntity( pos );
         if( block != null )

src/main/java/dan200/computercraft/shared/computer/blocks/TileComputerBase.java

@@ -39,7 +39,6 @@
 import net.minecraft.util.text.StringTextComponent;
 import net.minecraft.util.text.TranslationTextComponent;
 import net.minecraft.world.World;
-import net.minecraftforge.common.util.LazyOptional;
 import net.minecraftforge.common.util.NonNullConsumer;
 
 import javax.annotation.Nonnull;
@@ -58,7 +57,7 @@ public abstract class TileComputerBase extends TileGeneric implements IComputerT
     private boolean on = false;
     boolean startOn = false;
     private boolean fresh = false;
-    private final NonNullConsumer<LazyOptional<IPeripheral>>[] invalidate;
+    private final NonNullConsumer<Object>[] invalidate;
 
     private final ComputerFamily family;
 
@@ -69,7 +68,7 @@ public TileComputerBase( TileEntityType<? extends TileGeneric> type, ComputerFam
 
         // We cache these so we can guarantee we only ever register one listener for adjacent capabilities.
         @SuppressWarnings( { "unchecked", "rawtypes" } )
-        NonNullConsumer<LazyOptional<IPeripheral>>[] invalidate = this.invalidate = new NonNullConsumer[6];
+        NonNullConsumer<Object>[] invalidate = this.invalidate = new NonNullConsumer[6];
         for( Direction direction : Direction.values() )
         {
             invalidate[direction.ordinal()] = o -> updateInput( direction );

src/main/java/dan200/computercraft/shared/peripheral/generic/GenericPeripheralProvider.java

@@ -9,6 +9,7 @@
 import dan200.computercraft.api.peripheral.PeripheralType;
 import dan200.computercraft.core.asm.NamedMethod;
 import dan200.computercraft.core.asm.PeripheralMethod;
+import dan200.computercraft.shared.util.CapabilityUtil;
 import net.minecraft.tileentity.TileEntity;
 import net.minecraft.util.Direction;
 import net.minecraft.util.math.BlockPos;
@@ -34,7 +35,7 @@ public static synchronized void addCapability( Capability<?> capability )
     }
 
     @Nullable
-    public static IPeripheral getPeripheral( @Nonnull World world, @Nonnull BlockPos pos, @Nonnull Direction side, NonNullConsumer<LazyOptional<IPeripheral>> invalidate )
+    public static IPeripheral getPeripheral( @Nonnull World world, @Nonnull BlockPos pos, @Nonnull Direction side, NonNullConsumer<Object> invalidate )
     {
         TileEntity tile = world.getBlockEntity( pos );
         if( tile == null ) return null;
@@ -52,7 +53,7 @@ public static IPeripheral getPeripheral( @Nonnull World world, @Nonnull BlockPos
                 if( capabilityMethods.isEmpty() ) return;
 
                 saturated.addMethods( contents, capabilityMethods );
-                wrapper.addListener( cast( invalidate ) );
+                CapabilityUtil.addListener( wrapper, invalidate );
             } );
         }
 
@@ -91,10 +92,4 @@ void addMethods( Object target, List<NamedMethod<PeripheralMethod>> methods )
             }
         }
     }
-
-    @SuppressWarnings( { "unchecked", "rawtypes" } )
-    private static <T> NonNullConsumer<T> cast( NonNullConsumer<?> consumer )
-    {
-        return (NonNullConsumer) consumer;
-    }
 }

src/main/java/dan200/computercraft/shared/peripheral/modem/wired/WiredModemLocalPeripheral.java

@@ -15,7 +15,6 @@
 import net.minecraft.util.math.BlockPos;
 import net.minecraft.world.World;
 import net.minecraftforge.common.util.Constants;
-import net.minecraftforge.common.util.LazyOptional;
 import net.minecraftforge.common.util.NonNullConsumer;
 
 import javax.annotation.Nonnull;
@@ -38,7 +37,7 @@ public final class WiredModemLocalPeripheral
     private String type;
 
     private IPeripheral peripheral;
-    private final NonNullConsumer<LazyOptional<IPeripheral>> invalidate;
+    private final NonNullConsumer<Object> invalidate;
 
     public WiredModemLocalPeripheral( @Nonnull Runnable invalidate )
     {

src/main/java/dan200/computercraft/shared/util/CapabilityUtil.java

@@ -35,12 +35,20 @@ public static <T> void invalidate( @Nullable LazyOptional<T>[] caps )
         }
     }
 
+    public static <T> void addListener( LazyOptional<T> p, NonNullConsumer<? super LazyOptional<T>> invalidate )
+    {
+        // We can make this safe with invalidate::accept, but then we're allocating it's just kind of absurd.
+        @SuppressWarnings( "unchecked" )
+        NonNullConsumer<LazyOptional<T>> safeInvalidate = (NonNullConsumer<LazyOptional<T>>) invalidate;
+        p.addListener( safeInvalidate );
+    }
+
     @Nullable
-    public static <T> T unwrap( LazyOptional<T> p, NonNullConsumer<LazyOptional<T>> invalidate )
+    public static <T> T unwrap( LazyOptional<T> p, NonNullConsumer<? super LazyOptional<T>> invalidate )
     {
         if( !p.isPresent() ) return null;
 
-        p.addListener( invalidate );
+        addListener( p, invalidate );
         return p.orElseThrow( NullPointerException::new );
     }