NaturesAura/src/main/java/de/ellpeck/naturesaura/commands/CommandAura.java

package de.ellpeck.naturesaura.commands;

import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.arguments.IntegerArgumentType;
import de.ellpeck.naturesaura.api.aura.chunk.IAuraChunk;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.Commands;
import net.minecraft.core.BlockPos;
import net.minecraft.network.chat.TextComponent;

public final class CommandAura {

    public static void register(CommandDispatcher<CommandSourceStack> dispatcher) {
        dispatcher.register(Commands.literal("naaura").requires(s -> s.hasPermission(2))
                .then(Commands.literal("add").then(Commands.argument("amount", IntegerArgumentType.integer(1)).executes(context -> {
                    int amount = IntegerArgumentType.getInteger(context, "amount");
                    CommandSourceStack source = context.getSource();
                    BlockPos pos = new BlockPos(source.getPosition());
                    while (amount > 0) {
                        BlockPos spot = IAuraChunk.getLowestSpot(source.getLevel(), pos, 35, pos);
                        amount -= IAuraChunk.getAuraChunk(source.getLevel(), spot).storeAura(spot, amount);
                    }
                    source.sendSuccess(new TextComponent("Added aura to area"), true);
                    return 0;
                })))
                .then(Commands.literal("remove").then(Commands.argument("amount", IntegerArgumentType.integer(1)).executes(context -> {
                    int amount = IntegerArgumentType.getInteger(context, "amount");
                    CommandSourceStack source = context.getSource();
                    BlockPos pos = new BlockPos(source.getPosition());
                    while (amount > 0) {
                        BlockPos spot = IAuraChunk.getHighestSpot(source.getLevel(), pos, 35, pos);
                        amount -= IAuraChunk.getAuraChunk(source.getLevel(), spot).drainAura(spot, amount);
                    }
                    source.sendSuccess(new TextComponent("Removed aura from area"), true);
                    return 0;
                })))
                .then(Commands.literal("reset").executes(context -> {
                    CommandSourceStack source = context.getSource();
                    BlockPos pos = new BlockPos(source.getPosition());
                    IAuraChunk.getSpotsInArea(source.getLevel(), pos, 35, (spot, amount) -> {
                        IAuraChunk chunk = IAuraChunk.getAuraChunk(source.getLevel(), spot);
                        if (amount > 0)
                            chunk.drainAura(spot, amount);
                        else
                            chunk.storeAura(spot, -amount);
                    });
                    source.sendSuccess(new TextComponent("Reset aura in area"), true);
                    return 0;
                })));
    }
}