Impl: REST calls to update a workspace to latest template

fioan89 · fioan89 · commit 2a9a5bac556e · 2022-08-17T22:19:33.000+03:00
- also includes a new REST call for retrieving templates
diff --git a/src/main/kotlin/com/coder/gateway/sdk/CoderRestClientService.kt b/src/main/kotlin/com/coder/gateway/sdk/CoderRestClientService.kt
@@ -2,11 +2,13 @@ package com.coder.gateway.sdk
 
 import com.coder.gateway.sdk.convertors.InstantConverter
 import com.coder.gateway.sdk.ex.AuthenticationResponseException
+import com.coder.gateway.sdk.ex.TemplateResponseException
 import com.coder.gateway.sdk.ex.WorkspaceResourcesResponseException
 import com.coder.gateway.sdk.ex.WorkspaceResponseException
 import com.coder.gateway.sdk.v2.CoderV2RestFacade
 import com.coder.gateway.sdk.v2.models.BuildInfo
 import com.coder.gateway.sdk.v2.models.CreateWorkspaceBuildRequest
+import com.coder.gateway.sdk.v2.models.Template
 import com.coder.gateway.sdk.v2.models.User
 import com.coder.gateway.sdk.v2.models.Workspace
 import com.coder.gateway.sdk.v2.models.WorkspaceAgent
@@ -115,6 +117,14 @@ class CoderRestClientService {
         return workspaceResourcesResponse.body()!!.flatMap { it.agents ?: emptyList() }
     }
 
+    private fun template(templateID: UUID): Template {
+        val templateResponse = retroRestClient.template(templateID).execute()
+        if (!templateResponse.isSuccessful) {
+            throw TemplateResponseException("Failed to retrieve template with id: $templateID, reason: ${templateResponse.message()}")
+        }
+        return templateResponse.body()!!
+    }
+
     fun startWorkspace(workspaceID: UUID, workspaceName: String): WorkspaceBuild {
         val buildRequest = CreateWorkspaceBuildRequest(null, "start", null, null, null)
         val buildResponse = retroRestClient.createWorkspaceBuild(workspaceID, buildRequest).execute()
@@ -134,4 +144,16 @@ class CoderRestClientService {
 
         return buildResponse.body()!!
     }
+
+    fun updateWorkspace(workspaceID: UUID, workspaceName: String, lastWorkspaceTransition: String, templateID: UUID): WorkspaceBuild {
+        val template = template(templateID)
+
+        val buildRequest = CreateWorkspaceBuildRequest(template.activeVersionID, lastWorkspaceTransition, null, null, null)
+        val buildResponse = retroRestClient.createWorkspaceBuild(workspaceID, buildRequest).execute()
+        if (buildResponse.code() != HTTP_CREATED) {
+            throw WorkspaceResponseException("Failed to update workspace ${workspaceName}: ${buildResponse.code()}, reason: ${buildResponse.message()}")
+        }
+
+        return buildResponse.body()!!
+    }
 }
diff --git a/src/main/kotlin/com/coder/gateway/sdk/ex/exceptions.kt b/src/main/kotlin/com/coder/gateway/sdk/ex/exceptions.kt
@@ -6,4 +6,6 @@ class AuthenticationResponseException(reason: String) : IOException(reason)
 
 class WorkspaceResponseException(reason: String) : IOException(reason)
 
-class WorkspaceResourcesResponseException(reason: String) : IOException(reason)
+class WorkspaceResourcesResponseException(reason: String) : IOException(reason)
+
+class TemplateResponseException(reason: String) : IOException(reason)
diff --git a/src/main/kotlin/com/coder/gateway/sdk/v2/CoderV2RestFacade.kt b/src/main/kotlin/com/coder/gateway/sdk/v2/CoderV2RestFacade.kt
@@ -2,6 +2,7 @@ package com.coder.gateway.sdk.v2
 
 import com.coder.gateway.sdk.v2.models.BuildInfo
 import com.coder.gateway.sdk.v2.models.CreateWorkspaceBuildRequest
+import com.coder.gateway.sdk.v2.models.Template
 import com.coder.gateway.sdk.v2.models.User
 import com.coder.gateway.sdk.v2.models.Workspace
 import com.coder.gateway.sdk.v2.models.WorkspaceBuild
@@ -39,4 +40,7 @@ interface CoderV2RestFacade {
      */
     @POST("api/v2/workspaces/{workspaceID}/builds")
     fun createWorkspaceBuild(@Path("workspaceID") workspaceID: UUID, @Body createWorkspaceBuildRequest: CreateWorkspaceBuildRequest): Call<WorkspaceBuild>
+
+    @GET("api/v2/templates/{templateID}")
+    fun template(@Path("templateID") templateID: UUID): Call<Template>
 }
diff --git a/src/main/kotlin/com/coder/gateway/sdk/v2/models/Template.kt b/src/main/kotlin/com/coder/gateway/sdk/v2/models/Template.kt
@@ -0,0 +1,21 @@
+package com.coder.gateway.sdk.v2.models
+
+import com.google.gson.annotations.SerializedName
+import java.time.Instant
+import java.util.UUID
+
+data class Template(
+    @SerializedName("id") val id: UUID,
+    @SerializedName("created_at") val createdAt: Instant,
+    @SerializedName("updated_at") val updatedAt: Instant,
+    @SerializedName("organization_id") val organizationIterator: UUID,
+    @SerializedName("name") val name: String,
+    @SerializedName("provisioner") val provisioner: String,
+    @SerializedName("active_version_id") val activeVersionID: UUID,
+    @SerializedName("workspace_owner_count") val workspaceOwnerCount: Int,
+    @SerializedName("description") val description: String,
+    @SerializedName("max_ttl_ms") val maxTTLMillis: Long,
+    @SerializedName("min_autostart_interval_ms") val minAutostartIntervalMillis: Long,
+    @SerializedName("created_by_id") val createdByID: UUID,
+    @SerializedName("created_by_name") val createdByName: String,
+)
