From d2391ec8e70ce70118279b38a4be084e431891db Mon Sep 17 00:00:00 2001
From: WakelessSloth56 <wsloth56@foxmail.com>
Date: Tue, 28 Jun 2022 03:07:03 +0800
Subject: [PATCH] feat(utils): safe random teleporter

---
 .../utils/game/RandomTeleporter.java          | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)

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
index 4868263a..d98e392a 100644
--- 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()));
     }