From 743072d8e31bc946118055f3a4edc30c84cbe9ad Mon Sep 17 00:00:00 2001
From: ChomeNS <95471003+ChomeNS@users.noreply.github.com>
Date: Sat, 26 Oct 2024 18:11:28 +0700
Subject: [PATCH] FindAltsCommand

---
 .../chomens_bot/commands/FindAltsCommand.java | 88 +++++++++++++++++++
 .../plugins/CommandHandlerPlugin.java         |  1 +
 2 files changed, 89 insertions(+)
 create mode 100644 src/main/java/me/chayapak1/chomens_bot/commands/FindAltsCommand.java

diff --git a/src/main/java/me/chayapak1/chomens_bot/commands/FindAltsCommand.java b/src/main/java/me/chayapak1/chomens_bot/commands/FindAltsCommand.java
new file mode 100644
index 0000000..d53b815
--- /dev/null
+++ b/src/main/java/me/chayapak1/chomens_bot/commands/FindAltsCommand.java
@@ -0,0 +1,88 @@
+package me.chayapak1.chomens_bot.commands;
+
+import com.google.gson.JsonElement;
+import com.google.gson.JsonObject;
+import me.chayapak1.chomens_bot.Bot;
+import me.chayapak1.chomens_bot.command.Command;
+import me.chayapak1.chomens_bot.command.CommandContext;
+import me.chayapak1.chomens_bot.command.CommandException;
+import me.chayapak1.chomens_bot.command.TrustLevel;
+import me.chayapak1.chomens_bot.data.PlayerEntry;
+import me.chayapak1.chomens_bot.plugins.PlayersPersistentDataPlugin;
+import me.chayapak1.chomens_bot.util.ColorUtilities;
+import net.kyori.adventure.text.Component;
+
+import java.util.Map;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Stream;
+
+public class FindAltsCommand extends Command {
+    public FindAltsCommand() {
+        super(
+                "findalts",
+                "Finds players with the same IP address",
+                new String[] { "<player>", "<ip>" },
+                new String[] { "alts", "sameip" },
+                TrustLevel.PUBLIC,
+                false
+        );
+    }
+
+    @Override
+    public Component execute(CommandContext context) throws Exception {
+        final Bot bot = context.bot;
+
+        String targetIP;
+
+        final String player = context.getString(true, true);
+
+        final PlayerEntry playerEntry = bot.players.getEntry(player);
+
+        if (playerEntry == null) targetIP = player;
+        else {
+            final CompletableFuture<String> future = bot.players.getPlayerIP(playerEntry);
+
+            targetIP = future.get(5, TimeUnit.SECONDS);
+        }
+
+        if (targetIP == null) throw new CommandException(Component.text("Could not find player's IP address"));
+
+        final Stream<String> matches = PlayersPersistentDataPlugin.playersObject.deepCopy().entrySet() // is calling deepCopy necessary?
+                .stream()
+                .filter(
+                        entry -> {
+                            final JsonObject ipsObject = entry
+                                    .getValue().getAsJsonObject().getAsJsonObject("ips");
+
+                            if (ipsObject == null) return false;
+
+                            final JsonElement currentServerIP = ipsObject.get(bot.host + ":" + bot.port);
+
+                            if (currentServerIP == null) return false;
+
+                            return currentServerIP.getAsString().equals(targetIP);
+                        }
+                )
+                .map(Map.Entry::getKey);
+
+        Component component = Component
+                .translatable("Possible alts for the player %s:")
+                .color(ColorUtilities.getColorByString(bot.config.colorPalette.defaultColor))
+                .arguments(Component.text(player).color(ColorUtilities.getColorByString(bot.config.colorPalette.username)))
+                .appendNewline();
+
+        int i = 0;
+        for (String name : matches.toList()) {
+            component = component
+                    .append(
+                            Component
+                                    .text(name)
+                                    .color((i++ & 1) == 0 ? ColorUtilities.getColorByString(bot.config.colorPalette.primary) : ColorUtilities.getColorByString(bot.config.colorPalette.secondary))
+                    )
+                    .appendSpace();
+        }
+
+        return component;
+    }
+}
diff --git a/src/main/java/me/chayapak1/chomens_bot/plugins/CommandHandlerPlugin.java b/src/main/java/me/chayapak1/chomens_bot/plugins/CommandHandlerPlugin.java
index c7074e3..43991a9 100644
--- a/src/main/java/me/chayapak1/chomens_bot/plugins/CommandHandlerPlugin.java
+++ b/src/main/java/me/chayapak1/chomens_bot/plugins/CommandHandlerPlugin.java
@@ -58,6 +58,7 @@ public class CommandHandlerPlugin {
         registerCommand(new IPFilterCommand());
         registerCommand(new StopCommand());
         registerCommand(new GrepLogCommand());
+        registerCommand(new FindAltsCommand());
     }
 
     public boolean disabled = false;