package com.nomoreactors.command; import com.mojang.brigadier.Command; import com.mojang.brigadier.CommandDispatcher; import com.mojang.brigadier.arguments.DoubleArgumentType; import com.mojang.brigadier.arguments.IntegerArgumentType; import com.mojang.brigadier.arguments.StringArgumentType; import com.mojang.brigadier.context.CommandContext; import com.nomoreactors.NoMoreActors; import com.nomoreactors.entity.BotEntity; import com.nomoreactors.entity.ModEntities; import com.nomoreactors.manager.BotManager; import net.minecraft.command.argument.EntityArgumentType; import net.minecraft.command.argument.ItemStackArgumentType; import net.minecraft.entity.EquipmentSlot; import net.minecraft.item.ItemStack; import net.minecraft.item.Items; import net.minecraft.server.command.CommandManager; import net.minecraft.server.command.ServerCommandSource; import net.minecraft.server.network.ServerPlayerEntity; import net.minecraft.text.Text; import net.minecraft.util.math.Vec3d; import java.util.List; public class BotCommand { public static void register(CommandDispatcher dispatcher) { dispatcher.register(CommandManager.literal("bot") .requires(source -> source.hasPermissionLevel(2)) .then(CommandManager.literal("spawn") .then(CommandManager.argument("count", IntegerArgumentType.integer(1, 500)) .executes(BotCommand::spawnBots) ) ) .then(CommandManager.literal("remove") .then(CommandManager.literal("all") .executes(BotCommand::removeAllBots) ) .then(CommandManager.literal("nearest") .executes(BotCommand::removeNearestBot) ) ) .then(CommandManager.literal("attack") .then(CommandManager.literal("on") .executes(ctx -> setAttackMode(ctx, true)) ) .then(CommandManager.literal("off") .executes(ctx -> setAttackMode(ctx, false)) ) .then(CommandManager.literal("target") .then(CommandManager.argument("target", EntityArgumentType.entity()) .executes(BotCommand::setTarget) ) ) ) .then(CommandManager.literal("equip") .then(CommandManager.literal("weapon") .then(CommandManager.argument("item", ItemStackArgumentType.itemStack(null)) .executes(BotCommand::equipWeapon) ) ) .then(CommandManager.literal("armor") .then(CommandManager.argument("slot", StringArgumentType.word()) .then(CommandManager.argument("item", ItemStackArgumentType.itemStack(null)) .executes(BotCommand::equipArmor) ) ) ) .then(CommandManager.literal("preset") .then(CommandManager.argument("preset", StringArgumentType.word()) .executes(BotCommand::equipPreset) ) ) ) .then(CommandManager.literal("formation") .then(CommandManager.literal("square") .then(CommandManager.argument("spacing", DoubleArgumentType.doubleArg(1.0, 10.0)) .executes(ctx -> setFormation(ctx, BotManager.FormationType.SQUARE)) ) ) .then(CommandManager.literal("line") .then(CommandManager.argument("spacing", DoubleArgumentType.doubleArg(1.0, 10.0)) .executes(ctx -> setFormation(ctx, BotManager.FormationType.LINE)) ) ) .then(CommandManager.literal("circle") .then(CommandManager.argument("spacing", DoubleArgumentType.doubleArg(1.0, 10.0)) .executes(ctx -> setFormation(ctx, BotManager.FormationType.CIRCLE)) ) ) .then(CommandManager.literal("wedge") .then(CommandManager.argument("spacing", DoubleArgumentType.doubleArg(1.0, 10.0)) .executes(ctx -> setFormation(ctx, BotManager.FormationType.WEDGE)) ) ) .then(CommandManager.literal("wall") .then(CommandManager.argument("spacing", DoubleArgumentType.doubleArg(1.0, 10.0)) .executes(ctx -> setFormation(ctx, BotManager.FormationType.WALL)) ) ) .then(CommandManager.literal("clear") .executes(BotCommand::clearFormation) ) ) .then(CommandManager.literal("info") .executes(BotCommand::showInfo) ) ); } private static int spawnBots(CommandContext ctx) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); int count = IntegerArgumentType.getInteger(ctx, "count"); BotManager manager = NoMoreActors.getBotManager(); int spawned = 0; Vec3d playerPos = player.getPos(); for (int i = 0; i < count; i++) { if (!manager.canSpawnBot(player)) { break; } BotEntity bot = new BotEntity(ModEntities.BOT_ENTITY, player.getServerWorld()); double angle = (2 * Math.PI * i) / count; double radius = 3.0; double x = playerPos.x + Math.cos(angle) * radius; double z = playerPos.z + Math.sin(angle) * radius; bot.refreshPositionAndAngles(x, playerPos.y, z, 0.0f, 0.0f); bot.setOwner(player); player.getServerWorld().spawnEntity(bot); manager.addBot(player, bot); spawned++; } source.sendFeedback(() -> Text.literal("Spawned " + spawned + " bot(s)"), false); return Command.SINGLE_SUCCESS; } catch (Exception e) { source.sendError(Text.literal("Failed to spawn bots: " + e.getMessage())); return 0; } } private static int removeAllBots(CommandContext ctx) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); BotManager manager = NoMoreActors.getBotManager(); int count = manager.getBotCount(player); manager.removeAllBots(player); source.sendFeedback(() -> Text.literal("Removed " + count + " bot(s)"), false); return Command.SINGLE_SUCCESS; } catch (Exception e) { source.sendError(Text.literal("Failed to remove bots: " + e.getMessage())); return 0; } } private static int removeNearestBot(CommandContext ctx) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); BotManager manager = NoMoreActors.getBotManager(); List bots = manager.getBots(player); BotEntity nearest = null; double minDist = Double.MAX_VALUE; for (BotEntity bot : bots) { double dist = bot.squaredDistanceTo(player); if (dist < minDist) { minDist = dist; nearest = bot; } } if (nearest != null) { nearest.discard(); source.sendFeedback(() -> Text.literal("Removed nearest bot"), false); } else { source.sendError(Text.literal("No bots found")); } return Command.SINGLE_SUCCESS; } catch (Exception e) { source.sendError(Text.literal("Failed to remove bot: " + e.getMessage())); return 0; } } private static int setAttackMode(CommandContext ctx, boolean attack) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); BotManager manager = NoMoreActors.getBotManager(); List bots = manager.getBots(player); for (BotEntity bot : bots) { bot.setAttacking(attack); } String mode = attack ? "ON" : "OFF"; source.sendFeedback(() -> Text.literal("Attack mode: " + mode), false); return Command.SINGLE_SUCCESS; } catch (Exception e) { source.sendError(Text.literal("Failed to set attack mode: " + e.getMessage())); return 0; } } private static int setTarget(CommandContext ctx) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); var target = EntityArgumentType.getEntity(ctx, "target"); BotManager manager = NoMoreActors.getBotManager(); if (target instanceof net.minecraft.entity.LivingEntity livingTarget) { List bots = manager.getBots(player); for (BotEntity bot : bots) { bot.setTarget(livingTarget); bot.setAttacking(true); } source.sendFeedback(() -> Text.literal("Bots targeting: " + target.getName().getString()), false); return Command.SINGLE_SUCCESS; } else { source.sendError(Text.literal("Target must be a living entity")); return 0; } } catch (Exception e) { source.sendError(Text.literal("Failed to set target: " + e.getMessage())); return 0; } } private static int equipWeapon(CommandContext ctx) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); ItemStack item = ItemStackArgumentType.getItemStackArgument(ctx, "item").createStack(1, false); BotManager manager = NoMoreActors.getBotManager(); List bots = manager.getBots(player); for (BotEntity bot : bots) { bot.equipItem(EquipmentSlot.MAINHAND, item); } source.sendFeedback(() -> Text.literal("Equipped weapon: " + item.getName().getString()), false); return Command.SINGLE_SUCCESS; } catch (Exception e) { source.sendError(Text.literal("Failed to equip weapon: " + e.getMessage())); return 0; } } private static int equipArmor(CommandContext ctx) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); String slotName = StringArgumentType.getString(ctx, "slot"); ItemStack item = ItemStackArgumentType.getItemStackArgument(ctx, "item").createStack(1, false); BotManager manager = NoMoreActors.getBotManager(); EquipmentSlot slot = switch (slotName.toLowerCase()) { case "head", "helmet" -> EquipmentSlot.HEAD; case "chest", "chestplate" -> EquipmentSlot.CHEST; case "legs", "leggings" -> EquipmentSlot.LEGS; case "feet", "boots" -> EquipmentSlot.FEET; default -> null; }; if (slot == null) { source.sendError(Text.literal("Invalid slot: " + slotName)); return 0; } List bots = manager.getBots(player); for (BotEntity bot : bots) { bot.equipItem(slot, item); } source.sendFeedback(() -> Text.literal("Equipped " + slotName + ": " + item.getName().getString()), false); return Command.SINGLE_SUCCESS; } catch (Exception e) { source.sendError(Text.literal("Failed to equip armor: " + e.getMessage())); return 0; } } private static int equipPreset(CommandContext ctx) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); String preset = StringArgumentType.getString(ctx, "preset"); BotManager manager = NoMoreActors.getBotManager(); ItemStack helmet, chest, legs, boots, weapon; switch (preset.toLowerCase()) { case "leather" -> { helmet = new ItemStack(Items.LEATHER_HELMET); chest = new ItemStack(Items.LEATHER_CHESTPLATE); legs = new ItemStack(Items.LEATHER_LEGGINGS); boots = new ItemStack(Items.LEATHER_BOOTS); weapon = new ItemStack(Items.IRON_SWORD); } case "iron" -> { helmet = new ItemStack(Items.IRON_HELMET); chest = new ItemStack(Items.IRON_CHESTPLATE); legs = new ItemStack(Items.IRON_LEGGINGS); boots = new ItemStack(Items.IRON_BOOTS); weapon = new ItemStack(Items.IRON_SWORD); } case "diamond" -> { helmet = new ItemStack(Items.DIAMOND_HELMET); chest = new ItemStack(Items.DIAMOND_CHESTPLATE); legs = new ItemStack(Items.DIAMOND_LEGGINGS); boots = new ItemStack(Items.DIAMOND_BOOTS); weapon = new ItemStack(Items.DIAMOND_SWORD); } case "netherite" -> { helmet = new ItemStack(Items.NETHERITE_HELMET); chest = new ItemStack(Items.NETHERITE_CHESTPLATE); legs = new ItemStack(Items.NETHERITE_LEGGINGS); boots = new ItemStack(Items.NETHERITE_BOOTS); weapon = new ItemStack(Items.NETHERITE_SWORD); } case "archer" -> { helmet = new ItemStack(Items.LEATHER_HELMET); chest = new ItemStack(Items.LEATHER_CHESTPLATE); legs = new ItemStack(Items.LEATHER_LEGGINGS); boots = new ItemStack(Items.LEATHER_BOOTS); weapon = new ItemStack(Items.BOW); } default -> { source.sendError(Text.literal("Unknown preset: " + preset)); return 0; } } List bots = manager.getBots(player); for (BotEntity bot : bots) { bot.equipFullSet(helmet, chest, legs, boots, weapon); } source.sendFeedback(() -> Text.literal("Equipped preset: " + preset), false); return Command.SINGLE_SUCCESS; } catch (Exception e) { source.sendError(Text.literal("Failed to equip preset: " + e.getMessage())); return 0; } } private static int setFormation(CommandContext ctx, BotManager.FormationType type) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); double spacing = DoubleArgumentType.getDouble(ctx, "spacing"); BotManager manager = NoMoreActors.getBotManager(); manager.setFormation(player, type, spacing); source.sendFeedback(() -> Text.literal("Formation set: " + type.name()), false); return Command.SINGLE_SUCCESS; } catch (Exception e) { source.sendError(Text.literal("Failed to set formation: " + e.getMessage())); return 0; } } private static int clearFormation(CommandContext ctx) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); BotManager manager = NoMoreActors.getBotManager(); List bots = manager.getBots(player); for (BotEntity bot : bots) { bot.setFormationPosition(null, -1); bot.setFormationType(0); } source.sendFeedback(() -> Text.literal("Formation cleared"), false); return Command.SINGLE_SUCCESS; } catch (Exception e) { source.sendError(Text.literal("Failed to clear formation: " + e.getMessage())); return 0; } } private static int showInfo(CommandContext ctx) { ServerCommandSource source = ctx.getSource(); try { ServerPlayerEntity player = source.getPlayerOrThrow(); BotManager manager = NoMoreActors.getBotManager(); int count = manager.getBotCount(player); source.sendFeedback(() -> Text.literal("Bot Info:"), false); source.sendFeedback(() -> Text.literal(" Active Bots: " + count + "/" + BotManager.MAX_BOTS_PER_PLAYER), false); source.sendFeedback(() -> Text.literal(" Server Limit: " + BotManager.MAX_BOTS_TOTAL), false); return Command.SINGLE_SUCCESS; } catch (Exception e) { source.sendError(Text.literal("Failed to get info: " + e.getMessage())); return 0; } } }