Fix link command code argument

GeyserMC · Jan 17, 2025 · 4cf344a · 4cf344a
1 parent 31e40a9
commit 4cf344a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/main/java/org/geysermc/globallinkserver/GlobalLinkServer.java b/src/main/java/org/geysermc/globallinkserver/GlobalLinkServer.java
@@ -11,6 +11,7 @@
 import java.util.logging.Logger;
 
 import com.destroystokyo.paper.event.server.PaperServerListPingEvent;
+import com.mojang.brigadier.arguments.IntegerArgumentType;
 import io.papermc.paper.command.brigadier.Commands;
 import io.papermc.paper.command.brigadier.argument.ArgumentTypes;
 import io.papermc.paper.event.player.AsyncChatEvent;
@@ -82,7 +83,7 @@ public void onEnable() {
                     Commands.literal("link")
                             .requires(ctx -> ctx.getSender() instanceof Player)
                             .executes(commandUtils::startLink)
-                            .then(Commands.argument("code", ArgumentTypes.integerRange())
+                            .then(Commands.argument("code", IntegerArgumentType.integer())
                                     .executes(commandUtils::linkWithCode)
                             )
                             .build(),

diff --git a/src/main/java/org/geysermc/globallinkserver/util/CommandUtils.java b/src/main/java/org/geysermc/globallinkserver/util/CommandUtils.java
@@ -5,6 +5,7 @@
  */
 package org.geysermc.globallinkserver.util;
 
+import com.mojang.brigadier.arguments.IntegerArgumentType;
 import com.mojang.brigadier.context.CommandContext;
 import io.papermc.paper.command.brigadier.CommandSourceStack;
 import net.kyori.adventure.text.Component;
@@ -47,7 +48,7 @@ public int startLink(CommandContext<CommandSourceStack> ctx) {
     }
 
     public int linkWithCode(CommandContext<CommandSourceStack> ctx) {
-        int linkId = ctx.getArgument("code", Integer.class);
+        int linkId = IntegerArgumentType.getInteger(ctx, "code");
 
         Player player = getPlayer(ctx);