Add a text field for entering multiple remote repositories at once

CONTRIBUTING.md

@@ -56,8 +56,8 @@ Few tips and tricks regarding versioning of the tooling:
 - Use `node` version `18` (newer versions causes hard to diagnose errors with `ERR_INVALID_THIS`).
 - If you changed `pnpm` or `node` version after running gradle you need to kill gradle daemon with `./gradlew --stop`.
   Otherwise you won't see effects of your changes.
-- Running `:runIde PplatformRuntimeVersion=X.Y` for the first time might fail due to missing IntelliJ installation. You
-  can fix it by running `:runIde PplatformVersion=X.Y` once - even if compilation fails it fixes your caches.
+- Running `./gradlew -PplatformRuntimeVersion=X.Y :runIde` for the first time might fail due to missing IntelliJ installation. You
+  can fix it by running `./gradlew -PplatformVersion=X.Y :runIde` once - even if compilation fails it fixes your caches.
 
 ## Using Nightly channel releases
 

src/main/java/com/sourcegraph/Icons.java

@@ -7,4 +7,8 @@ public interface Icons {
   Icon SourcegraphLogo = IconLoader.getIcon("/icons/sourcegraphLogo.svg", Icons.class);
   Icon CodyLogo = IconLoader.getIcon("/icons/codyLogo.svg", Icons.class);
   Icon GearPlain = IconLoader.getIcon("/icons/gearPlain.svg", Icons.class);
+  Icon RepoHostBitbucket = IconLoader.getIcon("/icons/repo-host-bitbucket.svg", Icons.class);
+  Icon RepoHostGeneric = IconLoader.getIcon("/icons/repo-host-generic.svg", Icons.class);
+  Icon RepoHostGitHub = IconLoader.getIcon("/icons/repo-host-github.svg", Icons.class);
+  Icon RepoHostGitlab = IconLoader.getIcon("/icons/repo-host-gitlab.svg", Icons.class);
 }

src/main/java/com/sourcegraph/cody/agent/CodyAgentClient.java

@@ -2,10 +2,7 @@
 
 import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.diagnostic.Logger;
-import com.sourcegraph.cody.agent.protocol.DebugMessage;
-import com.sourcegraph.cody.agent.protocol.EditTask;
-import com.sourcegraph.cody.agent.protocol.TextDocumentEditParams;
-import com.sourcegraph.cody.agent.protocol.WorkspaceEditParams;
+import com.sourcegraph.cody.agent.protocol.*;
 import java.util.concurrent.CompletableFuture;
 import java.util.function.Consumer;
 import org.eclipse.lsp4j.jsonrpc.services.JsonNotification;
@@ -54,6 +51,24 @@ public CompletableFuture<Void> editTaskDidDelete(EditTask params) {
     return acceptOnEventThread("editTask/didDelete", onEditTaskDidDelete, params);
   }
 
