Module-LibsDisguises/src/main/java/dev/plex/listener/DisguiseListener.java

package dev.plex.listener;

import dev.plex.LibsDisguises;
import dev.plex.cache.DataUtils;
import dev.plex.player.PlexPlayer;
import dev.plex.util.PlexLog;
import java.util.ArrayList;
import java.util.List;
import me.libraryaddict.disguise.DisguiseAPI;
import me.libraryaddict.disguise.disguisetypes.DisguiseType;
import me.libraryaddict.disguise.disguisetypes.PlayerDisguise;
import me.libraryaddict.disguise.disguisetypes.watchers.AreaEffectCloudWatcher;
import me.libraryaddict.disguise.disguisetypes.watchers.EnderDragonWatcher;
import me.libraryaddict.disguise.disguisetypes.watchers.PhantomWatcher;
import me.libraryaddict.disguise.disguisetypes.watchers.SlimeWatcher;
import me.libraryaddict.disguise.disguisetypes.watchers.WitherWatcher;
import me.libraryaddict.disguise.events.DisguiseEvent;
import net.kyori.adventure.text.Component;
import net.kyori.adventure.text.format.NamedTextColor;
import net.kyori.adventure.text.serializer.legacy.LegacyComponentSerializer;
import net.kyori.adventure.text.serializer.plain.PlainTextComponentSerializer;
import org.bukkit.Bukkit;
import org.bukkit.command.Command;
import org.bukkit.command.PluginCommandYamlParser;
import org.bukkit.entity.Player;
import org.bukkit.event.EventHandler;
import org.bukkit.event.EventPriority;
import org.bukkit.event.player.PlayerCommandPreprocessEvent;
import org.bukkit.plugin.Plugin;

public class DisguiseListener extends PlexListener
{
    private static float safeYMod(float f)
    {
        return Math.max(-256f, Math.min(256f, f));
    }

    @EventHandler
    public void onDisguiseEvent(DisguiseEvent event)
    {
        event.setCancelled(true);
        PlexPlayer plexPlayer = null;
        if (event.getCommandSender() instanceof Player player)
        {
            plexPlayer = DataUtils.getPlayer(player.getUniqueId());
        }
        if (event.getDisguise().getType() == DisguiseType.FISHING_HOOK)
        {
            event.getCommandSender().sendMessage(Component.text("You cannot use Fishing Hook disguises").color(NamedTextColor.RED));
            return;
        }
        String name = event.getDisguise().getWatcher().getCustomName();
        if (name != null)
        {
            int noColorLen = PlainTextComponentSerializer.plainText().serialize(LegacyComponentSerializer.legacySection().deserialize(name)).length();
            // each color code counts as one char rather than two, for flexibility
            if (((name.length() - noColorLen) / 2) + noColorLen > 32)
            {
                event.getCommandSender().sendMessage(Component.text("Your disguise name is too long").color(NamedTextColor.RED));
                return;
            }
        }
        if (event.getDisguise().getWatcher() instanceof EnderDragonWatcher watcher && watcher.getPhase() == 7)
        {
            watcher.setPhase(6);
        }
        if (event.getDisguise().getWatcher() instanceof WitherWatcher watcher && watcher.getInvulnerability() > 2048)
        {
            watcher.setInvulnerability(2048);
        }
        if (event.getDisguise().isPlayerDisguise() && plexPlayer != null && !plexPlayer.getPlayer().hasPermission("plex.libsdisguises.player"))
        {
            PlayerDisguise playerDisguise = (PlayerDisguise)event.getDisguise();
            String targetName = playerDisguise.getName();
            String origName = event.getDisguised().getName();
            playerDisguise.setName(origName);
            playerDisguise.setNameVisible(true);
            playerDisguise.getWatcher().setNameYModifier(0);
            playerDisguise.setSkin(targetName);
            playerDisguise.setDisplayedInTab(false);
            playerDisguise.setTablistName(origName);
        }
        if (event.getDisguise().isHidePlayer())
        {
            event.getDisguise().setHidePlayer(false);
        }
        if (event.getDisguise().getWatcher() instanceof AreaEffectCloudWatcher watcher)
        {
            if (watcher.getRadius() > 5)
            {
                watcher.setRadius(5);
            }
            else if (watcher.getRadius() < 0)
            {
                watcher.setRadius(0);
            }
        }
        event.getDisguise().getWatcher().setNameYModifier(safeYMod(event.getDisguise().getWatcher().getNameYModifier()));
        event.getDisguise().getWatcher().setYModifier(safeYMod(event.getDisguise().getWatcher().getYModifier()));
        if (event.getDisguise().getWatcher() instanceof SlimeWatcher watcher && watcher.getSize() > 10)
        {
            watcher.setSize(10);
        }
        if (event.getDisguise().getWatcher() instanceof PhantomWatcher watcher)
        {
            if (watcher.getSize() > 20)
            {
                watcher.setSize(20);
            }
            else if (watcher.getSize() < -36)
            {
                watcher.setSize(-36);
            }
        }
        event.setCancelled(false);
    }

    final List<Command> commands = new ArrayList<>();

    @EventHandler
    public void onDisguiseToggle(UndisguiseEvent event)
    {
        for (Player player : Bukkit.getOnlinePlayers())
        {
            if (event.isUndisguiseAdmins())
            {
                DisguiseAPI.undisguiseToAll(player);
            }
            else
            {
                PlexPlayer plexPlayer = DataUtils.getPlayer(player.getUniqueId());
                if (!player.hasPermission("plex.libsdisguises.bypass"))
                {
                    DisguiseAPI.undisguiseToAll(player);
                }
            }
        }
    }

    @EventHandler(priority = EventPriority.HIGHEST)
    public void onPlayerCommandPreprocess(PlayerCommandPreprocessEvent event)
    {
        String message = event.getMessage();
        // Don't check the arguments
        message = message.replaceAll("\\s.*", "").replaceFirst("/", "");
        if (!LibsDisguises.enabled)
        {
            for (Command command : commands)
            {
                for (String commandAliases : command.getAliases())
                {
                    if (message.equalsIgnoreCase(command.getName()) || message.equalsIgnoreCase(commandAliases))
                    {
                        event.getPlayer().sendMessage(Component.text("LibsDisguises is currently disabled.").color(NamedTextColor.RED));
                        event.setCancelled(true);
                        break;
                    }
                }
            }
        }
    }

    public List<Command> getCommands()
    {
        for (Plugin plugin : Bukkit.getPluginManager().getPlugins())
        {
            if (plugin.getName().equals("LibsDisguises"))
            {
                List<Command> commandList = PluginCommandYamlParser.parse(plugin);
                commands.addAll(commandList);
            }
        }
        PlexLog.log("Successfully fetched all LibsDisguises commands!");
        return commands;
    }
}