Multiverse-Core/src/main/java/com/onarandombox/MultiverseCore/listeners/MVPlayerListener.java

package com.onarandombox.MultiverseCore.listeners;

import com.dumptruckman.minecraft.util.Logging;
import com.onarandombox.MultiverseCore.MultiverseCore;
import com.onarandombox.MultiverseCore.api.MVWorld;
import com.onarandombox.MultiverseCore.api.MVWorldManager;
import com.onarandombox.MultiverseCore.api.SafeTTeleporter;
import com.onarandombox.MultiverseCore.config.MVCoreConfig;
import com.onarandombox.MultiverseCore.economy.MVEconomist;
import com.onarandombox.MultiverseCore.event.MVRespawnEvent;
import com.onarandombox.MultiverseCore.inject.InjectableListener;
import com.onarandombox.MultiverseCore.teleportation.TeleportQueue;
import com.onarandombox.MultiverseCore.utils.permissions.PermissionsChecker;
import com.onarandombox.MultiverseCore.utils.result.ResultGroup;
import com.onarandombox.MultiverseCore.world.entrycheck.BlacklistResult;
import com.onarandombox.MultiverseCore.world.entrycheck.EntryFeeResult;
import com.onarandombox.MultiverseCore.world.entrycheck.PlayerLimitResult;
import com.onarandombox.MultiverseCore.world.entrycheck.WorldAccessResult;
import com.onarandombox.MultiverseCore.world.entrycheck.WorldEntryCheckerProvider;
import io.vavr.control.Option;
import jakarta.inject.Inject;
import jakarta.inject.Provider;
import org.bukkit.GameMode;
import org.bukkit.Location;
import org.bukkit.Material;
import org.bukkit.Server;
import org.bukkit.World;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Player;
import org.bukkit.event.EventHandler;
import org.bukkit.event.EventPriority;
import org.bukkit.event.player.PlayerChangedWorldEvent;
import org.bukkit.event.player.PlayerJoinEvent;
import org.bukkit.event.player.PlayerPortalEvent;
import org.bukkit.event.player.PlayerRespawnEvent;
import org.bukkit.event.player.PlayerTeleportEvent;
import org.jetbrains.annotations.NotNull;
import org.jvnet.hk2.annotations.Service;

@Service
public class MVPlayerListener implements InjectableListener {
    private final @NotNull MultiverseCore plugin;
    private final @NotNull MVCoreConfig config;
    private final @NotNull Server server;
    private final @NotNull TeleportQueue teleportQueue;
    private final @NotNull Provider<MVWorldManager> worldManagerProvider;
    private final @NotNull WorldEntryCheckerProvider worldEntryCheckerProvider;
    private final @NotNull MVEconomist economist;
    private final @NotNull PermissionsChecker permissionsChecker;
    private final @NotNull SafeTTeleporter safeTTeleporter;

    @Inject
    MVPlayerListener(
            @NotNull MultiverseCore plugin,
            @NotNull MVCoreConfig config,
            @NotNull Server server,
            @NotNull TeleportQueue teleportQueue,
            @NotNull Provider<MVWorldManager> worldManagerProvider,
            @NotNull WorldEntryCheckerProvider worldEntryCheckerProvider,
            @NotNull MVEconomist economist,
            @NotNull PermissionsChecker permissionsChecker,
            @NotNull SafeTTeleporter safeTTeleporter
            ) {
        this.plugin = plugin;
        this.config = config;
        this.server = server;
        this.teleportQueue = teleportQueue;
        this.worldManagerProvider = worldManagerProvider;
        this.worldEntryCheckerProvider = worldEntryCheckerProvider;
        this.economist = economist;
        this.permissionsChecker = permissionsChecker;
        this.safeTTeleporter = safeTTeleporter;
    }

