fix problems with paper mappings

bukkit/src/main/java/revxrsal/commands/bukkit/brigadier/MinecraftArgumentType.java

@@ -31,6 +31,7 @@
 
 import java.lang.reflect.Constructor;
 import java.util.Arrays;
+import java.util.List;
 import java.util.Optional;
 import java.util.stream.Collectors;
 
@@ -49,95 +50,95 @@ public enum MinecraftArgumentType {
      * - boolean single
      * - boolean playerOnly
      */
-    ENTITY("ArgumentEntity", boolean.class, boolean.class),
+    ENTITY(new String[]{"ArgumentEntity", "EntityArgument"}, boolean.class, boolean.class),
 
     /**
      * A player, online or not. Can also use a selector, which may match one or more
      * players (but not entities).
      */
-    GAME_PROFILE("ArgumentProfile"),
+    GAME_PROFILE("ArgumentProfile", "GameProfileArgument"),
 
     /**
      * A chat color. One of the names from <a href="https://wiki.vg/Chat#Colors">colors</a>, or {@code reset}.
      * Case-insensitive.
      */
-    COLOR("ArgumentChatFormat"),
+    COLOR("ArgumentChatFormat", "ColorArgument"),
 
     /**
      * A JSON Chat component.
      */
-    COMPONENT("ArgumentChatComponent"),
+    COMPONENT("ArgumentChatComponent", "ComponentArgument"),
 
     /**
      * A regular message, potentially including selectors.
      */
-    MESSAGE("ArgumentChat"),
+    MESSAGE("ArgumentChat", "MessageArgument"),
 
     /**
      * An NBT value, parsed using JSON-NBT rules. This represents a full NBT tag.
      */
-    NBT("ArgumentNBTTag"),
+    NBT("ArgumentNBTTag", "CompoundTagArgument"),
 
     /**
      * Represents a partial NBT tag, usable in data modify command.
      */
-    NBT_TAG("ArgumentNBTBase"),
+    NBT_TAG("ArgumentNBTBase", "NbtTagArgument"),
 
     /**
      * A path within an NBT value, allowing for array and member accesses.
      */
-    NBT_PATH("ArgumentNBTKey"),
+    NBT_PATH("ArgumentNBTKey", "NbtPathArgument"),
 
     /**
      * A scoreboard objective.
      */
-    SCOREBOARD_OBJECTIVE("ArgumentScoreboardObjective"),
+    SCOREBOARD_OBJECTIVE("ArgumentScoreboardObjective", "ObjectiveArgument"),
 
     /**
      * A single score criterion.
      */
-    OBJECTIVE_CRITERIA("ArgumentScoreboardCriteria"),
+    OBJECTIVE_CRITERIA("ArgumentScoreboardCriteria", "ObjectiveCriteriaArgument"),
 
     /**
      * A scoreboard operator.
      */
-    SCOREBOARD_SLOT("ArgumentScoreboardSlot"),
+    SCOREBOARD_SLOT("ArgumentScoreboardSlot", "SlotArgument"),
 
     /**
      * Something that can join a team. Allows selectors and *.
      */
-    SCORE_HOLDER("ArgumentScoreholder"),
+    SCORE_HOLDER("ArgumentScoreholder", "ScoreHolderArgument"),
 
     /**
      * The name of a team. Parsed as an unquoted string.
      */
-    TEAM("ArgumentScoreboardTeam"),
+    TEAM("ArgumentScoreboardTeam", "TeamArgument"),
 
     /**
      * A scoreboard operator.
      */
-    OPERATION("ArgumentMathOperation"),
+    OPERATION("ArgumentMathOperation", "OperationArgument"),
 
     /**
      * A particle effect (an identifier with extra information following it for
      * specific particles, mirroring the Particle packet)
      */
-    PARTICLE("ArgumentParticle"),
+    PARTICLE("ArgumentParticle", "ParticleArgument"),
 
     /**
      * Represents an angle.
      */
-    ANGLE("ArgumentAngle"),
+    ANGLE("ArgumentAngle", "AngleArgument"),
 
     /**
      * A name for an inventory slot.
      */
-    ITEM_SLOT("ArgumentInventorySlot"),
+    ITEM_SLOT("ArgumentInventorySlot", "SlotArgument"),
 
     /**
      * An Identifier.
      */
-    RESOURCE_LOCATION("ArgumentMinecraftKeyRegistered"),
+    RESOURCE_LOCATION("ArgumentMinecraftKeyRegistered", "ResourceLocationArgument"),
 
     /**
      * A potion effect.
@@ -157,95 +158,95 @@ public enum MinecraftArgumentType {
     /**
      * Represents a dimension.
      */
-    DIMENSION("ArgumentDimension"),
+    DIMENSION("ArgumentDimension", "DimensionArgument"),
 
     /**
      * Represents a time duration.
      */
-    TIME("ArgumentTime"),
+    TIME("ArgumentTime", "TimeArgument"),
 
     /**
      * Represents a UUID value.
      *
      * @since Minecraft 1.16
      */
-    UUID("ArgumentUUID"),
+    UUID("ArgumentUUID", "UuidArgument"),
 
     /**
      * A location, represented as 3 numbers (which must be integers). May use relative locations
      * with ~
      */
-    BLOCK_POS("coordinates.ArgumentPosition"),
+    BLOCK_POS("coordinates.ArgumentPosition", "coordinates.BlockPosArgument"),
 
     /**
-     * A column location, represented as 3 numbers (which must be integers). May use relative locations
+     * A column location, represented as 2 numbers (which must be integers). May use relative locations
      * with ~.
      */
-    COLUMN_POS("coordinates.ArgumentVec2I"),
+    COLUMN_POS("coordinates.ArgumentVec2I", "coordinates.ColumnPosArgument"),
 
     /**
      * A location, represented as 3 numbers (which may have a decimal point, but will be moved to the
      * center of a block if none is specified). May use relative locations with ~.
      */
-    VECTOR_3("coordinates.ArgumentVec3"),
+    VECTOR_3("coordinates.ArgumentVec3", "coordinates.Vec3Argument"),
 
     /**
      * A location, represented as 2 numbers (which may have a decimal point, but will be moved to the center
      * of a block if none is specified). May use relative locations with ~.
      */
-    VECTOR_2("coordinates.ArgumentVec2"),
+    VECTOR_2("coordinates.ArgumentVec2", "coordinates.Vec2Argument"),
 
     /**
      * An angle, represented as 2 numbers (which may have a decimal point, but will be moved to the
      * center of a block if none is specified). May use relative locations with ~.
      */
-    ROTATION("coordinates.ArgumentRotation"),
+    ROTATION("coordinates.ArgumentRotation", "coordinates.RotationArgument"),
 
     /**
      * A collection of up to 3 axes.
      */
-    SWIZZLE("coordinates.ArgumentRotationAxis"),
+    SWIZZLE("coordinates.ArgumentRotationAxis", "coordinates.SwizzleArgument"),
 
     /**
      * A block state, optionally including NBT and state information.
      */
-    BLOCK_STATE("blocks.ArgumentTile"),
+    BLOCK_STATE("blocks.ArgumentTile", "blocks.BlockStateArgument"),
 
     /**
      * A block, or a block tag.
      */
-    BLOCK_PREDICATE("blocks.ArgumentBlockPredicate"),
+    BLOCK_PREDICATE("blocks.ArgumentBlockPredicate", "blocks.BlockPredicateArgument"),
 
     /**
      * An item, optionally including NBT.
      */
-    ITEM_STACK("item.ArgumentItemStack"),
+    ITEM_STACK("item.ArgumentItemStack", "item.ItemArgument"),
 
     /**
      * An item, or an item tag.
      */
-    ITEM_PREDICATE("item.ArgumentItemPredicate"),
+    ITEM_PREDICATE("item.ArgumentItemPredicate", "item.ItemPredicateArgument"),
 
     /**
      * A function.
      */
-    FUNCTION("item.ArgumentTag"),
+    FUNCTION("item.ArgumentTag", "item.FunctionArgument"),
 
     /**
      * The entity anchor related to the facing argument in the teleport command,
      * is feet or eyes.
      */
-    ENTITY_ANCHOR("ArgumentAnchor"),
+    ENTITY_ANCHOR("ArgumentAnchor", "EntityAnchorArgument"),
 
     /**
      * An integer range of values with a min and a max.
      */
-    INT_RANGE("ArgumentCriterionValue$b"),
+    INT_RANGE("ArgumentCriterionValue$b", "RangeArgument$Ints"),
 
     /**
      * A floating-point range of values with a min and a max.
      */
-    FLOAT_RANGE("ArgumentCriterionValue$a"),
+    FLOAT_RANGE("ArgumentCriterionValue$a", "RangeArgument$Floats"),
 
     /**
      * Template mirror
@@ -265,8 +266,17 @@ public enum MinecraftArgumentType {
     private @Nullable ArgumentType<?> argumentType;
     private @Nullable Constructor<? extends ArgumentType> argumentConstructor;
 
-    MinecraftArgumentType(String name, Class<?>... parameters) {
-        Class<?> argumentClass = resolveArgumentClass(name);
+    MinecraftArgumentType(String... names) {
+        this(names, new Class[0]);
+    }
+
+    MinecraftArgumentType(String[] names, Class<?>... parameters) {
+        Class<?> argumentClass = null;
+        for (String name : names) {
+            argumentClass = resolveArgumentClass(name);
+            if (argumentClass != null)
+                break;
+        }
         this.parameters = parameters;
         if (argumentClass == null) {
             argumentType = null;
@@ -288,25 +298,32 @@ public enum MinecraftArgumentType {
         }
     }
 
+    static class Data {
+        private static final List<String> POSSIBLE_CLASS_NAMES = Arrays.asList(
+                "net.minecraft.server.{name}",
+                "net.minecraft.server.{version}.{name}",
+                "net.minecraft.commands.arguments.{name}",
+                "net.minecraft.server.{version}.{stripped_name}"
+        );
+    }
+
     private static @Nullable Class<?> resolveArgumentClass(String name) {
-        try {
-            if (BukkitVersion.supports(1, 17)) {
-                try {
-                    return BukkitVersion.findNmsClass("commands.arguments." + name);
-                } catch (Throwable ignored) {
-                    return Class.forName("net.minecraft.commands.arguments." + name);
-                }
-            } else {
-                String stripped;
-                if (name.lastIndexOf('.') != -1)
-                    stripped = name.substring(name.lastIndexOf('.') + 1);
-                else
-                    stripped = name;
-                return BukkitVersion.findNmsClass(stripped);
+        String strippedName;
+        if (name.lastIndexOf('.') != -1)
+            strippedName = name.substring(name.lastIndexOf('.') + 1);
+        else
+            strippedName = name;
+        for (String s : Data.POSSIBLE_CLASS_NAMES) {
+            String className = s
+                    .replace("{version}", BukkitVersion.version())
+                    .replace("{name}", name)
+                    .replace("{stripped_name}", strippedName);
+            try {
+                return Class.forName(className);
+            } catch (ClassNotFoundException ignored) {
             }
-        } catch (Throwable t) {
-            return null;
         }
+        return null;
     }
 
     /**

bukkit/src/main/java/revxrsal/commands/bukkit/util/BukkitVersion.java

@@ -22,7 +22,7 @@ public final class BukkitVersion {
     private static final int MAJOR_VERSION, MINOR_VERSION, PATCH_NUMBER;
 
     /**
-     * The current version string, for example 1_17_R1
+     * The current version string, for example v1_17_R1
      */
     private static final String VERSION = fetchVersion();
 
@@ -120,6 +120,15 @@ public static int patchNumber() {
         return PATCH_NUMBER;
     }
 
+    /**
+     * Returns the current version string, for example v1_17_R1
+     *
+     * @return The current version string
+     */
+    public static @NotNull String version() {
+        return VERSION;
+    }
+
     /**
      * Returns the NMS class with the given name. The name must not contain
      * the net.minecraft.server prefix.