diff options
| author | 2019-01-30 21:05:32 +0200 | |
|---|---|---|
| committer | 2019-01-30 21:05:32 +0200 | |
| commit | ba7a354efae7d49833c887cf147ac940c975a1fa (patch) | |
| tree | 02e14fda81dd5984e24f2df392c57c6e829fc875 /src/main/java/cuchaz/enigma/gui/SourceRemapper.java | |
| parent | Rewrite the Jenkinsfile to use the new declarative pipeline syntax, lets hope... (diff) | |
| download | enigma-fork-ba7a354efae7d49833c887cf147ac940c975a1fa.tar.gz enigma-fork-ba7a354efae7d49833c887cf147ac940c975a1fa.tar.xz enigma-fork-ba7a354efae7d49833c887cf147ac940c975a1fa.zip | |
Remap sources (#106)
* Source remapping beginnings
* Fix navigation to remapped classes
* Translate identifier info reference
* Remap local variables with default names in source
* Caching translator
* Fix lack of highlighting for first opened class
* Fix unicode variable names
* Unicode checker shouldn't be checking just alphanumeric
* Fix package tree being built from obf names
* Don't index `this` as method call for method::reference
* Apply proposed names
* Fix source export issues
* Replace unicode var names at bytecode level uniquely
* Drop imports from editor source
* Class selector fixes
* Delta keep track of base mappings to enable lookup of old names
* Optimize source remapping by remapping source with a StringBuffer instead of copying
* Bump version
Diffstat (limited to 'src/main/java/cuchaz/enigma/gui/SourceRemapper.java')
| -rw-r--r-- | src/main/java/cuchaz/enigma/gui/SourceRemapper.java | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/src/main/java/cuchaz/enigma/gui/SourceRemapper.java b/src/main/java/cuchaz/enigma/gui/SourceRemapper.java new file mode 100644 index 0000000..f38f44e --- /dev/null +++ b/src/main/java/cuchaz/enigma/gui/SourceRemapper.java | |||
| @@ -0,0 +1,64 @@ | |||
| 1 | package cuchaz.enigma.gui; | ||
| 2 | |||
| 3 | import cuchaz.enigma.analysis.Token; | ||
| 4 | |||
| 5 | import java.util.HashMap; | ||
| 6 | import java.util.Map; | ||
| 7 | |||
| 8 | public class SourceRemapper { | ||
| 9 | private final String source; | ||
| 10 | private final Iterable<Token> tokens; | ||
| 11 | |||
| 12 | public SourceRemapper(String source, Iterable<Token> tokens) { | ||
| 13 | this.source = source; | ||
| 14 | this.tokens = tokens; | ||
| 15 | } | ||
| 16 | |||
| 17 | public Result remap(Remapper remapper) { | ||
| 18 | StringBuffer remappedSource = new StringBuffer(source); | ||
| 19 | Map<Token, Token> remappedTokens = new HashMap<>(); | ||
| 20 | |||
| 21 | int accumulatedOffset = 0; | ||
| 22 | for (Token token : tokens) { | ||
| 23 | Token movedToken = token.move(accumulatedOffset); | ||
| 24 | |||
| 25 | String remappedName = remapper.remap(token, movedToken); | ||
| 26 | if (remappedName != null) { | ||
| 27 | accumulatedOffset += movedToken.getRenameOffset(remappedName); | ||
| 28 | movedToken.rename(remappedSource, remappedName); | ||
| 29 | } | ||
| 30 | |||
| 31 | if (!token.equals(movedToken)) { | ||
| 32 | remappedTokens.put(token, movedToken); | ||
| 33 | } | ||
| 34 | } | ||
| 35 | |||
| 36 | return new Result(remappedSource.toString(), remappedTokens); | ||
| 37 | } | ||
| 38 | |||
| 39 | public static class Result { | ||
| 40 | private final String remappedSource; | ||
| 41 | private final Map<Token, Token> remappedTokens; | ||
| 42 | |||
| 43 | Result(String remappedSource, Map<Token, Token> remappedTokens) { | ||
| 44 | this.remappedSource = remappedSource; | ||
| 45 | this.remappedTokens = remappedTokens; | ||
| 46 | } | ||
| 47 | |||
| 48 | public String getSource() { | ||
| 49 | return remappedSource; | ||
| 50 | } | ||
| 51 | |||
| 52 | public Token getRemappedToken(Token token) { | ||
| 53 | return remappedTokens.getOrDefault(token, token); | ||
| 54 | } | ||
| 55 | |||
| 56 | public boolean isEmpty() { | ||
| 57 | return remappedTokens.isEmpty(); | ||
| 58 | } | ||
| 59 | } | ||
| 60 | |||
| 61 | public interface Remapper { | ||
| 62 | String remap(Token token, Token movedToken); | ||
| 63 | } | ||
| 64 | } | ||