Add support for local symf context

src/main/java/com/sourcegraph/cody/CodyFocusChangeListener.java

@@ -35,7 +35,6 @@ public void focusGained(@NotNull Editor editor) {
     if (!ConfigUtil.isCodyEnabled()) {
       return;
     }
-
     Project project = editor.getProject();
     if (project == null) {
       return;
@@ -48,7 +47,15 @@ public void focusGained(@NotNull Editor editor) {
 
     CodyAgentService.applyAgentOnBackgroundThread(
         project,
-        agent -> agent.getServer().textDocumentDidFocus(TextDocument.fromPath(file.getPath())));
+        agent -> {
+          try {
+            // TODO: This is bad but needed to avoid race with file context of commands executed
+            // through context menu
+            Thread.sleep(100);
+          } catch (InterruptedException ignored) {
+          }
+          agent.getServer().textDocumentDidFocus(TextDocument.fromPath(file.getPath()));
+        });
 
     CodyAgentCodebase.getInstance(project).onFileOpened(project, file);
   }

src/main/java/com/sourcegraph/cody/LowerPanel.kt

@@ -1,26 +1,20 @@
 package com.sourcegraph.cody
 
-import com.intellij.ui.ColorUtil
-import com.intellij.util.ui.UIUtil
-import com.sourcegraph.cody.context.EmbeddingStatusView
+import com.intellij.openapi.ui.VerticalFlowLayout
+import com.sourcegraph.cody.context.ui.EnhancedContextPanel
 import java.awt.BorderLayout
-import javax.swing.BorderFactory
-import javax.swing.BoxLayout
 import javax.swing.JPanel
 
 class LowerPanel(
     stopGeneratingButtonPanel: JPanel,
     promptPanel: PromptPanel,
-    embeddingStatusView: EmbeddingStatusView
+    contextView: EnhancedContextPanel
 ) : JPanel(BorderLayout()) {
   init {
-    val borderColor = ColorUtil.brighter(UIUtil.getPanelBackground(), 3)
-    val topBorder = BorderFactory.createMatteBorder(1, 0, 0, 0, borderColor)
-    embeddingStatusView.border = topBorder
-    border = embeddingStatusView.border
-    layout = BoxLayout(this, BoxLayout.Y_AXIS)
+    border = contextView.border
+    layout = VerticalFlowLayout()
     add(stopGeneratingButtonPanel)
     add(promptPanel)
-    add(embeddingStatusView)
+    add(contextView)
   }
 }

src/main/java/com/sourcegraph/cody/PromptPanel.kt

@@ -15,7 +15,9 @@ import java.awt.event.ComponentAdapter
 import java.awt.event.ComponentEvent
 import java.awt.event.KeyAdapter
 import java.awt.event.KeyEvent
-import javax.swing.*
+import javax.swing.JButton
+import javax.swing.JLayeredPane
+import javax.swing.KeyStroke
 import javax.swing.border.EmptyBorder
 import javax.swing.event.DocumentEvent
 import javax.swing.text.DefaultEditorKit
@@ -28,7 +30,7 @@ class PromptPanel(
 ) : JLayeredPane() {
 
   private var isInHistoryMode = true
-  private val autoGrowingTextArea = AutoGrowingTextArea(3, 9, this)
+  private val autoGrowingTextArea = AutoGrowingTextArea(5, 9, this)
   private val scrollPane = autoGrowingTextArea.scrollPane
   private val margin = 14
   val textArea = autoGrowingTextArea.textArea

src/main/java/com/sourcegraph/cody/SubscriptionTabPanel.kt

@@ -54,7 +54,7 @@ class SubscriptionTabPanel : JPanel() {
               "</tr>" +
               "</table>")
     }
-    if (ApplicationInfo.getInstance().getBuild().baselineVersion <= 223) {
+    if (ApplicationInfo.getInstance().build.baselineVersion <= 223) {
       separator()
     } else {
       row {

src/main/java/com/sourcegraph/cody/vscode/InlineAutocompleteTriggerKind.java

@@ -2,5 +2,5 @@
 
 public enum InlineAutocompleteTriggerKind {
   Invoke,
-  Automatic;
+  Automatic
 }

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

@@ -21,7 +21,7 @@ import com.sourcegraph.cody.commands.ui.CommandsTabPanel
 import com.sourcegraph.cody.config.CodyAccount
 import com.sourcegraph.cody.config.CodyApplicationSettings
 import com.sourcegraph.cody.config.CodyAuthenticationManager
-import com.sourcegraph.cody.context.EmbeddingStatusView
+import com.sourcegraph.cody.context.ui.EnhancedContextPanel
 import com.sourcegraph.cody.ui.ChatScrollPane
 import com.sourcegraph.cody.ui.SendButton
 import com.sourcegraph.cody.vscode.CancellationToken
@@ -48,8 +48,12 @@ class CodyToolWindowContent(private val project: Project) : UpdatableChat {
   private val stopGeneratingButton =
       JButton("Stop generating", IconUtil.desaturate(AllIcons.Actions.Suspend))
   private val commandsPanel: CommandsTabPanel =
-      CommandsTabPanel(project) { cmdId: CommandId -> sendMessage(project, message = null, cmdId) }
-  val embeddingStatusView: EmbeddingStatusView
+      CommandsTabPanel(project) { cmdId: CommandId ->
+        ApplicationManager.getApplication().invokeLater {
+          sendMessage(project, message = null, cmdId)
+        }
+      }
+  val contextView: EnhancedContextPanel
   override var isChatVisible = false
   override var id: String? = null
   private var codyOnboardingGuidancePanel: CodyOnboardingGuidancePanel? = null
@@ -85,9 +89,7 @@ class CodyToolWindowContent(private val project: Project) : UpdatableChat {
             sendButton,
             isGenerating = stopGeneratingButton::isVisible)
     val stopGeneratingButtonPanel = JPanel(FlowLayout(FlowLayout.CENTER, 0, 5))
-    stopGeneratingButtonPanel.preferredSize =
-        Dimension(Short.MAX_VALUE.toInt(), stopGeneratingButton.getPreferredSize().height + 10)
-
+    stopGeneratingButtonPanel.minimumSize = Dimension(Short.MAX_VALUE.toInt(), 0)
     stopGeneratingButton.addActionListener {
       inProgressChat.abort()
       stopGeneratingButton.isVisible = false
@@ -99,8 +101,8 @@ class CodyToolWindowContent(private val project: Project) : UpdatableChat {
     stopGeneratingButton.isVisible = false
     stopGeneratingButtonPanel.add(stopGeneratingButton)
     stopGeneratingButtonPanel.isOpaque = false
-    embeddingStatusView = EmbeddingStatusView(project)
-    val lowerPanel = LowerPanel(stopGeneratingButtonPanel, promptPanel, embeddingStatusView)
+    contextView = EnhancedContextPanel(project)
+    val lowerPanel = LowerPanel(stopGeneratingButtonPanel, promptPanel, contextView)
 
     // Main content panel
     contentPanel.layout = BorderLayout(0, 0)
@@ -128,6 +130,7 @@ class CodyToolWindowContent(private val project: Project) : UpdatableChat {
     CodyAgentService.applyAgentOnBackgroundThread(project) { agent ->
       fetchSubscriptionPanelData(project, agent.server).thenAccept {
         if (it != null) {
+
           ApplicationManager.getApplication().invokeLater { refreshSubscriptionTab(it) }
         }
       }
@@ -136,8 +139,10 @@ class CodyToolWindowContent(private val project: Project) : UpdatableChat {
 
   @RequiresEdt
   private fun refreshSubscriptionTab(data: SubscriptionTabPanelData) {
+    val isSubscriptionTabPresent = tabbedPane.tabCount >= SUBSCRIPTION_TAB_INDEX + 1
+
     if (data.isDotcomAccount && data.codyProFeatureFlag) {
-      if (tabbedPane.tabCount < SUBSCRIPTION_TAB_INDEX + 1) {
+      if (!isSubscriptionTabPresent) {
         tabbedPane.insertTab(
             /* title = */ "Subscription",
             /* icon = */ null,
@@ -146,7 +151,7 @@ class CodyToolWindowContent(private val project: Project) : UpdatableChat {
             SUBSCRIPTION_TAB_INDEX)
       }
       subscriptionPanel.update(data.isCurrentUserPro)
-    } else if (SUBSCRIPTION_TAB_INDEX < tabbedPane.tabCount) {
+    } else if (isSubscriptionTabPresent) {
       tabbedPane.remove(SUBSCRIPTION_TAB_INDEX)
     }
   }
@@ -347,7 +352,13 @@ class CodyToolWindowContent(private val project: Project) : UpdatableChat {
     ApplicationManager.getApplication().executeOnPooledThread {
       val chat = Chat()
       try {
-        chat.sendMessageViaAgent(project, humanMessage, commandId, this, inProgressChat)
+        chat.sendMessageViaAgent(
+            project,
+            humanMessage,
+            commandId,
+            this,
+            inProgressChat,
+            contextView.isEnhancedContextEnabled.get())
       } catch (e: Exception) {
         logger.error("Error sending message '$humanMessage' to chat", e)
         addMessageToChat(

src/main/kotlin/com/sourcegraph/cody/agent/CodyAgentCodebase.kt

@@ -5,64 +5,32 @@ import com.intellij.openapi.components.Service
 import com.intellij.openapi.components.service
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.vfs.VirtualFile
-import com.sourcegraph.cody.CodyToolWindowContent
 import com.sourcegraph.cody.config.CodyProjectSettings
-import com.sourcegraph.common.ProjectFileUtils
 import com.sourcegraph.config.ConfigUtil
 import com.sourcegraph.vcs.RepoUtil
 
 @Service(Service.Level.PROJECT)
 class CodyAgentCodebase(val project: Project) {
 
-  fun initializeRepoName() {
-    ApplicationManager.getApplication().executeOnPooledThread {
-      onRepositoryNameChange(RepoUtil.findRepositoryName(project, null))
-    }
-  }
-
   // TODO: Support list of repository names instead of just one.
   private val application = ApplicationManager.getApplication()
   private val settings = CodyProjectSettings.getInstance(project)
   private var inferredUrl: String? = null
 
-  fun setUrl(url: String) {
-    settings.remoteUrl = url
-    onPropagateConfiguration()
-  }
-
   fun getUrl(): String? = settings.remoteUrl ?: inferredUrl
 
-  fun onFileOpened(project: Project, file: VirtualFile) {
+  fun onFileOpened(project: Project, file: VirtualFile?) {
     application.executeOnPooledThread {
-      onRepositoryNameChange(RepoUtil.findRepositoryName(project, file))
-      application.runReadAction {
-        val openedFileName = file.name
-        val relativeFilePath: String? = ProjectFileUtils.getRelativePathToProjectRoot(project, file)
-        ApplicationManager.getApplication().invokeLater {
-          CodyToolWindowContent.executeOnInstanceIfNotDisposed(project) {
-            embeddingStatusView.setOpenedFileName(openedFileName, relativeFilePath)
-          }
+      val repositoryName = RepoUtil.findRepositoryName(project, file)
+      if (repositoryName != null && inferredUrl != repositoryName) {
+        inferredUrl = repositoryName
+        CodyAgentService.applyAgentOnBackgroundThread(project) {
+          it.server.configurationDidChange(ConfigUtil.getAgentConfiguration(project))
         }
       }
     }
   }
 
-  private fun onPropagateConfiguration() {
-    CodyToolWindowContent.executeOnInstanceIfNotDisposed(project) {
-      embeddingStatusView.updateEmbeddingStatus()
-    }
-    CodyAgentService.applyAgentOnBackgroundThread(project) {
-      it.server.configurationDidChange(ConfigUtil.getAgentConfiguration(project))
-    }
-  }
-
-  private fun onRepositoryNameChange(repositoryName: String?) {
-    if (repositoryName != null && inferredUrl != repositoryName) {
-      inferredUrl = repositoryName
-      onPropagateConfiguration()
-    }
-  }
-
   companion object {
     @JvmStatic
     fun getInstance(project: Project): CodyAgentCodebase {

src/main/kotlin/com/sourcegraph/cody/agent/protocol/ChatSubmitMessageParams.kt

@@ -2,4 +2,4 @@ package com.sourcegraph.cody.agent.protocol
 
 import com.sourcegraph.cody.agent.WebviewMessage
 
-data class ChatSubmitMessageParams(val id: String, val message: WebviewMessage) {}
+data class ChatSubmitMessageParams(val id: String, val message: WebviewMessage)

src/main/kotlin/com/sourcegraph/cody/autocomplete/CodyEditorFactoryListener.kt

@@ -131,7 +131,7 @@ class CodyEditorFactoryListener : EditorFactoryListener {
         val current = editorManager.getSelectedTextEditor(true)
         return current != null && current == editor
       }
-      val current = editorManager.getSelectedEditor()
+      val current = editorManager.selectedEditor
       return current is TextEditor && editor == current.editor
     }