From ba7a354efae7d49833c887cf147ac940c975a1fa Mon Sep 17 00:00:00 2001
From: Gegy
Date: Wed, 30 Jan 2019 21:05:32 +0200
Subject: Remap sources (#106)

* Source remapping beginnings

* Fix navigation to remapped classes

* Translate identifier info reference

* Remap local variables with default names in source

* Caching translator

* Fix lack of highlighting for first opened class

* Fix unicode variable names

* Unicode checker shouldn't be checking just alphanumeric

* Fix package tree being built from obf names

* Don't index `this` as method call for method::reference

* Apply proposed names

* Fix source export issues

* Replace unicode var names at bytecode level uniquely

* Drop imports from editor source

* Class selector fixes

* Delta keep track of base mappings to enable lookup of old names

* Optimize source remapping by remapping source with a StringBuffer instead of copying

* Bump version
---
 src/main/java/cuchaz/enigma/SourceProvider.java | 97 +++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 src/main/java/cuchaz/enigma/SourceProvider.java

(limited to 'src/main/java/cuchaz/enigma/SourceProvider.java')

diff --git a/src/main/java/cuchaz/enigma/SourceProvider.java b/src/main/java/cuchaz/enigma/SourceProvider.java
new file mode 100644
index 0000000..48e5a59
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/SourceProvider.java
@@ -0,0 +1,97 @@
+package cuchaz.enigma;
+
+import com.strobel.assembler.metadata.ITypeLoader;
+import com.strobel.assembler.metadata.MetadataSystem;
+import com.strobel.assembler.metadata.TypeDefinition;
+import com.strobel.assembler.metadata.TypeReference;
+import com.strobel.decompiler.DecompilerContext;
+import com.strobel.decompiler.DecompilerSettings;
+import com.strobel.decompiler.PlainTextOutput;
+import com.strobel.decompiler.languages.java.JavaOutputVisitor;
+import com.strobel.decompiler.languages.java.ast.AstBuilder;
+import com.strobel.decompiler.languages.java.ast.CompilationUnit;
+import com.strobel.decompiler.languages.java.ast.InsertParenthesesVisitor;
+import com.strobel.decompiler.languages.java.ast.transforms.IAstTransform;
+import cuchaz.enigma.utils.Utils;
+import oml.ast.transformers.*;
+
+import java.io.StringWriter;
+import java.io.Writer;
+import java.util.Arrays;
+import java.util.List;
+
+public class SourceProvider {
+	private final DecompilerSettings settings;
+
+	private final ITypeLoader typeLoader;
+	private final MetadataSystem metadataSystem;
+
+	public SourceProvider(DecompilerSettings settings, ITypeLoader typeLoader, MetadataSystem metadataSystem) {
+		this.settings = settings;
+		this.typeLoader = typeLoader;
+		this.metadataSystem = metadataSystem;
+	}
+
+	public SourceProvider(DecompilerSettings settings, ITypeLoader typeLoader) {
+		this(settings, typeLoader, new Deobfuscator.NoRetryMetadataSystem(typeLoader));
+	}
+
+	public static DecompilerSettings createSettings() {
+		DecompilerSettings settings = DecompilerSettings.javaDefaults();
+		settings.setMergeVariables(Utils.getSystemPropertyAsBoolean("enigma.mergeVariables", true));
+		settings.setForceExplicitImports(Utils.getSystemPropertyAsBoolean("enigma.forceExplicitImports", true));
+		settings.setForceExplicitTypeArguments(Utils.getSystemPropertyAsBoolean("enigma.forceExplicitTypeArguments", true));
+		settings.setShowDebugLineNumbers(Utils.getSystemPropertyAsBoolean("enigma.showDebugLineNumbers", false));
+		settings.setShowSyntheticMembers(Utils.getSystemPropertyAsBoolean("enigma.showSyntheticMembers", false));
+
+		return settings;
+	}
+
+	public CompilationUnit getSources(String name) {
+		TypeReference type = metadataSystem.lookupType(name);
+		if (type == null) {
+			throw new Error(String.format("Unable to find desc: %s", name));
+		}
+
+		TypeDefinition resolvedType = type.resolve();
+
+		settings.setTypeLoader(typeLoader);
+
+		// decompile it!
+		DecompilerContext context = new DecompilerContext();
+		context.setCurrentType(resolvedType);
+		context.setSettings(settings);
+
+		AstBuilder builder = new AstBuilder(context);
+		builder.addType(resolvedType);
+		builder.runTransformations(null);
+		runCustomTransforms(builder, context);
+
+		return builder.getCompilationUnit();
+	}
+
+	public void writeSource(Writer writer, CompilationUnit sourceTree) {
+		// render the AST into source
+		sourceTree.acceptVisitor(new InsertParenthesesVisitor(), null);
+		sourceTree.acceptVisitor(new JavaOutputVisitor(new PlainTextOutput(writer), settings), null);
+	}
+
+	public String writeSourceToString(CompilationUnit sourceTree) {
+		StringWriter writer = new StringWriter();
+		writeSource(writer, sourceTree);
+		return writer.toString();
+	}
+
+	private static void runCustomTransforms(AstBuilder builder, DecompilerContext context) {
+		List<IAstTransform> transformers = Arrays.asList(
+				new ObfuscatedEnumSwitchRewriterTransform(context),
+				new VarargsFixer(context),
+				new RemoveObjectCasts(context),
+				new Java8Generics(),
+				new InvalidIdentifierFix()
+		);
+		for (IAstTransform transform : transformers) {
+			transform.run(builder.getCompilationUnit());
+		}
+	}
+}
-- 
cgit v1.2.3