BuckSyntaxHighlighter.java example

Explorer
buck-master
/*
 * Copyright 2015-present Facebook, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may
 * not use this file except in compliance with the License. You may obtain
 * a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */

package com.facebook.buck.intellij.ideabuck.highlight;

import static com.intellij.openapi.editor.colors.TextAttributesKey.createTextAttributesKey;

import com.facebook.buck.intellij.ideabuck.lang.BuckLexerAdapter;
import com.facebook.buck.intellij.ideabuck.lang.psi.BuckTypes;
import com.intellij.lexer.Lexer;
import com.intellij.openapi.editor.DefaultLanguageHighlighterColors;
import com.intellij.openapi.editor.HighlighterColors;
import com.intellij.openapi.editor.colors.CodeInsightColors;
import com.intellij.openapi.editor.colors.TextAttributesKey;
import com.intellij.openapi.fileTypes.SyntaxHighlighterBase;
import com.intellij.psi.TokenType;
import com.intellij.psi.tree.IElementType;

/** Syntax highlighter for buck PSI elements. */
public class BuckSyntaxHighlighter extends SyntaxHighlighterBase {

  public static final TextAttributesKey BUCK_KEYWORD =
      createTextAttributesKey("BUCK_KEY", DefaultLanguageHighlighterColors.KEYWORD);
  public static final TextAttributesKey BUCK_NUMBER =
      createTextAttributesKey("BUCK_NUMBER", DefaultLanguageHighlighterColors.NUMBER);
  public static final TextAttributesKey BUCK_COMMENT =
      createTextAttributesKey("BUCK_COMMENT", DefaultLanguageHighlighterColors.LINE_COMMENT);
  public static final TextAttributesKey BUCK_RULE_NAME =
      createTextAttributesKey(
          "BUCK_RULE_NAME", DefaultLanguageHighlighterColors.FUNCTION_DECLARATION);
  public static final TextAttributesKey BUCK_GLOB =
      createTextAttributesKey("BUCK_GLOB", CodeInsightColors.ANNOTATION_NAME_ATTRIBUTES);
  public static final TextAttributesKey FUNCTION_CALL =
      createTextAttributesKey("FUNCTION_CALL", CodeInsightColors.ANNOTATION_NAME_ATTRIBUTES);
  public static final TextAttributesKey BUCK_STRING =
      createTextAttributesKey("BUCK_STRING", DefaultLanguageHighlighterColors.STRING);
  public static final TextAttributesKey BUCK_MACRO =
      createTextAttributesKey("BUCK_MACRO", DefaultLanguageHighlighterColors.STATIC_FIELD);

  private static final TextAttributesKey[] BAD_CHAR_KEYS =
      new TextAttributesKey[] {HighlighterColors.BAD_CHARACTER};
  private static final TextAttributesKey[] KEY_KEYS = new TextAttributesKey[] {BUCK_KEYWORD};
  private static final TextAttributesKey[] NUMBER_KEYS = new TextAttributesKey[] {BUCK_NUMBER};
  private static final TextAttributesKey[] GLOB_KEYS = new TextAttributesKey[] {BUCK_GLOB};
  private static final TextAttributesKey[] FUNCTION_CALL_KEY =
      new TextAttributesKey[] {FUNCTION_CALL};
  private static final TextAttributesKey[] COMMENT_KEYS = new TextAttributesKey[] {BUCK_COMMENT};
  private static final TextAttributesKey[] STRING_KEYS = new TextAttributesKey[] {BUCK_STRING};
  private static final TextAttributesKey[] MACROS_KEYS = new TextAttributesKey[] {BUCK_MACRO};
  private static final TextAttributesKey[] EMPTY_KEYS = new TextAttributesKey[0];

  @Override
  public Lexer getHighlightingLexer() {
    return new BuckLexerAdapter();
  }

  @Override
  public TextAttributesKey[] getTokenHighlights(IElementType tokenType) {
    if (tokenType.equals(BuckTypes.DOUBLE_QUOTED_STRING)
        || tokenType.equals(BuckTypes.SINGLE_QUOTED_STRING)) {
      return STRING_KEYS;
    } else if (tokenType.equals(BuckTypes.GLOB_KEYWORD)
        || tokenType.equals(BuckTypes.GLOB_EXCLUDES_KEYWORD)) {
      return GLOB_KEYS;
    } else if (tokenType.equals(BuckTypes.BOOLEAN) || tokenType.equals(BuckTypes.NONE)) {
      return KEY_KEYS;
    } else if (tokenType.equals(BuckTypes.NUMBER)) {
      return NUMBER_KEYS;
    } else if (tokenType.equals(BuckTypes.LINE_COMMENT)) {
      return COMMENT_KEYS;
    } else if (tokenType.equals(BuckTypes.MACROS)) {
      return MACROS_KEYS;
    } else if (tokenType.equals(BuckTypes.FUNCTION_CALL)) {
      return FUNCTION_CALL_KEY;
    } else if (tokenType.equals(TokenType.BAD_CHARACTER)) {
      return BAD_CHAR_KEYS;
    } else {
      return EMPTY_KEYS;
    }
  }
}