fix(gui): revert contextual keywords to identifiers (#1394)

jadx-gui/src/main/java/jadx/gui/ui/codearea/JadxTokenMaker.java

@@ -1,5 +1,7 @@
 package jadx.gui.ui.codearea;
 
+import java.util.Set;
+
 import javax.swing.text.Segment;
 
 import org.fife.ui.rsyntaxtextarea.Token;
@@ -14,6 +16,8 @@ import org.slf4j.LoggerFactory;
 import jadx.api.JavaClass;
 import jadx.api.JavaNode;
 
+import static jadx.api.plugins.utils.Utils.constSet;
+
 public final class JadxTokenMaker extends JavaTokenMaker {
 	private static final Logger LOG = LoggerFactory.getLogger(JadxTokenMaker.class);
 
@@ -42,6 +46,10 @@ public final class JadxTokenMaker extends JavaTokenMaker {
 		while (current != null) {
 			if (prev != null) {
 				switch (current.getType()) {
+					case TokenTypes.RESERVED_WORD:
+						fixContextualKeyword(current);
+						break;
+
 					case TokenTypes.IDENTIFIER:
 						current = mergeLongClassNames(prev, current, false);
 						break;
@@ -56,6 +64,17 @@ public final class JadxTokenMaker extends JavaTokenMaker {
 		}
 	}
 
+	private static final Set<String> CONTEXTUAL_KEYWORDS = constSet(
+			"exports", "module", "non-sealed", "open", "opens", "permits", "provides", "record",
+			"requires", "sealed", "to", "transitive", "uses", "var", "with", "yield");
+
+	private static void fixContextualKeyword(Token token) {
+		String lexeme = token.getLexeme(); // TODO: create new string every call, better to avoid
+		if (lexeme != null && CONTEXTUAL_KEYWORDS.contains(lexeme)) {
+			token.setType(TokenTypes.IDENTIFIER);
+		}
+	}
+
 	@NotNull
 	private Token mergeLongClassNames(Token prev, Token current, boolean annotation) {
 		int offset = current.getTextOffset();