src/main/java/li/cil/occ/mods/appeng/DriverGridNode.java

package li.cil.occ.mods.appeng;

import appeng.api.networking.IGrid;
import appeng.api.networking.IGridNode;
import appeng.api.networking.energy.IEnergyGrid;
import li.cil.oc.api.network.Arguments;
import li.cil.oc.api.network.Callback;
import li.cil.oc.api.network.Context;
import li.cil.oc.api.network.ManagedEnvironment;
import li.cil.oc.api.prefab.DriverTileEntity;
import li.cil.occ.mods.ManagedTileEntityEnvironment;
import net.minecraft.world.World;

public class DriverGridNode extends DriverTileEntity {
	@Override
	public Class<?> getTileEntityClass() {
		return IGridNode.class;
	}

	@Override
	public ManagedEnvironment createEnvironment(World world, int x, int y, int z) {
		return new Environment((IGridNode) world.getTileEntity(x,y,z));
	}

	public static final class Environment extends ManagedTileEntityEnvironment<IGridNode> {

		public Environment(IGridNode tileEntity) {
			super(tileEntity, "me_node");
		}

		@Callback(doc = "function():number -- Returns the average of energy usage over the last 10 ticks.")
		public Object[] getAvgPowerUsage(final Context context, final Arguments args){
			final IGrid grid = tileEntity.getGrid();
			return grid!=null
					? new Object[]{((IEnergyGrid)grid.getCache(IEnergyGrid.class)).getAvgPowerUsage()}
					: new Object[]{null, "no grid"};
		}

		@Callback(doc = "function():number -- Returns the total amount of power available.")
		public Object[] getStoredPower(final Context context, final Arguments args){
			final IGrid grid = tileEntity.getGrid();
			return grid!=null
					? new Object[]{((IEnergyGrid)grid.getCache(IEnergyGrid.class)).getStoredPower()}
					: new Object[]{null, "no grid"};
		}

		@Callback(doc = "function():boolean -- Returns whether the network is powered.")
		public Object[] isNetworkPowered(final Context context, final Arguments args){
			final IGrid grid = tileEntity.getGrid();
			return grid!=null
					? new Object[]{((IEnergyGrid)grid.getCache(IEnergyGrid.class)).isNetworkPowered()}
					: new Object[]{null, "no grid"};
		}
	}
}