summaryrefslogtreecommitdiff
path: root/src/main/java/cuchaz/enigma/ProposingTranslator.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/cuchaz/enigma/ProposingTranslator.java')
-rw-r--r--src/main/java/cuchaz/enigma/ProposingTranslator.java53
1 files changed, 0 insertions, 53 deletions
diff --git a/src/main/java/cuchaz/enigma/ProposingTranslator.java b/src/main/java/cuchaz/enigma/ProposingTranslator.java
deleted file mode 100644
index 018fbfd..0000000
--- a/src/main/java/cuchaz/enigma/ProposingTranslator.java
+++ /dev/null
@@ -1,53 +0,0 @@
1package cuchaz.enigma;
2
3import cuchaz.enigma.api.service.NameProposalService;
4import cuchaz.enigma.translation.Translatable;
5import cuchaz.enigma.translation.Translator;
6import cuchaz.enigma.translation.mapping.EntryRemapper;
7import cuchaz.enigma.translation.mapping.ResolutionStrategy;
8import cuchaz.enigma.translation.representation.entry.Entry;
9
10import java.util.Arrays;
11import java.util.Optional;
12
13public class ProposingTranslator implements Translator {
14 private final EntryRemapper mapper;
15 private final NameProposalService[] nameProposalServices;
16
17 public ProposingTranslator(EntryRemapper mapper, NameProposalService[] nameProposalServices) {
18 this.mapper = mapper;
19 this.nameProposalServices = nameProposalServices;
20 }
21
22 @Override
23 @SuppressWarnings("unchecked")
24 public <T extends Translatable> T translate(T translatable) {
25 if (translatable == null) {
26 return null;
27 }
28
29 T deobfuscated = mapper.deobfuscate(translatable);
30
31 if (translatable instanceof Entry && ((Entry) deobfuscated).getName().equals(((Entry<?>) translatable).getName())) {
32 return mapper.getObfResolver()
33 .resolveEntry((Entry<?>) translatable, ResolutionStrategy.RESOLVE_ROOT)
34 .stream()
35 .map(this::proposeName)
36 .filter(Optional::isPresent)
37 .map(Optional::get)
38 .findFirst()
39 .map(newName -> (T) ((Entry) deobfuscated).withName(newName))
40 .orElse(deobfuscated);
41 }
42
43 return deobfuscated;
44 }
45
46 private Optional<String> proposeName(Entry<?> entry) {
47 return Arrays.stream(nameProposalServices)
48 .map(service -> service.proposeName(entry, mapper))
49 .filter(Optional::isPresent)
50 .map(Optional::get)
51 .findFirst();
52 }
53}