Add a uuid command

Author: sleepy-evelyn

src/main/java/dev/ithundxr/railwaystweaks/commands/RailwaysTweaksCommands.java

@@ -1,12 +1,14 @@
 package dev.ithundxr.railwaystweaks.commands;
 
 import com.mojang.brigadier.Command;
+import com.mojang.brigadier.arguments.StringArgumentType;
 import com.mojang.brigadier.builder.ArgumentBuilder;
 import com.mojang.brigadier.context.CommandContext;
 import com.simibubi.create.Create;
 import com.simibubi.create.content.trains.entity.CarriageContraptionEntity;
 import dev.ithundxr.railwaystweaks.RailwaysTweaks;
 import dev.ithundxr.railwaystweaks.mixin.compat.tconstruct.SimpleChannelAccessor;
+import dev.ithundxr.railwaystweaks.utils.UUIDFinder;
 import me.pepperbell.simplenetworking.C2SPacket;
 import me.pepperbell.simplenetworking.S2CPacket;
 import net.fabricmc.fabric.api.command.v2.CommandRegistrationCallback;
@@ -40,10 +42,18 @@ public static void init() {
                     .executes(ctx -> avgMSPT(ctx.getSource())));
         });
 
+        CommandRegistrationCallback.EVENT.register((dispatcher, registryAccess, environment) -> {
+            dispatcher.register(literal("uuid")
+                    .then(Commands.argument("player_name", StringArgumentType.string())
+                            .executes(RailwaysTweaksCommands::getPlayerUUID)
+                    )
+            );
+        });
+
         CommandRegistrationCallback.EVENT.register((dispatcher, registryAccess, environment) -> {
             dispatcher.register(literal("railwaystweaks")
                     .then(
-                        literal("opac-party")
+                            literal("opac-party")
                                 .then(Commands.argument("player_uuid", UuidArgument.uuid())
                                         .executes(RailwaysTweaksCommands::getPlayerPartyName)
                                 )
@@ -132,11 +142,23 @@ private static int getPlayerPartyName(CommandContext<CommandSourceStack> ctx) {
         IServerPartyAPI partyAPI = api.getPartyManager().getPartyByMember(uuid);
 
         if (partyAPI != null) {
-            ctx.getSource().sendSuccess(() -> Component.literal(partyAPI.getDefaultName() + "\n" + partyAPI.getId()), true);
+            ctx.getSource().sendSuccess(() -> Component.literal(partyAPI.getDefaultName() + "\n" + partyAPI.getId()), false);
             return 0;
         } else {
             ctx.getSource().sendFailure(Component.literal("Failed to get a party uuid from this player"));
             return 1;
         }
     }
+
+    private static int getPlayerUUID(CommandContext<CommandSourceStack> ctx) {
+        String name = StringArgumentType.getString(ctx, "player_name");
+
+        if (name != null) {
+            UUIDFinder.findUuid(name, (uuid) -> {
+                if (uuid != null)
+                    ctx.getSource().sendSuccess(() -> Component.literal(uuid.toString()), false);
+            });
+        }
+        return 0;
+    }
 }

src/main/java/dev/ithundxr/railwaystweaks/utils/UUIDFinder.java

@@ -0,0 +1,61 @@
+package dev.ithundxr.railwaystweaks.utils;
+
+import com.google.gson.JsonObject;
+import com.google.gson.JsonParser;
+
+import java.io.IOException;
+import java.net.HttpURLConnection;
+import java.net.URL;
+import java.util.Scanner;
+import java.util.UUID;
+import java.util.concurrent.CompletableFuture;
+import java.util.function.Consumer;
+
+public class UUIDFinder {
+    private static final String MOJANG_UUID_API = "https://api.mojang.com/users/profiles/minecraft/";
+    private static final int TIMEOUT_IN_SECS = 3;
+
+    private static CompletableFuture<UUID> uuidFuture;
+
+    /**
+     * Takes a users name and finds a matching UUID
+     */
+    public static void findUuid(String playerName, Consumer<UUID> finish) {
+        // Asynchronously search for a users UUID
+        uuidFuture = CompletableFuture.supplyAsync(() -> UUIDFinder.getUuidFromPlayer(playerName));
+        uuidFuture.thenAccept(finish);
+    }
+
+    private static UUID getUuidFromPlayer(String playerName) {
+        UUID uuid;
+
+        try {
+            URL uuidGetRequest = new URL(MOJANG_UUID_API + playerName);
+            var httpURLConnection = (HttpURLConnection) uuidGetRequest.openConnection();
+            httpURLConnection.setRequestMethod("GET");
+
+            int responseCode = httpURLConnection.getResponseCode();
+            if (responseCode != 200)
+                uuidFuture.completeExceptionally(new IOException("Failed to get a valid UUID for the player"));
+
+            var stringBuilder = new StringBuilder();
+            var scanner = new Scanner(uuidGetRequest.openStream());
+            while (scanner.hasNext())
+                stringBuilder.append(scanner.nextLine());
+            scanner.close();
+
+            var uuidObject = (JsonObject) JsonParser.parseString(stringBuilder.toString());
+            String id = uuidObject.get("id").getAsString();
+
+            // Format into a proper uuid (accepted) uuid format complete with '-' characters
+            String idFormatted = id.substring(0, 8) + "-" + id.substring(8, 12) + "-"
+                    + id.substring(12, 16) + "-" + id.substring(16, 20) + "-" + id.substring(20, 32);
+
+            uuid = UUID.fromString(idFormatted);
+
+        } catch (IOException e) {
+            uuid = null;
+        }
+        return uuid;
+    }
+}