fix(jetbrains): avoid connection loops

Author: andreafalzetti

components/ide/jetbrains/gateway-plugin/src/main/kotlin/io/gitpod/jetbrains/gateway/GitpodConnectionProvider.kt

@@ -31,6 +31,8 @@ import com.jetbrains.gateway.ssh.ClientOverSshTunnelConnector
 import com.jetbrains.gateway.ssh.SshHostTunnelConnector
 import com.jetbrains.gateway.thinClientLink.ThinClientHandle
 import com.jetbrains.rd.util.URI
+import com.jetbrains.rd.util.addUnique
+import com.jetbrains.rd.util.concurrentMapOf
 import com.jetbrains.rd.util.lifetime.Lifetime
 import io.gitpod.gitpodprotocol.api.entities.WorkspaceInstance
 import io.gitpod.jetbrains.icons.GitpodIcons
@@ -47,9 +49,12 @@ import javax.swing.JComponent
 import javax.swing.JLabel
 import kotlin.coroutines.coroutineContext
 
+class GitpodConnection(val gitpodHost: String, val workspaceId: String, val backendPort: String?) {}
+
 @Suppress("UnstableApiUsage", "OPT_IN_USAGE")
 class GitpodConnectionProvider : GatewayConnectionProvider {
-
+//    private val activeConnections = mutableListOf<GitpodConnection>()
+    private val activeConnectionsC = concurrentMapOf<String, Boolean>()
     private val gitpod = service<GitpodConnectionService>()
 
     private val httpClient = HttpClient.newBuilder()
@@ -74,6 +79,16 @@ class GitpodConnectionProvider : GatewayConnectionProvider {
             parameters["workspaceId"]!!,
             parameters["backendPort"]
         )
+
+        val connectionKeyId = "${connectParams.gitpodHost}-${connectParams.workspaceId}-${connectParams.backendPort}"
+
+        if (activeConnectionsC.containsKey(connectionKeyId)) {
+            thisLogger().warn("THE CONNECTION ALREADY EXISTS!!")
+            throw IllegalArgumentException("THE CONNECTION ALREADY EXISTS!!")
+        } else {
+            thisLogger().warn("NEW CONNECTION!!!!!")
+        }
+
         val client = gitpod.obtainClient(connectParams.gitpodHost)
         val connectionLifetime = Lifetime.Eternal.createNested()
         val updates = client.listenToWorkspace(connectionLifetime, connectParams.workspaceId)
@@ -230,6 +245,15 @@ class GitpodConnectionProvider : GatewayConnectionProvider {
                                         setErrorMessage("failed to fetch JetBrains Gateway Join Link.")
                                         return@launch
                                     }
+
+                                    // TODO: is this the right place?
+//                                    val newConnection = GitpodConnection(connectParams.gitpodHost, connectParams.workspaceId, connectParams.backendPort)
+//                                    activeConnections.add(newConnection)
+                                    val connectionKeyId = "${connectParams.gitpodHost}-${connectParams.workspaceId}-${connectParams.backendPort}"
+                                    activeConnectionsC[connectionKeyId] = true
+
+                                    thisLogger().warn("Storing connection for ${connectParams.gitpodHost} - ${connectParams.workspaceId} - ${connectParams.backendPort}")
+
                                     val connector = ClientOverSshTunnelConnector(
                                         connectionLifetime,
                                         SshHostTunnelConnector(credentials),
@@ -239,6 +263,7 @@ class GitpodConnectionProvider : GatewayConnectionProvider {
                                     clientHandle.clientClosed.advise(connectionLifetime) {
                                         application.invokeLater {
                                             connectionLifetime.terminate()
+                                            // todo: remove this connection from activeConnections
                                         }
                                     }
                                     clientHandle.onClientPresenceChanged.advise(connectionLifetime) {