From 1d7cf6a3752c6dd4f329b224cf9e05d12dfd1e0d Mon Sep 17 00:00:00 2001
From: BlayTheNinth <1933180+BlayTheNinth@users.noreply.github.com>
Date: Wed, 28 Feb 2024 20:41:02 +0100
Subject: [PATCH] feat: Add setBurnTime hook

---
 .../java/net/blay09/mods/balm/fabric/FabricBalmHooks.java    | 4 ++--
 .../main/java/net/blay09/mods/balm/forge/ForgeBalmHooks.java | 5 ++---
 shared/src/main/java/net/blay09/mods/balm/api/BalmHooks.java | 3 ++-
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmHooks.java b/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmHooks.java
index 6e2ea555..ad276177 100644
--- a/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmHooks.java
+++ b/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmHooks.java
@@ -104,8 +104,8 @@ public int getBurnTime(ItemStack itemStack) {
     }
 
     @Override
-    public void setBurnTime(ItemStack itemStack, int burnTime) {
-        FuelRegistry.INSTANCE.add(itemStack.getItem(), burnTime);
+    public void setBurnTime(Item item, int burnTime) {
+        FuelRegistry.INSTANCE.add(item, burnTime);
     }
 
     @Override
diff --git a/forge/src/main/java/net/blay09/mods/balm/forge/ForgeBalmHooks.java b/forge/src/main/java/net/blay09/mods/balm/forge/ForgeBalmHooks.java
index d4f2791b..d7d989a6 100644
--- a/forge/src/main/java/net/blay09/mods/balm/forge/ForgeBalmHooks.java
+++ b/forge/src/main/java/net/blay09/mods/balm/forge/ForgeBalmHooks.java
@@ -23,7 +23,6 @@
 import net.minecraft.world.level.levelgen.feature.ConfiguredFeature;
 import net.minecraft.world.phys.BlockHitResult;
 import net.minecraftforge.common.ForgeHooks;
-import net.minecraftforge.common.ForgeMod;
 import net.minecraftforge.common.MinecraftForge;
 import net.minecraftforge.common.ToolActions;
 import net.minecraftforge.common.util.FakePlayer;
@@ -117,8 +116,8 @@ public int getBurnTime(ItemStack itemStack) {
     }
 
     @Override
-    public void setBurnTime(ItemStack itemStack, int burnTime) {
-        burnTimes.put(itemStack.getItem(), burnTime);
+    public void setBurnTime(Item item, int burnTime) {
+        burnTimes.put(item, burnTime);
     }
 
     @Override
diff --git a/shared/src/main/java/net/blay09/mods/balm/api/BalmHooks.java b/shared/src/main/java/net/blay09/mods/balm/api/BalmHooks.java
index 289e95ea..92d391e5 100644
--- a/shared/src/main/java/net/blay09/mods/balm/api/BalmHooks.java
+++ b/shared/src/main/java/net/blay09/mods/balm/api/BalmHooks.java
@@ -12,6 +12,7 @@
 import net.minecraft.world.entity.Pose;
 import net.minecraft.world.entity.player.Player;
 import net.minecraft.world.item.DyeColor;
+import net.minecraft.world.item.Item;
 import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.level.Level;
 import net.minecraft.world.level.block.state.BlockState;
@@ -56,7 +57,7 @@ default CompoundTag getPersistentData(Player player) {
 
     int getBurnTime(ItemStack itemStack);
 
-    void setBurnTime(ItemStack itemStack, int burnTime);
+    void setBurnTime(Item item, int burnTime);
 
     void firePlayerCraftingEvent(Player player, ItemStack crafted, Container craftMatrix);