    @EventHandler
    public void playerJoin(PlayerJoinEvent event) {
        Player player = event.getPlayer();
        Location spawnLocation = getWorldManager().getFirstSpawnWorld().getSpawnLocation();

        if (!player.hasPlayedBefore()) {
            Logging.finer("Player joined for the FIRST time!");
            if (config.getFirstSpawnOverride()) {
                Logging.fine("Moving NEW player to spawn location: %s", spawnLocation);
                oneTickLater(() -> player.teleport(spawnLocation));
            }
            return;
        }

        Logging.finer("Player joined AGAIN!");
        MVWorld world = getWorldManager().getMVWorld(player.getWorld());
        if (world == null) {
            // We don't want to do anything if the player is not in a world managed by MV.
            Logging.fine("Player joined in a world not managed by MV.");
            return;
        }

        worldEntryCheckerProvider.forWorld(player, world)
                .canStayInWorld()
                .onSuccess(() -> oneTickLater(() -> handleGameModeAndFlight(player, world)))
                .onFailure(() -> {
                    player.sendMessage("[MV] - Sorry you can't be in this world anymore!");
                    oneTickLater(() -> player.teleport(spawnLocation));
                });
    }

    @EventHandler(priority = EventPriority.LOW)
    public void playerRespawn(PlayerRespawnEvent event) {
        MVWorld playerWorld = getWorldManager().getMVWorld(event.getPlayer().getWorld());
        if (playerWorld == null) {
            // We don't want to do anything if the player is not in a world managed by MV.
            Logging.fine("Player respawned in a world not managed by MV.");
            return;
        }

        if (playerWorld.getBedRespawn() && (event.isBedSpawn() || event.isAnchorSpawn())) {
            Logging.fine("Spawning %s at their %s.", event.getPlayer().getName(), event.isBedSpawn() ? "BED" : "ANCHOR");
            return;
        }

        Location respawnLocation = getMostAccurateRespawnLocation(playerWorld.getRespawnToWorld());
        MVRespawnEvent respawnEvent = new MVRespawnEvent(respawnLocation, event.getPlayer(), "compatability"); //TODO: Update this event with proper respawn method
        this.server.getPluginManager().callEvent(respawnEvent);
        event.setRespawnLocation(respawnEvent.getPlayersRespawnLocation());
    }

    private Location getMostAccurateRespawnLocation(@NotNull World respawnWorld) {
        return Option.of(getWorldManager().getMVWorld(respawnWorld))
                .map(MVWorld::getSpawnLocation)
                .getOrElse(respawnWorld.getSpawnLocation());
    }

    @EventHandler(priority = EventPriority.LOWEST)
    public void playerPortalCheck(PlayerPortalEvent event) {
        if (event.isCancelled()) {
            return;
        }

        // REMEMBER! getTo MAY be NULL HERE!!!
        // If the player was actually outside of the portal, adjust the from location
        if (event.getFrom().getWorld().getBlockAt(event.getFrom()).getType() != Material.NETHER_PORTAL) {
            Location newloc = safeTTeleporter.findPortalBlockNextTo(event.getFrom());
            // TODO: Fix this. Currently, we only check for PORTAL blocks. I'll have to figure out what
            // TODO: we want to do here.
            if (newloc != null) {
                event.setFrom(newloc);
            }
        }
    }

