RandomTPAPI.java

package com.gmail.picono435.randomtp.api;

import com.gmail.picono435.randomtp.RandomTPMod;
import com.gmail.picono435.randomtp.config.Config;
import com.gmail.picono435.randomtp.config.Messages;
import com.mojang.datafixers.util.Pair;
import dev.architectury.injectables.annotations.ExpectPlatform;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.core.BlockPos;
import net.minecraft.core.Holder;
import net.minecraft.core.registries.Registries;
import net.minecraft.network.chat.Component;
import net.minecraft.resources.ResourceKey;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.server.MinecraftServer;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.server.level.ServerPlayer;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.biome.Biome;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.block.CactusBlock;
import net.minecraft.world.level.block.LiquidBlock;
import net.minecraft.world.level.portal.TeleportTransition;
import net.minecraft.world.phys.Vec3;

import java.util.Map;
import java.util.Random;
import java.util.Set;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;

public class RandomTPAPI {

    private static final ExecutorService executorService = Executors.newFixedThreadPool(10);

    public static Future<Boolean> randomTeleport(ServerPlayer player, ServerLevel world) {
        return randomTeleport(player, world, null);
    }

    public static Future<Boolean> randomTeleport(ServerPlayer player, ServerLevel world, ResourceKey<Biome> biomeResourceKey) {
        return executorService.submit(() -> {
            try  {
                Random random = new Random();
                BlockPos.MutableBlockPos mutableBlockPos = new BlockPos.MutableBlockPos();
                if(biomeResourceKey == null) {
                    Pair<Integer, Integer> coordinates = generateCoordinates(world, player, random);
                    int x = coordinates.getFirst();
                    int z = coordinates.getSecond();

                    mutableBlockPos.setX(x);
                    mutableBlockPos.setY(50);
                    mutableBlockPos.setZ(z);
                } else {
                    Pair<BlockPos, Holder<Biome>> pair = world.findClosestBiome3d(biomeHolder -> biomeHolder.is(biomeResourceKey), player.getOnPos(), 6400, 32, 64);
                    if(pair == null) {
                        Component msg = Component.literal(Messages.getMaxTries().replaceAll("\\{playerName\\}", player.getName().getString()).replaceAll("&", "§"));
                        player.sendSystemMessage(msg, false);
                        return false;
                    }
                    mutableBlockPos.setX(pair.getFirst().getX());
                    mutableBlockPos.setY(50);
                    mutableBlockPos.setZ(pair.getFirst().getZ());
                    if(!world.getWorldBorder().isWithinBounds(mutableBlockPos)) {
                        Component msg = Component.literal(Messages.getMaxTries().replaceAll("\\{playerName\\}", player.getName().getString()).replaceAll("&", "§"));
                        player.sendSystemMessage(msg, false);
                        return false;
                    }
                }
                int maxTries = Config.getMaxTries();
                int y = mutableBlockPos.getY();
                while (!isSafe(world, mutableBlockPos) && (maxTries == -1 || maxTries > 0)) {
                    y++;
                    mutableBlockPos.setY(y);
                    if(mutableBlockPos.getY() >= 200 || !isInBiomeWhitelist(world.getBiome(mutableBlockPos.immutable()).unwrapKey().get().location())) {
                        if(biomeResourceKey != null) {
                            Pair<BlockPos, Holder<Biome>> pair = world.findClosestBiome3d(biomeHolder -> biomeHolder.is(biomeResourceKey), player.getOnPos(), 6400, 32, 64);
                            if(pair == null) {
                                Component msg = Component.literal(Messages.getMaxTries().replaceAll("\\{playerName\\}", player.getName().getString()).replaceAll("&", "§"));
                                player.sendSystemMessage(msg, false);
                                return false;
                            }
                            mutableBlockPos.setX(pair.getFirst().getX());
                            mutableBlockPos.setY(50);
                            mutableBlockPos.setZ(pair.getFirst().getZ());
                            if(!world.getWorldBorder().isWithinBounds(mutableBlockPos)) {
                                Component msg = Component.literal(Messages.getMaxTries().replaceAll("\\{playerName\\}", player.getName().getString()).replaceAll("&", "§"));
                                player.sendSystemMessage(msg, false);
                                return false;
                            }
                            continue;
                        }
                        Pair<Integer, Integer> coordinates = generateCoordinates(world, player, random);
                        int x = coordinates.getFirst();
                        int z = coordinates.getSecond();

                        mutableBlockPos.setX(x);
                        mutableBlockPos.setY(50);
                        mutableBlockPos.setZ(z);
                        continue;
                    }
                    if(maxTries > 0){
                        maxTries--;
                    }
                    if(maxTries == 0) {
                        Component msg = Component.literal(Messages.getMaxTries().replaceAll("\\{playerName\\}", player.getName().getString()).replaceAll("&", "§"));
                        player.sendSystemMessage(msg, false);
                        return false;
                    }
                }

                player.getServer().submit(() -> {
                    TeleportTransition teleportTransition = new TeleportTransition(world, mutableBlockPos.getCenter(), Vec3.ZERO, player.getYRot(), player.getXRot(), false, false, Set.of(), TeleportTransition.DO_NOTHING);
                    player.teleport(teleportTransition);
                    Component successful = Component.literal(Messages.getSuccessful().replaceAll("\\{playerName\\}", player.getName().getString()).replaceAll("\\{blockX\\}", "" + (int)player.position().x).replaceAll("\\{blockY\\}", "" + (int)player.position().y).replaceAll("\\{blockZ\\}", "" + (int)player.position().z).replaceAll("&", "§"));
                    player.sendSystemMessage(successful, false);
                });
                return true;
            } catch(Exception ex) {
                RandomTPMod.getLogger().info("Error executing command.");
                ex.printStackTrace();
                return false;
            }
        });
    }

