src/main/java/cuchaz/enigma/ProposingTranslator.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

package cuchaz.enigma;

import cuchaz.enigma.api.service.NameProposalService;
import cuchaz.enigma.translation.Translatable;
import cuchaz.enigma.translation.Translator;
import cuchaz.enigma.translation.mapping.*;
import cuchaz.enigma.translation.representation.entry.Entry;

import java.util.Optional;

public class ProposingTranslator implements Translator {
    private final EntryRemapper mapper;
    private final NameProposalService nameProposalService;

    public ProposingTranslator(EntryRemapper mapper, NameProposalService nameProposalService) {
        this.mapper = mapper;
        this.nameProposalService = nameProposalService;
    }

    @Override
    @SuppressWarnings("unchecked")
    public <T extends Translatable> T translate(T translatable) {
        if (translatable == null) {
            return null;
        }

        T deobfuscated = mapper.deobfuscate(translatable);

        if (translatable instanceof Entry && ((Entry) deobfuscated).getName().equals(((Entry<?>) translatable).getName())) {
            return mapper.getObfResolver()
                    .resolveEntry((Entry<?>) translatable, ResolutionStrategy.RESOLVE_ROOT)
                    .stream()
                    .map(e1 -> nameProposalService.proposeName(e1, mapper))
                    .filter(Optional::isPresent)
                    .map(Optional::get)
                    .findFirst()
                    .map(newName -> (T) ((Entry) deobfuscated).withName(newName))
                    .orElse(deobfuscated);
        }

        return deobfuscated;
    }
}