Add annotator

devkt-plugins · Apr 18, 2018 · 0455667 · 0455667
1 parent 6da7b84
commit 0455667
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 4 deletions.
diff --git a/src/com/tang/intellij/devkt/lua/lang/LuaLanguage.java b/src/com/tang/intellij/devkt/lua/lang/LuaLanguage.java
@@ -16,7 +16,9 @@
 
 package com.tang.intellij.devkt.lua.lang;
 
+import org.jetbrains.annotations.NotNull;
 import org.jetbrains.kotlin.com.intellij.lang.Language;
+import org.jetbrains.kotlin.com.intellij.openapi.fileTypes.LanguageFileType;
 
 /**
  * Created by TangZhiXu on 2015/11/15.
@@ -28,6 +30,12 @@ public class LuaLanguage extends Language {
 
 	public static final LuaLanguage INSTANCE = new LuaLanguage();
 
+	@Override
+	public @NotNull
+	LanguageFileType getAssociatedFileType() {
+		return LuaFileType.INSTANCE;
+	}
+
 	public LuaLanguage() {
 		super("Lua");
 	}

diff --git a/src/com/tang/intellij/devkt/lua/lua-devkt.kt b/src/com/tang/intellij/devkt/lua/lua-devkt.kt
@@ -3,10 +3,9 @@ package com.tang.intellij.devkt.lua
 import com.tang.intellij.devkt.lua.highlighting.*
 import com.tang.intellij.devkt.lua.lang.LuaLanguage
 import com.tang.intellij.devkt.lua.lang.LuaParserDefinition
-import com.tang.intellij.devkt.lua.psi.LuaStringTypes
-import com.tang.intellij.devkt.lua.psi.LuaTypes
-import org.ice1000.devkt.openapi.ColorScheme
-import org.ice1000.devkt.openapi.ExtendedDevKtLanguage
+import com.tang.intellij.devkt.lua.psi.*
+import org.ice1000.devkt.openapi.*
+import org.jetbrains.kotlin.com.intellij.psi.PsiElement
 import org.jetbrains.kotlin.com.intellij.psi.tree.IElementType
 
 class EmmyLua<T> : ExtendedDevKtLanguage<T>(
@@ -41,5 +40,19 @@ class EmmyLua<T> : ExtendedDevKtLanguage<T>(
 		else -> super.attributesOf(type, colorScheme)
 	}
 
+	override fun annotate(element: PsiElement, document: AnnotationHolder<T>, colorScheme: ColorScheme<T>) {
+		super.annotate(element, document, colorScheme)
+		when (element) {
+			is LuaFuncDef -> function(element, document, colorScheme)
+			is LuaLocalDef -> {
+				element.nameList?.run { nameDefList.forEach {
+					document.highlight(it, colorScheme.variable)
+				} }
+			}
+		}
+	}
 
+	private fun function(element: LuaFuncDef, document: AnnotationHolder<T>, colorScheme: ColorScheme<T>) {
+		element.id?.let { document.highlight(it, colorScheme.function) }
+	}
 }