From c5fbd67510a8e274a0d52963f59100a04b143a93 Mon Sep 17 00:00:00 2001
From: ChomeNS <95471003+chomens@users.noreply.github.com>
Date: Fri, 21 Mar 2025 17:51:51 +0700
Subject: [PATCH] refactor: make `.username set <username>` only a string
 instead of a greedy string so we can set our username to an empty string for
 the funnies

---
 .../chipmunkmod/commands/UsernameCommand.java         | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/src/main/java/land/chipmunk/chipmunkmod/commands/UsernameCommand.java b/src/main/java/land/chipmunk/chipmunkmod/commands/UsernameCommand.java
index eb96209..7860032 100644
--- a/src/main/java/land/chipmunk/chipmunkmod/commands/UsernameCommand.java
+++ b/src/main/java/land/chipmunk/chipmunkmod/commands/UsernameCommand.java
@@ -4,8 +4,7 @@ import com.mojang.brigadier.Command;
 import com.mojang.brigadier.CommandDispatcher;
 import com.mojang.brigadier.context.CommandContext;
 
-import static com.mojang.brigadier.arguments.StringArgumentType.greedyString;
-import static com.mojang.brigadier.arguments.StringArgumentType.getString;
+import static com.mojang.brigadier.arguments.StringArgumentType.*;
 import static land.chipmunk.chipmunkmod.command.CommandManager.literal;
 import static land.chipmunk.chipmunkmod.command.CommandManager.argument;
 
@@ -35,8 +34,8 @@ public class UsernameCommand {
                         .then(
                                 literal("set")
                                         .then(
-                                                argument("username", greedyString())
-                                                        .executes(c -> updateUsername(c))
+                                                argument("username", string())
+                                                        .executes(UsernameCommand::updateUsername)
                                         )
                         )
                         .then(
@@ -53,7 +52,7 @@ public class UsernameCommand {
         return updateSession(context, session);
     }
 
-    public static int updateSession(CommandContext<FabricClientCommandSource> context, Session session) throws CommandSyntaxException {
+    public static int updateSession(CommandContext<FabricClientCommandSource> context, Session session) {
         final FabricClientCommandSource source = context.getSource();
 
         final MinecraftClient client = source.getClient();
@@ -62,7 +61,7 @@ public class UsernameCommand {
 
         // TODO: Put this in a separate class
         final ServerInfo info = client.getCurrentServerEntry();
-        client.world.disconnect();
+        if (client.world != null) client.world.disconnect();
         client.disconnect();
         ConnectScreen.connect(new TitleScreen(), client, ServerAddress.parse(info.address), info, false, null);