impl: enhanced workflow for network disruptions

fioan89 · fioan89 · commit acd4f227344d · 2025-07-23T23:45:31.000+03:00
Currently, when the network connection drops, the Coder TBX plugin resets itself, redirects users to the authentication page, and terminates active SSH sessions to remote IDEs. This disrupts the user experience, forcing users to manually reconnect once the network is restored. Additionally, since the SSH session to the remote IDE is lost, the JBClient is unable to re-establish a connection with the remote backend.

This PR aims to improve that experience by adopting a behavior similar to the SSH plugin. Instead of clearing the list of workspaces or dropping existing SSH sessions during a network outage, we retain them. Once the network is restored, the plugin will automatically reinitialize the HTTP client and regenerate the SSH configuration—only if the number of workspaces has changed during the disconnection—without requiring user intervention.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 ## Unreleased
 
+### Changed
+
+- improved workflow when network connection is flaky
+
 ## 0.5.2 - 2025-07-22
 
 ### Fixed
diff --git a/src/main/kotlin/com/coder/toolbox/CoderRemoteProvider.kt b/src/main/kotlin/com/coder/toolbox/CoderRemoteProvider.kt
@@ -166,17 +166,17 @@ class CoderRemoteProvider(
                     context.logger.error(ex, "workspace polling error encountered, trying to auto-login")
                     if (ex is APIResponseException && ex.isTokenExpired) {
                         WorkspaceConnectionManager.shouldEstablishWorkspaceConnections = true
+                        close()
+                        // force auto-login
+                        firstRun = true
+                        context.envPageManager.showPluginEnvironmentsPage()
+                        break
                     }
-                    close()
-                    // force auto-login
-                    firstRun = true
-                    context.envPageManager.showPluginEnvironmentsPage()
-                    break
                 }
             }
 
             // TODO: Listening on a web socket might be better?
-            select<Unit> {
+            select {
                 onTimeout(POLL_INTERVAL) {
                     context.logger.trace("workspace poller waked up by the $POLL_INTERVAL timeout")
                 }

Original file line number	Diff line number	Diff line change
`@@ -166,17 +166,17 @@ class CoderRemoteProvider(`
`166`	`166`	`context.logger.error(ex, "workspace polling error encountered, trying to auto-login")`
`167`	`167`	`if (ex is APIResponseException && ex.isTokenExpired) {`
`168`	`168`	`WorkspaceConnectionManager.shouldEstablishWorkspaceConnections = true`
	`169`	`+ close()`
	`170`	`+ // force auto-login`
	`171`	`+ firstRun = true`
	`172`	`+ context.envPageManager.showPluginEnvironmentsPage()`
	`173`	`+ break`
`169`	`174`	`}`
`170`		`- close()`
`171`		`- // force auto-login`
`172`		`- firstRun = true`
`173`		`- context.envPageManager.showPluginEnvironmentsPage()`
`174`		`- break`
`175`	`175`	`}`
`176`	`176`	`}`
`177`	`177`
`178`	`178`	`// TODO: Listening on a web socket might be better?`
`179`		`- select<Unit> {`
	`179`	`+ select {`
`180`	`180`	`onTimeout(POLL_INTERVAL) {`
`181`	`181`	`context.logger.trace("workspace poller waked up by the $POLL_INTERVAL timeout")`
`182`	`182`	`}`