Done.

devkt-plugins · Apr 18, 2018 · 9a8e5f8 · 9a8e5f8
1 parent 0455667
commit 9a8e5f8
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/com/tang/intellij/devkt/lua/lua-devkt.kt b/src/com/tang/intellij/devkt/lua/lua-devkt.kt
@@ -44,10 +44,14 @@ class EmmyLua<T> : ExtendedDevKtLanguage<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)
-				} }
+			is LuaLocalDef -> localVar(element, document, colorScheme)
+		}
+	}
+
+	private fun localVar(element: LuaLocalDef, document: AnnotationHolder<T>, colorScheme: ColorScheme<T>) {
+		element.nameList?.run {
+			nameDefList.forEach {
+				document.highlight(it, colorScheme.variable)
 			}
 		}
 	}