@Override
    public TextAttributesKey[] getTokenHighlights(IElementType tokenType) {
        if (tokenType.equals(SdTypes.IDENTIFIER_VAL) || tokenType.equals(SdTypes.IDENTIFIER_WITH_DASH_VAL)) {
            return IDENTIFIER_KEYS;
        } else if (keyWordsSet.contains(tokenType)) {
            return KEY_KEYS;
        } else if (tokenType.equals(SdTypes.SYMBOL) || tokenType.equals(SdTypes.COMMA)) {
            return SYMBOL_KEYS;
        } else if (tokenType.equals(SdTypes.STRING_REG)) {
            return STRING_KEYS;
        } else if (tokenType.equals(SdTypes.INTEGER_REG) || tokenType.equals(SdTypes.FLOAT_REG)) {
            return NUMBER_KEYS;
        } else if (tokenType.equals(SdTypes.COMMENT)) {
            return COMMENT_KEYS;
        } else if (constantsSet.contains(tokenType)) {
            return CONSTANT_KEYS;
        } else if (tokenType.equals(TokenType.BAD_CHARACTER)) {
            return BAD_CHAR_KEYS;
        } else {
            return EMPTY_KEYS;
        }
    }
Defines the syntax highlighting of an SD file.

@author Shahar Ariel
integration/intellij/src/main/java/ai/vespa/intellij/schema/SdSyntaxHighlighter.java