[ type handler ] Handle typing event in Java and Kotlin

ice1000 · Apr 20, 2018 · 527aa57 · 527aa57
1 parent 0946490
commit 527aa57
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/common/src/org/ice1000/devkt/lang/languages.kt b/common/src/org/ice1000/devkt/lang/languages.kt
@@ -13,6 +13,7 @@ import org.jetbrains.kotlin.com.intellij.openapi.fileTypes.PlainTextLanguage
 import org.jetbrains.kotlin.com.intellij.openapi.project.Project
 import org.jetbrains.kotlin.com.intellij.pom.java.LanguageLevel
 import org.jetbrains.kotlin.com.intellij.psi.PsiElement
+import org.jetbrains.kotlin.com.intellij.psi.PsiWhiteSpace
 import org.jetbrains.kotlin.idea.KotlinLanguage
 import org.jetbrains.kotlin.lexer.KotlinLexer
 import javax.swing.Icon
@@ -82,6 +83,13 @@ sealed class BuiltinDevKtLanguage<TextAttributes>(
 	override fun satisfies(fileName: String) = false
 	override val lineCommentStart = "//"
 	override val blockComment: Pair<String, String>? = "/*" to "*/"
+	override fun handleTyping(offset: Int, text: String?, element: PsiElement?, document: IDevKtDocumentHandler<TextAttributes>) {
+		element.takeIf { text == "\n" }?.run {
+			val whitespace = containingFile.findElementAt(document.startOffsetOf(document.lineOf(offset)))
+					as? PsiWhiteSpace ?: return@run null
+			super.handleTyping(offset, whitespace.text, element, document)
+		} ?: super.handleTyping(offset, text, element, document)
+	}
 }
 
 /**