refactor: return the agents instead the full list of resources

fioan89 · fioan89 · commit a3d2ecc9fcfc · 2022-06-28T15:04:33.000+03:00
diff --git a/src/main/kotlin/com/coder/gateway/sdk/CoderRestClientService.kt b/src/main/kotlin/com/coder/gateway/sdk/CoderRestClientService.kt
@@ -8,7 +8,7 @@ import com.coder.gateway.sdk.v2.CoderV2RestFacade
 import com.coder.gateway.sdk.v2.models.BuildInfo
 import com.coder.gateway.sdk.v2.models.User
 import com.coder.gateway.sdk.v2.models.Workspace
-import com.coder.gateway.sdk.v2.models.WorkspaceResource
+import com.coder.gateway.sdk.v2.models.WorkspaceAgent
 import com.google.gson.Gson
 import com.google.gson.GsonBuilder
 import com.intellij.openapi.components.Service
@@ -97,15 +97,17 @@ class CoderRestClientService {
     }
 
     /**
-     * Retrieves the workspace resources (a workspace is a collection of ojects like, VMs, containers, cloud DBs, etc...)
+     * Retrieves the workspace agents. A workspace is a collection of objects like, VMs, containers, cloud DBs, etc...
+     * Agents run on compute hosts like VMs or containers.
+     *
      * @throws WorkspaceResourcesResponseException if workspace resources could not be retrieved.
      */
-    fun workspaceResources(workspace: Workspace): List<WorkspaceResource> {
+    fun workspaceAgents(workspace: Workspace): List<WorkspaceAgent> {
         val workspaceResourcesResponse = retroRestClient.workspaceResourceByBuild(workspace.latestBuild.id).execute()
         if (!workspaceResourcesResponse.isSuccessful) {
-            throw WorkspaceResourcesResponseException("Could not retrieve resources for ${workspace.name} workspace :${workspaceResourcesResponse.code()}, reason: ${workspaceResourcesResponse.message()}")
+            throw WorkspaceResourcesResponseException("Could not retrieve agents for ${workspace.name} workspace :${workspaceResourcesResponse.code()}, reason: ${workspaceResourcesResponse.message()}")
         }
 
-        return workspaceResourcesResponse.body()!!
+        return workspaceResourcesResponse.body()!!.flatMap { it.agents ?: emptyList() }
     }
 }
