Registry changes & additions

worldedit-bukkit/src/main/java/com/sk89q/worldedit/bukkit/BukkitBlockCategoryRegistry.java

@@ -0,0 +1,52 @@
+/*
+ * WorldEdit, a Minecraft world manipulation toolkit
+ * Copyright (C) sk89q <http://www.sk89q.com>
+ * Copyright (C) WorldEdit team and contributors
+ *
+ * This program is free software: you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as published by the
+ * Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
+ * for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package com.sk89q.worldedit.bukkit;
+
+import com.sk89q.worldedit.registry.Category;
+import com.sk89q.worldedit.world.block.BlockType;
+import com.sk89q.worldedit.world.registry.BlockCategoryRegistry;
+import org.bukkit.Bukkit;
+import org.bukkit.Material;
+import org.bukkit.NamespacedKey;
+import org.bukkit.Tag;
+
+import java.util.Set;
+import java.util.stream.Collectors;
+
+public class BukkitBlockCategoryRegistry implements BlockCategoryRegistry {
+
+    private Set<BlockType> getFromBukkitTag(Tag<Material> tag) {
+        return tag.getValues().stream().map(BukkitAdapter::asBlockType).collect(Collectors.toSet());
+    }
+
+    @Override
+    public Set<BlockType> getCategorisedByName(String category) {
+        String[] split = category.split(":");
+        String namespace = split.length > 1 ? split[0] : "minecraft";
+        String key =  split.length > 1 ? split[1] : category;
+        Tag<Material> tag = Bukkit.getTag(Tag.REGISTRY_BLOCKS, new NamespacedKey(namespace, key), Material.class);
+        return getFromBukkitTag(tag);
+    }
+
+    @Override
+    public Set<BlockType> getAll(Category<BlockType> category) {
+        return getCategorisedByName(category.getId());
+    }
+}

worldedit-bukkit/src/main/java/com/sk89q/worldedit/bukkit/BukkitItemCategoryRegistry.java

@@ -0,0 +1,52 @@
+/*
+ * WorldEdit, a Minecraft world manipulation toolkit
+ * Copyright (C) sk89q <http://www.sk89q.com>
+ * Copyright (C) WorldEdit team and contributors
+ *
+ * This program is free software: you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as published by the
+ * Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
+ * for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package com.sk89q.worldedit.bukkit;
+
+import com.sk89q.worldedit.registry.Category;
+import com.sk89q.worldedit.world.item.ItemType;
+import com.sk89q.worldedit.world.registry.ItemCategoryRegistry;
+import org.bukkit.Bukkit;
+import org.bukkit.Material;
+import org.bukkit.NamespacedKey;
+import org.bukkit.Tag;
+
+import java.util.Set;
+import java.util.stream.Collectors;
+
+public class BukkitItemCategoryRegistry implements ItemCategoryRegistry {
+
+    private Set<ItemType> getFromBukkitTag(Tag<Material> tag) {
+        return tag.getValues().stream().map(BukkitAdapter::asItemType).collect(Collectors.toSet());
+    }
+
+    @Override
+    public Set<ItemType> getCategorisedByName(String category) {
+        String[] split = category.split(":");
+        String namespace = split.length > 1 ? split[0] : "minecraft";
+        String key =  split.length > 1 ? split[1] : category;
+        Tag<Material> tag = Bukkit.getTag(Tag.REGISTRY_ITEMS, new NamespacedKey(namespace, key), Material.class);
+        return getFromBukkitTag(tag);
+    }
+
+    @Override
+    public Set<ItemType> getAll(Category<ItemType> category) {
+        return getCategorisedByName(category.getId());
+    }
+}

worldedit-bukkit/src/main/java/com/sk89q/worldedit/bukkit/BukkitRegistries.java

@@ -31,6 +31,8 @@ class BukkitRegistries extends BundledRegistries {
     private final ItemRegistry itemRegistry = new BukkitItemRegistry();
     private final BiomeRegistry biomeRegistry = new BukkitBiomeRegistry();
     private final EntityRegistry entityRegistry = new BukkitEntityRegistry();
+    private final BlockCategoryRegistry blockCategoryRegistry = new BukkitBlockCategoryRegistry();
+    private final ItemCategoryRegistry itemCategoryRegistry = new BukkitItemCategoryRegistry();
 
     /**
      * Create a new instance.
@@ -42,6 +44,11 @@ class BukkitRegistries extends BundledRegistries {
     public BlockRegistry getBlockRegistry() {
         return blockRegistry;
     }
+    
+    @Override
+    public BlockCategoryRegistry getBlockCategoryRegistry() {
+    	return blockCategoryRegistry;
+    }
 
     @Override
     public BiomeRegistry getBiomeRegistry() {
@@ -52,6 +59,11 @@ class BukkitRegistries extends BundledRegistries {
     public ItemRegistry getItemRegistry() {
         return itemRegistry;
     }
+    
+    @Override
+    public ItemCategoryRegistry getItemCategoryRegistry() {
+    	return itemCategoryRegistry;
+    }
 
     @Override
     public EntityRegistry getEntityRegistry() {