fix(gui): make link for full class names (#378)

2018-10-28 18:24:09 +03:00
parent a8a31643f1
commit a3464d7184
24 changed files with 602 additions and 463 deletions
@@ -0,0 +1,105 @@
+package jadx.gui.ui.codearea;
+
+import javax.swing.text.Segment;
+
+import org.fife.ui.rsyntaxtextarea.Token;
+import org.fife.ui.rsyntaxtextarea.TokenImpl;
+import org.fife.ui.rsyntaxtextarea.TokenTypes;
+import org.fife.ui.rsyntaxtextarea.modes.JavaTokenMaker;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import jadx.api.JavaClass;
+import jadx.api.JavaNode;
+import jadx.gui.treemodel.JClass;
+
+public final class JadxTokenMaker extends JavaTokenMaker {
+	private static final Logger LOG = LoggerFactory.getLogger(JadxTokenMaker.class);
+
+	private final CodeArea codeArea;
+	private final JClass jCls;
+
+	public JadxTokenMaker(CodeArea codeArea, JClass jCls) {
+		this.codeArea = codeArea;
+		this.jCls = jCls;
+	}
+
+	@Override
+	public Token getTokenList(Segment text, int initialTokenType, int startOffset) {
+		Token tokens = super.getTokenList(text, initialTokenType, startOffset);
+		if (startOffset > 0 && tokens.getType() != TokenTypes.NULL) {
+			try {
+				processTokens(tokens);
+			} catch (Exception e) {
+				LOG.error("Process tokens failed for text: {}", text, e);
+			}
+		}
+		return tokens;
+	}
+
+	private void processTokens(Token tokens) {
+		Token prev = null;
+		Token current = tokens;
+		while (current != null) {
+			if (prev != null) {
+				int tokenType = current.getType();
+				if (tokenType == TokenTypes.IDENTIFIER) {
+					current = mergeLongClassNames(prev, current, false);
+				} else if (tokenType == TokenTypes.ANNOTATION) {
+					current = mergeLongClassNames(prev, current, true);
+				}
+			}
+			prev = current;
+			current = current.getNextToken();
+		}
+	}
+
+	@NotNull
+	private Token mergeLongClassNames(Token prev, Token current, boolean annotation) {
+		int offset = current.getOffset();
+		if (annotation) {
+			offset++;
+		}
+		JavaNode javaNode = codeArea.getJavaNodeAtOffset(jCls, offset);
+		if (javaNode instanceof JavaClass) {
+			String name = javaNode.getName();
+			String lexeme = current.getLexeme();
+			if (annotation && lexeme.length() > 1) {
+				lexeme = lexeme.substring(1);
+			}
+			if (!lexeme.equals(name) && javaNode.getFullName().startsWith(lexeme)) {
+				// try to replace long class name with one token
+				Token replace = concatTokensUntil(current, name);
+				if (replace != null && prev instanceof TokenImpl) {
+					TokenImpl impl = ((TokenImpl) prev);
+					impl.setNextToken(replace);
+					current = replace;
+				}
+			}
+		}
+		return current;
+	}
+
+	@Nullable
+	private Token concatTokensUntil(Token start, String endText) {
+		StringBuilder sb = new StringBuilder();
+		Token current = start;
+		while (current != null && current.getType() != TokenTypes.NULL) {
+			String text = current.getLexeme();
+			if (text != null) {
+				sb.append(text);
+				if (text.equals(endText)) {
+					char[] line = sb.toString().toCharArray();
+					TokenImpl token = new TokenImpl(line, 0, line.length - 1, start.getOffset(),
+							start.getType(), start.getLanguageIndex());
+					token.setNextToken(current.getNextToken());
+					return token;
+				}
+			}
+			current = current.getNextToken();
+		}
+		return null;
+	}
+}