Use .kts for in-memory script source files

Author: fwcd

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

@@ -15,7 +15,7 @@ import org.jetbrains.kotlin.psi.psiUtil.parentsWithSelf
 import org.jetbrains.kotlin.resolve.BindingContext
 import org.jetbrains.kotlin.resolve.scopes.LexicalScope
 import org.jetbrains.kotlin.types.KotlinType
-import java.nio.file.Path
+import java.nio.file.Paths
 
 class CompiledFile(
     val content: String,
@@ -24,6 +24,7 @@ class CompiledFile(
     val container: ComponentProvider,
     val sourcePath: Collection<KtFile>,
     val classPath: CompilerClassPath,
+    val isScript: Boolean = false,
     val kind: CompilationKind = CompilationKind.DEFAULT
 ) {
     /**
@@ -122,7 +123,7 @@ class CompiledFile(
         }
 
         val padOffset = " ".repeat(offset)
-        val recompile = classPath.compiler.createFile(padOffset + surroundingContent)
+        val recompile = classPath.compiler.createFile(padOffset + surroundingContent, Paths.get(if (isScript) "dummy.virtual.kts" else "dummy.virtual.kt"))
         return recompile.findElementAt(cursor)?.findParent<KtElement>()
     }
 

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

@@ -26,6 +26,7 @@ class SourcePath(
         var compiledContainer: ComponentProvider? = null,
         val isTemporary: Boolean = false // A temporary source file will not be returned by .all()
     ) {
+        val isScript: Boolean = uri.toString().endsWith(".kts")
         val kind: CompilationKind =
             if (path?.fileName?.toString()?.endsWith(".gradle.kts") ?: false) CompilationKind.BUILD_SCRIPT
             else CompilationKind.DEFAULT
@@ -71,7 +72,7 @@ class SourcePath(
                 parseIfChanged().compileIfNull().doPrepareCompiledFile()
 
         private fun doPrepareCompiledFile(): CompiledFile =
-                CompiledFile(content, compiledFile!!, compiledContext!!, compiledContainer!!, allIncludingThis(), cp, kind)
+                CompiledFile(content, compiledFile!!, compiledContext!!, compiledContainer!!, allIncludingThis(), cp, isScript, kind)
 
         private fun allIncludingThis(): Collection<KtFile> = parseIfChanged().let {
             if (isTemporary) (all().asSequence() + sequenceOf(parsed!!)).toList()