class SmaliSyntaxHighlighter extends SyntaxHighlighterBase {

  @Override
  @NotNull
  public TextAttributesKey[] getTokenHighlights(IElementType tokenType) {
    if (KEYWORD_TOKENS.contains(tokenType) || ACCESS_MODIFIER_TOKENS.contains(tokenType)) {
      return KEYWORD_ATTR_KEYS;
    }
    if (COMMENT_TOKENS.contains(tokenType)) {
      return COMMENT_ATTR_KEYS;
    }
    if (tokenType.equals(JAVA_IDENTIFIER)) {
      return JAVA_IDENTIFIER_ATTR_KEYS;
    }
    if (STRING_TOKENS.contains(tokenType)) {
      return STRING_ATTR_KEYS;
    }
    if (NUMBER_TOKENS.contains(tokenType)) {
      return NUMBER_ATTR_KEYS;
    }
    if (BRACES_TOKENS.contains(tokenType)) {
      return BRACES_ATTR_KEYS;
    }
    if (PARENTHESES_TOKENS.contains(tokenType)) {
      return PARENTHESES_ATTR_KEYS;
    }
    return EMPTY_KEYS;
  }

}