impl: support for uri handling (2)

- reworked the main env provider to be able to close all existing resources and re-initialize with a different deployment
- the re-initialization is needed in order to navigate to an env page with the targeted workspace. If the workspace is not from the current deployment there will be no env page to navigate to.

Author: fioan89

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

@@ -47,15 +47,12 @@ 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 = CoderProtocolHandler(context, settings, httpClient, dialogUi, isInitialized)
 
     // The REST client, if we are signed in
     private var client: CoderRestClient? = null
@@ -67,6 +64,9 @@ class CoderRemoteProvider(
 
     // On the first load, automatically log in if we can.
     private var firstRun = true
+    private val isInitialized: MutableStateFlow<Boolean> = MutableStateFlow(false)
+    private var coderHeaderPager = NewEnvironmentPage(context, context.i18n.pnotr(getDeploymentURL()?.first ?: ""))
+    private val linkHandler = CoderProtocolHandler(context, settings, httpClient, dialogUi, isInitialized)
     override val environments: MutableStateFlow<LoadableState<List<RemoteProviderEnvironment>>> = MutableStateFlow(
         LoadableState.Value(emptyList())
     )
@@ -176,9 +176,10 @@ class CoderRemoteProvider(
      */
     override fun close() {
         pollJob?.cancel()
-        client = null
+        client?.close()
         lastEnvironments = null
         environments.value = LoadableState.Value(emptyList())
+        isInitialized.update { false }
     }
 
     override val svgIcon: SvgIcon =
@@ -213,8 +214,7 @@ class CoderRemoteProvider(
      * Just displays the deployment URL at the moment, but we could use this as
      * a form for creating new environments.
      */
-    override fun getNewEnvironmentUiPage(): UiPage =
-        NewEnvironmentPage(context, context.i18n.pnotr(getDeploymentURL()?.first ?: ""))
+    override fun getNewEnvironmentUiPage(): UiPage = coderHeaderPager
 
     /**
      * We always show a list of environments.
@@ -234,7 +234,14 @@ class CoderRemoteProvider(
      */
     override suspend fun handleUri(uri: URI) {
         val params = uri.toQueryParameters()
-        linkHandler.handle(params)
+        linkHandler.handle(params) { restClient, cli ->
+            // stop polling and de-initialize resources
+            close()
+            // start initialization with the new settings
+            this@CoderRemoteProvider.client = restClient
+            coderHeaderPager = NewEnvironmentPage(context, context.i18n.pnotr(restClient.url.toString()))
+            poll(restClient, cli)
+        }
     }
 
     /**

src/main/kotlin/com/coder/toolbox/sdk/CoderRestClient.kt

@@ -272,4 +272,12 @@ open class CoderRestClient(
         }
         return buildResponse.body()!!
     }
+
+    fun close() {
+        httpClient.apply {
+            dispatcher.executorService.shutdown()
+            connectionPool.evictAll()
+            cache?.close()
+        }
+    }
 }

src/main/kotlin/com/coder/toolbox/util/CoderProtocolHandler.kt

@@ -1,6 +1,7 @@
 package com.coder.toolbox.util
 
 import com.coder.toolbox.CoderToolboxContext
+import com.coder.toolbox.cli.CoderCLIManager
 import com.coder.toolbox.cli.ensureCLI
 import com.coder.toolbox.models.WorkspaceAndAgentStatus
 import com.coder.toolbox.plugin.PluginManager
@@ -33,7 +34,10 @@ open class CoderProtocolHandler(
      * Throw if required arguments are not supplied or the workspace is not in a
      * connectable state.
      */
-    suspend fun handle(parameters: Map<String, String>) {
+    suspend fun handle(
+        parameters: Map<String, String>,
+        onReinitialize: suspend (CoderRestClient, CoderCLIManager) -> Unit
+    ) {
         val deploymentURL =
             parameters.url() ?: dialogUi.ask(
                 context.i18n.ptrl("Deployment URL"),
@@ -49,7 +53,7 @@ open class CoderProtocolHandler(
         } else {
             null
         }
-        val client = try {
+        val restClient = try {
             authenticate(deploymentURL, queryToken)
         } catch (ex: MissingArgumentException) {
             throw MissingArgumentException("Query parameter \"$TOKEN\" is missing", ex)
@@ -59,7 +63,7 @@ open class CoderProtocolHandler(
         val workspaceName =
             parameters.workspace() ?: throw MissingArgumentException("Query parameter \"$WORKSPACE\" is missing")
 
-        val workspaces = client.workspaces()
+        val workspaces = restClient.workspaces()
         val workspace =
             workspaces.firstOrNull {
                 it.name == workspaceName
@@ -117,23 +121,24 @@ open class CoderProtocolHandler(
             ensureCLI(
                 context,
                 deploymentURL.toURL(),
-                client.buildInfo().version,
+                restClient.buildInfo().version,
                 settings
             )
 
         // We only need to log in if we are using token-based auth.
-        if (client.token != null) {
+        if (restClient.token != null) {
             context.logger.info("Authenticating Coder CLI...")
-            cli.login(client.token)
+            cli.login(restClient.token)
         }
 
         context.logger.info("Configuring Coder CLI...")
-        cli.configSsh(client.agentNames(workspaces))
+        cli.configSsh(restClient.agentNames(workspaces))
 
-        isInitialized.waitForTrue()
+        onReinitialize(restClient, cli)
         context.cs.launch {
             context.ui.showWindow()
-            yield()
+            context.envPageManager.showPluginEnvironmentsPage(true)
+            isInitialized.waitForTrue()
             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