From 038669a3ce711a29fc30aa3c91ccd7fa10af7829 Mon Sep 17 00:00:00 2001
From: Naz <bluspringindohq@gmail.com>
Date: Thu, 13 Jul 2023 10:30:19 +0800
Subject: [PATCH] Try to make the remapper even less prone to failing

---
 .../kilt/loader/remap/KiltAsmRemapper.kt      | 22 ++++++++++++++++++-
 .../kilt/loader/remap/KiltRemapper.kt         | 12 ++++++----
 2 files changed, 29 insertions(+), 5 deletions(-)

diff --git a/src/main/kotlin/xyz/bluspring/kilt/loader/remap/KiltAsmRemapper.kt b/src/main/kotlin/xyz/bluspring/kilt/loader/remap/KiltAsmRemapper.kt
index a7a3aa81..2d7961f6 100644
--- a/src/main/kotlin/xyz/bluspring/kilt/loader/remap/KiltAsmRemapper.kt
+++ b/src/main/kotlin/xyz/bluspring/kilt/loader/remap/KiltAsmRemapper.kt
@@ -1,5 +1,6 @@
 package xyz.bluspring.kilt.loader.remap
 
+import net.fabricmc.loader.api.FabricLoader
 import org.objectweb.asm.commons.Remapper
 import org.objectweb.asm.signature.SignatureReader
 import org.objectweb.asm.signature.SignatureWriter
@@ -73,7 +74,10 @@ class KiltAsmRemapper(
         val mappedPair = methodMappings[name] ?: return name
         val mapped = mappedPair.first
 
-        if (KiltRemapper.remapDescriptor(descriptor).replaceAfter(")", "") != mappedPair.second.replaceAfter(")", ""))
+        val remappedDescriptor = KiltRemapper.remapDescriptor(descriptor)
+        val remappedDescriptorNoEnd = remappedDescriptor.replaceAfter(")", "")
+
+        if (remappedDescriptorNoEnd != mappedPair.second.replaceAfter(")", ""))
             return name
 
         if ((!name.startsWith("m_") && !name.startsWith("f_")) && !name.endsWith("_")) {
@@ -81,6 +85,22 @@ class KiltAsmRemapper(
             if (ownerWhitelist.any { owner.startsWith(it) })
                 return name
 
+            // Try to see if these mappings can be used to help
+            val srgClassTree = KiltRemapper.srgIntermediaryTree.classes.firstOrNull { it.getName("searge") == owner }
+
+            if (srgClassTree != null) {
+                val matchingMethod = srgClassTree.methods.firstOrNull { it.getName("searge") == name && it.getDescriptor("intermediary").replaceAfter(")", "") == remappedDescriptorNoEnd }
+
+                if (matchingMethod != null) {
+                    val intermediaryName = matchingMethod.getName("intermediary")
+
+                    return if (KiltRemapper.useNamed)
+                        FabricLoader.getInstance().mappingResolver.mapMethodName("intermediary", srgClassTree.getName("intermediary"), intermediaryName, KiltRemapper.remapDescriptor(descriptor, toIntermediary = true))
+                    else
+                        intermediaryName
+                }
+            }
+
             // Because of the way the remapper works, it remaps every name sharing the same f_num_ and m_num_ names
             // without checking the owners, because it functions under the assumption that every intermediate name
             // is of the same mapped name. However, that assumption fails when it reaches a non-intermediate name,
diff --git a/src/main/kotlin/xyz/bluspring/kilt/loader/remap/KiltRemapper.kt b/src/main/kotlin/xyz/bluspring/kilt/loader/remap/KiltRemapper.kt
index 99024cb5..3354856e 100644
--- a/src/main/kotlin/xyz/bluspring/kilt/loader/remap/KiltRemapper.kt
+++ b/src/main/kotlin/xyz/bluspring/kilt/loader/remap/KiltRemapper.kt
@@ -30,7 +30,7 @@ object KiltRemapper {
     // Keeps track of the remapper changes, so every time I update the remapper,
     // it remaps all the mods following the remapper changes.
     // this can update by like 12 versions in 1 update, so don't worry too much about it.
-    const val REMAPPER_VERSION = 34
+    const val REMAPPER_VERSION = 35
 
     private val logger = LoggerFactory.getLogger("Kilt Remapper")
     // This is created automatically using https://github.com/BluSpring/srg2intermediary
@@ -405,9 +405,13 @@ object KiltRemapper {
         return exceptions
     }
 
-    fun remapClass(name: String): String {
+    fun remapClass(name: String, toIntermediary: Boolean = false): String {
         val workaround = kiltWorkaroundTree.classes.firstOrNull { it.getRawName("forge") == name }?.getRawName("kilt")
 
+        if (toIntermediary) {
+            return workaround ?: srgIntermediaryTree.classes.firstOrNull { it.getName("searge") == name }?.getName("intermediary") ?: name
+        }
+
         return workaround ?: classMappings[name] ?: name
     }
 
@@ -415,7 +419,7 @@ object KiltRemapper {
         return classMappings.entries.firstOrNull { it.value == name }?.key ?: name
     }
 
-    fun remapDescriptor(descriptor: String, reverse: Boolean = false): String {
+    fun remapDescriptor(descriptor: String, reverse: Boolean = false, toIntermediary: Boolean = false): String {
         var formedString = ""
 
         var incompleteString = ""
@@ -434,7 +438,7 @@ object KiltRemapper {
 
                     val name = incompleteString.removePrefix("L").removeSuffix(";")
                     formedString += if (!reverse)
-                        remapClass(name)
+                        remapClass(name, toIntermediary)
                     else
                         unmapClass(name)