Color unobfuscated tokens in blue (#569)

2 files changed, 14 insertions, 2 deletions

diff --git a/enigma/src/main/java/cuchaz/enigma/source/DecompiledClassSource.java b/enigma/src/main/java/cuchaz/enigma/source/DecompiledClassSource.java
index 4f78e9c..69692db 100644
--- a/enigma/src/main/java/cuchaz/enigma/source/DecompiledClassSource.java
+++ b/enigma/src/main/java/cuchaz/enigma/source/DecompiledClassSource.java
@@ -12,6 +12,7 @@ import cuchaz.enigma.EnigmaProject;
 import cuchaz.enigma.EnigmaServices;
 import cuchaz.enigma.analysis.EntryReference;
 import cuchaz.enigma.api.service.NameProposalService;
+import cuchaz.enigma.api.service.ObfuscationTestService;
 import cuchaz.enigma.translation.LocalNameGenerator;
 import cuchaz.enigma.translation.TranslateResult;
 import cuchaz.enigma.translation.Translator;
@@ -72,7 +73,11 @@ public class DecompiledClassSource {
                                         return proposedName.get();
                                 }
 
-                                target.add(RenamableTokenType.OBFUSCATED, movedToken);
+                                if (isUnobfuscated(project, entry)) {
+                                        target.add(RenamableTokenType.UNOBFUSCATED, movedToken);
+                                } else {
+                                        target.add(RenamableTokenType.OBFUSCATED, movedToken);
+                                }
                         }
                 }
 
@@ -96,6 +101,12 @@ public class DecompiledClassSource {
                 }).findFirst();
         }
 
+        private static boolean isUnobfuscated(EnigmaProject project, Entry<?> entry) {
+                EnigmaServices services = project.getEnigma().getServices();
+
+                return services.get(ObfuscationTestService.TYPE).stream().anyMatch(service -> service.testDeobfuscated(entry));
+        }
+
         @Nullable
         private String generateDefaultName(Entry<?> entry) {
                 if (entry instanceof LocalVariableDefEntry) {
diff --git a/enigma/src/main/java/cuchaz/enigma/source/RenamableTokenType.java b/enigma/src/main/java/cuchaz/enigma/source/RenamableTokenType.java
index c63aad9..aad3ada 100644
--- a/enigma/src/main/java/cuchaz/enigma/source/RenamableTokenType.java
+++ b/enigma/src/main/java/cuchaz/enigma/source/RenamableTokenType.java
@@ -3,5 +3,6 @@ package cuchaz.enigma.source;
 public enum RenamableTokenType {
         OBFUSCATED,
         DEOBFUSCATED,
-        PROPOSED
+        PROPOSED,
+        UNOBFUSCATED,
 }