+  @Nullable Consumer<Void> onRemoteRepoDidChange;
+
+  @JsonNotification("remoteRepo/didChange")
+  public void remoteRepoDidChange() {
+    if (onRemoteRepoDidChange != null) {
+      onRemoteRepoDidChange.accept(null);
+    }
+  }
+
+  @Nullable Consumer<RemoteRepoFetchState> onRemoteRepoDidChangeState;
+
+  @JsonNotification("remoteRepo/didChangeState")
+  public void remoteRepoDidChangeState(RemoteRepoFetchState state) {
+    if (onRemoteRepoDidChangeState != null) {
+      onRemoteRepoDidChangeState.accept(state);
+    }
+  }
+
   @JsonRequest("textDocument/edit")
   public CompletableFuture<Void> textDocumentEdit(TextDocumentEditParams params) {
     return acceptOnEventThread("textDocument/edit", onTextDocumentEdit, params);

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

@@ -1,28 +1,6 @@
 package com.sourcegraph.cody.agent
 
-import com.sourcegraph.cody.agent.protocol.AttributionSearchParams
-import com.sourcegraph.cody.agent.protocol.AttributionSearchResponse
-import com.sourcegraph.cody.agent.protocol.AutocompleteParams
-import com.sourcegraph.cody.agent.protocol.AutocompleteResult
-import com.sourcegraph.cody.agent.protocol.ChatHistoryResponse
-import com.sourcegraph.cody.agent.protocol.ChatModelsParams
-import com.sourcegraph.cody.agent.protocol.ChatModelsResponse
-import com.sourcegraph.cody.agent.protocol.ChatRestoreParams
-import com.sourcegraph.cody.agent.protocol.ChatSubmitMessageParams
-import com.sourcegraph.cody.agent.protocol.ClientInfo
-import com.sourcegraph.cody.agent.protocol.CompletionItemParams
-import com.sourcegraph.cody.agent.protocol.CurrentUserCodySubscription
-import com.sourcegraph.cody.agent.protocol.EditTask
-import com.sourcegraph.cody.agent.protocol.Event
-import com.sourcegraph.cody.agent.protocol.GetFeatureFlag
-import com.sourcegraph.cody.agent.protocol.GetFoldingRangeParams
-import com.sourcegraph.cody.agent.protocol.GetFoldingRangeResult
-import com.sourcegraph.cody.agent.protocol.GetRepoIdsParam
-import com.sourcegraph.cody.agent.protocol.GetRepoIdsResponse
-import com.sourcegraph.cody.agent.protocol.InlineEditParams
-import com.sourcegraph.cody.agent.protocol.ProtocolTextDocument
-import com.sourcegraph.cody.agent.protocol.ServerInfo
-import com.sourcegraph.cody.agent.protocol.TaskIdParam
+import com.sourcegraph.cody.agent.protocol.*
 import com.sourcegraph.cody.chat.ConnectionId
 import java.util.concurrent.CompletableFuture
 import org.eclipse.lsp4j.jsonrpc.services.JsonNotification
@@ -87,6 +65,11 @@ interface CodyAgentServer {
   @JsonNotification("autocomplete/completionAccepted")
   fun completionAccepted(logID: CompletionItemParams)
 
+  @JsonNotification("remoteRepo/didChange") fun remoteRepoDidChange()
+
+  @JsonNotification("remoteRepo/didChangeState")
+  fun remoteRepoDidChangeState(state: RemoteRepoFetchState)
+
   @JsonRequest("webview/receiveMessage")
   fun webviewReceiveMessage(params: WebviewReceiveMessageParams): CompletableFuture<Any?>
 
@@ -130,4 +113,10 @@ interface CodyAgentServer {
   fun attributionSearch(
       params: AttributionSearchParams
   ): CompletableFuture<AttributionSearchResponse>
+
+  @JsonRequest("remoteRepo/has")
+  fun remoteRepoHas(params: RemoteRepoHasParams): CompletableFuture<RemoteRepoHasResponse>
+
+  @JsonRequest("remoteRepo/list")
+  fun remoteRepoList(params: RemoteRepoListParams): CompletableFuture<RemoteRepoListResponse>
 }

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

@@ -6,17 +6,20 @@ import com.intellij.openapi.components.Service
 import com.intellij.openapi.components.service
 import com.intellij.openapi.diagnostic.Logger
 import com.intellij.openapi.fileEditor.FileEditorManager
+import com.intellij.openapi.progress.ProcessCanceledException
 import com.intellij.openapi.project.Project
 import com.sourcegraph.cody.CodyFileEditorListener
 import com.sourcegraph.cody.chat.AgentChatSessionService
 import com.sourcegraph.cody.config.CodyApplicationSettings
+import com.sourcegraph.cody.context.RemoteRepoSearcher
 import com.sourcegraph.cody.edit.FixupService
 import com.sourcegraph.cody.statusbar.CodyStatusService
 import java.util.concurrent.CompletableFuture
 import java.util.concurrent.TimeUnit
 import java.util.concurrent.TimeoutException
 import java.util.concurrent.atomic.AtomicReference
 import java.util.function.Consumer
+import kotlinx.coroutines.runBlocking
 
 @Service(Service.Level.PROJECT)
 class CodyAgentService(project: Project) : Disposable {
@@ -61,6 +64,14 @@ class CodyAgentService(project: Project) : Disposable {
         FixupService.getInstance(project).getActiveSession()?.performInlineEdits(params.edits)
       }
 
+      agent.client.onRemoteRepoDidChange = Consumer {
+        RemoteRepoSearcher.getInstance(project).remoteRepoDidChange()
+      }
+
+      agent.client.onRemoteRepoDidChangeState = Consumer { state ->
+        RemoteRepoSearcher.getInstance(project).remoteRepoDidChangeState(state)
+      }
+
       if (!project.isDisposed) {
         AgentChatSessionService.getInstance(project).restoreAllSessions(agent)
         val fileEditorManager = FileEditorManager.getInstance(project)
@@ -156,19 +167,15 @@ class CodyAgentService(project: Project) : Disposable {
     ) {
       if (CodyApplicationSettings.instance.isCodyEnabled) {
         ApplicationManager.getApplication().executeOnPooledThread {
-          try {
-            val instance = getInstance(project)
-            val isReadyButNotFunctional = instance.codyAgent.getNow(null)?.isConnected() == false
-            val agent =
-                if (isReadyButNotFunctional && restartIfNeeded) instance.restartAgent(project)
-                else instance.codyAgent
-
-            callback.accept(agent.get())
-            setAgentError(project, null)
-          } catch (e: Exception) {
-            logger.warn("Failed to execute call to agent", e)
-            onFailure.accept(e)
-            if (restartIfNeeded) getInstance(project).restartAgent(project)
+          runBlocking {
+            val task: suspend (CodyAgent) -> Unit = { agent ->
+              try {
+                callback.accept(agent)
+              } catch (e: Exception) {
+                onFailure.accept(e)
+              }
+            }
+            coWithAgent(project, restartIfNeeded, task)
           }
         }
       }
@@ -201,5 +208,34 @@ class CodyAgentService(project: Project) : Disposable {
         false
       }
     }
+
+    suspend fun <T> coWithAgent(project: Project, callback: suspend (CodyAgent) -> T) =
+        coWithAgent(project, false, callback)
+
+    suspend fun <T> coWithAgent(
+        project: Project,
+        restartIfNeeded: Boolean,
+        callback: suspend (CodyAgent) -> T
+    ): T {
+      if (!CodyApplicationSettings.instance.isCodyEnabled) {
+        throw Exception("Cody is not enabled")
+      }
+      try {
+        val instance = CodyAgentService.getInstance(project)
+        val isReadyButNotFunctional = instance.codyAgent.getNow(null)?.isConnected() == false
+        val agent =
+            if (isReadyButNotFunctional && restartIfNeeded) instance.restartAgent(project)
+            else instance.codyAgent
+        val result = callback(agent.get())
+        setAgentError(project, null)
+        return result
+      } catch (e: Exception) {
+        logger.warn("Failed to execute call to agent", e)
+        if (restartIfNeeded && e !is ProcessCanceledException) {
+          getInstance(project).restartAgent(project)
+        }
+        throw e
+      }
+    }
   }
 }

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

@@ -0,0 +1,29 @@
+package com.sourcegraph.cody.agent.protocol
+
+import com.sourcegraph.cody.error.CodyError
+
+data class RemoteRepoHasParams(
+    val repoName: String,
+)
+
+data class RemoteRepoHasResponse(
+    val result: Boolean,
+)
+
+data class RemoteRepoListParams(
+    val query: String?,
+    val first: Int,
+    val after: String?,
+)
+
+data class RemoteRepoListResponse(
+    val startIndex: Int,
+    val count: Int,
+    val repos: List<Repo>,
+    val state: RemoteRepoFetchState,
+)
+
+data class RemoteRepoFetchState(
+    val state: String, // one of: "paused", "fetching", "errored", "complete"
+    val error: CodyError?,
+)

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

@@ -40,7 +40,7 @@ class ChatPanel(
   private val messagesPanel = MessagesPanel(project, chatSession)
   private val chatPanel = ChatScrollPane(messagesPanel)
 
-  private val contextView: EnhancedContextPanel = EnhancedContextPanel(project, chatSession)
+  private val contextView: EnhancedContextPanel = EnhancedContextPanel.create(project, chatSession)
 
   private val stopGeneratingButton =
       object : JButton("Stop generating", IconUtil.desaturate(AllIcons.Actions.Suspend)) {
@@ -75,7 +75,7 @@ class ChatPanel(
     promptPanel.focus()
   }
 
-  fun isEnhancedContextEnabled(): Boolean = contextView.isEnhancedContextEnabled.get()
+  fun isEnhancedContextEnabled(): Boolean = contextView.isEnhancedContextEnabled
 
   @RequiresEdt
   fun addOrUpdateMessage(message: ChatMessage, index: Int) {

src/main/kotlin/com/sourcegraph/cody/config/SourcegraphServerPath.kt

@@ -4,6 +4,7 @@ import com.intellij.util.xmlb.annotations.Attribute
 import com.intellij.util.xmlb.annotations.Tag
 import com.sourcegraph.cody.auth.ServerPath
 import com.sourcegraph.config.ConfigUtil
+import java.net.URI
 import java.util.regex.Pattern
 
 @Tag("server")
@@ -22,6 +23,9 @@ data class SourcegraphServerPath(
     return url + GRAPHQL_API_SUFFIX
   }
 
+  val displayName: String
+    get() = URI.create(url).host
+
   companion object {
     const val DEFAULT_HOST = ConfigUtil.DOTCOM_URL