Add Polyfrost mods patch & fix LP patch

CleanroomMC · Sep 13, 2024 · cefde7b · cefde7b
1 parent 072c159
commit cefde7b
Show file tree

Hide file tree

Showing 5 changed files with 50 additions and 0 deletions.
diff --git a/src/main/java/com/cleanroommc/fugue/common/FugueLoadingPlugin.java b/src/main/java/com/cleanroommc/fugue/common/FugueLoadingPlugin.java
@@ -175,6 +175,13 @@ public class FugueLoadingPlugin implements IFMLLoadingPlugin {
         if (FugueConfig.modPatchConfig.enableCrossbow) {
             TransformerDelegate.registerExplicitTransformerByInstance(new TransformerEntityArrowTransformer(), "git.jbredwards.crossbow.mod.asm.transformer.TransformerEntityArrow");
         }
+        if (FugueConfig.modPatchConfig.enablePolyForst) {
+            TransformerDelegate.registerExplicitTransformerByInstance(
+                    new LaunchWrapperTweakerTransformer(),
+                    "cc.polyfrost.oneconfig.loader.stage0.LaunchWrapperTweaker",
+                    "cc.polyfrost.oneconfig.loader.OneConfigLoader"
+            );
+        }
         if (FugueConfig.getCodeSourcePatchTargets.length > 0) {
             TransformerDelegate.registerExplicitTransformerByInstance(new ITweakerTransformer(), FugueConfig.getCodeSourcePatchTargets);
         }

diff --git a/src/main/java/com/cleanroommc/fugue/config/ModPatchConfig.java b/src/main/java/com/cleanroommc/fugue/config/ModPatchConfig.java
@@ -112,4 +112,6 @@ public class ModPatchConfig {
     public boolean enableInfLib = true;
     @Config.Name("Enable Crossbow(jbredwards) Patch")
     public boolean enableCrossbow = true;
+    @Config.Name("Enable Patch to PolyForst mods")
+    public boolean enablePolyForst = true;
 }
diff --git a/src/main/java/com/cleanroommc/fugue/helper/HookHelper.java b/src/main/java/com/cleanroommc/fugue/helper/HookHelper.java
@@ -8,6 +8,7 @@
 import top.outlands.foundation.TransformerDelegate;
 import top.outlands.foundation.boot.ActualClassLoader;
 
+import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
 import java.lang.reflect.Field;
@@ -189,4 +190,18 @@ public static Object computeIfAbsent(Map<Object, Object> map, Object key, Functi
             return value;
         }
     }
+
+    public static boolean isInitialized(File file) {
+        try {
+            return Arrays.asList(Launch.classLoader.getURLs()).contains(file.toURI().toURL());
+        } catch (MalformedURLException e) {
+            return false;
+        }
+    }
+
+    public static void addToClasspath(File file) {
+        try {
+            Launch.classLoader.addURL(file.toURI().toURL());
+        } catch (MalformedURLException ignored) {}
+    }
 }
diff --git a/src/main/java/com/cleanroommc/fugue/transformer/LaunchWrapperTweakerTransformer.java b/src/main/java/com/cleanroommc/fugue/transformer/LaunchWrapperTweakerTransformer.java
@@ -0,0 +1,23 @@
+package com.cleanroommc.fugue.transformer;
+
+import com.cleanroommc.fugue.common.Fugue;
+import javassist.ClassPool;
+import javassist.CtClass;
+import top.outlands.foundation.IExplicitTransformer;
+
+import java.io.ByteArrayInputStream;
+
+public class LaunchWrapperTweakerTransformer implements IExplicitTransformer {
+    @Override
+    public byte[] transform(byte[] bytes) {
+        try {
+            CtClass cc = ClassPool.getDefault().makeClass(new ByteArrayInputStream(bytes));
+            cc.getDeclaredMethod("isInitialized").setBody("{return com.cleanroommc.fugue.helper.HookHelper#isInitialized($$);}");
+            cc.getDeclaredMethod("addToClasspath").setBody("{com.cleanroommc.fugue.helper.HookHelper#addToClasspath($$);}");
+            bytes = cc.toBytecode();
+        } catch (Throwable t) {
+            Fugue.LOGGER.error("Exception {} on {}", t, this.getClass().getSimpleName());
+        }
+        return bytes;
+    }
+}
diff --git a/src/main/java/com/cleanroommc/fugue/transformer/logisticpipes/LogisticsPipesTrigger.java b/src/main/java/com/cleanroommc/fugue/transformer/logisticpipes/LogisticsPipesTrigger.java
@@ -1,14 +1,17 @@
 package com.cleanroommc.fugue.transformer.logisticpipes;
 
+import com.cleanroommc.fugue.common.Fugue;
 import net.minecraft.launchwrapper.Launch;
 import top.outlands.foundation.IExplicitTransformer;
 
 public class LogisticsPipesTrigger implements IExplicitTransformer {
     @Override
     public byte[] transform(byte[] bytes) {
+        Fugue.LOGGER.info("Loading LP classes...");
         try {
             Launch.classLoader.findClass("logisticspipes.network.packets.routingdebug.RoutingUpdateUntrace");
             Launch.classLoader.findClass("logisticspipes.network.packets.modules.SupplierPipeMode");
+            Launch.classLoader.findClass("logisticspipes.network.packets.module.ModuleBasedItemSinkList");
             Launch.classLoader.findClass("logisticspipes.network.packets.pipe.FluidSupplierMinMode");
             Launch.classLoader.findClass("logisticspipes.network.abstractguis.ModuleCoordinatesGuiProvider");
             Launch.classLoader.findClass("logisticspipes.network.guis.upgrade.SneakyUpgradeConfigGuiProvider");