Use Java SDK defined on JAVA_HOME by default on JetBrains IDEs

Author: felladrin

components/ide/jetbrains/backend-plugin/src/main/kotlin/io/gitpod/jetbrains/remote/GitpodProjectManager.kt

@@ -10,17 +10,20 @@ import com.intellij.openapi.module.Module
 import com.intellij.openapi.module.ModuleManager
 import com.intellij.openapi.project.ModuleListener
 import com.intellij.openapi.project.Project
-import com.intellij.openapi.projectRoots.ProjectJdkTable
-import com.intellij.openapi.projectRoots.Sdk
+import com.intellij.openapi.projectRoots.*
+import com.intellij.openapi.projectRoots.impl.SdkConfigurationUtil
 import com.intellij.openapi.roots.ModuleRootModificationUtil
 import com.intellij.openapi.roots.ProjectRootManager
+import com.intellij.openapi.vfs.VirtualFileManager
 import com.intellij.util.application
 import kotlinx.coroutines.GlobalScope
 import kotlinx.coroutines.future.await
 import kotlinx.coroutines.launch
+import java.nio.file.Path
 import java.util.concurrent.CompletableFuture
 
 
+@Suppress("UnstableApiUsage", "OPT_IN_USAGE")
 class GitpodProjectManager(
         private val project: Project
 ) {
@@ -29,20 +32,39 @@ class GitpodProjectManager(
         configureSdks()
     }
 
-    /**
-     * It is a workaround for https://youtrack.jetbrains.com/issue/GTW-88
-     */
     private fun configureSdks() {
         if (application.isHeadlessEnvironment) {
             return
         }
         val pendingSdk = CompletableFuture<Sdk>()
         application.invokeLaterOnWriteThread {
+            application.invokeAndWait { ProjectJdkTable.getInstance().preconfigure() }
+
             application.runWriteAction {
+                val jdkTable = ProjectJdkTable.getInstance()
+
                 try {
-                    ProjectJdkTable.getInstance().preconfigure()
-                    pendingSdk.complete(ProjectJdkTable.getInstance().allJdks.firstOrNull())
+                    var sdk: Sdk? = null
+                    val javaHome = System.getenv("JAVA_HOME")
+                    val sdkType = SdkType.findByName(jdkTable.defaultSdkType.name)
+
+                    if (javaHome !== null && sdkType != null && JdkUtil.checkForJdk(javaHome)) {
+                        val javaHomeVirtualFile = VirtualFileManager.getInstance().findFileByNioPath(Path.of(javaHome))
+
+                        if (javaHomeVirtualFile != null) {
+                            sdk = SdkConfigurationUtil.setupSdk(jdkTable.allJdks, javaHomeVirtualFile, sdkType, false, null, null)
+                        }
+                    }
+
+                    if (sdk != null && jdkTable.findJdk(sdk.name) == null) {
+                        jdkTable.addJdk(sdk)
+                        pendingSdk.complete(sdk)
+                    } else {
+                        jdkTable.preconfigure()
+                        pendingSdk.complete(jdkTable.allJdks.firstOrNull())
+                    }
                 } catch (t: Throwable) {
+                    jdkTable.preconfigure()
                     pendingSdk.completeExceptionally(t)
                 }
             }
@@ -58,24 +80,23 @@ class GitpodProjectManager(
             configureSdk(sdk)
         }
     }
-
     private fun configureSdk(sdk: Sdk) {
         application.invokeLaterOnWriteThread {
             application.runWriteAction {
                 val projectRootManager = ProjectRootManager.getInstance(project)
-                if (projectRootManager.projectSdk == null) {
+                if (projectRootManager.projectSdk != sdk) {
                     projectRootManager.projectSdk = sdk
                     thisLogger().warn("gitpod: '${project.name}' project: SDK was auto preconfigured: $sdk")
                 }
             }
         }
         for (module in ModuleManager.getInstance(project).modules) {
             ModuleRootModificationUtil.updateModel(module) { m ->
-                if (m.sdk == null) {
+                if (m.sdk != sdk) {
                     m.sdk = sdk
                     thisLogger().warn("gitpod: '${module.name}' module: SDK was auto preconfigured: $sdk")
                 }
             }
         }
     }
-}
+}