    @EventHandler(priority = EventPriority.HIGH)
    public void playerPortal(PlayerPortalEvent event) {
        if (event.isCancelled()) {
            // We don't want to do anything if the event is already cancelled.
            return;
        }

        Location toLocation = event.getTo();
        if (toLocation == null) {
            // We don't want to do anything if the destination is null.
            Logging.fine("Player '%s' entering a portal with null destination location.", event.getPlayer().getName());
            return;
        }

        Player player = event.getPlayer();

        MVWorld fromWorld = getWorldManager().getMVWorld(event.getFrom().getWorld());
        MVWorld toWorld = getWorldManager().getMVWorld(toLocation.getWorld());

        if (toWorld == null) {
            // We don't want to do anything if the destination world is not managed by MV.
            Logging.fine("Player '%s' is teleporting to world '%s' which is not managed by Multiverse-Core. No further actions will be taken by Multiverse-Core.",
                    player.getName(), toLocation.getWorld());
            return;
        }
        if (toWorld.equals(fromWorld)) {
            // We don't want to do anything if the destination world is the same as the origin world.
            Logging.fine("Player '%s' is teleporting to the same world.", player.getName());
            return;
        }

        ResultGroup worldEntryResult = worldEntryCheckerProvider.forWorld(player, toWorld).canEnterWorld(fromWorld)
                .onSuccess(() -> {
                    Logging.fine("Player '%s' is allowed to use portals to enter world '%s'.", player.getName(), toWorld.getName());
                    if (!config.isUsingCustomPortalSearch()) {
                        event.setSearchRadius(config.getCustomPortalSearchRadius());
                    }
                })
                .onFailure(() -> {
                    event.setCancelled(true);
                    Logging.fine("Player '%s' is not allowed to use portals to enter world '%s'.", player.getName(), toWorld.getName());
                });

        Logging.finer("Portal entry result for player '%s', from '%s' to '%s': %s",
                player.getName(), fromWorld == null ? "null" : fromWorld.getName(), toWorld.getName(), worldEntryResult);

        sendWorldEntryMessage(player, player, fromWorld, toWorld, worldEntryResult);
    }

    @EventHandler(priority = EventPriority.HIGHEST)
    public void playerTeleport(PlayerTeleportEvent event) {
        if (event.isCancelled()) {
            // We don't want to do anything if the event is already cancelled.
            return;
        }

        Player teleportee = event.getPlayer();

        Option<String> teleporterName = teleportQueue.popFromQueue(teleportee.getName());
        if (teleporterName.isEmpty() && !config.getTeleportIntercept()) {
            // We don't want to do anything if teleport interception is disabled, and teleport is not by MV.
            return;
        }

        CommandSender teleporter = teleporterName.map(name -> name.equals("CONSOLE")
                        ? server.getConsoleSender()
                        : server.getPlayerExact(name))
                .getOrElse(teleportee);

        Logging.fine("Inferred teleporter '%s' for teleportee '%s'.", teleporter.getName(), teleportee.getName());

        Option<MVWorld> fromWorld = Option.of(getWorldManager().getMVWorld(event.getFrom().getWorld()));
        MVWorld toWorld = Option.of(event.getTo()).map(to -> getWorldManager().getMVWorld(to.getWorld())).getOrNull();

        if (toWorld == null) {
            // We don't want to do anything if the destination world is not managed by MV.
            Logging.fine("Player '%s' is teleporting to world '%s' which is not managed by Multiverse-Core. No further actions will be taken by Multiverse-Core.",
                    teleportee.getName(), event.getTo().getWorld());
            return;
        }
        if (fromWorld.filter(world -> world.equals(toWorld)).isDefined()) {
            // We don't want to do anything if the destination world is the same as the origin world.
            Logging.fine("Player '%s' is teleporting to the same world.", teleportee.getName());
            return;
        }

        ResultGroup worldEntryResult = worldEntryCheckerProvider.forWorld(teleportee, toWorld)
                .canEnterWorld(fromWorld.getOrNull())
                .onSuccess(() -> Logging.fine("MV-Core is allowing '%s' to go to '%s'.", teleportee.getName(), toWorld.getName()))
                .onSuccessReason(EntryFeeResult.Success.ENOUGH_MONEY, () -> {
                    economist.payEntryFee((Player) teleporter, toWorld);
                    //TODO send payment message
                })
                .onFailure(() -> {
                    event.setCancelled(true);
                    Logging.fine("MV-Core is denying '%s' from going to '%s'.", teleportee.getName(), toWorld.getName());
                });

        Logging.fine("World entry result for player '%s', from '%s' to '%s': %s",
                teleportee.getName(), fromWorld.map(MVWorld::getName).getOrNull(), toWorld.getName(), worldEntryResult);

        sendWorldEntryMessage(teleporter, teleportee, fromWorld.getOrNull(), toWorld, worldEntryResult);
    }

