Break out toAgentModels

We will need this in the recent connections view as well.

Author: code-asher

src/main/kotlin/com/coder/gateway/sdk/v2/models/Workspace.kt

@@ -1,5 +1,10 @@
 package com.coder.gateway.sdk.v2.models
 
+import com.coder.gateway.models.WorkspaceAgentModel
+import com.coder.gateway.models.WorkspaceAndAgentStatus
+import com.coder.gateway.models.WorkspaceVersionStatus
+import com.coder.gateway.sdk.Arch
+import com.coder.gateway.sdk.OS
 import com.google.gson.annotations.SerializedName
 import java.time.Instant
 import java.util.UUID
@@ -24,4 +29,48 @@ data class Workspace(
     @SerializedName("autostart_schedule") val autostartSchedule: String?,
     @SerializedName("ttl_ms") val ttlMillis: Long?,
     @SerializedName("last_used_at") val lastUsedAt: Instant,
-)
+)
+
+fun Workspace.toAgentModels(): Set<WorkspaceAgentModel> {
+    val wam = this.latestBuild.resources.filter { it.agents != null }.flatMap { it.agents!! }.map { agent ->
+        val workspaceWithAgentName = "${this.name}.${agent.name}"
+        val wm = WorkspaceAgentModel(
+            this.id,
+            this.name,
+            workspaceWithAgentName,
+            this.templateID,
+            this.templateName,
+            this.templateIcon,
+            null,
+            WorkspaceVersionStatus.from(this),
+            this.latestBuild.status,
+            WorkspaceAndAgentStatus.from(this, agent),
+            this.latestBuild.transition,
+            OS.from(agent.operatingSystem),
+            Arch.from(agent.architecture),
+            agent.expandedDirectory ?: agent.directory,
+        )
+
+        wm
+    }.toSet()
+    if (wam.isNullOrEmpty()) {
+        val wm = WorkspaceAgentModel(
+            this.id,
+            this.name,
+            this.name,
+            this.templateID,
+            this.templateName,
+            this.templateIcon,
+            null,
+            WorkspaceVersionStatus.from(this),
+            this.latestBuild.status,
+            WorkspaceAndAgentStatus.from(this),
+            this.latestBuild.transition,
+            null,
+            null,
+            null
+        )
+        return setOf(wm)
+    }
+    return wam
+}

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

@@ -7,7 +7,6 @@ import com.coder.gateway.models.TokenSource
 import com.coder.gateway.models.WorkspaceAgentModel
 import com.coder.gateway.models.WorkspaceAndAgentStatus
 import com.coder.gateway.models.WorkspaceVersionStatus
-import com.coder.gateway.sdk.Arch
 import com.coder.gateway.sdk.CoderCLIManager
 import com.coder.gateway.sdk.CoderRestClientService
 import com.coder.gateway.sdk.CoderSemVer
@@ -20,8 +19,8 @@ import com.coder.gateway.sdk.ex.AuthenticationResponseException
 import com.coder.gateway.sdk.ex.TemplateResponseException
 import com.coder.gateway.sdk.ex.WorkspaceResponseException
 import com.coder.gateway.sdk.toURL
-import com.coder.gateway.sdk.v2.models.Workspace
 import com.coder.gateway.sdk.v2.models.WorkspaceStatus
+import com.coder.gateway.sdk.v2.models.toAgentModels
 import com.coder.gateway.sdk.withPath
 import com.coder.gateway.services.CoderSettingsState
 import com.intellij.ide.ActivityTracker
@@ -659,7 +658,15 @@ class CoderWorkspacesStepView(val setNextButtonEnabled: (Boolean) -> Unit) : Cod
             val timeBeforeRequestingWorkspaces = System.currentTimeMillis()
             try {
                 val ws = clientService.client.workspaces()
-                val ams = ws.flatMap { it.toAgentModels() }.toSet()
+                val ams = ws.flatMap { it.toAgentModels() }
+                ams.forEach {
+                    cs.launch(Dispatchers.IO) {
+                        it.templateIcon = iconDownloader.load(it.templateIconPath, it.name)
+                        withContext(Dispatchers.Main) {
+                            tableOfWorkspaces.updateUI()
+                        }
+                    }
+                }
                 val timeAfterRequestingWorkspaces = System.currentTimeMillis()
                 logger.info("Retrieving the workspaces took: ${timeAfterRequestingWorkspaces - timeBeforeRequestingWorkspaces} millis")
                 return@withContext ams
@@ -675,62 +682,6 @@ class CoderWorkspacesStepView(val setNextButtonEnabled: (Boolean) -> Unit) : Cod
         }
     }
 
-    private fun Workspace.toAgentModels(): Set<WorkspaceAgentModel> {
-        val wam = this.latestBuild.resources.filter { it.agents != null }.flatMap { it.agents!! }.map { agent ->
-            val workspaceWithAgentName = "${this.name}.${agent.name}"
-            val wm = WorkspaceAgentModel(
-                this.id,
-                this.name,
-                workspaceWithAgentName,
-                this.templateID,
-                this.templateName,
-                this.templateIcon,
-                null,
-                WorkspaceVersionStatus.from(this),
-                this.latestBuild.status,
-                WorkspaceAndAgentStatus.from(this, agent),
-                this.latestBuild.transition,
-                OS.from(agent.operatingSystem),
-                Arch.from(agent.architecture),
-                agent.expandedDirectory ?: agent.directory,
-            )
-            cs.launch(Dispatchers.IO) {
-                wm.templateIcon = iconDownloader.load(wm.templateIconPath, wm.name)
-                withContext(Dispatchers.Main) {
-                    tableOfWorkspaces.updateUI()
-                }
-            }
-            wm
-        }.toSet()
-
-        if (wam.isNullOrEmpty()) {
-            val wm = WorkspaceAgentModel(
-                this.id,
-                this.name,
-                this.name,
-                this.templateID,
-                this.templateName,
-                this.templateIcon,
-                null,
-                WorkspaceVersionStatus.from(this),
-                this.latestBuild.status,
-                WorkspaceAndAgentStatus.from(this),
-                this.latestBuild.transition,
-                null,
-                null,
-                null
-            )
-            cs.launch(Dispatchers.IO) {
-                wm.templateIcon = iconDownloader.load(wm.templateIconPath, wm.name)
-                withContext(Dispatchers.Main) {
-                    tableOfWorkspaces.updateUI()
-                }
-            }
-            return setOf(wm)
-        }
-        return wam
-    }
-
     override fun onPrevious() {
         super.onPrevious()
         logger.info("Going back to the main view")