also translate source file attributes so it's easy to browse decompiled sources

1 files changed, 8 insertions, 0 deletions

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));
+                }
         }
 }