Utiliser snapTo seulement sur la même region

Petite explication: Sur Folia, un déplacement d'entité effectué via snapTo déclenche une mise à jour des sections qui necessite d'être exécutée sur le thread propriétaire de la région initiale.
Cependant, il est possible que ce dernier ne soit pas dans la même region initiale, de ce fait Folia rejette la demande en générant l'erreur: Cannot move entity off-main qu'on se mette sur le thread de l'entité, ou le thread de la future localisation.
De ce fait, je fais une vérification au préalable si c'est le même monde et que la region tické pour l'entité et la future localisation est le même, et si c'est le cas, snapTo fonctionne, sinon il faut lancer une teleportation async.

De ce fait, quand tout est réuni, je garde le fonctionnement initial, sinon on est obligé de teleporter l'entité

Author: Euphillya

v1_20_R4/src/main/java/net/citizensnpcs/nms/v1_20_R4/util/NMSImpl.java

@@ -1618,9 +1618,15 @@ public void setKnockbackResistance(org.bukkit.entity.LivingEntity entity, double
 
     @Override
     public void setLocationDirectly(org.bukkit.entity.Entity entity, Location location) {
-        // Todo temp teleport
         if (net.citizensnpcs.api.util.SpigotUtil.isFoliaServer()) {
-            net.citizensnpcs.api.util.SpigotUtil.teleportAsync(entity, location);
+            if (location.getWorld() == entity.getWorld()
+                    && CitizensAPI.getScheduler().isOnOwnerThread(entity)
+                    && CitizensAPI.getScheduler().isOnOwnerThread(location)) {
+                getHandle(entity).moveTo(location.getX(), location.getY(), location.getZ(), location.getYaw(),
+                        location.getPitch());
+            } else {
+                net.citizensnpcs.api.util.SpigotUtil.teleportAsync(entity, location);
+            }
             return;
         }
         getHandle(entity).moveTo(location.getX(), location.getY(), location.getZ(), location.getYaw(),

v1_21_R5/src/main/java/net/citizensnpcs/nms/v1_21_R5/util/NMSImpl.java

@@ -1611,7 +1611,14 @@ public void setHeadYaw(org.bukkit.entity.Entity entity, float yaw) {
     @Override
     public void setLocationDirectly(org.bukkit.entity.Entity entity, Location location) {
         if (net.citizensnpcs.api.util.SpigotUtil.isFoliaServer()) {
-            net.citizensnpcs.api.util.SpigotUtil.teleportAsync(entity, location);
+            if (location.getWorld() == entity.getWorld()
+                    && CitizensAPI.getScheduler().isOnOwnerThread(entity)
+                    && CitizensAPI.getScheduler().isOnOwnerThread(location)) {
+                getHandle(entity).snapTo(location.getX(), location.getY(), location.getZ(), location.getYaw(),
+                        location.getPitch());
+            } else {
+                net.citizensnpcs.api.util.SpigotUtil.teleportAsync(entity, location);
+            }
             return;
         }
         getHandle(entity).snapTo(location.getX(), location.getY(), location.getZ(), location.getYaw(),

v1_21_R6/src/main/java/net/citizensnpcs/nms/v1_21_R6/util/NMSImpl.java

@@ -1647,7 +1647,14 @@ public void setHeadYaw(org.bukkit.entity.Entity entity, float yaw) {
     public void setLocationDirectly(org.bukkit.entity.Entity entity, Location location) {
         // Todo - temp teleport
         if (net.citizensnpcs.api.util.SpigotUtil.isFoliaServer()) {
-            net.citizensnpcs.api.util.SpigotUtil.teleportAsync(entity, location);
+            if (location.getWorld() == entity.getWorld()
+                    && CitizensAPI.getScheduler().isOnOwnerThread(entity)
+                    && CitizensAPI.getScheduler().isOnOwnerThread(location)) {
+                getHandle(entity).snapTo(location.getX(), location.getY(), location.getZ(), location.getYaw(),
+                        location.getPitch());
+            } else {
+                net.citizensnpcs.api.util.SpigotUtil.teleportAsync(entity, location);
+            }
             return;
         }
         getHandle(entity).snapTo(location.getX(), location.getY(), location.getZ(), location.getYaw(),