created some more argument types

Author: DatAsianBoi123

src/main/java/com/datasiqn/commandcore/argument/type/ArgumentType.java

@@ -5,8 +5,13 @@
 import com.datasiqn.commandcore.command.CommandContext;
 import com.datasiqn.resultapi.None;
 import com.datasiqn.resultapi.Result;
+import org.bukkit.Bukkit;
+import org.bukkit.Location;
 import org.bukkit.Material;
+import org.bukkit.World;
+import org.bukkit.entity.EntityType;
 import org.bukkit.entity.Player;
+import org.bukkit.loot.LootTable;
 import org.bukkit.util.Vector;
 import org.jetbrains.annotations.Contract;
 import org.jetbrains.annotations.NotNull;
@@ -64,6 +69,33 @@ public interface ArgumentType<T> {
      */
     ArgumentType<Vector> VECTOR = new VectorArgumentType();
 
+    /**
+     * {@code ArgumentType} that represents a loaded world
+     */
+    ArgumentType<World> WORLD = new WorldArgumentType();
+
+    /**
+     * {@code ArgumentType} that represents an entity
+     */
+    ArgumentType<EntityType> ENTITY = new EnumArgumentType<>(EntityType.class, "entity");
+
+    /**
+     * {@code ArgumentType} that represents an entity that is living
+     */
+    ArgumentType<EntityType> LIVING_ENTITY = new FilteredEnumArgumentType<>(EntityType.class, EntityType::isAlive, "living entity");
+
+    /**
+     * {@code ArgumentType} that represents an entity that can be spawned using {@link org.bukkit.World#spawnEntity(Location, EntityType)}.
+     * <br>
+     * This is the same as the {@code ENTITY} argument type, except that this omits the {@code Player} entity type
+     */
+    ArgumentType<EntityType> SPAWNABLE_ENTITY = new FilteredEnumArgumentType<>(EntityType.class, entityType -> entityType != EntityType.PLAYER, "living entity");
+
+    /**
+     * {@code ArgumentType} that represents a loot table
+     */
+    ArgumentType<LootTable> LOOT_TABLE = new LootTableArgumentType();
+
     /**
      * {@code ArgumentType} that represents a material
      */

src/main/java/com/datasiqn/commandcore/argument/type/LootTableArgumentType.java

@@ -0,0 +1,31 @@
+package com.datasiqn.commandcore.argument.type;
+
+import com.datasiqn.commandcore.command.CommandContext;
+import com.datasiqn.resultapi.None;
+import com.datasiqn.resultapi.Result;
+import org.bukkit.loot.LootTable;
+import org.bukkit.loot.LootTables;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.Arrays;
+import java.util.List;
+import java.util.stream.Collectors;
+
+class LootTableArgumentType implements SimpleArgumentType<LootTable> {
+    private final List<String> tabCompletes = Arrays.stream(LootTables.values()).map(LootTables::name).collect(Collectors.toList());
+
+    @Override
+    public @NotNull String getTypeName() {
+        return "loot table";
+    }
+
+    @Override
+    public @NotNull Result<LootTable, None> parseWord(String word) {
+        return Result.resolve(() -> LootTables.valueOf(word.toUpperCase())).map(LootTables::getLootTable);
+    }
+
+    @Override
+    public @NotNull List<String> getTabComplete(@NotNull CommandContext context) {
+        return tabCompletes;
+    }
+}

src/main/java/com/datasiqn/commandcore/argument/type/WorldArgumentType.java

@@ -0,0 +1,28 @@
+package com.datasiqn.commandcore.argument.type;
+
+import com.datasiqn.commandcore.command.CommandContext;
+import com.datasiqn.resultapi.None;
+import com.datasiqn.resultapi.Result;
+import org.bukkit.Bukkit;
+import org.bukkit.World;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+class WorldArgumentType implements SimpleArgumentType<World> {
+    @Override
+    public @NotNull String getTypeName() {
+        return "world";
+    }
+
+    @Override
+    public @NotNull Result<World, None> parseWord(String word) {
+        return Result.ofNullable(Bukkit.getWorld(word), None.NONE);
+    }
+
+    @Override
+    public @NotNull List<String> getTabComplete(@NotNull CommandContext context) {
+        return Bukkit.getWorlds().stream().map(World::getName).collect(Collectors.toList());
+    }
+}