Stage 12: Data storage system

src/main/java/dev/loki/killer/KillerCarry.java

@@ -0,0 +1,189 @@
+package dev.loki.killer;
+
+import dev.loki.config.GameConfig;
+import dev.loki.player.Killer;
+import dev.loki.player.Survivor;
+import org.bukkit.Bukkit;
+import org.bukkit.Location;
+import org.bukkit.entity.Player;
+import org.bukkit.plugin.Plugin;
+import org.bukkit.scheduler.BukkitTask;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.UUID;
+
+public class KillerCarry {
+    private final Plugin plugin;
+    private final GameConfig config;
+    private final Map<UUID, CarryData> activeCarries;
+    private BukkitTask carryTask;
+
+    private static class CarryData {
+        final Killer killer;
+        final Survivor survivor;
+        double wiggleProgress;
+
+        CarryData(Killer killer, Survivor survivor) {
+            this.killer = killer;
+            this.survivor = survivor;
+            this.wiggleProgress = 0;
+        }
+    }
+
+    public KillerCarry(Plugin plugin, GameConfig config) {
+        this.plugin = plugin;
+        this.config = config;
+        this.activeCarries = new HashMap<>();
+    }
+
+    public boolean tryPickup(Killer killer, Survivor survivor) {
+        if (!canPickup(killer, survivor)) {
+            return false;
+        }
+
+        Player killerPlayer = killer.getPlayer();
+        Player survivorPlayer = survivor.getPlayer();
+
+        if (killerPlayer == null || survivorPlayer == null) return false;
+
+        if (!HitValidator.isValidCarryPosition(killerPlayer, survivorPlayer)) {
+            return false;
+        }
+
+        startCarry(killer, survivor);
+        return true;
+    }
+
+    public boolean canPickup(Killer killer, Survivor survivor) {
+        if (killer.isCarrying()) return false;
+        if (!survivor.getState().name().equals("DOWNED")) return false;
+
+        Player killerPlayer = killer.getPlayer();
+        Player survivorPlayer = survivor.getPlayer();
+
+        return killerPlayer != null && survivorPlayer != null && survivorPlayer.isOnline();
+    }
+
+    private void startCarry(Killer killer, Survivor survivor) {
+        killer.setCarrying(true);
+
+        Player survivorPlayer = survivor.getPlayer();
+        if (survivorPlayer != null) {
+            survivorPlayer.setInvisible(true);
+        }
+
+        CarryData data = new CarryData(killer, survivor);
+        activeCarries.put(killer.getUuid(), data);
+
+        if (carryTask == null) {
+            startCarryTask();
+        }
+    }
+
+    public boolean drop(Killer killer) {
+        CarryData data = activeCarries.remove(killer.getUuid());
+        if (data == null) return false;
+
+        stopCarry(data);
+        return true;
+    }
+
+    private void stopCarry(CarryData data) {
+        data.killer.setCarrying(false);
+
+        Player survivorPlayer = data.survivor.getPlayer();
+        if (survivorPlayer != null) {
+            survivorPlayer.setInvisible(false);
+            survivorPlayer.teleport(data.killer.getPlayer().getLocation());
+        }
+
+        if (activeCarries.isEmpty() && carryTask != null) {
+            carryTask.cancel();
+            carryTask = null;
+        }
+    }
+
+    private void startCarryTask() {
+        carryTask = Bukkit.getScheduler().runTaskTimer(plugin, () -> {
+            activeCarries.values().removeIf(data -> {
+                if (!isValidCarry(data)) {
+                    stopCarry(data);
+                    return true;
+                }
+
+                updateCarryPosition(data);
+                updateWiggle(data);
+
+                return false;
+            });
+        }, 1L, 1L);
+    }
+
+    private boolean isValidCarry(CarryData data) {
+        Player killerPlayer = data.killer.getPlayer();
+        Player survivorPlayer = data.survivor.getPlayer();
+
+        if (killerPlayer == null || survivorPlayer == null) return false;
+        if (!killerPlayer.isOnline() || !survivorPlayer.isOnline()) return false;
+
+        return true;
+    }
+
+    private void updateCarryPosition(CarryData data) {
+        Player killerPlayer = data.killer.getPlayer();
+        Player survivorPlayer = data.survivor.getPlayer();
+
+        Location carryLoc = killerPlayer.getLocation().clone().add(0, -0.5, 0);
+        survivorPlayer.teleport(carryLoc);
+    }
+
+    private void updateWiggle(CarryData data) {
+        Player survivorPlayer = data.survivor.getPlayer();
+        if (survivorPlayer == null) return;
+
+        if (survivorPlayer.isSneaking()) {
+            data.wiggleProgress += 1.5;
+        }
+
+        if (data.wiggleProgress >= 100) {
+            forceDrop(data.killer);
+        }
+    }
+
+    public void forceDrop(Killer killer) {
+        CarryData data = activeCarries.remove(killer.getUuid());
+        if (data == null) return;
+
+        Player survivorPlayer = data.survivor.getPlayer();
+        if (survivorPlayer != null) {
+            Location dropLoc = survivorPlayer.getLocation();
+            dropLoc.add(killer.getPlayer().getLocation().getDirection().multiply(1.5));
+            survivorPlayer.teleport(dropLoc);
+        }
+
+        stopCarry(data);
+    }
+
+    public boolean isCarrying(Killer killer) {
+        return activeCarries.containsKey(killer.getUuid());
+    }
+
+    public Survivor getCarried(Killer killer) {
+        CarryData data = activeCarries.get(killer.getUuid());
+        return data != null ? data.survivor : null;
+    }
+
+    public double getWiggleProgress(Killer killer) {
+        CarryData data = activeCarries.get(killer.getUuid());
+        return data != null ? data.wiggleProgress : 0;
+    }
+
+    public void shutdown() {
+        if (carryTask != null) {
+            carryTask.cancel();
+        }
+        activeCarries.values().forEach(this::stopCarry);
+        activeCarries.clear();
+    }
+}