package com.example.aicompanion.monitoring; import com.example.aicompanion.AICompanionMod; import com.example.aicompanion.bot.AIBot; import com.example.aicompanion.data.AIProfile; import com.example.aicompanion.storage.ProfileStorage; import net.minecraft.block.entity.BlockEntity; import net.minecraft.block.entity.ChestBlockEntity; import net.minecraft.item.Item; import net.minecraft.item.Items; import net.minecraft.server.MinecraftServer; import net.minecraft.server.network.ServerPlayerEntity; import net.minecraft.server.world.ServerWorld; import net.minecraft.text.Text; import net.minecraft.util.math.BlockPos; import net.minecraft.world.World; import java.util.*; public class ChestMonitor { private static final Map> chestSnapshots = new HashMap<>(); private static int tickCounter = 0; private static final int SCAN_INTERVAL = 20 * 500; public static void tick(MinecraftServer server) { tickCounter++; if (tickCounter % SCAN_INTERVAL == 0) { scanChests(server); } } private static void scanChests(MinecraftServer server) { for (AIProfile profile : ProfileStorage.getAllProfiles().values()) { if (!profile.isActive()) continue; ServerPlayerEntity bot = AIBot.getBot(profile.getName()); if (bot == null) continue; ServerWorld world = bot.getServerWorld(); BlockPos basePos = profile.getBasePos(); scanAreaChests(world, basePos, profile); } } private static void scanAreaChests(ServerWorld world, BlockPos center, AIProfile profile) { int radius = 16; Map currentSnapshots = new HashMap<>(); for (int x = -radius; x <= radius; x++) { for (int y = -radius; y <= radius; y++) { for (int z = -radius; z <= radius; z++) { BlockPos pos = center.add(x, y, z); BlockEntity blockEntity = world.getBlockEntity(pos); if (blockEntity instanceof ChestBlockEntity chest) { ChestSnapshot snapshot = ChestSnapshot.create(chest); currentSnapshots.put(pos, snapshot); checkForGrief(profile, pos, snapshot); } } } } chestSnapshots.put(profile.getName(), currentSnapshots); } private static void checkForGrief(AIProfile profile, BlockPos pos, ChestSnapshot current) { Map previous = chestSnapshots.get(profile.getName()); if (previous == null) return; ChestSnapshot old = previous.get(pos); if (old == null) return; if (current.totalItems < old.totalItems) { ServerPlayerEntity bot = AIBot.getBot(profile.getName()); if (bot != null) { bot.sendMessage(Text.literal("§c[System Alert] Chest mismatch! Saaman gayab hai."), false); AICompanionMod.LOGGER.warn("Chest grief detected for {} at {}", profile.getName(), pos); } } } private static class ChestSnapshot { int totalItems; Map itemCounts; static ChestSnapshot create(ChestBlockEntity chest) { ChestSnapshot snapshot = new ChestSnapshot(); snapshot.totalItems = 0; snapshot.itemCounts = new HashMap<>(); for (int i = 0; i < chest.size(); i++) { var stack = chest.getStack(i); if (!stack.isEmpty()) { snapshot.totalItems += stack.getCount(); snapshot.itemCounts.merge(stack.getItem(), stack.getCount(), Integer::sum); } } return snapshot; } ChestType classify() { if (itemCounts.containsKey(Items.DIAMOND) || itemCounts.containsKey(Items.EMERALD)) { return ChestType.VALUABLES; } else if (itemCounts.containsKey(Items.IRON_INGOT) || itemCounts.containsKey(Items.COAL)) { return ChestType.RESOURCES; } else { return ChestType.TRADEABLES; } } } private enum ChestType { VALUABLES, RESOURCES, TRADEABLES } }