    private void sendWorldEntryMessage(CommandSender teleporter, Player teleportee, MVWorld fromWorld, MVWorld toWorld, ResultGroup worldEntryResult) {
        if (worldEntryResult.isSuccess()) {
            return;
        }

        if (teleportee.equals(teleporter)) {
            teleporter.sendMessage("You are unable to teleport to world '%s'".formatted(toWorld.getName()));
        } else {
            teleporter.sendMessage("You are unable to teleport '%s' to world '%s'".formatted(teleportee.getName(), toWorld.getName()));
        }

        worldEntryResult
                .onFailureReason(WorldAccessResult.Failure.class, reason -> {
                    teleporter.sendMessage("You do not have permission to access world '%s'".formatted(toWorld.getName()));
                })
                .onFailureReason(PlayerLimitResult.Failure.class, reason -> {
                    teleporter.sendMessage("World '%s' is full".formatted(toWorld.getName()));
                })
                .onFailureReason(EntryFeeResult.Failure.class, reason -> {
                    teleporter.sendMessage("You do not have enough money to pay world '%s' entry fee.".formatted(toWorld.getName()));
                    //TODO Formatted entry fee amount
                })
                .onFailureReason(BlacklistResult.Failure.class, reason -> {
                    teleporter.sendMessage("World '%s' is blacklisted by world '%s'".formatted(fromWorld.getName(), toWorld.getName()));
                });
    }

    @EventHandler(priority = EventPriority.MONITOR)
    public void playerChangedWorld(PlayerChangedWorldEvent event) {
        MVWorld toWorld = getWorldManager().getMVWorld(event.getPlayer().getWorld());
        if (toWorld == null) {
            // We don't want to do anything if the destination world is not managed by MV.
            Logging.fine("Player '%s' is changing to world '%s' which is not managed by Multiverse-Core. No further actions will be taken by Multiverse-Core.",
                    event.getPlayer().getName(), event.getPlayer().getWorld());
            return;
        }
        this.handleGameModeAndFlight(event.getPlayer(), toWorld);
    }

    public void handleGameModeAndFlight(Player player, MVWorld world) {
        if (!config.getEnforceGameMode()) {
            Logging.fine("GameMode enforcement is disabled. Not changing game mode.");
            return;
        }
        if (!player.getWorld().equals(world.getCBWorld())) {
            Logging.fine("Player '%s' is not in world '%s'. Not changing game mode or flight.", player.getName(), world.getName());
            return;
        }
        if (permissionsChecker.hasGameModeBypassPermission(player, world)) {
            Logging.fine("Player '%s' has bypass permission. Not changing game mode or flight.", player.getName());
            return;
        }

        GameMode targetGameMode = world.getGameMode();
        Logging.fine("Handling gamemode for player %s: Changing to %s", player.getName(), targetGameMode.toString());
        player.setGameMode(targetGameMode);

        // TODO need a override permission for this
        if (player.getAllowFlight() && !world.getAllowFlight() && player.getGameMode() != GameMode.CREATIVE) {
            player.setAllowFlight(false);
            if (player.isFlying()) {
                player.setFlying(false);
            }
        } else if (world.getAllowFlight()) {
            if (player.getGameMode() == GameMode.CREATIVE) {
                player.setAllowFlight(true);
            }
        }
    }

    private void oneTickLater(Runnable runnable) {
        server.getScheduler().scheduleSyncDelayedTask(plugin, runnable, 1L);
    }

    private MVWorldManager getWorldManager() {
        return worldManagerProvider.get();
    }
}