Handle runtime missing JCEF (#2451)

Resolves
https://linear.app/sourcegraph/issue/CODY-3753/make-it-possible-to-automatically-switch-user-runtime-to-one-which.

## Test plan 

### Dev run (`:customRunIde`)
1. Modify `build.gradle.kts`. Change
  ```
dependencies {
  intellijPlatform {
    jetbrainsRuntime()
...
  ```
  to
  ```
dependencies {
  intellijPlatform {
    jetbrainsRuntimeExplicit("jbr-21.0.3-osx-aarch64-b446.1")
...
  ```
so that the IDE will use no jcef runtime at the start. 
2. `:customRunIde`
3. Verify the notification appears
4. Verify the proper panel appears

Author: mkondratek

src/main/kotlin/com/sourcegraph/cody/CodyToolWindowContent.kt

@@ -8,9 +8,12 @@ import com.intellij.util.concurrency.annotations.RequiresEdt
 import com.intellij.util.ui.UIUtil
 import com.sourcegraph.cody.chat.SignInWithSourcegraphPanel
 import com.sourcegraph.cody.chat.ui.CodyOnboardingGuidancePanel
+import com.sourcegraph.cody.chat.ui.ErrorPanel
+import com.sourcegraph.cody.chat.ui.MissingJcefPanel
 import com.sourcegraph.cody.config.CodyAccount
 import com.sourcegraph.cody.config.CodyApplicationSettings
 import com.sourcegraph.cody.config.CodyAuthenticationManager
+import com.sourcegraph.cody.initialization.VerifyJavaBootRuntimeVersion.Companion.isCurrentRuntimeMissingJcef
 import com.sourcegraph.cody.ui.web.CodyToolWindowContentWebviewHost
 import com.sourcegraph.cody.ui.web.WebUIService
 import java.awt.CardLayout
@@ -21,7 +24,7 @@ import javax.swing.JComponent
 import javax.swing.JPanel
 
 @Service(Service.Level.PROJECT)
-class CodyToolWindowContent(project: Project) {
+class CodyToolWindowContent(val project: Project) {
   private val cardLayout = CardLayout()
   private val cardPanel = JPanel(cardLayout)
   val allContentPanel: JComponent = JPanel(GridLayout(1, 1))
@@ -37,11 +40,13 @@ class CodyToolWindowContent(project: Project) {
     cardPanel.add(codyOnboardingGuidancePanel, ONBOARDING_PANEL, ONBOARDING_PANEL_INDEX)
 
     // Because the webview may be created lazily, populate a placeholder control.
-    val placeholder = JPanel(GridBagLayout())
+    val spinnerPlaceholder = JPanel(GridBagLayout())
     val spinnerLabel =
         JBLabel("Starting Cody...", Icons.StatusBar.CompletionInProgress, JBLabel.CENTER)
-    placeholder.add(spinnerLabel, GridBagConstraints())
-    cardPanel.add(placeholder, LOADING_PANEL, LOADING_PANEL_INDEX)
+    spinnerPlaceholder.add(spinnerLabel, GridBagConstraints())
+    cardPanel.add(spinnerPlaceholder, LOADING_PANEL, LOADING_PANEL_INDEX)
+    cardPanel.add(MissingJcefPanel(), CHANGE_RUNTIME_PANEL, CHANGE_RUNTIME_PANEL_INDEX)
+    cardPanel.add(ErrorPanel(), ERROR_PANEL, ERROR_INDEX)
 
     WebUIService.getInstance(project).views.provideCodyToolWindowContent(this)
 
@@ -67,6 +72,18 @@ class CodyToolWindowContent(project: Project) {
       showView(cardPanel)
       return
     }
+    val errorOnProxyCreation = WebUIService.getInstance(project).proxyCreationException.get()
+    if (errorOnProxyCreation != null) {
+      if (isCurrentRuntimeMissingJcef()) {
+        cardLayout.show(cardPanel, CHANGE_RUNTIME_PANEL)
+        showView(cardPanel)
+      } else {
+        cardLayout.show(cardPanel, ERROR_PANEL)
+        showView(cardPanel)
+        logger.error(errorOnProxyCreation)
+      }
+      return
+    }
     cardLayout.show(cardPanel, LOADING_PANEL)
     showView(webview?.proxy?.component ?: cardPanel)
   }
@@ -99,10 +116,14 @@ class CodyToolWindowContent(project: Project) {
     const val ONBOARDING_PANEL = "onboardingPanel"
     const val SIGN_IN_PANEL = "signInWithSourcegraphPanel"
     const val LOADING_PANEL = "loadingPanel"
+    const val CHANGE_RUNTIME_PANEL = "changeRuntime"
+    const val ERROR_PANEL = "error"
 
     const val SIGN_IN_PANEL_INDEX = 0
     const val ONBOARDING_PANEL_INDEX = 1
     const val LOADING_PANEL_INDEX = 2
+    const val CHANGE_RUNTIME_PANEL_INDEX = 3
+    const val ERROR_INDEX = 4
 
     var logger = Logger.getInstance(CodyToolWindowContent::class.java)
 

src/main/kotlin/com/sourcegraph/cody/chat/ui/ErrorPanel.kt

@@ -0,0 +1,39 @@
+package com.sourcegraph.cody.chat.ui
+
+import com.intellij.ui.SimpleTextAttributes
+import com.intellij.util.ui.JBUI
+import com.sourcegraph.Icons
+import com.sourcegraph.common.CodyBundle
+import java.awt.GridBagConstraints
+import java.awt.GridBagLayout
+import javax.swing.JLabel
+import javax.swing.JPanel
+import javax.swing.JTextPane
+import javax.swing.text.SimpleAttributeSet
+import javax.swing.text.StyleConstants
+
+class ErrorPanel : JPanel(GridBagLayout()) {
+  private val description =
+      JTextPane().apply {
+        text = CodyBundle.getString("ErrorPanel.content")
+        foreground = SimpleTextAttributes.GRAY_ATTRIBUTES.fgColor
+        val center = SimpleAttributeSet()
+        StyleConstants.setAlignment(center, StyleConstants.ALIGN_CENTER)
+        styledDocument.setParagraphAttributes(0, styledDocument.length, center, false)
+      }
+
+  private val label =
+      JLabel(CodyBundle.getString("ErrorPanel.label")).apply { icon = Icons.CodyLogoSlash }
+
+  init {
+    val constraints =
+        GridBagConstraints().apply {
+          fill = GridBagConstraints.HORIZONTAL
+          gridx = 0
+          insets = JBUI.insets(20)
+        }
+
+    add(label, constraints)
+    add(description, constraints)
+  }
+}

src/main/kotlin/com/sourcegraph/cody/chat/ui/MissingJcefPanel.kt

@@ -0,0 +1,41 @@
+package com.sourcegraph.cody.chat.ui
+
+import com.intellij.openapi.projectRoots.impl.jdkDownloader.RuntimeChooserUtil
+import com.intellij.ui.SimpleTextAttributes
+import com.intellij.util.ui.JBUI
+import com.sourcegraph.common.CodyBundle
+import java.awt.GridBagConstraints
+import java.awt.GridBagLayout
+import javax.swing.JButton
+import javax.swing.JPanel
+import javax.swing.JTextPane
+import javax.swing.text.SimpleAttributeSet
+import javax.swing.text.StyleConstants
+
+class MissingJcefPanel : JPanel(GridBagLayout()) {
+  private val jcefDescription =
+      JTextPane().apply {
+        text = CodyBundle.getString("MissingJcefPanel.content")
+        foreground = SimpleTextAttributes.GRAY_ATTRIBUTES.fgColor
+        val center = SimpleAttributeSet()
+        StyleConstants.setAlignment(center, StyleConstants.ALIGN_CENTER)
+        styledDocument.setParagraphAttributes(0, styledDocument.length, center, false)
+      }
+
+  private val jcefButton =
+      JButton(CodyBundle.getString("chooseRuntimeWithJcef.button")).apply {
+        addActionListener { RuntimeChooserUtil.showRuntimeChooserPopup() }
+      }
+
+  init {
+    val constraints =
+        GridBagConstraints().apply {
+          fill = GridBagConstraints.HORIZONTAL
+          gridx = 0
+          insets = JBUI.insets(20)
+        }
+
+    add(jcefDescription, constraints)
+    add(jcefButton, constraints)
+  }
+}

src/main/kotlin/com/sourcegraph/cody/initialization/PostStartupActivity.kt

@@ -29,6 +29,7 @@ class PostStartupActivity : ProjectActivity {
   // doing something wrong, which may be slowing down agent startup. Not fixing it now but this
   // deserves more investigation.
   override suspend fun execute(project: Project) {
+    VerifyJavaBootRuntimeVersion().runActivity(project)
     SettingsMigration().runActivity(project)
     CodyAuthNotificationActivity().runActivity(project)
     CodyAuthenticationManager.getInstance().addAuthChangeListener(project)

src/main/kotlin/com/sourcegraph/cody/initialization/VerifyJavaBootRuntimeVersion.kt

@@ -0,0 +1,51 @@
+package com.sourcegraph.cody.initialization
+
+import com.intellij.notification.Notification
+import com.intellij.notification.NotificationAction
+import com.intellij.notification.NotificationType
+import com.intellij.notification.impl.NotificationFullContent
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.projectRoots.impl.jdkDownloader.RuntimeChooserCurrentItem
+import com.intellij.openapi.projectRoots.impl.jdkDownloader.RuntimeChooserUtil
+import com.intellij.openapi.projectRoots.impl.jdkDownloader.currentRuntime
+import com.sourcegraph.Icons
+import com.sourcegraph.common.CodyBundle
+import com.sourcegraph.common.NotificationGroups
+
+class VerifyJavaBootRuntimeVersion : Activity {
+
+  override fun runActivity(project: Project) {
+    if (isCurrentRuntimeMissingJcef()) {
+      JcefRuntimeNotification().notify(project)
+    }
+  }
+
+  companion object {
+    fun isCurrentRuntimeMissingJcef(): Boolean {
+      val model = RuntimeChooserCurrentItem.currentRuntime()
+      val doesNameContainJcefSuffix = model.version?.endsWith("-jcef") ?: true
+      return !doesNameContainJcefSuffix
+    }
+  }
+}
+
+class JcefRuntimeNotification :
+    Notification(
+        NotificationGroups.SOURCEGRAPH_ERRORS,
+        CodyBundle.getString("JcefRuntimeNotification.title"),
+        CodyBundle.getString("JcefRuntimeNotification.content"),
+        NotificationType.WARNING),
+    NotificationFullContent {
+
+  init {
+    icon = Icons.CodyLogoSlash
+
+    addAction(
+        object : NotificationAction(CodyBundle.getString("chooseRuntimeWithJcef.button")) {
+          override fun actionPerformed(anActionEvent: AnActionEvent, notification: Notification) {
+            RuntimeChooserUtil.showRuntimeChooserPopup()
+          }
+        })
+  }
+}

src/main/kotlin/com/sourcegraph/cody/ui/web/WebUIService.kt

@@ -6,7 +6,9 @@ import com.intellij.openapi.components.Service
 import com.intellij.openapi.components.service
 import com.intellij.openapi.diagnostic.Logger
 import com.intellij.openapi.project.Project
+import com.jetbrains.rd.util.AtomicReference
 import com.jetbrains.rd.util.ConcurrentHashMap
+import com.sourcegraph.cody.CodyToolWindowContent
 import com.sourcegraph.cody.agent.ConfigFeatures
 import com.sourcegraph.cody.agent.CurrentConfigFeatures
 import com.sourcegraph.cody.agent.protocol.WebviewCreateWebviewPanelParams
@@ -45,6 +47,8 @@ class WebUIService(private val project: Project) {
     return panels.reset()
   }
 
+  val proxyCreationException = AtomicReference<IllegalStateException?>(null)
+
   private fun <T> withCreationGate(name: String, action: (gate: WebUIProxyCreationGate) -> T): T {
     val gate =
         proxies.computeIfAbsent(name) {
@@ -112,7 +116,8 @@ class WebUIService(private val project: Project) {
                 portMapping = emptyList(),
                 enableFindWidget = false,
                 retainContextWhenHidden = false))
-    val proxy = WebUIProxy.create(delegate)
+
+    val proxy = createWebUIProxy(delegate) ?: return
     delegate.view = createView(proxy)
     proxy.updateTheme(themeController.getTheme())
     withCreationGate(handle) {
@@ -122,6 +127,16 @@ class WebUIService(private val project: Project) {
     }
   }
 
+  private fun createWebUIProxy(delegate: WebUIHost): WebUIProxy? =
+      try {
+        proxyCreationException.getAndSet(null)
+        WebUIProxy.create(delegate)
+      } catch (e: IllegalStateException) {
+        proxyCreationException.getAndSet(e)
+        CodyToolWindowContent.executeOnInstanceIfNotDisposed(project) { refreshPanelsVisibility() }
+        null
+      }
+
   internal fun createWebviewPanel(params: WebviewCreateWebviewPanelParams) {
     runInEdt {
       val delegate = WebUIHostImpl(project, params.handle, params.options)

src/main/resources/CodyBundle.properties

@@ -98,6 +98,14 @@ EndOfTrialNotification.do-not-show-again=Don't show again
 TrialEndedNotification.ignore=cody.ignore.notification.trial-ended
 TrialEndedNotification.ended.title=Your Cody Pro trial has ended
 TrialEndedNotification.ended.content=<html>Thank you for trying Cody Pro! Your trial period has ended. To keep enjoying all the features of Cody Pro, subscribe to our monthly plan. You can cancel anytime.</html>
+JcefRuntimeNotification.title=Cody requires a runtime with JCEF
+JcefRuntimeNotification.content=Your current runtime does not support Java Chromium Embedded Framework (JCEF), which is essential for Cody to work properly.
+MissingJcefPanel.content=Cody relies on Java Chromium Embedded Framework (JCEF) to function. To continue using Cody in your IDE, please switch to a runtime that includes JCEF.
+chooseRuntimeWithJcef.button=Choose a runtime with JCEF...
+
+ErrorPanel.content=Check the IDE logs for errors
+ErrorPanel.label=Cody encountered an unexpected error
+
 export.failed=Cody: Chat export failed. Please retry...
 
 # GotItTooltip