fix: URI handling race condition when creating polling jobs

- when opening a URI, multiple polling jobs could be triggered on different
  Coder deployments if Toolbox starts from scratch. This happens because Toolbox
  takes longer to complete its initial plugin initialization, while the URI
  handling logic runs faster and doesn't wait properly for the plugin to be ready,
  leading to an early polling job. Meanwhile, once Toolbox finishes its initialization,
  it also triggers another polling job.
- this patch properly waits for the plugin initialization and properly cancel the initial
  polling job, which is then replaced by the URI handling polling job.

Author: fioan89

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

@@ -122,6 +122,12 @@ class CoderRemoteProvider(
                 environments.update {
                     LoadableState.Value(resolvedEnvironments.toList())
                 }
+                if (isInitialized.value == false) {
+                    context.logger.info("Environments for ${client.url} are now initialized")
+                    isInitialized.update {
+                        true
+                    }
+                }
 
                 lastEnvironments = resolvedEnvironments
             } catch (_: CancellationException) {
@@ -238,7 +244,7 @@ class CoderRemoteProvider(
             // start initialization with the new settings
             this@CoderRemoteProvider.client = restClient
             coderHeaderPager = NewEnvironmentPage(context, context.i18n.pnotr(restClient.url.toString()))
-            poll(restClient, cli)
+            pollJob = poll(restClient, cli)
         }
     }
 
@@ -324,7 +330,6 @@ class CoderRemoteProvider(
         pollJob?.cancel()
         pollJob = poll(client, cli)
         goToEnvironmentsPage()
-        isInitialized.update { true }
     }
 
     /**

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

@@ -13,7 +13,7 @@ import com.coder.toolbox.settings.CoderSettings
 import kotlinx.coroutines.TimeoutCancellationException
 import kotlinx.coroutines.delay
 import kotlinx.coroutines.flow.StateFlow
-import kotlinx.coroutines.flow.filter
+import kotlinx.coroutines.flow.first
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.time.withTimeout
 import kotlinx.coroutines.yield
@@ -41,7 +41,7 @@ open class CoderProtocolHandler(
      */
     suspend fun handle(
         uri: URI,
-        onReinitialize: suspend (CoderRestClient, CoderCLIManager) -> Unit
+        reInitialize: suspend (CoderRestClient, CoderCLIManager) -> Unit
     ) {
         val params = uri.toQueryParameters()
 
@@ -148,11 +148,12 @@ open class CoderProtocolHandler(
         context.logger.info("Configuring Coder CLI...")
         cli.configSsh(restClient.agentNames(workspaces))
 
-        onReinitialize(restClient, cli)
+        isInitialized.waitForTrue()
+        reInitialize(restClient, cli)
+
         context.cs.launch {
             context.ui.showWindow()
             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
@@ -366,8 +367,9 @@ internal fun getMatchingAgent(
     return agent
 }
 
-fun StateFlow<Boolean>.waitForTrue() {
-    this.filter { it }
-}
+/**
+ * Suspends the coroutine until first true value is received.
+ */
+suspend fun StateFlow<Boolean>.waitForTrue() = this.first { it }
 
 class MissingArgumentException(message: String, ex: Throwable? = null) : IllegalArgumentException(message, ex)