Updated a cache to use UUID instead of Player

src/main/java/re/imc/geysermodelengine/listener/ModelListener.java

@@ -43,14 +43,14 @@ public class ModelListener implements Listener {
         ModelEntityData model = map.get(event.getVehicle());
 
         if (model != null && event.getPassenger() instanceof Player player) {
-            plugin.getModelManager().getDriversCache().put(player, Pair.of(event.getVehicle(), event.getSeat()));
+            plugin.getModelManager().getDriversCache().put(player.getUniqueId(), Pair.of(event.getVehicle(), event.getSeat()));
         }
     }
 
     @EventHandler(priority = EventPriority.MONITOR)
     public void onModelDismount(ModelDismountEvent event) {
         if (event.getPassenger() instanceof Player player) {
-            plugin.getModelManager().getDriversCache().remove(player);
+            plugin.getModelManager().getDriversCache().remove(player.getUniqueId());
         }
     }
 

src/main/java/re/imc/geysermodelengine/listener/MountPacketListener.java

@@ -28,7 +28,7 @@ public class MountPacketListener implements PacketListener {
         Player player = event.getPlayer();
 
         WrapperPlayClientEntityAction action = new WrapperPlayClientEntityAction(event);
-        Pair<ActiveModel, Mount> seat = plugin.getModelManager().getDriversCache().get(player);
+        Pair<ActiveModel, Mount> seat = plugin.getModelManager().getDriversCache().get(player.getUniqueId());
 
         if (seat == null) return;
         if (action.getAction() != WrapperPlayClientEntityAction.Action.START_SNEAKING) return;

src/main/java/re/imc/geysermodelengine/managers/model/ModelManager.java

@@ -22,7 +22,7 @@ public class ModelManager {
     private final ConcurrentHashMap<Integer, Map<ActiveModel, ModelEntityData>> entitiesCache = new ConcurrentHashMap<>();
     private final ConcurrentHashMap<Integer, ModelEntityData> modelEntitiesCache = new ConcurrentHashMap<>();
 
-    private final ConcurrentHashMap<Player, Pair<ActiveModel, Mount>> driversCache = new ConcurrentHashMap<>();
+    private final ConcurrentHashMap<UUID, Pair<ActiveModel, Mount>> driversCache = new ConcurrentHashMap<>();
 
     public ModelManager(GeyserModelEngine plugin) {
         this.plugin = plugin;
@@ -65,7 +65,7 @@ public class ModelManager {
         return modelEntitiesCache;
     }
 
-    public ConcurrentHashMap<Player, Pair<ActiveModel, Mount>> getDriversCache() {
+    public ConcurrentHashMap<UUID, Pair<ActiveModel, Mount>> getDriversCache() {
         return driversCache;
     }
 }

src/main/java/re/imc/geysermodelengine/runnables/BedrockMountControlRunnable.java

@@ -28,7 +28,7 @@ public class BedrockMountControlRunnable implements Consumer<ScheduledTask> {
             if (!FloodgateApi.getInstance().isFloodgatePlayer(player.getUniqueId())) continue;
 
             float pitch = player.getLocation().getPitch();
-            Pair<ActiveModel, Mount> seat = plugin.getModelManager().getDriversCache().get(player);
+            Pair<ActiveModel, Mount> seat = plugin.getModelManager().getDriversCache().get(player.getUniqueId());
 
             if (seat == null) continue;