    private static Pair<Integer, Integer> generateCoordinates(ServerLevel world, Player player, Random random) {
        // Calculating X coordinates
        int x;
        if(random.nextInt(2) == 1) {
            // Calculating X coordinates from min to max
            int maxDistance = Config.getMaxDistance() == 0 ? (int) world.getWorldBorder().getMinX() : (int) (player.getX() + Config.getMaxDistance());
            if(maxDistance < world.getWorldBorder().getMinX()) maxDistance = (int) world.getWorldBorder().getMinX();
            int minDistance = (int) (player.getX() - Config.getMinDistance());
            if(minDistance < world.getWorldBorder().getMinX()) minDistance = (int) (world.getWorldBorder().getMinX() + 10);
            if(maxDistance < minDistance) maxDistance = maxDistance ^ minDistance ^ (minDistance = maxDistance);
            if(maxDistance == minDistance) minDistance = minDistance - 1;
            x = random.nextInt(maxDistance - minDistance) + minDistance;
        } else {
            // Calculating X coordinates from max to min
            int maxDistance = Config.getMaxDistance() == 0 ? (int) world.getWorldBorder().getMaxX() : (int) (player.getX() - Config.getMaxDistance());
            if(maxDistance > world.getWorldBorder().getMaxX()) maxDistance = (int) world.getWorldBorder().getMaxX();
            int minDistance = (int) (player.getX() + Config.getMinDistance());
            if(minDistance > world.getWorldBorder().getMaxX()) minDistance = (int) (world.getWorldBorder().getMaxX() - 10);
            if(maxDistance < minDistance) maxDistance = maxDistance ^ minDistance ^ (minDistance = maxDistance);
            if(maxDistance == minDistance) minDistance = minDistance - 1;
            x = random.nextInt(maxDistance - minDistance) + minDistance;
        }
        int z;
        if(random.nextInt(2) == 1) {
            // Calculating Z coordinates from min to max
            int maxDistance = Config.getMaxDistance() == 0 ? (int) world.getWorldBorder().getMinZ() : (int) (player.getZ() + Config.getMaxDistance());
            if(maxDistance < world.getWorldBorder().getMinZ()) maxDistance = (int) world.getWorldBorder().getMinZ();
            int minDistance = (int) (player.getZ() - Config.getMinDistance());
            if(minDistance < world.getWorldBorder().getMinZ()) minDistance = (int) (world.getWorldBorder().getMinZ() + 10);
            if(maxDistance < minDistance) maxDistance = maxDistance ^ minDistance ^ (minDistance = maxDistance);
            if(maxDistance == minDistance) minDistance = minDistance - 1;
            z = random.nextInt(maxDistance - minDistance) + minDistance;
        } else {
            // Calculating Z coordinates from max to min
            int maxDistance = Config.getMaxDistance() == 0 ? (int) world.getWorldBorder().getMaxZ() : (int) (player.getZ() - Config.getMaxDistance());
            if(maxDistance > world.getWorldBorder().getMaxZ()) maxDistance = (int) world.getWorldBorder().getMaxZ();
            int minDistance = (int) (player.getZ() + Config.getMinDistance());
            if(minDistance > world.getWorldBorder().getMaxZ()) minDistance = (int) (world.getWorldBorder().getMaxZ() - 10);
            if(maxDistance < minDistance) maxDistance = maxDistance ^ minDistance ^ (minDistance = maxDistance);
            if(maxDistance == minDistance) minDistance = minDistance - 1;
            z = random.nextInt(maxDistance - minDistance) + minDistance;
        }
        return new Pair<>(x, z);
    }

