impl: support uri handling (1)

- reacts to uris like: jetbrains://gateway/com.coder.toolbox?url=https%3A%2F%2Fdev.coder.com&token=....&workspace=bobiverse-bill
- the handling still does not work correctly when Toolbox is already running.

Author: fioan89

src/main/kotlin/com/coder/toolbox/CoderRemoteProvider.kt

@@ -7,8 +7,8 @@ import com.coder.toolbox.services.CoderSecretsService
 import com.coder.toolbox.services.CoderSettingsService
 import com.coder.toolbox.settings.CoderSettings
 import com.coder.toolbox.settings.Source
+import com.coder.toolbox.util.CoderProtocolHandler
 import com.coder.toolbox.util.DialogUi
-import com.coder.toolbox.util.LinkHandler
 import com.coder.toolbox.util.toQueryParameters
 import com.coder.toolbox.views.Action
 import com.coder.toolbox.views.CoderSettingsPage
@@ -47,13 +47,15 @@ class CoderRemoteProvider(
     private var pollJob: Job? = null
     private var lastEnvironments: Set<CoderRemoteEnvironment>? = null
 
+    private val isInitialized: MutableStateFlow<Boolean> = MutableStateFlow(false)
+
     // Create our services from the Toolbox ones.
     private val settingsService = CoderSettingsService(context.settingsStore)
     private val settings: CoderSettings = CoderSettings(settingsService, context.logger)
     private val secrets: CoderSecretsService = CoderSecretsService(context.secretsStore)
     private val settingsPage: CoderSettingsPage = CoderSettingsPage(context, settingsService)
     private val dialogUi = DialogUi(context, settings)
-    private val linkHandler = LinkHandler(context, settings, httpClient, dialogUi)
+    private val linkHandler = CoderProtocolHandler(context, settings, httpClient, dialogUi, isInitialized)
 
     // The REST client, if we are signed in
     private var client: CoderRestClient? = null
@@ -65,7 +67,6 @@ class CoderRemoteProvider(
 
     // On the first load, automatically log in if we can.
     private var firstRun = true
-
     override val environments: MutableStateFlow<LoadableState<List<RemoteProviderEnvironment>>> = MutableStateFlow(
         LoadableState.Value(emptyList())
     )
@@ -234,11 +235,7 @@ class CoderRemoteProvider(
      */
     override suspend fun handleUri(uri: URI) {
         val params = uri.toQueryParameters()
-        context.cs.launch {
-            val name = linkHandler.handle(params)
-            // TODO@JB: Now what?  How do we actually connect this workspace?
-            context.logger.debug("External request for $name: $uri")
-        }
+        linkHandler.handle(params)
     }
 
     /**
@@ -323,6 +320,7 @@ class CoderRemoteProvider(
         pollJob?.cancel()
         pollJob = poll(client, cli)
         goToEnvironmentsPage()
+        isInitialized.update { true }
     }
 
     /**

src/main/kotlin/com/coder/toolbox/util/LinkHandler.kt renamed to src/main/kotlin/com/coder/toolbox/util/CoderProtocolHandler.kt

@@ -11,15 +11,20 @@ import com.coder.toolbox.sdk.v2.models.WorkspaceAgent
 import com.coder.toolbox.sdk.v2.models.WorkspaceStatus
 import com.coder.toolbox.settings.CoderSettings
 import com.coder.toolbox.settings.Source
+import kotlinx.coroutines.flow.StateFlow
+import kotlinx.coroutines.flow.filter
+import kotlinx.coroutines.launch
+import kotlinx.coroutines.yield
 import okhttp3.OkHttpClient
 import java.net.HttpURLConnection
 import java.net.URL
 
-open class LinkHandler(
+open class CoderProtocolHandler(
     private val context: CoderToolboxContext,
     private val settings: CoderSettings,
     private val httpClient: OkHttpClient?,
     private val dialogUi: DialogUi,
+    private val isInitialized: StateFlow<Boolean>,
 ) {
     /**
      * Given a set of URL parameters, prepare the CLI then return a workspace to
@@ -31,7 +36,7 @@ open class LinkHandler(
     suspend fun handle(
         parameters: Map<String, String>,
         indicator: ((t: String) -> Unit)? = null,
-    ): String {
+    ) {
         val deploymentURL =
             parameters.url() ?: dialogUi.ask(
                 context.i18n.ptrl("Deployment URL"),
@@ -129,10 +134,16 @@ open class LinkHandler(
         indicator?.invoke("Configuring Coder CLI...")
         cli.configSsh(client.agentNames(workspaces))
 
-        val name = "${workspace.name}.${agent.name}"
-        // TODO@JB: Can we ask for the IDE and project path or how does
-        //          this work?
-        return name
+        isInitialized.waitForTrue()
+        context.cs.launch {
+            context.ui.showWindow()
+            yield()
+            context.envPageManager.showEnvironmentPage("${workspace.name}.${agent.name}", false)
+            // without a yield or a delay(0) the env page does not show up. My assumption is that
+            // the coroutine is finishing too fast without giving enough time to compose main thread
+            // to catch the state change. Yielding gives other coroutines the chance to run
+            yield()
+        }
     }
 
     /**
@@ -332,4 +343,8 @@ internal fun getMatchingAgent(
     return agent
 }
 
+fun StateFlow<Boolean>.waitForTrue() {
+    this.filter { it }
+}
+
 class MissingArgumentException(message: String, ex: Throwable? = null) : IllegalArgumentException(message, ex)