1 files changed, 23 insertions, 0 deletions
diff --git a/src/main/java/world/anhgelus/lifesteal/mixin/ItemMixin.java b/src/main/java/world/anhgelus/lifesteal/mixin/ItemMixin.java
new file mode 100644
index 0000000..a405637
--- /dev/null
+++ b/src/main/java/world/anhgelus/lifesteal/mixin/ItemMixin.java
@@ -0,0 +1,23 @@
+package world.anhgelus.lifesteal.mixin;
+
+import net.minecraft.entity.player.PlayerEntity;
+import net.minecraft.item.Item;
+import net.minecraft.item.ItemStack;
+import net.minecraft.item.Items;
+import net.minecraft.server.network.ServerPlayerEntity;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import world.anhgelus.lifesteal.LifeStealer;
+
+@Mixin(Item.class)
+public class ItemMixin {
+    @Inject(at = @At("RETURN"), method = "onCraftByPlayer")
+    public void craft(ItemStack is, PlayerEntity player, CallbackInfo ci) {
+        if (!is.isOf(Items.NETHER_STAR)) return; // assuming that the nether star cannot be crafted
+        final var lifestealer = LifeStealer.Manager.getLifeStealer((ServerPlayerEntity) player);
+        player.getInventory().removeOne(is);
+        player.giveOrDropStack(lifestealer.getHeart());
+    }
+}