Add support for .tsrg mappings (McpConfig)

Legacy-Fabric · Dec 4, 2021 · 8e33846 · 8e33846
1 parent f710945
commit 8e33846
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -6,7 +6,7 @@ plugins {
 }
 
 group = "net.legacyfabric"
-version = "1.0.0"
+version = "1.0.1"
 
 repositories {
     mavenCentral()

diff --git a/src/main/java/net/legacyfabric/mcpmatcher/util/MappingUtils.java b/src/main/java/net/legacyfabric/mcpmatcher/util/MappingUtils.java
@@ -43,12 +43,11 @@ private static void readJarFile(Path jarFile, SignatureVisitor visitor) throws I
     }
 
     public static MappingSet readMappingsFile(String arg) throws IOException {
-        //TODO: could use a map here and support more formats.
-        if (arg.endsWith(".csrg")) {
-            return MappingFormats.CSRG.read(Paths.get(arg));
-        } else {
-            return MappingFormats.SRG.read(Paths.get(arg));
-        }
+        return switch (arg.split("\\.")[1]) {
+            case "csrg" -> MappingFormats.CSRG.read(Paths.get(arg));
+            case "tsrg" -> MappingFormats.TSRG.read(Paths.get(arg));
+            default -> MappingFormats.SRG.read(Paths.get(arg));
+        };
     }
 
     private static class SignatureVisitor extends ClassVisitor {