avoid duplicate logic

2 files changed, 31 insertions, 29 deletions

diff --git a/enigma/src/main/java/cuchaz/enigma/EnigmaProject.java b/enigma/src/main/java/cuchaz/enigma/EnigmaProject.java
index fcd2c96..a01eca1 100644
--- a/enigma/src/main/java/cuchaz/enigma/EnigmaProject.java
+++ b/enigma/src/main/java/cuchaz/enigma/EnigmaProject.java
@@ -18,6 +18,7 @@ import java.util.stream.Stream;
 
 import com.google.common.base.Functions;
 import com.google.common.base.Preconditions;
+import cuchaz.enigma.api.service.ObfuscationTestService;
 import cuchaz.enigma.classprovider.ObfuscationFixClassProvider;
 import org.objectweb.asm.ClassWriter;
 import org.objectweb.asm.tree.ClassNode;
@@ -158,6 +159,35 @@ public class EnigmaProject {
                 return obfReference.isNamed() && isRenamable(obfReference.getNameableEntry());
         }
 
+        public boolean isObfuscated(Entry<?> entry) {
+                String name = entry.getName();
+
+                List<ObfuscationTestService> obfuscationTestServices = this.getEnigma().getServices().get(ObfuscationTestService.TYPE);
+                if (!obfuscationTestServices.isEmpty()) {
+                        for (ObfuscationTestService service : obfuscationTestServices) {
+                                if (service.testDeobfuscated(entry)) {
+                                        return false;
+                                }
+                        }
+                }
+
+                List<NameProposalService> nameProposalServices = this.getEnigma().getServices().get(NameProposalService.TYPE);
+                if (!nameProposalServices.isEmpty()) {
+                        for (NameProposalService service : nameProposalServices) {
+                                if (service.proposeName(entry, mapper).isPresent()) {
+                                        return false;
+                                }
+                        }
+                }
+
+                String mappedName = mapper.deobfuscate(entry).getName();
+                if (mappedName != null && !mappedName.isEmpty() && !mappedName.equals(name)) {
+                        return false;
+                }
+
+                return true;
+        }
+
         public JarExport exportRemappedJar(ProgressListener progress) {
                 Collection<ClassEntry> classEntries = jarIndex.getEntryIndex().getClasses();
                 ClassProvider fixingClassProvider = new ObfuscationFixClassProvider(classProvider, jarIndex);
diff --git a/enigma/src/main/java/cuchaz/enigma/analysis/StructureTreeNode.java b/enigma/src/main/java/cuchaz/enigma/analysis/StructureTreeNode.java
index 876b5ca..f310aa7 100644
--- a/enigma/src/main/java/cuchaz/enigma/analysis/StructureTreeNode.java
+++ b/enigma/src/main/java/cuchaz/enigma/analysis/StructureTreeNode.java
@@ -2,7 +2,6 @@ package cuchaz.enigma.analysis;
 
 import cuchaz.enigma.EnigmaProject;
 import cuchaz.enigma.api.service.NameProposalService;
-import cuchaz.enigma.api.service.ObfuscationTestService;
 import cuchaz.enigma.translation.TranslateResult;
 import cuchaz.enigma.translation.mapping.EntryRemapper;
 import cuchaz.enigma.translation.representation.TypeDescriptor;
@@ -43,7 +42,7 @@ public class StructureTreeNode extends DefaultMutableTreeNode {
             }
 
             // don't add deobfuscated members if hideDeobfuscated is true, unless it's an inner class
-            if (hideDeobfuscated && this.isDeobfuscated(project, child) && !(child instanceof ClassEntry)) {
+            if (hideDeobfuscated && !project.isObfuscated(child) && !(child instanceof ClassEntry)) {
                 continue;
             }
 
@@ -56,33 +55,6 @@ public class StructureTreeNode extends DefaultMutableTreeNode {
         }
     }
 
-    private boolean isDeobfuscated(EnigmaProject project, ParentedEntry child) {
-        List<ObfuscationTestService> obfuscationTestServices = project.getEnigma().getServices().get(ObfuscationTestService.TYPE);
-
-        if (!obfuscationTestServices.isEmpty()) {
-            for (ObfuscationTestService service : obfuscationTestServices) {
-                if (service.testDeobfuscated(child)) {
-                    return true;
-                }
-            }
-        }
-
-        if (!this.nameProposalServices.isEmpty()) {
-            for (NameProposalService service : this.nameProposalServices) {
-                if (service.proposeName(child, this.mapper).isPresent()) {
-                    return true;
-                }
-            }
-        }
-
-        String mappedName = project.getMapper().deobfuscate(child).getName();
-        if (mappedName != null && !mappedName.isEmpty() && !mappedName.equals(child.getName())) {
-            return true;
-        }
-
-        return false;
-    }
-
     @Override
     public String toString() {
         TranslateResult<ParentedEntry> translateResult = this.mapper.extendedDeobfuscate(this.entry);