1 files changed, 27 insertions, 0 deletions
diff --git a/src/main/java/moe/nea/firmament/mixins/MixinItemModels.java b/src/main/java/moe/nea/firmament/mixins/MixinItemModels.java
new file mode 100644
index 0000000..2c2cc31
--- /dev/null
+++ b/src/main/java/moe/nea/firmament/mixins/MixinItemModels.java
@@ -0,0 +1,27 @@
+package moe.nea.firmament.mixins;
+
+import moe.nea.firmament.events.CustomItemModelEvent;
+import net.minecraft.client.render.item.ItemModels;
+import net.minecraft.client.render.model.BakedModel;
+import net.minecraft.client.render.model.BakedModelManager;
+import net.minecraft.item.ItemStack;
+import org.spongepowered.asm.mixin.Final;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
+
+@Mixin(ItemModels.class)
+public class MixinItemModels {
+    @Shadow
+    @Final
+    private BakedModelManager modelManager;
+
+    @Inject(method = "getModel(Lnet/minecraft/item/ItemStack;)Lnet/minecraft/client/render/model/BakedModel;", at = @At("HEAD"), cancellable = true)
+    public void onGetModel(ItemStack stack, CallbackInfoReturnable<BakedModel> cir) {
+        var model = CustomItemModelEvent.getModel(stack, modelManager);
+        if (model != null)
+            cir.setReturnValue(model);
+    }
+}