[analyzer] init config/system paths in bundle mode

GitOrigin-RevId: 5cd7ad3a6f3bbf5e66b9624cc34810598604fc69

Author: gregsh

kotlin-lsp/src/com/jetbrains/ls/kotlinLsp/KotlinLspServer.kt

@@ -2,6 +2,7 @@
 package com.jetbrains.ls.kotlinLsp
 
 import com.intellij.openapi.application.PathManager
+import com.intellij.openapi.diagnostic.fileLogger
 import com.jetbrains.ls.api.core.LSServer
 import com.jetbrains.ls.api.core.LSServerContext
 import com.jetbrains.ls.api.features.LSConfiguration
@@ -27,7 +28,9 @@ import org.jetbrains.kotlin.idea.base.plugin.artifacts.KotlinArtifacts
 import org.jetbrains.kotlin.idea.compiler.configuration.KotlinPluginLayoutMode
 import org.jetbrains.kotlin.idea.compiler.configuration.KotlinPluginLayoutModeProvider
 import org.jetbrains.kotlin.idea.compiler.configuration.isRunningFromSources
+import java.nio.file.Path
 import kotlin.io.path.absolutePathString
+import kotlin.io.path.createDirectories
 import kotlin.io.path.createTempDirectory
 import kotlin.system.exitProcess
 
@@ -45,10 +48,13 @@ fun main(args: Array<String>) {
     }
 }
 
+// use after `initKotlinLspLogger` call
+private val LOG by lazy { fileLogger() }
+
 private fun run(runConfig: KotlinLspServerRunConfig) {
     val mode = runConfig.mode
     initKotlinLspLogger(writeToStdOut = mode != KotlinLspServerMode.Stdio)
-    initIdeaPaths()
+    initIdeaPaths(runConfig.systemPath)
     setLspKotlinPluginModeIfRunningFromProductionLsp()
     val config = createConfiguration()
 
@@ -112,17 +118,21 @@ private suspend fun handleRequests(connection: LspConnection, runConfig: KotlinL
     }
 }
 
-private fun initIdeaPaths() {
+private fun initIdeaPaths(systemPath: Path?) {
     val fromSources = getIJPathIfRunningFromSources()
     if (fromSources != null) {
         System.setProperty("idea.home.path", fromSources)
         System.setProperty("idea.config.path", "$fromSources/config/idea")
         System.setProperty("idea.system.path", "$fromSources/system/idea")
     }
     else {
-        val tmp = createTempDirectory("idea-home").absolutePathString()
-        System.setProperty("idea.home.path", tmp)
+        val path = systemPath?.createDirectories() ?: createTempDirectory("idea-system")
+        System.setProperty("idea.home.path", "$path")
+        System.setProperty("idea.config.path", "$path/config")
+        System.setProperty("idea.system.path", "$path/system")
     }
+    LOG.info("idea.config.path=${System.getProperty("idea.config.path")}")
+    LOG.info("idea.system.path=${System.getProperty("idea.system.path")}")
 }
 
 private fun setLspKotlinPluginModeIfRunningFromProductionLsp() {

kotlin-lsp/src/com/jetbrains/ls/kotlinLsp/KotlinLspServerRunConfig.kt

@@ -2,16 +2,15 @@ package com.jetbrains.ls.kotlinLsp
 
 import com.github.ajalt.clikt.core.CliktCommand
 import com.github.ajalt.clikt.core.PrintHelpMessage
-import com.github.ajalt.clikt.parameters.options.default
-import com.github.ajalt.clikt.parameters.options.flag
-import com.github.ajalt.clikt.parameters.options.help
-import com.github.ajalt.clikt.parameters.options.option
-import com.github.ajalt.clikt.parameters.options.validate
+import com.github.ajalt.clikt.parameters.options.*
 import com.github.ajalt.clikt.parameters.types.int
+import com.github.ajalt.clikt.parameters.types.path
 import com.jetbrains.lsp.implementation.TcpConnectionConfig
+import java.nio.file.Path
 
 data class KotlinLspServerRunConfig(
-    val mode: KotlinLspServerMode
+    val mode: KotlinLspServerMode,
+    val systemPath: Path? = null,
 )
 
 sealed interface KotlinLspServerMode {
@@ -37,12 +36,15 @@ sealed class KotlinLspCommand {
 }
 
 private class Parser : CliktCommand(name = "kotlin-lsp") {
-    val socket: Int by option().int().default(9999).help("A port which will be used for a Kotlin LSP connection. Default is 9999")
+    val socket: Int by option().int().default(9999)
+        .help("A port which will be used for a Kotlin LSP connection. Default is 9999")
     val stdio: Boolean by option().flag()
         .help("Whether the Kotlin LSP server is used in stdio mode. If not set, server mode will be used with a port specified by `${::socket.name}`")
     val client: Boolean by option().flag()
         .help("Whether the Kotlin LSP server is used in client mode. If not set, server mode will be used with a port specified by `${::socket.name}`")
         .validate { if (it && stdio) fail("Can't use stdio mode with client mode") }
+    val systemPath: Path? by option().path()
+        .help("Path for Kotlin LSP caches and indexes")
 
     val multiclient: Boolean by option().flag()
         .help("Whether the Kotlin LSP server is used in multiclient mode. If not set, server will be shut down after the first client disconnects.`")
@@ -58,7 +60,7 @@ private class Parser : CliktCommand(name = "kotlin-lsp") {
             client -> KotlinLspServerMode.Socket(TcpConnectionConfig.Client(port = socket))
             else -> KotlinLspServerMode.Socket(TcpConnectionConfig.Server(port = socket, isMulticlient = multiclient))
         }
-        return KotlinLspServerRunConfig(mode)
+        return KotlinLspServerRunConfig(mode, systemPath)
     }
 
     override fun run() {}

kotlin-vscode/client/src/lspClient.ts

@@ -183,6 +183,7 @@ async function getRunningJavaServerLspOptions(): Promise<ServerOptions | null> {
     args.push(
         '-classpath', extractPath + path.sep + '*',
         'com.jetbrains.ls.kotlinLsp.KotlinLspServerKt', '--client',
+        '--system-path', getContext().globalStorageUri.fsPath,
     );
     if (runWithJavaSupport()) {
         args.push('--with-java')