diff options
Diffstat (limited to 'src/main/java/cuchaz/enigma/ProposingTranslator.java')
| -rw-r--r-- | src/main/java/cuchaz/enigma/ProposingTranslator.java | 53 |
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 @@ | |||
| 1 | package cuchaz.enigma; | ||
| 2 | |||
| 3 | import cuchaz.enigma.api.service.NameProposalService; | ||
| 4 | import cuchaz.enigma.translation.Translatable; | ||
| 5 | import cuchaz.enigma.translation.Translator; | ||
| 6 | import cuchaz.enigma.translation.mapping.EntryRemapper; | ||
| 7 | import cuchaz.enigma.translation.mapping.ResolutionStrategy; | ||
| 8 | import cuchaz.enigma.translation.representation.entry.Entry; | ||
| 9 | |||
| 10 | import java.util.Arrays; | ||
| 11 | import java.util.Optional; | ||
| 12 | |||
| 13 | public 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 | } | ||