Patch worse hurt time

Author: kappa-maintainer

src/main/java/com/cleanroommc/fugue/common/TransformerHelper.java

@@ -6,6 +6,7 @@
 import com.cleanroommc.fugue.modifiers.IC2ExtraFixer;
 import com.cleanroommc.fugue.transformer.advancedrocket.ClassTransformerTransformer;
 import com.cleanroommc.fugue.transformer.betterfc.HK_LoaderTransformer;
+import com.cleanroommc.fugue.transformer.betterhurttimer.DamageSpecialArmorMixinTransformer;
 import com.cleanroommc.fugue.transformer.betterportals.MixinEntityRendererTransformer;
 import com.cleanroommc.fugue.transformer.betterportals.ExtensionKtTransformer;
 import com.cleanroommc.fugue.transformer.calculator.GuiInfoCalculatorTransformer;
@@ -363,8 +364,7 @@ public static void registerTransformers() {
                     "com.shblock.integrated_proxy.mixin.MixinLoader"
             );
         }
-        RemoveMixinInitFromCotrTransformer instance = new RemoveMixinInitFromCotrTransformer(
-            );
+        RemoveMixinInitFromCotrTransformer instance = new RemoveMixinInitFromCotrTransformer();
         if (FugueConfig.modPatchConfig.enableThaumicFixesPatch) {
             TransformerDelegate.registerExplicitTransformer(instance,
                     "com.seriouscreeper.thaumicfixes.core.ThaumicFixesLoadingPlugin"
@@ -494,6 +494,12 @@ public static void registerTransformers() {
                     "io.github.lgatodu47.screenshot_viewer.ScreenshotViewer$ScreenshotViewerEvents"
             );
         }
+        if (FugueConfig.modPatchConfig.enableWorseHurtTime) {
+            MixinServiceLaunchWrapper.registerMixinClassTransformer(
+                    new DamageSpecialArmorMixinTransformer(),
+                    "arekkuusu.betterhurttimer.mixin.DamageSpecialArmorMixin"
+            );
+        }
 
         //Common patches below
 

src/main/java/com/cleanroommc/fugue/config/ModPatchConfig.java

@@ -161,4 +161,7 @@ public class ModPatchConfig {
     public boolean enableCorpse = true;
     @Config.Name("Enable ScreenshotViewer Patch")
     public boolean enableScreenshotViewer = true;
+    @Config.Name("Enable Worse Hurt Time Patch")
+    @Config.Comment("Should be compatible with better hurt time")
+    public boolean enableWorseHurtTime = true;
 }

src/main/java/com/cleanroommc/fugue/transformer/betterhurttimer/DamageSpecialArmorMixinTransformer.java

@@ -0,0 +1,37 @@
+package com.cleanroommc.fugue.transformer.betterhurttimer;
+
+import com.cleanroommc.fugue.common.Fugue;
+import org.objectweb.asm.ClassReader;
+import org.objectweb.asm.ClassWriter;
+import org.objectweb.asm.tree.*;
+import top.outlands.foundation.IExplicitTransformer;
+
+import java.util.Arrays;
+
+public class DamageSpecialArmorMixinTransformer implements IExplicitTransformer {
+    @Override
+    public byte[] transform(byte[] bytes) {
+        ClassNode classNode = new ClassNode();
+        ClassReader classReader = new ClassReader(bytes);
+        classReader.accept(classNode, 0);
+        if (classNode.methods != null)
+        {
+            for (MethodNode methodNode : classNode.methods)
+            {
+                if (methodNode.name.equals("storeValues"))
+                {
+                    if (methodNode.invisibleAnnotableParameterCount != 7) break;
+                    AnnotationNode local = methodNode.invisibleParameterAnnotations[5].getFirst();
+                    local.values.removeLast();
+                    local.values.add(0);
+                    break;
+                }
+
+            }
+        }
+        ClassWriter classWriter = new ClassWriter(0);
+
+        classNode.accept(classWriter);
+        return classWriter.toByteArray();
+    }
+}