From 3b8b8aec1453ee845167e9c848586dd7c041c06a Mon Sep 17 00:00:00 2001
From: Kacper Sawicki <kacper@coder.com>
Date: Mon, 14 Jul 2025 14:50:42 +0200
Subject: [PATCH] Set 'jetbrains_connection' as build reason on workspace start

---
 src/main/kotlin/com/coder/toolbox/sdk/CoderRestClient.kt  | 8 +++++++-
 .../toolbox/sdk/v2/models/CreateWorkspaceBuildRequest.kt  | 5 ++++-
 .../coder/toolbox/sdk/v2/models/WorkspaceBuildReason.kt   | 7 +++++++
 3 files changed, 18 insertions(+), 2 deletions(-)
 create mode 100644 src/main/kotlin/com/coder/toolbox/sdk/v2/models/WorkspaceBuildReason.kt

diff --git a/src/main/kotlin/com/coder/toolbox/sdk/CoderRestClient.kt b/src/main/kotlin/com/coder/toolbox/sdk/CoderRestClient.kt
index 9aa3dfb..325a204 100644
--- a/src/main/kotlin/com/coder/toolbox/sdk/CoderRestClient.kt
+++ b/src/main/kotlin/com/coder/toolbox/sdk/CoderRestClient.kt
@@ -15,6 +15,7 @@ import com.coder.toolbox.sdk.v2.models.User
 import com.coder.toolbox.sdk.v2.models.Workspace
 import com.coder.toolbox.sdk.v2.models.WorkspaceAgent
 import com.coder.toolbox.sdk.v2.models.WorkspaceBuild
+import com.coder.toolbox.sdk.v2.models.WorkspaceBuildReason
 import com.coder.toolbox.sdk.v2.models.WorkspaceResource
 import com.coder.toolbox.sdk.v2.models.WorkspaceStatus
 import com.coder.toolbox.sdk.v2.models.WorkspaceTransition
@@ -268,7 +269,12 @@ open class CoderRestClient(
      * @throws [APIResponseException].
      */
     suspend fun startWorkspace(workspace: Workspace): WorkspaceBuild {
-        val buildRequest = CreateWorkspaceBuildRequest(null, WorkspaceTransition.START)
+        val buildRequest = CreateWorkspaceBuildRequest(
+            null,
+            WorkspaceTransition.START,
+            null,
+            WorkspaceBuildReason.JETBRAINS_CONNECTION
+        )
         val buildResponse = retroRestClient.createWorkspaceBuild(workspace.id, buildRequest)
         if (buildResponse.code() != HttpURLConnection.HTTP_CREATED) {
             throw APIResponseException(
diff --git a/src/main/kotlin/com/coder/toolbox/sdk/v2/models/CreateWorkspaceBuildRequest.kt b/src/main/kotlin/com/coder/toolbox/sdk/v2/models/CreateWorkspaceBuildRequest.kt
index a2f1ca2..53c70c8 100644
--- a/src/main/kotlin/com/coder/toolbox/sdk/v2/models/CreateWorkspaceBuildRequest.kt
+++ b/src/main/kotlin/com/coder/toolbox/sdk/v2/models/CreateWorkspaceBuildRequest.kt
@@ -10,7 +10,8 @@ data class CreateWorkspaceBuildRequest(
     @Json(name = "template_version_id") val templateVersionID: UUID?,
     // Use to start, stop and delete the workspace.
     @Json(name = "transition") val transition: WorkspaceTransition,
-    @Json(name = "orphan") var orphan: Boolean? = null
+    @Json(name = "orphan") var orphan: Boolean? = null,
+    @Json(name = "reason") var reason: WorkspaceBuildReason? = null
 ) {
     override fun equals(other: Any?): Boolean {
         if (this === other) return true
@@ -21,6 +22,7 @@ data class CreateWorkspaceBuildRequest(
         if (templateVersionID != other.templateVersionID) return false
         if (transition != other.transition) return false
         if (orphan != other.orphan) return false
+        if (reason != other.reason) return false
         return true
     }
 
@@ -28,6 +30,7 @@ data class CreateWorkspaceBuildRequest(
         var result = orphan?.hashCode() ?: 0
         result = 31 * result + (templateVersionID?.hashCode() ?: 0)
         result = 31 * result + transition.hashCode()
+        result = 31 * result + (reason?.hashCode() ?: 0)
         return result
     }
 }
diff --git a/src/main/kotlin/com/coder/toolbox/sdk/v2/models/WorkspaceBuildReason.kt b/src/main/kotlin/com/coder/toolbox/sdk/v2/models/WorkspaceBuildReason.kt
new file mode 100644
index 0000000..390c934
--- /dev/null
+++ b/src/main/kotlin/com/coder/toolbox/sdk/v2/models/WorkspaceBuildReason.kt
@@ -0,0 +1,7 @@
+package com.coder.toolbox.sdk.v2.models
+
+import com.squareup.moshi.Json
+
+enum class WorkspaceBuildReason {
+    @Json(name = "jetbrains_connection") JETBRAINS_CONNECTION,
+}
\ No newline at end of file