TotalFreedomMod/commons/src/main/java/me/totalfreedom/totalfreedommod/command/Command_nickclean.java

package me.totalfreedom.totalfreedommod.command;

import me.totalfreedom.totalfreedommod.command.handling.CommandParameters;
import me.totalfreedom.totalfreedommod.command.handling.CommandPermissions;
import me.totalfreedom.totalfreedommod.command.handling.FreedomCommand;
import me.totalfreedom.totalfreedommod.command.handling.SourceType;
import me.totalfreedom.totalfreedommod.config.ConfigEntry;
import me.totalfreedom.totalfreedommod.util.FUtil;
import org.apache.commons.lang.StringUtils;
import org.bukkit.ChatColor;
import org.bukkit.Color;
import org.bukkit.command.Command;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Player;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

@CommandPermissions(permission = "nickclean", source = SourceType.BOTH)
@CommandParameters(description = "Essentials Interface Command - Remove illegal chatcodes from nicknames of one or all players on server.", usage = "/<command> [player]", aliases = "nc")
public class Command_nickclean extends FreedomCommand
{
    private static final Map<String, Color> colorCodes = populateHashMap();

    private static Map<String, Color> populateHashMap()
    {
        Map<String, Color> colorCodes = new HashMap<>();
        colorCodes.put("&0", Color.BLACK);
        colorCodes.put("&1", Color.BLUE);
        colorCodes.put("&2", Color.GREEN);
        colorCodes.put("&3", Color.TEAL);
        colorCodes.put("&4", Color.MAROON);
        colorCodes.put("&5", Color.FUCHSIA);
        colorCodes.put("&6", Color.OLIVE);
        colorCodes.put("&7", Color.SILVER);
        colorCodes.put("&8", Color.GRAY);
        colorCodes.put("&9", Color.NAVY);
        colorCodes.put("&a", Color.LIME);
        colorCodes.put("&b", Color.AQUA);
        colorCodes.put("&c", Color.RED);
        colorCodes.put("&d", Color.PURPLE);
        colorCodes.put("&e", Color.YELLOW);
        colorCodes.put("&f", Color.WHITE);
        return colorCodes;
    }

    @Override
    public boolean run(CommandSender sender, Player playerSender, Command cmd, String commandLabel, String[] args, boolean senderIsConsole)
    {
        if (args.length > 1)
        {
            Player player = getPlayer(args[0]);

            if (player == null)
            {
                msg(PLAYER_NOT_FOUND);
                return true;
            }

            FUtil.adminAction(sender.getName(), "Cleaning " + player.getName() + "'s nickname", false);
            cleanNickname(player);
            return true;
        }

        FUtil.adminAction(sender.getName(), "Cleaning all nicknames", false);
        for (final Player player : server.getOnlinePlayers())
        {
            cleanNickname(player);
        }
        return true;
    }

    public void cleanNickname(Player player)
    {
        final String playerName = player.getName();
        final String nickName = plugin.esb.getNickname(playerName);
        StringBuilder newNick = new StringBuilder();
        boolean nickChanged = false;
        if (nickName != null)
        {
            if (nickName.contains("§x"))
            {
                // Detects colors that are similar to blocked codes.
                spliterator:
                for (String split : nickName.split("§x"))
                {
                    List<Color> colors = new ArrayList<>();
                    String hexColorSub;
                    if (split.length() >= 12 && split.contains("§"))
                    {
                        hexColorSub = split.substring(0, 12);
                        split = String.valueOf(split.charAt(12));
                        String hexColorString = "#" + hexColorSub.replace("§", "");
                        java.awt.Color hexColor = java.awt.Color.decode(hexColorString);

                        // Get a range of nearby colors that are alike to the color blocked.
                        Color colorFirst;
                        Color colorSecond;

                        colorFirst = Color.fromRGB(Math.min(hexColor.getRed() + 20, 255), Math.min(hexColor.getGreen() + 20, 255), Math.min(hexColor.getBlue() + 20, 255));
                        colorSecond = Color.fromRGB(Math.max(hexColor.getRed() - 20, 0), Math.max(hexColor.getGreen() - 20, 0), Math.max(hexColor.getBlue() - 20, 0));
                        colors.addAll(FUtil.createColorGradient(colorFirst, colorSecond, 40));

                        colorFirst = Color.fromRGB(Math.min(hexColor.getRed() + 20, 255), Math.min(hexColor.getGreen(), 255), Math.min(hexColor.getBlue(), 255));
                        colorSecond = Color.fromRGB(Math.max(hexColor.getRed() - 20, 0), Math.max(hexColor.getGreen(), 0), Math.max(hexColor.getBlue(), 0));
                        colors.addAll(FUtil.createColorGradient(colorFirst, colorSecond, 40));

                        colorFirst = Color.fromRGB(Math.min(hexColor.getRed(), 255), Math.min(hexColor.getGreen() + 20, 255), Math.min(hexColor.getBlue(), 255));
                        colorSecond = Color.fromRGB(Math.max(hexColor.getRed(), 0), Math.max(hexColor.getGreen() - 20, 0), Math.max(hexColor.getBlue(), 0));
                        colors.addAll(FUtil.createColorGradient(colorFirst, colorSecond, 40));

                        colorFirst = Color.fromRGB(Math.min(hexColor.getRed(), 255), Math.min(hexColor.getGreen(), 255), Math.min(hexColor.getBlue() + 20, 255));
                        colorSecond = Color.fromRGB(Math.max(hexColor.getRed(), 0), Math.max(hexColor.getGreen(), 0), Math.max(hexColor.getBlue() - 20, 0));
                        colors.addAll(FUtil.createColorGradient(colorFirst, colorSecond, 40));

                        for (String colorCode : ConfigEntry.BLOCKED_CHATCODES.getString().split(","))
                        {
                            // Makes sure that there's hex colors in the split.
                            for (Color color : colors)
                            {
                                if (colorCodes.get(colorCode) != null && FUtil.colorClose(color, colorCodes.get(colorCode), 40))
                                {
                                    nickChanged = true;
                                    newNick.append(split);
                                    continue spliterator;
                                }
                            }

                        }
                        newNick.append("§x").append(hexColorSub).append(split);
                    }
                }
            } else
            {
                // Falls back on old code if hex isn't used.
                final Pattern REGEX = Pattern.compile(FUtil.colorize(ChatColor.COLOR_CHAR + "[" + StringUtils.join(ConfigEntry.BLOCKED_CHATCODES.getString().split(","), "") + "]"), Pattern.CASE_INSENSITIVE);
                if (!nickName.isEmpty() && !nickName.equalsIgnoreCase(playerName))
                {
                    final Matcher matcher = REGEX.matcher(nickName);
                    if (matcher.find())
                    {
                        nickChanged = true;
                        newNick.append(matcher.replaceAll(""));
                    }
                }
            }
        }

        if (nickChanged)
        {
            msg(ChatColor.RESET + playerName + ": \"" + nickName + ChatColor.RESET + "\" -> \"" + newNick.toString() + ChatColor.RESET + "\".");
        }
        plugin.esb.setNickname(playerName, newNick.toString());
    }
}