Set 'jetbrains_connection' as build reason on workspace start

Author: kacpersaw

src/main/kotlin/com/coder/gateway/cli/CoderCLIManager.kt

@@ -116,6 +116,7 @@ data class Features(
     val disableAutostart: Boolean = false,
     val reportWorkspaceUsage: Boolean = false,
     val wildcardSSH: Boolean = false,
+    val buildReason: Boolean = false,
 )
 
 /**
@@ -479,13 +480,21 @@ class CoderCLIManager(
      *
      * Throws if the command execution fails.
      */
-    fun startWorkspace(workspaceOwner: String, workspaceName: String): String = exec(
-        "--global-config",
-        coderConfigPath.toString(),
-        "start",
-        "--yes",
-        workspaceOwner + "/" + workspaceName,
-    )
+    fun startWorkspace(workspaceOwner: String, workspaceName: String, feats: Features = features): String {
+        val args = mutableListOf(
+            "--global-config",
+            coderConfigPath.toString(),
+            "start",
+            "--yes",
+            workspaceOwner + "/" + workspaceName
+        )
+
+        if (feats.buildReason) {
+            args.addAll(listOf("--reason", "jetbrains_connection"))
+        }
+
+        return exec(*args.toTypedArray())
+    }
 
     private fun exec(vararg args: String): String {
         val stdout =
@@ -511,6 +520,7 @@ class CoderCLIManager(
                     disableAutostart = version >= SemVer(2, 5, 0),
                     reportWorkspaceUsage = version >= SemVer(2, 13, 0),
                     wildcardSSH = version >= SemVer(2, 19, 0),
+                    buildReason = version >= SemVer(2, 25, 0),
                 )
             }
         }

src/main/kotlin/com/coder/gateway/sdk/CoderRestClient.kt

@@ -15,6 +15,7 @@ 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.coder.gateway.sdk.v2.models.WorkspaceBuildReason
 import com.coder.gateway.sdk.v2.models.WorkspaceResource
 import com.coder.gateway.sdk.v2.models.WorkspaceStatus
 import com.coder.gateway.sdk.v2.models.WorkspaceTransition
@@ -244,7 +245,7 @@ open class CoderRestClient(
      * @throws [APIResponseException].
      */
     fun stopWorkspace(workspace: Workspace): WorkspaceBuild {
-        val buildRequest = CreateWorkspaceBuildRequest(null, WorkspaceTransition.STOP)
+        val buildRequest = CreateWorkspaceBuildRequest(null, WorkspaceTransition.STOP, null)
         val buildResponse = retroRestClient.createWorkspaceBuild(workspace.id, buildRequest).execute()
         if (buildResponse.code() != HttpURLConnection.HTTP_CREATED) {
             throw APIResponseException("stop workspace ${workspace.name}", url, buildResponse)
@@ -265,7 +266,11 @@ open class CoderRestClient(
     fun updateWorkspace(workspace: Workspace): WorkspaceBuild {
         val template = template(workspace.templateID)
         val buildRequest =
-            CreateWorkspaceBuildRequest(template.activeVersionID, WorkspaceTransition.START)
+            CreateWorkspaceBuildRequest(
+                template.activeVersionID,
+                WorkspaceTransition.START,
+                WorkspaceBuildReason.JETBRAINS_CONNECTION
+            )
         val buildResponse = retroRestClient.createWorkspaceBuild(workspace.id, buildRequest).execute()
         if (buildResponse.code() != HttpURLConnection.HTTP_CREATED) {
             throw APIResponseException("update workspace ${workspace.name}", url, buildResponse)

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

@@ -10,6 +10,8 @@ data class CreateWorkspaceBuildRequest(
     @Json(name = "template_version_id") val templateVersionID: UUID?,
     // Use to start and stop the workspace.
     @Json(name = "transition") val transition: WorkspaceTransition,
+    // Use to set build reason for a workspace.
+    @Json(name = "reason") val reason: WorkspaceBuildReason?,
 ) {
     override fun equals(other: Any?): Boolean {
         if (this === other) return true
@@ -19,13 +21,15 @@ data class CreateWorkspaceBuildRequest(
 
         if (templateVersionID != other.templateVersionID) return false
         if (transition != other.transition) return false
+        if (reason != other.reason) return false
 
         return true
     }
 
     override fun hashCode(): Int {
         var result = templateVersionID?.hashCode() ?: 0
         result = 31 * result + transition.hashCode()
+        result = 31 * result + (reason?.hashCode() ?: 0)
         return result
     }
 }

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

@@ -0,0 +1,7 @@
+package com.coder.gateway.sdk.v2.models
+
+import com.squareup.moshi.Json
+
+enum class WorkspaceBuildReason {
+    @Json(name = "jetbrains_connection") JETBRAINS_CONNECTION,
+}