Replaced Redirect usage with ModifyArg

Fixes #73
Fixes #72

1 files changed, 17 insertions, 16 deletions

diff --git a/src/main/java/shcm/shsupercm/fabric/citresewn/mixin/citelytra/ElytraFeatureRendererMixin.java b/src/main/java/shcm/shsupercm/fabric/citresewn/mixin/citelytra/ElytraFeatureRendererMixin.java
index db2241e..b4151f8 100644
--- a/src/main/java/shcm/shsupercm/fabric/citresewn/mixin/citelytra/ElytraFeatureRendererMixin.java
+++ b/src/main/java/shcm/shsupercm/fabric/citresewn/mixin/citelytra/ElytraFeatureRendererMixin.java
@@ -12,7 +12,7 @@ import net.minecraft.util.Identifier;
 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.Redirect;
+import org.spongepowered.asm.mixin.injection.ModifyArg;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
 import shcm.shsupercm.fabric.citresewn.CITResewn;
 import shcm.shsupercm.fabric.citresewn.OptionalCompat;
@@ -25,7 +25,8 @@ public class ElytraFeatureRendererMixin {
     private WeakReference<ItemStack> elytraItemCached = new WeakReference<>(null);
     private WeakReference<LivingEntity> livingEntityCached = new WeakReference<>(null);
 
-    @Inject(method = "render(Lnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;ILnet/minecraft/entity/LivingEntity;FFFFFF)V", at = @At("HEAD"))
+    @Inject(method = "render(Lnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;ILnet/minecraft/entity/LivingEntity;FFFFFF)V", at =
+    @At("HEAD"))
     private void render(MatrixStack matrixStack, VertexConsumerProvider vertexConsumerProvider, int i, LivingEntity livingEntity, float f, float g, float h, float j, float k, float l, CallbackInfo ci) {
         if (!CITResewnConfig.INSTANCE().enabled || CITResewn.INSTANCE.activeCITs == null)
             return;
@@ -39,21 +40,21 @@ public class ElytraFeatureRendererMixin {
         this.livingEntityCached = new WeakReference<>(livingEntity);
     }
 
-    @Redirect(method = "render(Lnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;ILnet/minecraft/entity/LivingEntity;FFFFFF)V", at = @At(value = "INVOKE", target = "Lnet/minecraft/client/render/RenderLayer;getArmorCutoutNoCull(Lnet/minecraft/util/Identifier;)Lnet/minecraft/client/render/RenderLayer;"))
-    private RenderLayer getArmorCutoutNoCull(Identifier originalIdentifier) {
-        if (!CITResewnConfig.INSTANCE().enabled || CITResewn.INSTANCE.activeCITs == null)
-            return RenderLayer.getArmorCutoutNoCull(originalIdentifier);
-
-        ItemStack itemStack = this.elytraItemCached.get();
-        LivingEntity livingEntity = this.livingEntityCached.get();
-        if (itemStack != null && itemStack.isOf(Items.ELYTRA) && livingEntity != null) {
-            Identifier elytraTexture = CITResewn.INSTANCE.activeCITs.getElytraTextureCached(itemStack, livingEntity.world, livingEntity);
-            this.elytraItemCached = new WeakReference<>(null);
-            this.livingEntityCached = new WeakReference<>(null);
-            if (elytraTexture != null)
-                return RenderLayer.getArmorCutoutNoCull(elytraTexture);
+    @ModifyArg(method = "render(Lnet/minecraft/client/util/math/MatrixStack;Lnet/minecraft/client/render/VertexConsumerProvider;ILnet/minecraft/entity/LivingEntity;FFFFFF)V", at =
+    @At(value = "INVOKE", target = "Lnet/minecraft/client/render/item/ItemRenderer;getArmorGlintConsumer(Lnet/minecraft/client/render/VertexConsumerProvider;Lnet/minecraft/client/render/RenderLayer;ZZ)Lnet/minecraft/client/render/VertexConsumer;"), index = 1)
+    private RenderLayer getArmorCutoutNoCull(RenderLayer original) {
+        if (CITResewnConfig.INSTANCE().enabled && CITResewn.INSTANCE.activeCITs != null) {
+            ItemStack itemStack = this.elytraItemCached.get();
+            LivingEntity livingEntity = this.livingEntityCached.get();
+            if (itemStack != null && itemStack.isOf(Items.ELYTRA) && livingEntity != null) {
+                Identifier elytraTexture = CITResewn.INSTANCE.activeCITs.getElytraTextureCached(itemStack, livingEntity.world, livingEntity);
+                this.elytraItemCached = new WeakReference<>(null);
+                this.livingEntityCached = new WeakReference<>(null);
+                if (elytraTexture != null)
+                    return RenderLayer.getArmorCutoutNoCull(elytraTexture);
+            }
         }
 
-        return RenderLayer.getArmorCutoutNoCull(originalIdentifier);
+        return original;
     }
 }