|
| 1 | +package gregification.modules.computercraft.peripheral; |
| 2 | + |
| 3 | +import dan200.computercraft.api.lua.ILuaContext; |
| 4 | +import dan200.computercraft.api.peripheral.IComputerAccess; |
| 5 | +import gregtech.api.capability.GregtechTileCapabilities; |
| 6 | +import gregtech.api.capability.IControllable; |
| 7 | +import gregtech.api.capability.impl.AbstractRecipeLogic; |
| 8 | +import gregtech.api.metatileentity.MetaTileEntity; |
| 9 | +import gregtech.api.metatileentity.TieredMetaTileEntity; |
| 10 | + |
| 11 | +import java.util.HashMap; |
| 12 | +import java.util.Map; |
| 13 | + |
| 14 | +class MetaTileEntityPeripheralStatusFunction implements MetaTileEntityPeripheralFunction { |
| 15 | + |
| 16 | + private final TieredMetaTileEntity tieredMetaTileEntity; |
| 17 | + |
| 18 | + private final IControllable controllable; |
| 19 | + private final AbstractRecipeLogic recipeLogic; |
| 20 | + |
| 21 | + public MetaTileEntityPeripheralStatusFunction(MetaTileEntity metaTileEntity) { |
| 22 | + this.tieredMetaTileEntity = metaTileEntity instanceof TieredMetaTileEntity ? (TieredMetaTileEntity) metaTileEntity : null; |
| 23 | + this.controllable = metaTileEntity.getCapability(GregtechTileCapabilities.CAPABILITY_CONTROLLABLE, null); |
| 24 | + this.recipeLogic = metaTileEntity.getCapability(GregtechTileCapabilities.CAPABILITY_RECIPE_LOGIC, null); |
| 25 | + } |
| 26 | + |
| 27 | + @Override |
| 28 | + public boolean isEnabled() { |
| 29 | + return this.tieredMetaTileEntity != null || this.controllable != null || this.recipeLogic != null; |
| 30 | + } |
| 31 | + |
| 32 | + @Override |
| 33 | + public String getName() { |
| 34 | + return "getStatus"; |
| 35 | + } |
| 36 | + |
| 37 | + @Override |
| 38 | + public Object[] call(IComputerAccess iComputerAccess, ILuaContext iLuaContext, int i, Object[] objects) { |
| 39 | + Map<String, Object> data = new HashMap<>(); |
| 40 | + if (controllable != null) { |
| 41 | + data.put("enabled", controllable.isWorkingEnabled()); |
| 42 | + } |
| 43 | + if (tieredMetaTileEntity != null) { |
| 44 | + data.put("tier", tieredMetaTileEntity.getTier()); |
| 45 | + } |
| 46 | + if (recipeLogic != null) { |
| 47 | + data.put("active", recipeLogic.isActive()); |
| 48 | + data.put("progress", recipeLogic.getProgress()); |
| 49 | + data.put("maxProgress", recipeLogic.getMaxProgress()); |
| 50 | + } |
| 51 | + return new Object[]{data}; |
| 52 | + } |
| 53 | +} |
0 commit comments