EnergonRelics/src/main/java/com/thebrokenrail/energonrelics/block/entity/forcefield/FieldProjectorBlockEntity.java

package com.thebrokenrail.energonrelics.block.entity.forcefield;

import com.thebrokenrail.energonrelics.api.block.entity.core.EnergyReceiverBlockEntity;
import com.thebrokenrail.energonrelics.api.energy.Action;
import com.thebrokenrail.energonrelics.block.forcefield.util.AbstractFieldBlock;
import com.thebrokenrail.energonrelics.block.forcefield.util.FieldProjectorBlock;
import com.thebrokenrail.energonrelics.config.HardcodedConfig;
import net.minecraft.block.BlockState;
import net.minecraft.block.entity.BlockEntityType;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.Direction;

import java.util.function.Function;

public class FieldProjectorBlockEntity extends EnergyReceiverBlockEntity {
    private final Function<BlockState, AbstractFieldBlock> block;

    public FieldProjectorBlockEntity(BlockEntityType<?> type, Function<BlockState, AbstractFieldBlock> block) {
        super(type);
        this.block = block;
    }

    protected void setLastBlock(BlockPos pos, BlockState lastState) {
    }

    protected long getCost() {
        return HardcodedConfig.FIELD_PROJECTOR_ENERGY_REQUIRED;
    }

    @Override
    protected void energyTick() {
        assert getWorld() != null;

        addAction(Action.createBlockStatePropertyAction(getCost(), FieldProjectorBlock.POWERED, true, false));
        if (getCachedState().get(FieldProjectorBlock.POWERED)) {
            Direction facing = getCachedState().get(FieldProjectorBlock.FACING);
            BlockState state = block.apply(getCachedState()).getDefaultState().with(AbstractFieldBlock.FACING, facing);

            boolean valid = false;
            for (int i = 1; i < HardcodedConfig.FIELD_MAX_SIZE + 1; i++) {
                BlockPos targetPos = getPos().offset(facing, i);
                BlockState targetState = getWorld().getBlockState(targetPos);
                if (targetState.isAir() || (targetState != state && targetState.getBlock() instanceof AbstractFieldBlock && targetState.get(AbstractFieldBlock.FACING) == facing)) {
                    getWorld().setBlockState(targetPos, state);
                } else if (targetState != state) {
                    setLastBlock(targetPos, targetState);
                    valid = true;
                    break;
                }
            }
            if (!valid) {
                setLastBlock(null, null);
            }
        }
    }
}