EnergonRelics/src/main/java/com/thebrokenrail/energonrelics/energy/helper/EnergyGeneratorBlockEntity.java

package com.thebrokenrail.energonrelics.energy.helper;

import com.thebrokenrail.energonrelics.energy.core.EnergyProviderBlockEntity;
import com.thebrokenrail.energonrelics.energy.core.util.Action;
import net.minecraft.block.entity.BlockEntityType;

public abstract class EnergyGeneratorBlockEntity extends EnergyProviderBlockEntity implements EnergyGenerator {
    public EnergyGeneratorBlockEntity(BlockEntityType<?> type) {
        super(type);
    }

    private long energy = 0;

    @Override
    public boolean isEnergyProvider() {
        return true;
    }

    @Override
    public long getEnergy() {
        return energy;
    }

    @Override
    public void setEnergy(long value) {
        if (value < 0) {
            throw new UnsupportedOperationException();
        }
        energy = value;
    }

    @Override
    protected void handlePropagatedAction(Action.PropagatedAction action) {
        super.handlePropagatedAction(action);
        handlePropagatedActionWithGenerator(action);
    }

    @Override
    public abstract long getDisplayEnergy();
}