add setting for disable lsp completion

onriv · Oct 13, 2024 · af5fe66 · af5fe66
1 parent f925504
commit af5fe66
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/src/main/kotlin/lean4ij/lsp/LeanLSPCompletionFeature.kt b/src/main/kotlin/lean4ij/lsp/LeanLSPCompletionFeature.kt
@@ -0,0 +1,18 @@
+package lean4ij.lsp
+
+import com.intellij.openapi.components.service
+import com.intellij.psi.PsiFile
+import com.redhat.devtools.lsp4ij.client.features.LSPCompletionFeature
+import lean4ij.Lean4Settings
+
+/**
+ * Add an impl for disable lsp completion for lean
+ * for sometimes it's slow...
+ */
+class LeanLSPCompletionFeature : LSPCompletionFeature() {
+    private val lean4Settings = service<Lean4Settings>()
+
+    override fun isEnabled(file: PsiFile): Boolean {
+        return lean4Settings.enableLspCompletion
+    }
+}
diff --git a/src/main/kotlin/lean4ij/lsp/LeanLanguageServerFactory.kt b/src/main/kotlin/lean4ij/lsp/LeanLanguageServerFactory.kt
@@ -5,7 +5,6 @@ import com.redhat.devtools.lsp4ij.LanguageServerEnablementSupport
 import com.redhat.devtools.lsp4ij.LanguageServerFactory
 import com.redhat.devtools.lsp4ij.client.LanguageClientImpl
 import com.redhat.devtools.lsp4ij.client.features.LSPClientFeatures
-import com.redhat.devtools.lsp4ij.client.features.LSPCompletionFeature
 import com.redhat.devtools.lsp4ij.server.StreamConnectionProvider
 import org.eclipse.lsp4j.services.LanguageServer
 import java.util.concurrent.atomic.AtomicBoolean
@@ -51,4 +50,11 @@ class LeanLanguageServerFactory : LanguageServerFactory, LanguageServerEnablemen
     override fun getServerInterface(): Class<out LanguageServer> {
         return InternalLeanLanguageServer::class.java
     }
-}
+
+    override fun createClientFeatures(): LSPClientFeatures {
+        return LSPClientFeatures().apply {
+            completionFeature = LeanLSPCompletionFeature()
+        }
+    }
+}
+