Initial commit.

1 files changed, 112 insertions, 0 deletions

diff --git a/src/main/java/com/sk89q/warmroast/McpMapping.java b/src/main/java/com/sk89q/warmroast/McpMapping.java
new file mode 100644
index 0000000..5ef2ff9
--- /dev/null
+++ b/src/main/java/com/sk89q/warmroast/McpMapping.java
@@ -0,0 +1,112 @@
+/*
+ * WarmRoast
+ * Copyright (C) 2013 Albert Pham <http://www.sk89q.com>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package com.sk89q.warmroast;
+
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import org.apache.commons.io.FileUtils;
+
+import au.com.bytecode.opencsv.CSVReader;
+
+public class McpMapping {
+
+    private static final Pattern clPattern = 
+            Pattern.compile("CL: (?<obfuscated>[^ ]+) (?<actual>[^ ]+)");
+    private static final Pattern mdPattern = 
+            Pattern.compile("MD: (?<obfuscatedClass>[^ /]+)/(?<obfuscatedMethod>[^ ]+) " +
+            		"[^ ]+ (?<method>[^ ]+) [^ ]+");
+
+    private final Map<String, ClassMapping> classes = new HashMap<>();
+    private final Map<String, String> methods = new HashMap<>();
+    
+    public ClassMapping mapClass(String obfuscated) {
+        return classes.get(obfuscated);
+    }
+
+    public void read(File joinedFile, File methodsFile) throws IOException {
+        try (FileReader r = new FileReader(methodsFile)) {
+            try (CSVReader reader = new CSVReader(r)) {
+                List<String[]> entries = reader.readAll();
+                processMethodNames(entries);
+            }
+        }
+        
+        List<String> lines = FileUtils.readLines(joinedFile, "UTF-8");
+        processClasses(lines);
+        processMethods(lines);
+    }
+    
+    public String fromMethodId(String id) {
+        String method = methods.get(id);
+        if (method == null) {
+            return id;
+        }
+        return method;
+    }
+    
+    private void processMethodNames(List<String[]> entries) {
+        boolean first = true;
+        for (String[] entry : entries) {
+            if (entry.length < 2) {
+                continue;
+            }
+            if (first) { // Header
+                first = false;
+                continue;
+            }
+            methods.put(entry[0], entry[1]);
+        }
+    }
+    
+    private void processClasses(List<String> lines) {
+        for (String line : lines) {
+            Matcher m = clPattern.matcher(line);
+            if (m.matches()) {
+                String obfuscated = m.group("obfuscated");
+                String actual = m.group("actual").replace("/", ".");
+                classes.put(obfuscated, new ClassMapping(obfuscated, actual));
+            }
+        }
+    }
+    
+    private void processMethods(List<String> lines) {
+        for (String line : lines) {
+            Matcher m = mdPattern.matcher(line);
+            if (m.matches()) {
+                String obfuscatedClass = m.group("obfuscatedClass");
+                String obfuscatedMethod = m.group("obfuscatedMethod");
+                String method = m.group("method");
+                String methodId = method.substring(method.lastIndexOf('/') + 1);
+                ClassMapping mapping = mapClass(obfuscatedClass);
+                if (mapping != null) {
+                    mapping.addMethod(obfuscatedMethod, 
+                            fromMethodId(methodId));
+                }
+            }
+        }
+    }
+
+}