WIP: Port to compilation on 1.21.4

25 files changed, 1 insertions, 945 deletions

diff --git a/src/main/java/moe/nea/firmament/mixins/CustomModelEventPatch.java b/src/main/java/moe/nea/firmament/mixins/CustomModelEventPatch.java
deleted file mode 100644
index e0a7544..0000000
--- a/src/main/java/moe/nea/firmament/mixins/CustomModelEventPatch.java
+++ /dev/null
@@ -1,36 +0,0 @@
-
-
-package moe.nea.firmament.mixins;
-
-import moe.nea.firmament.events.CustomItemModelEvent;
-import moe.nea.firmament.features.texturepack.CustomGlobalTextures;
-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.Item;
-import net.minecraft.item.ItemStack;
-import net.minecraft.util.Identifier;
-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;
-
-import java.util.Map;
-
-@Mixin(ItemModels.class)
-public class CustomModelEventPatch {
-
-	@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 $this = (ItemModels) (Object) this;
-		var model = CustomItemModelEvent.getModel(stack, $this);
-		if (model == null) {
-			model = CustomGlobalTextures.replaceGlobalModel($this, stack);
-		}
-		if (model != null) {
-			cir.setReturnValue(model);
-		}
-	}
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/CustomSkullTexturePatch.java b/src/main/java/moe/nea/firmament/mixins/CustomSkullTexturePatch.java
deleted file mode 100644
index f3b616a..0000000
--- a/src/main/java/moe/nea/firmament/mixins/CustomSkullTexturePatch.java
+++ /dev/null
@@ -1,21 +0,0 @@
-
-
-package moe.nea.firmament.mixins;
-
-import moe.nea.firmament.features.texturepack.CustomSkyBlockTextures;
-import net.minecraft.block.SkullBlock;
-import net.minecraft.client.render.RenderLayer;
-import net.minecraft.client.render.block.entity.SkullBlockEntityRenderer;
-import net.minecraft.component.type.ProfileComponent;
-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.CallbackInfoReturnable;
-
-@Mixin(SkullBlockEntityRenderer.class)
-public class CustomSkullTexturePatch {
-	@Inject(method = "getRenderLayer", at = @At("HEAD"), cancellable = true)
-	private static void onGetRenderLayer(SkullBlock.SkullType type, ProfileComponent profile, CallbackInfoReturnable<RenderLayer> cir) {
-		CustomSkyBlockTextures.INSTANCE.modifySkullTexture(type, profile, cir);
-	}
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/IncomingPacketListenerPatches.java b/src/main/java/moe/nea/firmament/mixins/IncomingPacketListenerPatches.java
index 80a9fd5..a7c3875 100644
--- a/src/main/java/moe/nea/firmament/mixins/IncomingPacketListenerPatches.java
+++ b/src/main/java/moe/nea/firmament/mixins/IncomingPacketListenerPatches.java
@@ -18,7 +18,6 @@ import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
 @Mixin(ClientPlayNetworkHandler.class)
 public abstract class IncomingPacketListenerPatches {
 
-
     @ModifyExpressionValue(method = "onCommandTree", at = @At(value = "NEW", target = "(Lcom/mojang/brigadier/tree/RootCommandNode;)Lcom/mojang/brigadier/CommandDispatcher;", remap = false))
     public CommandDispatcher onOnCommandTree(CommandDispatcher dispatcher) {
         MaskCommands.Companion.publish(new MaskCommands(dispatcher));
@@ -31,7 +30,7 @@ public abstract class IncomingPacketListenerPatches {
             packet.getParameters(),
             new Vec3d(packet.getX(), packet.getY(), packet.getZ()),
             new Vector3f(packet.getOffsetX(), packet.getOffsetY(), packet.getOffsetZ()),
-            packet.isLongDistance(),
+            packet.isImportant(),
             packet.getCount(),
             packet.getSpeed()
         );
diff --git a/src/main/java/moe/nea/firmament/mixins/ReplaceTextColorInHandledScreen.java b/src/main/java/moe/nea/firmament/mixins/ReplaceTextColorInHandledScreen.java
deleted file mode 100644
index c9fb073..0000000
--- a/src/main/java/moe/nea/firmament/mixins/ReplaceTextColorInHandledScreen.java
+++ /dev/null
@@ -1,48 +0,0 @@
-package moe.nea.firmament.mixins;
-
-
-import com.llamalad7.mixinextras.injector.wrapoperation.Operation;
-import com.llamalad7.mixinextras.injector.wrapoperation.WrapOperation;
-import moe.nea.firmament.features.texturepack.CustomTextColors;
-import net.minecraft.client.font.TextRenderer;
-import net.minecraft.client.gui.DrawContext;
-import net.minecraft.client.gui.screen.ingame.AnvilScreen;
-import net.minecraft.client.gui.screen.ingame.BeaconScreen;
-import net.minecraft.client.gui.screen.ingame.CreativeInventoryScreen;
-import net.minecraft.client.gui.screen.ingame.HandledScreen;
-import net.minecraft.client.gui.screen.ingame.InventoryScreen;
-import net.minecraft.client.gui.screen.ingame.MerchantScreen;
-import net.minecraft.text.Text;
-import org.spongepowered.asm.mixin.Mixin;
-import org.spongepowered.asm.mixin.injection.At;
-
-@Mixin({HandledScreen.class, InventoryScreen.class, CreativeInventoryScreen.class, MerchantScreen.class,
-	AnvilScreen.class, BeaconScreen.class})
-public class ReplaceTextColorInHandledScreen {
-
-	// To my future self: double check those mixins, but don't be too concerned about errors. Some of the wrapopertions
-	// only apply in some of the specified subclasses.
-
-	@WrapOperation(
-		method = "drawForeground",
-		at = @At(
-			value = "INVOKE",
-			target = "Lnet/minecraft/client/gui/DrawContext;drawText(Lnet/minecraft/client/font/TextRenderer;Lnet/minecraft/text/Text;IIIZ)I"),
-		expect = 0,
-		require = 0)
-	private int replaceTextColorWithVariableShadow(DrawContext instance, TextRenderer textRenderer, Text text, int x, int y, int color, boolean shadow, Operation<Integer> original) {
-		return original.call(instance, textRenderer, text, x, y, CustomTextColors.INSTANCE.mapTextColor(text, color), shadow);
-	}
-
-	@WrapOperation(
-		method = "drawForeground",
-		at = @At(
-			value = "INVOKE",
-			target = "Lnet/minecraft/client/gui/DrawContext;drawTextWithShadow(Lnet/minecraft/client/font/TextRenderer;Lnet/minecraft/text/Text;III)I"),
-		expect = 0,
-		require = 0)
-	private int replaceTextColorWithShadow(DrawContext instance, TextRenderer textRenderer, Text text, int x, int y, int color, Operation<Integer> original) {
-		return original.call(instance, textRenderer, text, x, y, CustomTextColors.INSTANCE.mapTextColor(text, color));
-	}
-
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/custommodels/ApplyHeadModelInItemRenderer.java b/src/main/java/moe/nea/firmament/mixins/custommodels/ApplyHeadModelInItemRenderer.java
deleted file mode 100644
index dac65fe..0000000
--- a/src/main/java/moe/nea/firmament/mixins/custommodels/ApplyHeadModelInItemRenderer.java
+++ /dev/null
@@ -1,35 +0,0 @@
-
-package moe.nea.firmament.mixins.custommodels;
-
-import com.llamalad7.mixinextras.sugar.Local;
-import com.llamalad7.mixinextras.sugar.ref.LocalRef;
-import moe.nea.firmament.features.texturepack.BakedModelExtra;
-import net.minecraft.client.render.VertexConsumerProvider;
-import net.minecraft.client.render.item.ItemRenderer;
-import net.minecraft.client.render.model.BakedModel;
-import net.minecraft.client.util.math.MatrixStack;
-import net.minecraft.item.ItemStack;
-import net.minecraft.item.ModelTransformationMode;
-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;
-
-@Mixin(ItemRenderer.class)
-public class ApplyHeadModelInItemRenderer {
-	@Inject(method = "renderItem(Lnet/minecraft/item/ItemStack;Lnet/minecraft/item/ModelTransformationMode;ZLnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;IILnet/minecraft/client/render/model/BakedModel;ZF)V",
-		at = @At("HEAD"))
-	private void applyHeadModel(ItemStack stack, ModelTransformationMode transformationMode, boolean leftHanded,
-	                            MatrixStack matrices, VertexConsumerProvider vertexConsumers, int light, int overlay,
-	                            BakedModel model, boolean useInventoryModel, float z, CallbackInfo ci,
-	                            @Local(argsOnly = true) LocalRef<BakedModel> modelMut
-	) {
-		var extra = BakedModelExtra.cast(model);
-		if (transformationMode == ModelTransformationMode.HEAD && extra != null) {
-			var headModel = extra.getHeadModel_firmament();
-			if (headModel != null) {
-				modelMut.set(headModel);
-			}
-		}
-	}
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/custommodels/BakedModelDataHolderBasic.java b/src/main/java/moe/nea/firmament/mixins/custommodels/BakedModelDataHolderBasic.java
deleted file mode 100644
index 3ed2177..0000000
--- a/src/main/java/moe/nea/firmament/mixins/custommodels/BakedModelDataHolderBasic.java
+++ /dev/null
@@ -1,42 +0,0 @@
-
-package moe.nea.firmament.mixins.custommodels;
-
-import moe.nea.firmament.features.texturepack.BakedModelExtra;
-import moe.nea.firmament.features.texturepack.TintOverrides;
-import net.minecraft.client.render.model.BakedModel;
-import net.minecraft.client.render.model.BasicBakedModel;
-import org.jetbrains.annotations.Nullable;
-import org.spongepowered.asm.mixin.Mixin;
-import org.spongepowered.asm.mixin.Unique;
-
-@Mixin(BasicBakedModel.class)
-public class BakedModelDataHolderBasic implements BakedModelExtra {
-
-	@Unique
-	private BakedModel headModel;
-
-	@Unique
-	@Nullable
-	private TintOverrides tintOverrides;
-
-	@Nullable
-	@Override
-	public BakedModel getHeadModel_firmament() {
-		return headModel;
-	}
-
-	@Override
-	public void setHeadModel_firmament(@Nullable BakedModel headModel) {
-		this.headModel = headModel;
-	}
-
-	@Override
-	public @Nullable TintOverrides getTintOverrides_firmament() {
-		return tintOverrides;
-	}
-
-	@Override
-	public void setTintOverrides_firmament(@Nullable TintOverrides tintOverrides) {
-		this.tintOverrides = tintOverrides;
-	}
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/custommodels/BakedModelDataHolderBuiltin.java b/src/main/java/moe/nea/firmament/mixins/custommodels/BakedModelDataHolderBuiltin.java
deleted file mode 100644
index 87aecb1..0000000
--- a/src/main/java/moe/nea/firmament/mixins/custommodels/BakedModelDataHolderBuiltin.java
+++ /dev/null
@@ -1,43 +0,0 @@
-
-package moe.nea.firmament.mixins.custommodels;
-
-import moe.nea.firmament.features.texturepack.BakedModelExtra;
-import moe.nea.firmament.features.texturepack.TintOverrides;
-import net.minecraft.client.render.model.BakedModel;
-import net.minecraft.client.render.model.BuiltinBakedModel;
-import org.jetbrains.annotations.Nullable;
-import org.spongepowered.asm.mixin.Mixin;
-import org.spongepowered.asm.mixin.Unique;
-
-@Mixin(BuiltinBakedModel.class)
-public class BakedModelDataHolderBuiltin implements BakedModelExtra {
-
-	@Unique
-	@Nullable
-	private BakedModel headModel;
-
-	@Unique
-	@Nullable
-	private TintOverrides tintOverrides;
-
-	@Override
-	public @Nullable TintOverrides getTintOverrides_firmament() {
-		return tintOverrides;
-	}
-
-	@Override
-	public void setTintOverrides_firmament(@Nullable TintOverrides tintOverrides) {
-		this.tintOverrides = tintOverrides;
-	}
-
-	@Nullable
-	@Override
-	public BakedModel getHeadModel_firmament() {
-		return headModel;
-	}
-
-	@Override
-	public void setHeadModel_firmament(@Nullable BakedModel headModel) {
-		this.headModel = headModel;
-	}
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/custommodels/BakedOverrideDataHolder.java b/src/main/java/moe/nea/firmament/mixins/custommodels/BakedOverrideDataHolder.java
deleted file mode 100644
index 26972b1..0000000
--- a/src/main/java/moe/nea/firmament/mixins/custommodels/BakedOverrideDataHolder.java
+++ /dev/null
@@ -1,28 +0,0 @@
-
-package moe.nea.firmament.mixins.custommodels;
-
-import moe.nea.firmament.features.texturepack.BakedOverrideData;
-import moe.nea.firmament.features.texturepack.FirmamentModelPredicate;
-import net.minecraft.client.render.model.json.ModelOverrideList;
-import org.jetbrains.annotations.NotNull;
-import org.jetbrains.annotations.Nullable;
-import org.spongepowered.asm.mixin.Mixin;
-import org.spongepowered.asm.mixin.Unique;
-
-@Mixin(ModelOverrideList.BakedOverride.class)
-public class BakedOverrideDataHolder implements BakedOverrideData {
-
-    @Unique
-    private FirmamentModelPredicate[] firmamentOverrides;
-
-    @Nullable
-    @Override
-    public FirmamentModelPredicate[] getFirmamentOverrides() {
-        return firmamentOverrides;
-    }
-
-    @Override
-    public void setFirmamentOverrides(@NotNull FirmamentModelPredicate[] overrides) {
-        this.firmamentOverrides = overrides;
-    }
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/custommodels/HeadModelReplacerPatch.java b/src/main/java/moe/nea/firmament/mixins/custommodels/HeadModelReplacerPatch.java
deleted file mode 100644
index 26c331e..0000000
--- a/src/main/java/moe/nea/firmament/mixins/custommodels/HeadModelReplacerPatch.java
+++ /dev/null
@@ -1,57 +0,0 @@
-
-package moe.nea.firmament.mixins.custommodels;
-
-import com.llamalad7.mixinextras.injector.wrapoperation.Operation;
-import com.llamalad7.mixinextras.injector.wrapoperation.WrapOperation;
-import com.llamalad7.mixinextras.sugar.Local;
-import moe.nea.firmament.features.texturepack.BakedModelExtra;
-import net.minecraft.block.AbstractSkullBlock;
-import net.minecraft.block.Block;
-import net.minecraft.block.Blocks;
-import net.minecraft.client.render.VertexConsumerProvider;
-import net.minecraft.client.render.entity.LivingEntityRenderer;
-import net.minecraft.client.render.entity.feature.HeadFeatureRenderer;
-import net.minecraft.client.render.entity.model.EntityModel;
-import net.minecraft.client.render.entity.model.ModelWithHead;
-import net.minecraft.client.render.entity.state.LivingEntityRenderState;
-import net.minecraft.client.render.model.BakedModel;
-import net.minecraft.client.util.math.MatrixStack;
-import net.minecraft.entity.EquipmentSlot;
-import net.minecraft.item.BlockItem;
-import net.minecraft.item.ItemStack;
-import org.spongepowered.asm.mixin.Mixin;
-import org.spongepowered.asm.mixin.injection.At;
-
-@Mixin(HeadFeatureRenderer.class)
-public class HeadModelReplacerPatch<S extends LivingEntityRenderState, M extends EntityModel<S> & ModelWithHead> {
-	/**
-	 * This class serves to disable the replacing of head models with the vanilla block model. Vanilla first selects loads
-	 * the model containing the head model regularly in {@link LivingEntityRenderer#updateRenderState}, but then discards
-	 * the model in {@link HeadFeatureRenderer#render(MatrixStack, VertexConsumerProvider, int, LivingEntityRenderState, float, float)}
-	 * if it detects a skull block. This serves to disable that functionality if a head model override is present.
-	 */
-	@WrapOperation(method = "render(Lnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;ILnet/minecraft/client/render/entity/state/LivingEntityRenderState;FF)V",
-		at = @At(value = "INVOKE", target = "Lnet/minecraft/item/BlockItem;getBlock()Lnet/minecraft/block/Block;"))
-	private Block replaceSkull(BlockItem instance, Operation<Block> original, @Local BakedModel bakedModel) {
-		var oldBlock = original.call(instance);
-		if (oldBlock instanceof AbstractSkullBlock) {
-			var extra = BakedModelExtra.cast(bakedModel);
-			if (extra != null && extra.getHeadModel_firmament() != null)
-				return Blocks.ENCHANTING_TABLE; // Any non skull block. Let's choose the enchanting table because it is very distinct.
-		}
-		return oldBlock;
-	}
-
-	/**
-	 * We disable the has model override, since texture packs get precedent to server data.
-	 */
-	@WrapOperation(method = "render(Lnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;ILnet/minecraft/client/render/entity/state/LivingEntityRenderState;FF)V",
-		at = @At(value = "INVOKE", target = "Lnet/minecraft/client/render/entity/feature/ArmorFeatureRenderer;hasModel(Lnet/minecraft/item/ItemStack;Lnet/minecraft/entity/EquipmentSlot;)Z"))
-	private boolean replaceHasModel(ItemStack stack, EquipmentSlot slot, Operation<Boolean> original,
-	                                @Local BakedModel bakedModel) {
-		var extra = BakedModelExtra.cast(bakedModel);
-		if (extra != null && extra.getHeadModel_firmament() != null)
-			return false;
-		return original.call(stack, slot);
-	}
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/custommodels/ItemColorRemovalPatch.java b/src/main/java/moe/nea/firmament/mixins/custommodels/ItemColorRemovalPatch.java
deleted file mode 100644
index 8c76c60..0000000
--- a/src/main/java/moe/nea/firmament/mixins/custommodels/ItemColorRemovalPatch.java
+++ /dev/null
@@ -1,39 +0,0 @@
-package moe.nea.firmament.mixins.custommodels;
-
-import moe.nea.firmament.features.texturepack.TintOverrides;
-import moe.nea.firmament.init.ItemColorsSodiumRiser;
-import net.minecraft.client.color.item.ItemColorProvider;
-import net.minecraft.client.color.item.ItemColors;
-import net.minecraft.item.ItemStack;
-import org.jetbrains.annotations.Nullable;
-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.CallbackInfoReturnable;
-
-@Mixin(ItemColors.class)
-public class ItemColorRemovalPatch {
-
-	/**
-	 * @see ItemColorsSodiumRiser
-	 */
-	private @Nullable ItemColorProvider overrideSodium_firmament(@Nullable ItemColorProvider original) {
-		var tintOverrides = TintOverrides.Companion.getCurrentOverrides();
-		if (!tintOverrides.hasOverrides()) return original;
-		return (stack, tintIndex) -> {
-			var override = tintOverrides.getOverride(tintIndex);
-			if (override != null) return override;
-			if (original != null) return original.getColor(stack, tintIndex);
-			return -1;
-		};
-	}
-
-
-	@Inject(method = "getColor", at = @At("HEAD"), cancellable = true)
-	private void overrideGetColorCall(ItemStack item, int tintIndex, CallbackInfoReturnable<Integer> cir) {
-		var tintOverrides = TintOverrides.Companion.getCurrentOverrides();
-		var override = tintOverrides.getOverride(tintIndex);
-		if (override != null)
-			cir.setReturnValue(override);
-	}
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/custommodels/ItemModelGeneratorJsonUnbakedModelCopy.java b/src/main/java/moe/nea/firmament/mixins/custommodels/ItemModelGeneratorJsonUnbakedModelCopy.java
deleted file mode 100644
index 89d0411..0000000
--- a/src/main/java/moe/nea/firmament/mixins/custommodels/ItemModelGeneratorJsonUnbakedModelCopy.java
+++ /dev/null
@@ -1,22 +0,0 @@
-
-package moe.nea.firmament.mixins.custommodels;
-
-import com.llamalad7.mixinextras.injector.ModifyReturnValue;
-import com.llamalad7.mixinextras.sugar.Local;
-import moe.nea.firmament.features.texturepack.JsonUnbakedModelFirmExtra;
-import net.minecraft.client.render.model.json.ItemModelGenerator;
-import net.minecraft.client.render.model.json.JsonUnbakedModel;
-import org.spongepowered.asm.mixin.Mixin;
-import org.spongepowered.asm.mixin.injection.At;
-
-@Mixin(ItemModelGenerator.class)
-public class ItemModelGeneratorJsonUnbakedModelCopy {
-    @ModifyReturnValue(method = "create", at = @At("RETURN"))
-    private JsonUnbakedModel copyExtraModelData(JsonUnbakedModel original, @Local(argsOnly = true) JsonUnbakedModel oldModel) {
-        var extra = ((JsonUnbakedModelFirmExtra) original);
-		var oldExtra = ((JsonUnbakedModelFirmExtra) oldModel);
-		extra.setHeadModel_firmament(oldExtra.getHeadModel_firmament());
-		extra.setTintOverrides_firmament(oldExtra.getTintOverrides_firmament());
-        return original;
-    }
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/custommodels/ItemRendererTintContextPatch.java b/src/main/java/moe/nea/firmament/mixins/custommodels/ItemRendererTintContextPatch.java
deleted file mode 100644
index 8c5411b..0000000
--- a/src/main/java/moe/nea/firmament/mixins/custommodels/ItemRendererTintContextPatch.java
+++ /dev/null
@@ -1,35 +0,0 @@
-package moe.nea.firmament.mixins.custommodels;
-
-import moe.nea.firmament.features.texturepack.BakedModelExtra;
-import moe.nea.firmament.features.texturepack.TintOverrides;
-import net.minecraft.client.render.VertexConsumerProvider;
-import net.minecraft.client.render.item.ItemRenderer;
-import net.minecraft.client.render.model.BakedModel;
-import net.minecraft.client.util.math.MatrixStack;
-import net.minecraft.item.ItemStack;
-import net.minecraft.item.ModelTransformationMode;
-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;
-
-@Mixin(value = ItemRenderer.class, priority = 1010)
-public class ItemRendererTintContextPatch {
-	@Inject(method = "renderItem(Lnet/minecraft/item/ItemStack;Lnet/minecraft/item/ModelTransformationMode;ZLnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;IILnet/minecraft/client/render/model/BakedModel;ZF)V",
-		at = @At(value = "HEAD"), allow = 1)
-	private void onStartRendering(ItemStack stack, ModelTransformationMode transformationMode, boolean leftHanded, MatrixStack matrices, VertexConsumerProvider vertexConsumers, int light, int overlay, BakedModel model, boolean useInventoryModel, float z, CallbackInfo ci) {
-		var extra = BakedModelExtra.cast(model);
-		if (extra != null) {
-			TintOverrides.Companion.enter(extra.getTintOverrides_firmament());
-		}
-	}
-
-	@Inject(method = "renderItem(Lnet/minecraft/item/ItemStack;Lnet/minecraft/item/ModelTransformationMode;ZLnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;IILnet/minecraft/client/render/model/BakedModel;ZF)V",
-		at = @At("TAIL"), allow = 1)
-	private void onEndRendering(ItemStack stack, ModelTransformationMode transformationMode, boolean leftHanded, MatrixStack matrices, VertexConsumerProvider vertexConsumers, int light, int overlay, BakedModel model, boolean useInventoryModel, float z, CallbackInfo ci) {
-		var extra = BakedModelExtra.cast(model);
-		if (extra != null) {
-			TintOverrides.Companion.exit(extra.getTintOverrides_firmament());
-		}
-	}
-}
diff --git a/src/main/java/moe/nea/firmament/mixins/custommodels/JsonUnbakedModelDataHolder.java b/src/main/java/moe/nea/firmament/mixins/custommodels/JsonUnbakedModelDataHolder.java
deleted file mode 100644
index a5bb34f..0000000
--- a/src/main/java/moe/nea/firmament/mixins/custommodels/JsonUnbakedModelDataHolder.java
+++ /dev/null
@@ -1,130 +0,0 @@
-package moe.nea.firmament.mixins.custommodels;
-
-import com.llamalad7.mixinextras.injector.ModifyReturnValue;
-import com.llamalad7.mixinextras.sugar.Local;
-import moe.nea.firmament.features.texturepack.BakedModelExtra;
-import moe.nea.firmament.features.texturepack.JsonUnbakedModelFirmExtra;
-import moe.nea.firmament.features.texturepack.TintOverrides;
-import moe.nea.firmament.util.ErrorUtil;
-import net.minecraft.client.render.model.BakedModel;
-import net.minecraft.client.render.model.Baker;
-import net.minecraft.client.render.model.ModelRotation;
-import net.minecraft.client.render.model.UnbakedModel;
-import net.minecraft.client.render.model.json.JsonUnbakedModel;
-import net.minecraft.util.Identifier;
-import org.jetbrains.annotations.NotNull;
-import org.jetbrains.annotations.Nullable;
-import org.spongepowered.asm.mixin.Mixin;
-import org.spongepowered.asm.mixin.Shadow;
-import org.spongepowered.asm.mixin.Unique;
-import org.spongepowered.asm.mixin.injection.At;
-import org.spongepowered.asm.mixin.injection.Inject;
-import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
-
-import java.util.Objects;
-
-@Mixin(JsonUnbakedModel.class)
-public abstract class JsonUnbakedModelDataHolder implements JsonUnbakedModelFirmExtra {
-	@Shadow
-	@Nullable
-	protected JsonUnbakedModel parent;
-
-	@Shadow
-	public abstract String toString();
-
-	@Unique
-	@Nullable
-	public Identifier headModel;
-	@Unique
-	@Nullable
-	public TintOverrides tintOverrides;
-	@Unique
-	@Nullable
-	public TintOverrides mergedTintOverrides;
-
-	@Override
-	public void setTintOverrides_firmament(@Nullable TintOverrides tintOverrides) {
-		this.tintOverrides = tintOverrides;
-		this.mergedTintOverrides = null;
-	}
-
-	@Override
-	public @NotNull TintOverrides getTintOverrides_firmament() {
-		if (mergedTintOverrides != null)
-			return mergedTintOverrides;
-		var mergedTintOverrides = parent == null ? new TintOverrides()
-			: ((JsonUnbakedModelFirmExtra) parent).getTintOverrides_firmament();
-		if (tintOverrides != null)
-			mergedTintOverrides = tintOverrides.mergeWithParent(mergedTintOverrides);
-		this.mergedTintOverrides = mergedTintOverrides;
-		return mergedTintOverrides;
-	}
-
-	@Override
-	public void setHeadModel_firmament(@Nullable Identifier identifier) {
-		this.headModel = identifier;
-	}
-
-	@Override
-	public @Nullable Identifier getHeadModel_firmament() {
-		if (this.headModel != null) return this.headModel;
-		if (this.parent == null) return null;
-		return ((JsonUnbakedModelFirmExtra) this.parent).getHeadModel_firmament();
-	}
-
-	@Inject(method = "resolve", at = @At("HEAD"))
-	private void addDependencies(UnbakedModel.Resolver resolver, CallbackInfo ci) {
-		var headModel = getHeadModel_firmament();
-		if (headModel != null) {
-			resolver.resolve(headModel);
-		}
-	}
-
-	private void addExtraBakeInfo(BakedModel bakedModel, Baker baker) {
-		if (!this.toString().contains("minecraft") && this.toString().contains("crimson")) {
-			System.out.println("Found non minecraft model " + this);
-		}
-		var extra = BakedModelExtra.cast(bakedModel);
-		if (extra != null) {
-			var headModel = getHeadModel_firmament();
-			if (headModel != null) {
-				extra.setHeadModel_firmament(baker.bake(headModel, ModelRotation.X0_Y0));
-			}
-			if (getTintOverrides_firmament().hasOverrides()) {
-				extra.setTintOverrides_firmament(getTintOverrides_firmament());
-			}
-		}
-	}
-
-	/**
-	 * @see ProvideBakerToJsonUnbakedModelPatch
-	 */
-	@Override
-	public void storeExtraBaker_firmament(@NotNull Baker baker) {
-		this.storedBaker = baker;
-	}
-
-	@Unique
-	private Baker storedBaker;
-
-	@ModifyReturnValue(
-		method = "bake(Ljava/util/function/Function;Lnet/minecraft/client/render/model/ModelBakeSettings;Z)Lnet/minecraft/client/render/model/BakedModel;",
-		at = @At("RETURN"))
-	private BakedModel bakeExtraInfoWithoutBaker(BakedModel original) {
-		if (storedBaker != null) {
-			addExtraBakeInfo(original, storedBaker);
-			storedBaker = null;
-		}
-		return original;
-	}
-
-	@ModifyReturnValue(
-		method = {
-			"bake(Lnet/minecraft/client/render/model/Baker;Ljava/util/function/Function;Lnet/minecraft/client/render/model/ModelBakeSettings;)Lnet/minecraft/client/render/model/BakedModel;"
-		},
-		at = @At(value = "RETURN"))
-	private BakedModel bakeExtraInfo(BakedModel original, @Local(argsOnly = true) Baker baker) {
-		addExtraBakeInfo(original, baker);
-		return original;
-	}
-}
diff --git a/src/main/java