add golang support

swr1bm86 · Mar 6, 2016 · 89c97d7 · 89c97d7
1 parent bd862da
commit 89c97d7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/META-INF/plugin.xml b/META-INF/plugin.xml
@@ -54,6 +54,8 @@
     <annotator language="JavaScript" implementationClass="zjhmale.rainbow.RainbowHighliter"/>
     <annotator language="Erlang" implementationClass="zjhmale.rainbow.RainbowHighliter"/>
     <annotator language="Scala" implementationClass="zjhmale.rainbow.RainbowHighliter"/>
+    <annotator language="go" implementationClass="zjhmale.rainbow.RainbowHighliter"/>
+
 
     <applicationConfigurable implementation="zjhmale.rainbow.setting.RainbowConfigurable"/>
     <applicationService serviceInterface="zjhmale.rainbow.setting.RainbowSettings"

diff --git a/src/zjhmale/rainbow/RainbowHighliter.java b/src/zjhmale/rainbow/RainbowHighliter.java
@@ -147,6 +147,9 @@ public void annotate(@NotNull PsiElement element, @NotNull AnnotationHolder hold
                     && !t.startsWith("//")
                     && !(t.startsWith("/*") && t.endsWith("*/"))
                     && !scalaDocTokens.contains(type.toString());
+            boolean goPredicate = languageID.equals("go")
+                    && !t.startsWith("//")
+                    && !(t.startsWith("/*") && t.endsWith("*/"));
 
             boolean isParentheses = delimitersList.contains(t);
             boolean isString = (t.startsWith("\"") && t.endsWith("\"")) || (t.startsWith("\'") && t.endsWith("\'"));
@@ -159,7 +162,8 @@ public void annotate(@NotNull PsiElement element, @NotNull AnnotationHolder hold
                     || rustPredicate
                     || jsPredicate
                     || erlangPredicate
-                    || scalaPredicate)
+                    || scalaPredicate
+                    || goPredicate)
                     && !isParentheses
                     && !isString) {
                 TextAttributes attrs = getIdentifierAttributes(t, backgroundColor);