From 91460a865cc5d4f8d1e0c38fd08c1f071af147bb Mon Sep 17 00:00:00 2001
From: jeff
Date: Sun, 18 Jan 2015 17:17:27 -0500
Subject: also translate source file attributes so it's easy to browse
 decompiled sources

---
 src/cuchaz/enigma/bytecode/ClassTranslator.java | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src')

diff --git a/src/cuchaz/enigma/bytecode/ClassTranslator.java b/src/cuchaz/enigma/bytecode/ClassTranslator.java
index bc12405c..735a8fa3 100644
--- a/src/cuchaz/enigma/bytecode/ClassTranslator.java
+++ b/src/cuchaz/enigma/bytecode/ClassTranslator.java
@@ -18,6 +18,7 @@ import javassist.CtField;
 import javassist.CtMethod;
 import javassist.bytecode.ConstPool;
 import javassist.bytecode.Descriptor;
+import javassist.bytecode.SourceFileAttribute;
 
 import com.google.common.collect.Maps;
 
@@ -125,5 +126,12 @@ public class ClassTranslator {
 			}
 		}
 		ClassRenamer.renameClasses(c, map);
+		
+		// translate the source file attribute too
+		ClassEntry deobfClassEntry = map.get(classEntry);
+		if (deobfClassEntry != null) {
+			String sourceFile = Descriptor.toJvmName(deobfClassEntry.getOuterClassName()) + ".java";
+			c.getClassFile().addAttribute(new SourceFileAttribute(constants, sourceFile));
+		}
 	}
 }
-- 
cgit v1.2.3