fix: show status color for sorted row

- instead of the original row

Author: fioan89

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

@@ -3,12 +3,19 @@ package com.coder.gateway.models
 import com.coder.gateway.sdk.v2.models.ProvisionerJobStatus
 import com.coder.gateway.sdk.v2.models.Workspace
 import com.coder.gateway.sdk.v2.models.WorkspaceTransition
+import com.intellij.ui.JBColor
 
 enum class WorkspaceAgentStatus(val label: String) {
     QUEUED("◍ Queued"), STARTING("⦿ Starting"), STOPPING("◍ Stopping"), DELETING("⦸ Deleting"),
     RUNNING("⦿ Running"), STOPPED("◍ Stopped"), DELETED("⦸ Deleted"),
     CANCELING("◍ Canceling action"), CANCELED("◍ Canceled action"), FAILED("ⓧ Failed");
 
+    fun statusColor() = when (this) {
+        RUNNING -> JBColor.GREEN
+        FAILED -> JBColor.RED
+        else -> if (JBColor.isBright()) JBColor.LIGHT_GRAY else JBColor.DARK_GRAY
+    }
+
     companion object {
         fun from(workspace: Workspace) = when (workspace.latestBuild.job.status) {
             ProvisionerJobStatus.PENDING -> QUEUED
@@ -28,5 +35,7 @@ enum class WorkspaceAgentStatus(val label: String) {
             ProvisionerJobStatus.CANCELED -> CANCELED
             ProvisionerJobStatus.FAILED -> FAILED
         }
+
+        fun from(str: String) = WorkspaceAgentStatus.values().first { it.label.contains(str, true) }
     }
 }

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

@@ -41,7 +41,6 @@ import com.intellij.openapi.ui.panel.ComponentPanelBuilder
 import com.intellij.openapi.wm.impl.welcomeScreen.WelcomeScreenUIManager
 import com.intellij.ui.AnActionButton
 import com.intellij.ui.AppIcon
-import com.intellij.ui.JBColor
 import com.intellij.ui.RelativeFont
 import com.intellij.ui.ToolbarDecorator
 import com.intellij.ui.components.JBTextField
@@ -773,20 +772,14 @@ class CoderWorkspacesStepView(val enableNextButtonCallback: (Boolean) -> Unit) :
                     super.getTableCellRendererComponent(table, value, isSelected, hasFocus, row, column)
                     if (value is String) {
                         text = value
+                        foreground = WorkspaceAgentStatus.from(value).statusColor()
                     }
                     font = this@CoderWorkspacesStepView.tableOfWorkspaces.tableHeader.font
                     border = JBUI.Borders.empty(0, 8)
-                    foreground = (table.model as ListTableModel<WorkspaceAgentModel>).getRowValue(row).statusColor()
                     return this
                 }
             }
         }
-
-        private fun WorkspaceAgentModel.statusColor() = when (this.agentStatus) {
-            RUNNING -> JBColor.GREEN
-            FAILED -> JBColor.RED
-            else -> if (JBColor.isBright()) JBColor.LIGHT_GRAY else JBColor.DARK_GRAY
-        }
     }
 
     private fun TableView<WorkspaceAgentModel>.selectItem(workspaceName: String?) {