    public static ServerLevel getWorld(String world, MinecraftServer server) {
        try {
            ResourceLocation resourcelocation = ResourceLocation.tryParse(world);
            ResourceKey<Level> registrykey = ResourceKey.create(Registries.DIMENSION, resourcelocation);
            ServerLevel worldTo = server.getLevel(registrykey);
            return worldTo;
        } catch(Exception ex) {
            ex.printStackTrace();
            return null;
        }
    }

    public static boolean checkCooldown(ServerPlayer player, Map<String, Long> cooldowns) {
        int cooldownTime = Config.getCooldown();
        if(cooldowns.containsKey(player.getName().getString())) {
            long secondsLeft = ((cooldowns.get(player.getName().getString())/1000)+cooldownTime) - (System.currentTimeMillis()/1000);
            return secondsLeft <= 0;
        } else {
            return true;
        }
    }

    public static long getCooldownLeft(ServerPlayer player, Map<String, Long> cooldowns) {
        int cooldownTime = Config.getCooldown();
        long secondsLeft = ((cooldowns.get(player.getName().getString())/1000)+cooldownTime) - (System.currentTimeMillis()/1000);
        return secondsLeft;
    }

    @ExpectPlatform
    public static boolean hasPermission(ServerPlayer player, String permission) {
        throw new AssertionError();
    }

    @ExpectPlatform
    public static boolean hasPermission(CommandSourceStack source, String permission) {
        throw new AssertionError();
    }

    public static boolean isSafe(ServerLevel world, BlockPos.MutableBlockPos mutableBlockPos) {
        return isEmpty(world, mutableBlockPos) && !isDangerBlocks(world, mutableBlockPos) && world.getWorldBorder().isWithinBounds(mutableBlockPos);
    }

    public static boolean isEmpty(ServerLevel world, BlockPos.MutableBlockPos mutableBlockPos) {
        return world.isEmptyBlock(mutableBlockPos.offset(0, 1, 0)) && world.isEmptyBlock(mutableBlockPos);
    }

    public static boolean isDangerBlocks(ServerLevel world, BlockPos.MutableBlockPos mutableBlockPos) {
        if(isDangerBlock(world, mutableBlockPos) && isDangerBlock(world, mutableBlockPos.offset(0, 1, 0)) &&
                isDangerBlock(world, mutableBlockPos.offset(0, -1, 0))) {
            return true;
        }
        return world.getBlockState(mutableBlockPos.offset(0, -1, 0)).getBlock() == Blocks.AIR;
    }

    public static boolean isDangerBlock(ServerLevel world, BlockPos mutableBlockPos) {
        return world.getBlockState(mutableBlockPos).getBlock() instanceof LiquidBlock
                || world.getBlockState(mutableBlockPos).getBlock() instanceof CactusBlock;
    }

    private static boolean isInBiomeWhitelist(ResourceLocation biome) {
        //WHITELIST
        if(Config.useBiomeWhitelist()) {
            if(biome == null) {
                return false;
            }
            return Config.getAllowedBiomes().contains(biome.toString());
            //BLACKLIST
        } else {
            if(biome == null) {
                return true;
            }
            return !Config.getAllowedBiomes().contains(biome.toString());
        }
    }
}