DreamTeleport.java

Go to the documentation of this file.

/*
 * MIT License
 *
 * Copyright (c) 2025 Dreamfire Studio
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
package com.dreamfirestudios.dreamcore.DreamTeleport;
 
import net.kyori.adventure.text.Component;
import org.bukkit.Bukkit;
import org.bukkit.Location;
import org.bukkit.World;
import org.bukkit.entity.LivingEntity;
import org.bukkit.entity.Player;
import org.bukkit.plugin.Plugin;
import org.bukkit.scheduler.BukkitTask;
 
import java.lang.ref.WeakReference;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
 
public final class DreamTeleport implements Runnable {
 
    public static final long DEFAULT_PERIOD_TICKS = 20L;
 
    private final Plugin plugin;
    private final Map<UUID, Location> startLocations = new ConcurrentHashMap<>();
    private final WeakReference<LivingEntity> liveTarget;   // may be null
    private final Location fixedTarget;                     // may be null
    private final boolean showCountdown;
    private final boolean cancelOnMove;
    private final double moveToleranceSq;
    private final long periodTicks;
 
    private final int totalSeconds;
    private int secondsLeft;
 
    private BukkitTask task;
 
    public DreamTeleport(
            Plugin plugin,
            Collection<Player> players,
            LivingEntity liveTarget,
            Location fixedTarget,
            int seconds,
            long periodTicks,
            boolean showCountdown,
            boolean cancelOnMove,
            double moveTolerance
    ) {
        this.plugin = Objects.requireNonNull(plugin, "plugin");
        Objects.requireNonNull(players, "players");
        if (liveTarget == null && fixedTarget == null) {
            throw new IllegalArgumentException("Either liveTarget or fixedTarget must be provided");
        }
 
        for (Player p : players) {
            if (p != null && p.isOnline()) {
                startLocations.put(p.getUniqueId(), p.getLocation().clone());
            }
        }
 
        this.liveTarget = liveTarget == null ? null : new WeakReference<>(liveTarget);
        this.fixedTarget = fixedTarget == null ? null : fixedTarget.clone();
        this.totalSeconds = Math.max(0, seconds);
        this.secondsLeft = this.totalSeconds;
        this.periodTicks = Math.max(1L, periodTicks);
        this.showCountdown = showCountdown;
        this.cancelOnMove = cancelOnMove;
        this.moveToleranceSq = Math.max(0.0, moveTolerance) * Math.max(0.0, moveTolerance);
    }
 
    public void start() {
        if (task != null) return;
        task = Bukkit.getScheduler().runTaskTimer(plugin, this, 0L, periodTicks);
    }
 
    public void cancel() {
        if (task != null) task.cancel();
        task = null;
        startLocations.clear();
    }
 
    public boolean isEmpty() {
        return startLocations.isEmpty();
    }
 
    public boolean contains(Player player) {
        return player != null && startLocations.containsKey(player.getUniqueId());
    }
 
    public boolean remove(Player player) {
        if (player == null) return isEmpty();
        startLocations.remove(player.getUniqueId());
        return isEmpty();
    }
 
    @Override
    public void run() {
        // if no players remain, stop the task
        if (startLocations.isEmpty()) {
            cancel();
            return;
        }
 
        // Per-tick housekeeping & countdown display
        if (secondsLeft > 0) {
            // show actionbar countdown (optional)
            if (showCountdown) {
                forEachOnlinePlayer(p -> p.sendActionBar(Component.text(
                        "Teleporting in " + secondsLeft + " / " + totalSeconds + "…"
                )));
            }
 
            // cancel on move (if enabled)
            if (cancelOnMove) {
                final List<UUID> toRemove = new ArrayList<>();
                for (Map.Entry<UUID, Location> e : startLocations.entrySet()) {
                    Player p = Bukkit.getPlayer(e.getKey());
                    if (p == null || !p.isOnline()) {
                        toRemove.add(e.getKey());
                        continue;
                    }
                    Location start = e.getValue();
                    Location now = p.getLocation();
 
                    if (!sameWorld(start, now) || start.toVector().distanceSquared(now.toVector()) > moveToleranceSq) {
                        p.sendActionBar(Component.text("Teleport cancelled (you moved)."));
                        toRemove.add(e.getKey());
                    }
                }
                toRemove.forEach(startLocations::remove);
                if (startLocations.isEmpty()) {
                    cancel();
                    return;
                }
            }
 
            secondsLeft--;
            return;
        }
 
        // Time's up: teleport remaining players
        Location target = resolveTarget();
        if (target == null) {
            // live target vanished AND no fixed fallback; cancel safely
            forEachOnlinePlayer(p -> p.sendActionBar(Component.text("Teleport failed: no valid target.")));
            cancel();
            return;
        }
 
        forEachOnlinePlayer(p -> p.teleport(target));
        cancel(); // finished
    }
 
    private void forEachOnlinePlayer(java.util.function.Consumer<Player> consumer) {
        for (UUID id : new ArrayList<>(startLocations.keySet())) {
            Player p = Bukkit.getPlayer(id);
            if (p != null && p.isOnline()) {
                consumer.accept(p);
            } else {
                startLocations.remove(id);
            }
        }
    }
 
    private static boolean sameWorld(Location a, Location b) {
        World wa = a.getWorld();
        World wb = b.getWorld();
        return wa != null && wa.equals(wb);
    }
 
    private Location resolveTarget() {
        if (liveTarget != null) {
            LivingEntity e = liveTarget.get();
            if (e != null && e.isValid()) {
                return e.getLocation();
            }
        }
        return fixedTarget;
    }
 
    // --- Getters ---
    public int getSecondsLeft() { return secondsLeft; }
    public int getTotalSeconds() { return totalSeconds; }
}