package dev.eggchan.optimizer.prediction; import dev.eggchan.optimizer.util.ChatMessage; import dev.eggchan.optimizer.util.EggchanLogger; import net.minecraft.client.MinecraftClient; import net.minecraft.client.world.ClientWorld; import net.minecraft.entity.Entity; import net.minecraft.entity.decoration.EndCrystalEntity; import net.minecraft.util.math.BlockPos; import java.util.HashMap; import java.util.Iterator; import java.util.Map; public class EntityPredictor { private static final long PLACE_TTL_MS = 5000; private static final long REMOVAL_TTL_MS = 5000; private static final Map placedEntities = new HashMap<>(); private static final Map placedTimestamps = new HashMap<>(); private static final Map pendingRemovals = new HashMap<>(); public static void markRemoved(int entityId) { pendingRemovals.put(entityId, System.currentTimeMillis()); } public static boolean wasRemoved(int entityId) { return pendingRemovals.containsKey(entityId); } public static void clearRemoved(int entityId) { pendingRemovals.remove(entityId); } public static void trackPlaced(BlockPos pos, EndCrystalEntity entity) { placedEntities.put(pos, entity); placedTimestamps.put(pos, System.currentTimeMillis()); EggchanLogger.info("Tracking ghost crystal at " + pos); } public static void confirmPlaced(BlockPos serverPos, int serverEntityId, ClientWorld world) { placedTimestamps.remove(serverPos); EndCrystalEntity predicted = placedEntities.remove(serverPos); if (predicted != null) { predicted.remove(); EggchanLogger.info("Ghost crystal confirmed by server at " + serverPos + " (entity id=" + serverEntityId + ")"); ChatMessage.sendConfirmedMessage(serverPos); } } public static void rollbackPlaced(BlockPos pos) { placedTimestamps.remove(pos); EndCrystalEntity ghost = placedEntities.remove(pos); if (ghost != null) { ghost.remove(); EggchanLogger.warn("Ghost crystal rejected/rolled back at " + pos); ChatMessage.sendRejectionMessage(pos); } } public static void correctEntityPosition(int entityId, double x, double y, double z, ClientWorld world) { Entity entity = world.getEntityById(entityId); if (entity != null) { entity.updatePosition(x, y, z); EggchanLogger.info("Corrected entity " + entityId + " position to (" + x + ", " + y + ", " + z + ")"); } } public static void tickTTL(MinecraftClient client) { long now = System.currentTimeMillis(); Iterator> placeIt = placedTimestamps.entrySet().iterator(); while (placeIt.hasNext()) { Map.Entry entry = placeIt.next(); if (now - entry.getValue() > PLACE_TTL_MS) { BlockPos pos = entry.getKey(); placeIt.remove(); EndCrystalEntity ghost = placedEntities.remove(pos); if (ghost != null) ghost.remove(); EggchanLogger.warn("Ghost crystal TTL expired at " + pos + ", discarding"); ChatMessage.sendTimeoutMessage(pos); } } Iterator> removeIt = pendingRemovals.entrySet().iterator(); while (removeIt.hasNext()) { Map.Entry entry = removeIt.next(); if (now - entry.getValue() > REMOVAL_TTL_MS) { EggchanLogger.warn("Pending removal TTL expired for entity id=" + entry.getKey() + ", clearing"); removeIt.remove(); } } } public static void rollbackAllPending() { for (EndCrystalEntity entity : placedEntities.values()) { entity.remove(); } placedEntities.clear(); placedTimestamps.clear(); EggchanLogger.info("All pending ghost crystals rolled back"); } public static void clear() { pendingRemovals.clear(); placedEntities.clear(); placedTimestamps.clear(); } }