feat(utils): safe random teleporter

auioc · Jun 27, 2022 · d2391ec · d2391ec
1 parent 41b3775
commit d2391ec
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/src/main/java/org/auioc/mcmod/arnicalib/utils/game/RandomTeleporter.java b/src/main/java/org/auioc/mcmod/arnicalib/utils/game/RandomTeleporter.java
@@ -1,10 +1,42 @@
 package org.auioc.mcmod.arnicalib.utils.game;
 
+import org.auioc.mcmod.arnicalib.api.java.exception.HException;
+import net.minecraft.core.BlockPos;
 import net.minecraft.world.entity.LivingEntity;
 import net.minecraft.world.phys.Vec3;
 
 public class RandomTeleporter {
 
+    private static BlockPos findRandomSafePos(LivingEntity living, BlockPos center, int radius) throws HException {
+        var level = living.getLevel();
+        var pos = PositionUtils.random(center, radius, living.getRandom()).mutable();
+        if (PositionUtils.isInWorldBounds(pos, level)) {
+            PositionUtils.setToStandableY(pos, center.getY() - radius, level); // throws HException
+            if (AABBUtils.isEmpty(AABBUtils.moveTo(living.getBoundingBox(), pos), level)) {
+                return pos.immutable();
+            }
+        }
+        throw new HException();
+    }
+
+    public static boolean safeTeleport(LivingEntity living, BlockPos center, int radius, int maxTries) {
+        if (living.level.isClientSide) return false;
+
+        for (int t = 0; t < maxTries; t++) {
+            try {
+                EntityUtils.teleportTo(living, findRandomSafePos(living, center, radius));
+                return true;
+            } catch (HException e) {
+                continue;
+            }
+        }
+        return false;
+    }
+
+    public static boolean safeTeleport(LivingEntity living, int radius, int maxTries) {
+        return safeTeleport(living, living.blockPosition(), radius, maxTries);
+    }
+
     public static void unsafeTeleport(LivingEntity living, Vec3 center, double radius) {
         EntityUtils.teleportTo(living, PositionUtils.random(center, radius, living.getRandom()));
     }