From f967021ea4a78de9aea0873eeb3e1359a4664a2b Mon Sep 17 00:00:00 2001
From: gegy1000
Date: Tue, 3 Jul 2018 14:41:14 +0200
Subject: Output directly to file on source export

---
 src/main/java/cuchaz/enigma/Deobfuscator.java | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/main/java/cuchaz/enigma/Deobfuscator.java b/src/main/java/cuchaz/enigma/Deobfuscator.java
index ccd099c6..6fe6e643 100644
--- a/src/main/java/cuchaz/enigma/Deobfuscator.java
+++ b/src/main/java/cuchaz/enigma/Deobfuscator.java
@@ -11,7 +11,6 @@
 
 package cuchaz.enigma;
 
-import com.google.common.base.Charsets;
 import com.google.common.collect.Lists;
 import com.google.common.collect.Maps;
 import com.google.common.collect.Sets;
@@ -277,13 +276,14 @@ public class Deobfuscator {
 
 			try {
 				// get the source
-				String source = getSource(getSourceTree(obfClassEntry.getName()));
+				CompilationUnit sourceTree = getSourceTree(obfClassEntry.getName());
 
 				// write the file
 				File file = new File(dirOut, deobfClassEntry.getName().replace('.', '/') + ".java");
 				file.getParentFile().mkdirs();
-				try (OutputStreamWriter out = new OutputStreamWriter(new BufferedOutputStream(new FileOutputStream(file)), Charsets.UTF_8)) {
-					out.write(source);
+				try (Writer writer = new BufferedWriter(new FileWriter(file))) {
+					sourceTree.acceptVisitor(new InsertParenthesesVisitor(), null);
+					sourceTree.acceptVisitor(new JavaOutputVisitor(new PlainTextOutput(writer), this.settings), null);
 				}
 			} catch (Throwable t) {
 				// don't crash the whole world here, just log the error and keep going
-- 
cgit v1.2.3