71 lines
2.8 KiB
Java
71 lines
2.8 KiB
Java
/*
|
|
* This file is part of ComputerCraft - http://www.computercraft.info
|
|
* Copyright Daniel Ratcliffe, 2011-2020. Do not distribute without permission.
|
|
* Send enquiries to dratcliffe@gmail.com
|
|
*/
|
|
|
|
package dan200.computercraft.shared.peripheral.generic;
|
|
|
|
import dan200.computercraft.ComputerCraft;
|
|
import dan200.computercraft.api.peripheral.IPeripheral;
|
|
import dan200.computercraft.core.asm.NamedMethod;
|
|
import dan200.computercraft.core.asm.PeripheralMethod;
|
|
import net.minecraft.tileentity.TileEntity;
|
|
import net.minecraft.util.Direction;
|
|
import net.minecraft.util.math.BlockPos;
|
|
import net.minecraft.world.World;
|
|
import net.minecraftforge.common.capabilities.Capability;
|
|
import net.minecraftforge.common.util.LazyOptional;
|
|
import net.minecraftforge.energy.CapabilityEnergy;
|
|
import net.minecraftforge.fluids.capability.CapabilityFluidHandler;
|
|
import net.minecraftforge.items.CapabilityItemHandler;
|
|
|
|
import javax.annotation.Nonnull;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
public class GenericPeripheralProvider
|
|
{
|
|
private static final Capability<?>[] CAPABILITIES = new Capability<?>[] {
|
|
CapabilityItemHandler.ITEM_HANDLER_CAPABILITY,
|
|
CapabilityEnergy.ENERGY,
|
|
CapabilityFluidHandler.FLUID_HANDLER_CAPABILITY,
|
|
};
|
|
|
|
@Nonnull
|
|
public static LazyOptional<IPeripheral> getPeripheral( @Nonnull World world, @Nonnull BlockPos pos, @Nonnull Direction side )
|
|
{
|
|
TileEntity tile = world.getTileEntity( pos );
|
|
if( tile == null ) return LazyOptional.empty();
|
|
|
|
ArrayList<SaturatedMethod> saturated = new ArrayList<>( 0 );
|
|
LazyOptional<IPeripheral> peripheral = LazyOptional.of( () -> new GenericPeripheral( tile, saturated ) );
|
|
|
|
List<NamedMethod<PeripheralMethod>> tileMethods = PeripheralMethod.GENERATOR.getMethods( tile.getClass() );
|
|
if( !tileMethods.isEmpty() ) addSaturated( saturated, tile, tileMethods );
|
|
|
|
for( Capability<?> capability : CAPABILITIES )
|
|
{
|
|
LazyOptional<?> wrapper = tile.getCapability( capability );
|
|
wrapper.ifPresent( contents -> {
|
|
List<NamedMethod<PeripheralMethod>> capabilityMethods = PeripheralMethod.GENERATOR.getMethods( contents.getClass() );
|
|
if( capabilityMethods.isEmpty() ) return;
|
|
|
|
addSaturated( saturated, contents, capabilityMethods );
|
|
wrapper.addListener( x -> peripheral.invalidate() );
|
|
} );
|
|
}
|
|
|
|
return saturated.isEmpty() ? LazyOptional.empty() : peripheral;
|
|
}
|
|
|
|
private static void addSaturated( ArrayList<SaturatedMethod> saturated, Object target, List<NamedMethod<PeripheralMethod>> methods )
|
|
{
|
|
saturated.ensureCapacity( saturated.size() + methods.size() );
|
|
for( NamedMethod<PeripheralMethod> method : methods )
|
|
{
|
|
saturated.add( new SaturatedMethod( target, method ) );
|
|
}
|
|
}
|
|
}
|