feat:AxeMineable tag

Author: Addi3

src/main/java/dev/amble/lib/datagen/tag/AmbleBlockTagProvider.java

@@ -3,6 +3,7 @@
 import java.util.*;
 import java.util.concurrent.CompletableFuture;
 
+import dev.amble.lib.datagen.util.AxeMineable;
 import net.fabricmc.fabric.api.datagen.v1.FabricDataOutput;
 import net.fabricmc.fabric.api.datagen.v1.provider.FabricTagProvider;
 
@@ -38,8 +39,23 @@ protected void configure(RegistryWrapper.WrapperLookup wrapperLookup) {
                 }
             }
         });
+        this.blockClass.forEach(clazz -> {
+            FabricTagBuilder pickaxeBuilder = getOrCreateTagBuilder(BlockTags.AXE_MINEABLE);
+            HashMap<Block, Optional<AxeMineable>> axeBlocks = ReflectionUtil.getAnnotatedValues(clazz, Block.class, AxeMineable.class, false);
+
+            for (Block block : axeBlocks.keySet()) {
+                pickaxeBuilder.add(block);
+                AxeMineable annotation = axeBlocks.get(block).orElseThrow();
+
+                if (annotation.tool() != AxeMineable.Tool.NONE) {
+                    getOrCreateTagBuilder(annotation.tool().tag).add(block);
+                }
+            }
+        });
     }
 
+
+
     public AmbleBlockTagProvider withBlocks(Class<? extends BlockContainer>... blockClass) {
         // add all to queue
         this.blockClass.addAll(Arrays.asList(blockClass));

src/main/java/dev/amble/lib/datagen/util/AxeMineable.java

@@ -0,0 +1,29 @@
+package dev.amble.lib.datagen.util;
+
+import net.minecraft.block.Block;
+import net.minecraft.registry.tag.BlockTags;
+import net.minecraft.registry.tag.TagKey;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Retention(RetentionPolicy.RUNTIME)
+@Target(ElementType.FIELD)
+public @interface AxeMineable {
+    Tool tool() default Tool.NONE;
+
+    enum Tool {
+        NONE(null),
+        STONE(BlockTags.NEEDS_STONE_TOOL),
+        IRON(BlockTags.NEEDS_IRON_TOOL),
+        DIAMOND(BlockTags.NEEDS_DIAMOND_TOOL),;
+
+        public final TagKey<Block> tag;
+
+        Tool(TagKey<Block> tag) {
+            this.tag = tag;
+        }
+    }
+}