Reorganized Imports/Packages

src/main/java/org/betterx/bclib/util/MethodReplace.java

@@ -0,0 +1,46 @@
+package org.betterx.bclib.util;
+
+import net.minecraft.world.item.Item;
+import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.level.block.Block;
+import net.minecraft.world.level.block.state.BlockBehaviour.BlockStateBase;
+
+import java.util.function.Function;
+import org.jetbrains.annotations.Nullable;
+
+public class MethodReplace {
+    private static Function<ItemStack, Boolean> itemReplace;
+    private static Function<BlockStateBase, Boolean> blockReplace;
+    private static Block block;
+    private static Item item;
+
+    public static void addItemReplace(Item item, Function<ItemStack, Boolean> itemReplace) {
+        MethodReplace.itemReplace = itemReplace;
+        MethodReplace.item = item;
+    }
+
+    public static void addBlockReplace(Block block, Function<BlockStateBase, Boolean> blockReplace) {
+        MethodReplace.blockReplace = blockReplace;
+        MethodReplace.block = block;
+    }
+
+    @Nullable
+    public static Function<ItemStack, Boolean> getItemReplace(Item item) {
+        if (MethodReplace.item != item) {
+            return null;
+        }
+        Function<ItemStack, Boolean> replace = itemReplace;
+        itemReplace = null;
+        return replace;
+    }
+
+    @Nullable
+    public static Function<BlockStateBase, Boolean> getBlockReplace(Block block) {
+        if (MethodReplace.block != block) {
+            return null;
+        }
+        Function<BlockStateBase, Boolean> replace = blockReplace;
+        blockReplace = null;
+        return replace;
+    }
+}