Extract ready logic to WorkspaceAndAgentStatus

Author: code-asher

src/main/kotlin/com/coder/gateway/models/WorkspaceAndAgentStatus.kt

@@ -47,6 +47,14 @@ enum class WorkspaceAndAgentStatus(val label: String, val description: String) {
         else -> if (JBColor.isBright()) JBColor.LIGHT_GRAY else JBColor.DARK_GRAY
     }
 
+    /**
+     * Return true if the agent is in a connectable state.
+     */
+    fun ready(): Boolean {
+        return listOf(READY, START_ERROR, AGENT_STARTING_READY, START_TIMEOUT_READY)
+            .contains(this)
+    }
+
     // We want to check that the workspace is `running`, the agent is
     // `connected`, and the agent lifecycle state is `ready` to ensure the best
     // possible scenario for attempting a connection.

src/main/kotlin/com/coder/gateway/views/steps/CoderWorkspacesStepView.kt

@@ -132,14 +132,8 @@ class CoderWorkspacesStepView(val setNextButtonEnabled: (Boolean) -> Unit) : Cod
         setEmptyState("Disconnected")
         setSelectionMode(ListSelectionModel.SINGLE_SELECTION)
         selectionModel.addListSelectionListener {
-            val ready = listOf(
-                WorkspaceAndAgentStatus.READY,
-                WorkspaceAndAgentStatus.START_ERROR,
-                WorkspaceAndAgentStatus.AGENT_STARTING_READY,
-                WorkspaceAndAgentStatus.START_TIMEOUT_READY,
-            ).contains(selectedObject?.agentStatus)
-            setNextButtonEnabled(ready && selectedObject?.agentOS == OS.LINUX)
-            if (ready && selectedObject?.agentOS != OS.LINUX) {
+            setNextButtonEnabled(selectedObject?.agentStatus?.ready() == true && selectedObject?.agentOS == OS.LINUX)
+            if (selectedObject?.agentStatus?.ready() == true && selectedObject?.agentOS != OS.LINUX) {
                 notificationBanner.apply {
                     component.isVisible = true
                     showInfo(CoderGatewayBundle.message("gateway.connector.view.coder.workspaces.unsupported.os.info"))