Impl: add REST calls for starting/stopping workspaces

fioan89 · fioan89 · commit 86654c3e4d5e · 2022-08-13T00:40:02.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
@@ -6,9 +6,11 @@ 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.User
 import com.coder.gateway.sdk.v2.models.Workspace
 import com.coder.gateway.sdk.v2.models.WorkspaceAgent
+import com.coder.gateway.sdk.v2.models.WorkspaceBuild
 import com.google.gson.Gson
 import com.google.gson.GsonBuilder
 import com.intellij.openapi.components.Service
@@ -20,8 +22,10 @@ import okhttp3.logging.HttpLoggingInterceptor
 import retrofit2.Retrofit
 import retrofit2.converter.gson.GsonConverterFactory
 import java.net.CookieManager
+import java.net.HttpURLConnection.HTTP_CREATED
 import java.net.URL
 import java.time.Instant
+import java.util.UUID
 
 @Service(Service.Level.APP)
 class CoderRestClientService {
@@ -110,4 +114,24 @@ class CoderRestClientService {
 
         return workspaceResourcesResponse.body()!!.flatMap { it.agents ?: emptyList() }
     }
+
+    fun startWorkspace(workspaceID: UUID, workspaceName: String): WorkspaceBuild {
+        val buildRequest = CreateWorkspaceBuildRequest(null, "start", null, null, null)
+        val buildResponse = retroRestClient.createWorkspaceBuild(workspaceID, buildRequest).execute()
+        if (buildResponse.code() != HTTP_CREATED) {
+            throw WorkspaceResponseException("Failed to build workspace ${workspaceName}: ${buildResponse.code()}, reason: ${buildResponse.message()}")
+        }
+
+        return buildResponse.body()!!
+    }
+
+    fun stopWorkspace(workspaceID: UUID, workspaceName: String): WorkspaceBuild {
+        val buildRequest = CreateWorkspaceBuildRequest(null, "stop", null, null, null)
+        val buildResponse = retroRestClient.createWorkspaceBuild(workspaceID, buildRequest).execute()
+        if (buildResponse.code() != HTTP_CREATED) {
+            throw WorkspaceResponseException("Failed to stop workspace ${workspaceName}: ${buildResponse.code()}, reason: ${buildResponse.message()}")
+        }
+
+        return buildResponse.body()!!
+    }
 }
diff --git a/src/main/kotlin/com/coder/gateway/sdk/v2/CoderV2RestFacade.kt b/src/main/kotlin/com/coder/gateway/sdk/v2/CoderV2RestFacade.kt
@@ -1,11 +1,15 @@
 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.User
 import com.coder.gateway.sdk.v2.models.Workspace
+import com.coder.gateway.sdk.v2.models.WorkspaceBuild
 import com.coder.gateway.sdk.v2.models.WorkspaceResource
 import retrofit2.Call
+import retrofit2.http.Body
 import retrofit2.http.GET
+import retrofit2.http.POST
 import retrofit2.http.Path
 import retrofit2.http.Query
 import java.util.UUID
@@ -29,4 +33,10 @@ interface CoderV2RestFacade {
 
     @GET("api/v2/workspacebuilds/{buildID}/resources")
     fun workspaceResourceByBuild(@Path("buildID") build: UUID): Call<List<WorkspaceResource>>
+
+    /**
+     * Queues a new build to occur for a workspace.
+     */
+    @POST("api/v2/workspaces/{workspaceID}/builds")
+    fun createWorkspaceBuild(@Path("workspaceID") workspaceID: UUID, @Body createWorkspaceBuildRequest: CreateWorkspaceBuildRequest): Call<WorkspaceBuild>
 }
diff --git a/src/main/kotlin/com/coder/gateway/sdk/v2/models/CreateParameterRequest.kt b/src/main/kotlin/com/coder/gateway/sdk/v2/models/CreateParameterRequest.kt
@@ -0,0 +1,12 @@
+package com.coder.gateway.sdk.v2.models
+
+import com.google.gson.annotations.SerializedName
+import java.util.UUID
+
+data class CreateParameterRequest(
+    @SerializedName("copy_from_parameter") val cloneID: UUID?,
+    @SerializedName("name") val name: String,
+    @SerializedName("source_value") val sourceValue: String,
+    @SerializedName("source_scheme") val sourceScheme: String,
+    @SerializedName("destination_scheme") val destinationScheme: String
+)
diff --git a/src/main/kotlin/com/coder/gateway/sdk/v2/models/CreateWorkspaceBuildRequest.kt b/src/main/kotlin/com/coder/gateway/sdk/v2/models/CreateWorkspaceBuildRequest.kt
@@ -0,0 +1,44 @@
+package com.coder.gateway.sdk.v2.models
+
+import com.google.gson.annotations.SerializedName
+import java.util.UUID
+
+data class CreateWorkspaceBuildRequest(
+    @SerializedName("template_version_id") val templateVersionID: UUID?,
+    @SerializedName("transition") val transition: String,
+    @SerializedName("dry_run") val dryRun: Boolean?,
+    @SerializedName("state") val state: Array<Byte>?,
+    @SerializedName("parameter_values") val parameterValues: Array<CreateParameterRequest>?
+) {
+
+    override fun equals(other: Any?): Boolean {
+        if (this === other) return true
+        if (javaClass != other?.javaClass) return false
+
+        other as CreateWorkspaceBuildRequest
+
+        if (templateVersionID != other.templateVersionID) return false
+        if (transition != other.transition) return false
+        if (dryRun != other.dryRun) return false
+        if (state != null) {
+            if (other.state == null) return false
+            if (!state.contentEquals(other.state)) return false
+        } else if (other.state != null) return false
+        if (parameterValues != null) {
+            if (other.parameterValues == null) return false
+            if (!parameterValues.contentEquals(other.parameterValues)) return false
+        } else if (other.parameterValues != null) return false
+
+        return true
+    }
+
+    override fun hashCode(): Int {
+        var result = templateVersionID?.hashCode() ?: 0
+        result = 31 * result + transition.hashCode()
+        result = 31 * result + (dryRun?.hashCode() ?: 0)
+        result = 31 * result + (state?.contentHashCode() ?: 0)
+        result = 31 * result + (parameterValues?.contentHashCode() ?: 0)
+        return result
+    }
+}
+
