Remove error listeners from highlighting lexer

fixes #111
Kotlin · Jul 2, 2022 · 03ddf99 · 03ddf99
1 parent 7a96077
commit 03ddf99
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/ki-shell/src/main/kotlin/org/jetbrains/kotlinx/ki/shell/plugins/KotlinHighlighter.kt b/ki-shell/src/main/kotlin/org/jetbrains/kotlinx/ki/shell/plugins/KotlinHighlighter.kt
@@ -35,7 +35,6 @@ class KotlinHighlighter(private val styles: SyntaxPlugin.HighlightStyles): BaseH
             try {
                 with(KotlinParserForHighlighting(code)) {
                     addParseListener(listener)
-                    removeErrorListeners()
                     script()
                 }
             } catch (e: Throwable) {
@@ -68,10 +67,17 @@ class KotlinHighlighter(private val styles: SyntaxPlugin.HighlightStyles): BaseH
     override fun setErrorIndex(p0: Int) {}
 
     private class KotlinParserForHighlighting(code: String) :
-            KotlinParser(CommonTokenStream(KotlinLexer(CharStreams.fromString(code))))
+            KotlinParser(
+                CommonTokenStream(
+                    KotlinLexer(CharStreams.fromString(code)).also {
+                        it.removeErrorListeners()
+                    }
+                )
+            )
     {
         init {
             _buildParseTrees = false
+            removeErrorListeners()
         }
     }
 }