package net.chip_gkj.startermod.commands; import com.google.gson.Gson; import com.google.gson.GsonBuilder; import com.google.gson.reflect.TypeToken; import com.mojang.brigadier.CommandDispatcher; import com.mojang.brigadier.StringReader; import com.mojang.brigadier.arguments.ArgumentType; import com.mojang.brigadier.context.CommandContext; import com.mojang.brigadier.exceptions.CommandSyntaxException; import com.mojang.brigadier.suggestion.Suggestions; import com.mojang.brigadier.suggestion.SuggestionsBuilder; import net.minecraft.commands.CommandSourceStack; import net.minecraft.commands.Commands; import net.minecraft.core.registries.BuiltInRegistries; import net.minecraft.network.chat.Component; import net.minecraft.server.MinecraftServer; import net.minecraft.server.level.ServerLevel; import net.minecraft.server.level.ServerPlayer; import net.minecraft.world.entity.EntityType; import net.minecraft.world.entity.Mob; import net.neoforged.bus.api.SubscribeEvent; import net.neoforged.fml.common.EventBusSubscriber; import net.neoforged.neoforge.event.RegisterCommandsEvent; import net.neoforged.neoforge.event.server.ServerStartedEvent; import net.neoforged.neoforge.event.tick.ServerTickEvent; import net.neoforged.neoforge.server.ServerLifecycleHooks; import java.io.*; import java.lang.reflect.Type; import java.nio.file.Files; import java.nio.file.Path; import java.nio.file.Paths; import java.util.HashSet; import java.util.Set; import java.util.concurrent.CompletableFuture; import java.util.stream.Stream; @EventBusSubscriber(modid = "chipgkj_mob_remover") public class CommandHandler { private static final Set BLACKLIST = new HashSet<>(); private static final Path CONFIG_PATH = Paths.get("config", "chipgkj_mob_remover_data.json"); private static final Gson GSON = new GsonBuilder().setPrettyPrinting().create(); private static final Type TYPE = new TypeToken>() {}.getType(); private static int tickCounter = 0; private static boolean serverReady = false; // ============================================================ // ЗАГРУЗКА ПРИ СТАРТЕ // ============================================================ static { loadConfig(); } private static void loadConfig() { if (!Files.exists(CONFIG_PATH)) return; try (Reader reader = Files.newBufferedReader(CONFIG_PATH)) { Set loaded = GSON.fromJson(reader, TYPE); if (loaded != null) { BLACKLIST.clear(); BLACKLIST.addAll(loaded); } } catch (Exception ignored) {} } private static void saveConfig() { try (Writer writer = Files.newBufferedWriter(CONFIG_PATH)) { GSON.toJson(BLACKLIST, writer); } catch (Exception ignored) {} } // ============================================================ // АРГУМЕНТ С АВТОДОПОЛНЕНИЕМ (КАК В /SUMMON) // ============================================================ public static class MobArgumentType implements ArgumentType { public static MobArgumentType mob() { return new MobArgumentType(); } @Override public String parse(StringReader reader) throws CommandSyntaxException { int start = reader.getCursor(); while (reader.canRead() && reader.peek() != ' ') { reader.skip(); } return reader.getString().substring(start, reader.getCursor()); } @Override public CompletableFuture listSuggestions(CommandContext context, SuggestionsBuilder builder) { Stream mobNames = BuiltInRegistries.ENTITY_TYPE.stream() .filter(type -> { try { return type.create(null) instanceof Mob; } catch (Exception e) { return false; } }) .map(type -> BuiltInRegistries.ENTITY_TYPE.getKey(type).toString()); mobNames.forEach(builder::suggest); return builder.buildFuture(); } } @SubscribeEvent public static void registerCommands(RegisterCommandsEvent event) { CommandDispatcher dispatcher = event.getDispatcher(); dispatcher.register(Commands.literal("disable") .then(Commands.literal("add") .then(Commands.argument("mob", MobArgumentType.mob()) .executes(ctx -> { String mob = ctx.getArgument("mob", String.class); BLACKLIST.add(mob); saveConfig(); ctx.getSource().sendSuccess( () -> Component.literal("§aМоб " + mob + " добавлен в чёрный список!"), false ); return 1; }))) .then(Commands.literal("remove") .then(Commands.argument("mob", MobArgumentType.mob()) .executes(ctx -> { String mob = ctx.getArgument("mob", String.class); BLACKLIST.remove(mob); saveConfig(); ctx.getSource().sendSuccess( () -> Component.literal("§aМоб " + mob + " удалён из чёрного списка!"), false ); return 1; }))) .then(Commands.literal("list") .executes(ctx -> { if (BLACKLIST.isEmpty()) { ctx.getSource().sendSuccess( () -> Component.literal("§eСписок пуст!"), false ); } else { String list = String.join(", ", BLACKLIST); ctx.getSource().sendSuccess( () -> Component.literal("§aЗапрещены: §e" + list), false ); } return 1; })) .then(Commands.literal("clear") .executes(ctx -> { BLACKLIST.clear(); saveConfig(); ctx.getSource().sendSuccess( () -> Component.literal("§aСписок очищен!"), false ); return 1; })) ); dispatcher.register(Commands.literal("kill_mobs") .executes(ctx -> { ServerPlayer player = ctx.getSource().getPlayer(); if (player == null) { ctx.getSource().sendFailure(Component.literal("Только для игроков!")); return 0; } int count = removeAllMobs(); ctx.getSource().sendSuccess( () -> Component.literal("§aУдалено мобов: §e" + count), false ); return 1; }) ); } // ============================================================ // ВКЛЮЧАЕМ ТИКЕР ПОСЛЕ ЗАГРУЗКИ СЕРВЕРА // ============================================================ @SubscribeEvent public static void onServerStarted(ServerStartedEvent event) { serverReady = true; System.out.println("[Мод] Автоудаление включено!"); } // ============================================================ // АВТОУДАЛЕНИЕ (ТИКЕР) — РАБОТАЕТ ТОЛЬКО ПОСЛЕ ЗАГРУЗКИ // ============================================================ @SubscribeEvent public static void onServerTick(ServerTickEvent.Post event) { try { if (!serverReady) return; if (BLACKLIST.isEmpty()) return; tickCounter++; if (tickCounter < 200) return; tickCounter = 0; removeAllMobs(); } catch (Exception ignored) {} } // ============================================================ // УДАЛЕНИЕ МОБОВ // ============================================================ private static int removeAllMobs() { if (BLACKLIST.isEmpty()) return 0; MinecraftServer server = ServerLifecycleHooks.getCurrentServer(); if (server == null) return 0; int count = 0; for (ServerLevel level : server.getAllLevels()) { for (var entity : level.getEntities().getAll()) { if (entity instanceof Mob mob) { String name = EntityType.getKey(mob.getType()).toString(); if (BLACKLIST.contains(name)) { mob.discard(); count++; } } } } return count; } }