From a9e03fa0e75b5b338021de982acbbb8277e08706 Mon Sep 17 00:00:00 2001
From: Fudge
Date: Mon, 2 Dec 2019 15:43:23 +0200
Subject: Allow attaching class, method, field, and parameter javadocs (#185)

* bring liach pr to modern enigma

* bump version

* fuck off vscode

* switch to COMMENT and write comments before

* it was already after, what do you want

* oops

* put inner classes at the end

* remove indents and use all caps

* add refreshmappings command

* Update src/main/java/cuchaz/enigma/translation/mapping/serde/EnigmaMappingsWriter.java

* Delete RefreshEnigmaMappingsCommand.java

* Update CommandMain.java

* ok
---
 src/main/java/cuchaz/enigma/gui/GuiController.java | 35 ++++++++++++++++++++--
 1 file changed, 32 insertions(+), 3 deletions(-)

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

diff --git a/src/main/java/cuchaz/enigma/gui/GuiController.java b/src/main/java/cuchaz/enigma/gui/GuiController.java
index 69f12e2..0b2fe27 100644
--- a/src/main/java/cuchaz/enigma/gui/GuiController.java
+++ b/src/main/java/cuchaz/enigma/gui/GuiController.java
@@ -368,20 +368,29 @@ public class GuiController {
 	}
 
 	private void refreshCurrentClass(EntryReference<Entry<?>, Entry<?>> reference) {
+		refreshCurrentClass(reference, false);
+	}
+
+	private void refreshCurrentClass(EntryReference<Entry<?>, Entry<?>> reference, boolean forceDecomp) {
 		if (currentSource != null) {
 			loadClass(currentSource.getEntry(), () -> {
 				if (reference != null) {
 					showReference(reference);
 				}
-			});
+			}, forceDecomp);
 		}
 	}
 
 	private void loadClass(ClassEntry classEntry, Runnable callback) {
+		loadClass(classEntry, callback, false);
+	}
+
+	private void loadClass(ClassEntry classEntry, Runnable callback, boolean forceDecomp) {
 		ClassEntry targetClass = classEntry.getOutermostClass();
 
-		boolean requiresDecompile = currentSource == null || !currentSource.getEntry().equals(targetClass);
+		boolean requiresDecompile = forceDecomp || currentSource == null || !currentSource.getEntry().equals(targetClass);
 		if (requiresDecompile) {
+			currentSource = null; // Or the GUI may try to find a nonexistent token
 			gui.setEditorText("(decompiling...)");
 		}
 
@@ -402,7 +411,7 @@ public class GuiController {
 
 	private DecompiledClassSource decompileSource(ClassEntry targetClass) {
 		try {
-			CompilationUnit sourceTree = sourceProvider.getSources(targetClass.getFullName());
+			CompilationUnit sourceTree = (CompilationUnit) sourceProvider.getSources(targetClass.getFullName()).clone();
 			if (sourceTree == null) {
 				gui.setEditorText("Unable to find class: " + targetClass);
 				return DecompiledClassSource.text(targetClass, "Unable to find class");
@@ -410,6 +419,7 @@ public class GuiController {
 
 			DropImportAstTransform.INSTANCE.run(sourceTree);
 			DropVarModifiersAstTransform.INSTANCE.run(sourceTree);
+			new AddJavadocsAstTransform(project.getMapper()).run(sourceTree);
 
 			String sourceString = sourceProvider.writeSourceToString(sourceTree);
 
@@ -521,6 +531,25 @@ public class GuiController {
 		refreshCurrentClass(reference);
 	}
 
+	public void changeDocs(EntryReference<Entry<?>, Entry<?>> reference, String updatedDocs) {
+		changeDoc(reference.getNameableEntry(), updatedDocs);
+
+		refreshCurrentClass(reference, true);
+	}
+
+	public void changeDoc(Entry<?> obfEntry, String newDoc) {
+		EntryRemapper mapper = project.getMapper();
+		if (mapper.getDeobfMapping(obfEntry) == null) {
+			markAsDeobfuscated(obfEntry,false); // NPE
+		}
+		mapper.mapFromObf(obfEntry, mapper.getDeobfMapping(obfEntry).withDocs(newDoc), false);
+	}
+
+	public void markAsDeobfuscated(Entry<?> obfEntry, boolean renaming) {
+		EntryRemapper mapper = project.getMapper();
+		mapper.mapFromObf(obfEntry, new EntryMapping(mapper.deobfuscate(obfEntry).getName()), renaming);
+	}
+
 	public void markAsDeobfuscated(EntryReference<Entry<?>, Entry<?>> reference) {
 		EntryRemapper mapper = project.getMapper();
 		Entry<?> entry = reference.getNameableEntry();
-- 
cgit v1.2.3