From fa906f7f4f11b78779831b34b765d2e6adce5c7a Mon Sep 17 00:00:00 2001
From: yoyosource <yoyosource@nidido.de>
Date: Sun, 19 Mar 2023 14:11:09 +0100
Subject: [PATCH] Hotfix Xray

Signed-off-by: yoyosource <yoyosource@nidido.de>
---
 .../utils/PlayerMovementWrapper19.java        | 32 +++++++++++++++++++
 .../bausystem/features/xray/XrayCommand.java  |  1 +
 .../utils/PlayerMovementWrapper.java          |  2 ++
 3 files changed, 35 insertions(+)

diff --git a/BauSystem_19/src/de/steamwar/bausystem/utils/PlayerMovementWrapper19.java b/BauSystem_19/src/de/steamwar/bausystem/utils/PlayerMovementWrapper19.java
index fd9a892e..a3531e5e 100644
--- a/BauSystem_19/src/de/steamwar/bausystem/utils/PlayerMovementWrapper19.java
+++ b/BauSystem_19/src/de/steamwar/bausystem/utils/PlayerMovementWrapper19.java
@@ -21,19 +21,51 @@ package de.steamwar.bausystem.utils;
 
 import net.minecraft.network.protocol.game.PacketPlayInFlying;
 import net.minecraft.server.level.EntityPlayer;
+import org.bukkit.Location;
 import org.bukkit.craftbukkit.v1_19_R2.entity.CraftPlayer;
 import org.bukkit.entity.Player;
 
+import java.util.HashMap;
+import java.util.Map;
+import java.util.UUID;
+
 public class PlayerMovementWrapper19 implements PlayerMovementWrapper {
 
+    private static class Position {
+        private double x;
+        private double y;
+        private double z;
+        private float yaw;
+        private float pitch;
+    }
+
+    private Map<UUID, Position> playerLocationMap = new HashMap<>();
+
     @Override
     public void setPosition(Player player, Object object) {
+        Position position = playerLocationMap.computeIfAbsent(player.getUniqueId(), uuid -> new Position());
         PacketPlayInFlying packetPlayInFlying = ((PacketPlayInFlying) object);
         EntityPlayer entityPlayer = ((CraftPlayer) player).getHandle();
         if (packetPlayInFlying.h) {
             entityPlayer.b(packetPlayInFlying.a, packetPlayInFlying.b, packetPlayInFlying.c, packetPlayInFlying.d, packetPlayInFlying.e);
+            position.x = packetPlayInFlying.a;
+            position.y = packetPlayInFlying.b;
+            position.z = packetPlayInFlying.c;
+            position.yaw = packetPlayInFlying.d;
+            position.pitch = packetPlayInFlying.e;
         } else {
             entityPlayer.e(packetPlayInFlying.a, packetPlayInFlying.b, packetPlayInFlying.c);
+            position.x = packetPlayInFlying.a;
+            position.y = packetPlayInFlying.b;
+            position.z = packetPlayInFlying.c;
+        }
+    }
+
+    @Override
+    public void disable(Player player) {
+        Position position = playerLocationMap.remove(player.getUniqueId());
+        if (position != null) {
+            player.teleport(new Location(player.getWorld(), position.x, position.y, position.z, position.yaw, position.pitch));
         }
     }
 }
diff --git a/BauSystem_Main/src/de/steamwar/bausystem/features/xray/XrayCommand.java b/BauSystem_Main/src/de/steamwar/bausystem/features/xray/XrayCommand.java
index 22c19b67..f1232760 100644
--- a/BauSystem_Main/src/de/steamwar/bausystem/features/xray/XrayCommand.java
+++ b/BauSystem_Main/src/de/steamwar/bausystem/features/xray/XrayCommand.java
@@ -97,6 +97,7 @@ public class XrayCommand extends SWCommand implements Listener {
         techHiderCommand.disable(region, player);
         if (hidden.get(region).contains(player)) {
             hidden.get(region).remove(player);
+            PlayerMovementWrapper.impl.disable(player);
             BauSystem.MESSAGE.sendPrefixless("XRAY_OFF", player, ChatMessageType.ACTION_BAR);
         } else {
             hidden.get(region).add(player);
diff --git a/BauSystem_Main/src/de/steamwar/bausystem/utils/PlayerMovementWrapper.java b/BauSystem_Main/src/de/steamwar/bausystem/utils/PlayerMovementWrapper.java
index f8aa209d..4a6a6447 100644
--- a/BauSystem_Main/src/de/steamwar/bausystem/utils/PlayerMovementWrapper.java
+++ b/BauSystem_Main/src/de/steamwar/bausystem/utils/PlayerMovementWrapper.java
@@ -27,4 +27,6 @@ public interface PlayerMovementWrapper {
     PlayerMovementWrapper impl = VersionDependent.getVersionImpl(BauSystem.getInstance());
 
     void setPosition(Player player, Object object);
+    default void disable(Player player) {
+    }
 }