Make kts support configurable and disable it by default

Author: fwcd

server/src/main/kotlin/org/javacs/kt/Configuration.kt

@@ -30,6 +30,13 @@ public data class DiagnosticsConfiguration(
     var debounceTime: Long = 250L
 )
 
+public data class ScriptsConfiguration(
+    /** Whether .kts scripts are handled. */
+    var enabled: Boolean = false,
+    /** Whether .gradle.kts scripts are handled. Only considered if scripts are enabled in general. */
+    var buildScriptsEnabled: Boolean = false
+)
+
 public data class JVMConfiguration(
     /** Which JVM target the Kotlin compiler uses. See Compiler.jvmTargetFrom for possible values. */
     var target: String = "default"
@@ -91,6 +98,7 @@ public data class Configuration(
     val compiler: CompilerConfiguration = CompilerConfiguration(),
     val completion: CompletionConfiguration = CompletionConfiguration(),
     val diagnostics: DiagnosticsConfiguration = DiagnosticsConfiguration(),
+    val scripts: ScriptsConfiguration = ScriptsConfiguration(),
     val indexing: IndexingConfiguration = IndexingConfiguration(),
     val externalSources: ExternalSourcesConfiguration = ExternalSourcesConfiguration(),
     val hints: InlayHintsConfiguration = InlayHintsConfiguration()

server/src/main/kotlin/org/javacs/kt/KotlinLanguageServer.kt

@@ -30,7 +30,7 @@ class KotlinLanguageServer : LanguageServer, LanguageClientAware, Closeable {
     private val tempDirectory = TemporaryDirectory()
     private val uriContentProvider = URIContentProvider(ClassContentProvider(config.externalSources, classPath, tempDirectory, CompositeSourceArchiveProvider(JdkSourceArchiveProvider(classPath), ClassPathSourceArchiveProvider(classPath))))
     val sourcePath = SourcePath(classPath, uriContentProvider, config.indexing, databaseService)
-    val sourceFiles = SourceFiles(sourcePath, uriContentProvider)
+    val sourceFiles = SourceFiles(sourcePath, uriContentProvider, config.scripts)
 
     private val textDocuments = KotlinTextDocumentService(sourceFiles, sourcePath, config, tempDirectory, uriContentProvider, classPath)
     private val workspaces = KotlinWorkspaceService(sourceFiles, sourcePath, classPath, textDocuments, config)

server/src/main/kotlin/org/javacs/kt/SourceFiles.kt

@@ -65,7 +65,8 @@ private class NotifySourcePath(private val sp: SourcePath) {
  */
 class SourceFiles(
     private val sp: SourcePath,
-    private val contentProvider: URIContentProvider
+    private val contentProvider: URIContentProvider,
+    private val scriptsConfig: ScriptsConfiguration
 ) {
     private val workspaceRoots = mutableSetOf<Path>()
     private var exclusions = SourceExclusions(workspaceRoots)
@@ -175,6 +176,19 @@ class SourceFiles(
         updateExclusions()
     }
 
+    private fun findSourceFiles(root: Path): Set<URI> {
+        val glob = if (scriptsConfig.enabled) "*.{kt,kts}" else "*.kt"
+        val sourceMatcher = FileSystems.getDefault().getPathMatcher("glob:$glob")
+        return SourceExclusions(root)
+            .walkIncluded()
+            .filter {
+                sourceMatcher.matches(it.fileName)
+                && (scriptsConfig.buildScriptsEnabled || !it.endsWith(".gradle.kts"))
+            }
+            .map(Path::toUri)
+            .toSet()
+    }
+
     private fun updateExclusions() {
         exclusions = SourceExclusions(workspaceRoots)
     }
@@ -222,15 +236,6 @@ private fun patch(sourceText: String, change: TextDocumentContentChangeEvent): S
     }
 }
 
-private fun findSourceFiles(root: Path): Set<URI> {
-    val sourceMatcher = FileSystems.getDefault().getPathMatcher("glob:*.{kt,kts}")
-    return SourceExclusions(root)
-        .walkIncluded()
-        .filter { sourceMatcher.matches(it.fileName) }
-        .map(Path::toUri)
-        .toSet()
-}
-
 private fun logAdded(sources: Collection<URI>, rootPath: Path?) {
     LOG.info("Adding {} under {} to source path", describeURIs(sources), rootPath)
 }