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